void FortranAnalysis::visit(SgFunctionCallExp * fcall)
{
SgFunctionRefExp * fref = isSgFunctionRefExp(fcall->get_function());
if (fref != NULL) {
SgExpressionPtrList::iterator it = fcall->get_args()->get_expressions().begin();
std::string name = fref->get_symbol()->get_name().getString();
if (name == "interior" && it != fcall->get_args()->get_expressions().end()) {
SgVarRefExp * var = isSgVarRefExp(*it);
SgSymbol * sym = var->get_symbol();
sym->setAttribute("halo_attr", new AstTextAttribute("HALO_VAR"));
debug("SgFunctionCallExp: adding halo attribute to %s\n",
sym->get_name().getString().c_str());
}
}
}
void mlmFrontend::attachAttribute(SgPragmaDeclaration* pragma, AstAttribute* attr)
{
// attribute to specify memory level
if(isSgVariableDeclaration(getNextStatement(pragma)))
{
SgVariableDeclaration* decl = isSgVariableDeclaration(getNextStatement(pragma));
ROSE_ASSERT(decl);
SgInitializedNamePtrList nameList = decl->get_variables();
if(nameList.size() == 1)
{
SgInitializedName* initName = nameList[0];
SgSymbol* symbol = initName->get_symbol_from_symbol_table();
symbol->setAttribute("mlmAttribute", attr);
}
}
// attribute to specify tiling level
else if(isSgForStatement(getNextStatement(pragma)))
{
SgForStatement* forStmt = isSgForStatement(getNextStatement(pragma));
ROSE_ASSERT(forStmt);
forStmt->setAttribute("mlmAttribute", attr);
}
}
void FortranAnalysis::visit(SgFunctionDefinition * func_def)
{
SgFunctionDeclaration * func_decl = isSgFunctionDeclaration(func_def->get_declaration());
if (func_decl == NULL) return;
SgInitializedNamePtrList func_args = func_decl->get_parameterList()->get_args();
SgInitializedNamePtrList::iterator arg = func_args.begin();
// for (it_args = func_args.begin(); it_args != func_args.end(); it_args++) {
while (arg != func_args.end()) {
SgInitializedName * func_arg = isSgInitializedName(*arg++);
SgSymbol * sym = func_def->lookup_symbol(func_arg->get_name());
SgType * type = sym->get_type();
if (sym == NULL) {
printf("FortranAnalysis::visit: no symbol for name %s\n",
func_arg->get_name().getString().c_str());
}
else if (isSgArrayType(type) != NULL) {
printf("WARNING: arg %s must be a scalar\n", func_arg->get_name().str());
sym->setAttribute("dummy_attr", new AstTextAttribute("DUMMY_ARRAY_TYPE_ARG"));
}
else if (isSgNamedType(type)) {
sym->setAttribute("dummy_attr", new AstTextAttribute("DUMMY_NAMED_TYPE_ARG"));
}
else {
sym->setAttribute("dummy_attr", new AstTextAttribute("DUMMY_ARG"));
}
if (sym != NULL && isElementalArrayType(func_arg)) {
sym->setAttribute("elemental_attr", new AstTextAttribute("ELEMENTAL_ARRAY"));
sym->setAttribute("index_attr", new AstTextAttribute("idx"));
}
if (sym != NULL && hasArrayDescriptor(func_arg)) {
sym->setAttribute("descriptor_attr", new AstTextAttribute("desc_"+func_arg->get_name()));
sym->setAttribute("index_attr", new AstTextAttribute("idx_"+func_arg->get_name()));
}
}
}