Skip to content

Commit ec1e0c5

Browse files
LeporacanthicusLeporacanthicus
authored
[Flang][OMP]Add support for DECLARE MAPPER parsing and semantics (#115160)
Will hit a TODO in the lowering, which there are tests added to check for this happening.
1 parent caa9a82 commit ec1e0c5

File tree

17 files changed

+269
-5
lines changed

17 files changed

+269
-5
lines changed

‎flang/include/flang/Parser/dump-parse-tree.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -506,6 +506,7 @@ class ParseTreeDumper {
506506
NODE(parser, OmpDeclareTargetSpecifier)
507507
NODE(parser, OmpDeclareTargetWithClause)
508508
NODE(parser, OmpDeclareTargetWithList)
509+
NODE(parser, OmpDeclareMapperSpecifier)
509510
NODE(parser, OmpDefaultClause)
510511
NODE_ENUM(OmpDefaultClause, Type)
511512
NODE(parser, OmpDefaultmapClause)
@@ -621,6 +622,7 @@ class ParseTreeDumper {
621622
NODE(parser, OpenMPDeclareReductionConstruct)
622623
NODE(parser, OpenMPDeclareSimdConstruct)
623624
NODE(parser, OpenMPDeclareTargetConstruct)
625+
NODE(parser, OpenMPDeclareMapperConstruct)
624626
NODE(parser, OmpMemoryOrderClause)
625627
NODE(parser, OmpAtomicClause)
626628
NODE(parser, OmpAtomicClauseList)

‎flang/include/flang/Parser/parse-tree.h

Lines changed: 17 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3916,6 +3916,19 @@ struct OpenMPDeclareTargetConstruct {
39163916
std::tuple<Verbatim, OmpDeclareTargetSpecifier> t;
39173917
};
39183918

3919+
struct OmpDeclareMapperSpecifier {
3920+
TUPLE_CLASS_BOILERPLATE(OmpDeclareMapperSpecifier);
3921+
std::tuple<std::optional<Name>, TypeSpec, Name> t;
3922+
};
3923+
3924+
// OMP v5.2: 5.8.8
3925+
// declare-mapper -> DECLARE MAPPER ([mapper-name :] type :: var) map-clauses
3926+
struct OpenMPDeclareMapperConstruct {
3927+
TUPLE_CLASS_BOILERPLATE(OpenMPDeclareMapperConstruct);
3928+
CharBlock source;
3929+
std::tuple<Verbatim, OmpDeclareMapperSpecifier, OmpClauseList> t;
3930+
};
3931+
39193932
// 2.16 declare-reduction -> DECLARE REDUCTION (reduction-identifier : type-list
39203933
// : combiner) [initializer-clause]
39213934
struct OmpReductionCombiner {
@@ -3966,9 +3979,10 @@ struct OpenMPDeclarativeAllocate {
39663979
struct OpenMPDeclarativeConstruct {
39673980
UNION_CLASS_BOILERPLATE(OpenMPDeclarativeConstruct);
39683981
CharBlock source;
3969-
std::variant<OpenMPDeclarativeAllocate, OpenMPDeclareReductionConstruct,
3970-
OpenMPDeclareSimdConstruct, OpenMPDeclareTargetConstruct,
3971-
OpenMPThreadprivate, OpenMPRequiresConstruct>
3982+
std::variant<OpenMPDeclarativeAllocate, OpenMPDeclareMapperConstruct,
3983+
OpenMPDeclareReductionConstruct, OpenMPDeclareSimdConstruct,
3984+
OpenMPDeclareTargetConstruct, OpenMPThreadprivate,
3985+
OpenMPRequiresConstruct>
39723986
u;
39733987
};
39743988

‎flang/lib/Lower/OpenMP/OpenMP.cpp

Lines changed: 7 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2623,6 +2623,13 @@ genOMP(lower::AbstractConverter &converter, lower::SymMap &symTable,
26232623
TODO(converter.getCurrentLocation(), "OpenMPDeclareSimdConstruct");
26242624
}
26252625

2626+
static void
2627+
genOMP(lower::AbstractConverter &converter, lower::SymMap &symTable,
2628+
semantics::SemanticsContext &semaCtx, lower::pft::Evaluation &eval,
2629+
const parser::OpenMPDeclareMapperConstruct &declareMapperConstruct) {
2630+
TODO(converter.getCurrentLocation(), "OpenMPDeclareMapperConstruct");
2631+
}
2632+
26262633
static void
26272634
genOMP(lower::AbstractConverter &converter, lower::SymMap &symTable,
26282635
semantics::SemanticsContext &semaCtx, lower::pft::Evaluation &eval,

‎flang/lib/Parser/openmp-parsers.cpp

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -860,6 +860,15 @@ TYPE_PARSER(
860860
TYPE_PARSER(sourced(construct<OpenMPDeclareTargetConstruct>(
861861
verbatim("DECLARE TARGET"_tok), Parser<OmpDeclareTargetSpecifier>{})))
862862

863+
// declare-mapper-specifier
864+
TYPE_PARSER(construct<OmpDeclareMapperSpecifier>(
865+
maybe(name / ":" / !":"_tok), typeSpec / "::", name))
866+
867+
// OpenMP 5.2: 5.8.8 Declare Mapper Construct
868+
TYPE_PARSER(sourced(construct<OpenMPDeclareMapperConstruct>(
869+
verbatim("DECLARE MAPPER"_tok),
870+
"(" >> Parser<OmpDeclareMapperSpecifier>{} / ")", Parser<OmpClauseList>{})))
871+
863872
TYPE_PARSER(construct<OmpReductionCombiner>(Parser<AssignmentStmt>{}) ||
864873
construct<OmpReductionCombiner>(
865874
construct<OmpReductionCombiner::FunctionCombiner>(
@@ -968,6 +977,8 @@ TYPE_PARSER(startOmpLine >>
968977
withMessage("expected OpenMP construct"_err_en_US,
969978
sourced(construct<OpenMPDeclarativeConstruct>(
970979
Parser<OpenMPDeclareReductionConstruct>{}) ||
980+
construct<OpenMPDeclarativeConstruct>(
981+
Parser<OpenMPDeclareMapperConstruct>{}) ||
971982
construct<OpenMPDeclarativeConstruct>(
972983
Parser<OpenMPDeclareSimdConstruct>{}) ||
973984
construct<OpenMPDeclarativeConstruct>(

‎flang/lib/Parser/unparse.cpp

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2652,6 +2652,22 @@ class UnparseVisitor {
26522652
EndOpenMP();
26532653
return false;
26542654
},
2655+
[&](const OpenMPDeclareMapperConstruct &z) {
2656+
Word("DECLARE MAPPER (");
2657+
const auto &spec{std::get<OmpDeclareMapperSpecifier>(z.t)};
2658+
if (auto mapname{std::get<std::optional<Name>>(spec.t)}) {
2659+
Walk(mapname);
2660+
Put(":");
2661+
}
2662+
Walk(std::get<TypeSpec>(spec.t));
2663+
Put("::");
2664+
Walk(std::get<Name>(spec.t));
2665+
Put(")");
2666+
2667+
Walk(std::get<OmpClauseList>(z.t));
2668+
Put("\n");
2669+
return false;
2670+
},
26552671
[&](const OpenMPDeclareReductionConstruct &) {
26562672
Word("DECLARE REDUCTION ");
26572673
return true;

‎flang/lib/Semantics/check-omp-structure.cpp

Lines changed: 15 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1472,6 +1472,21 @@ void OmpStructureChecker::Leave(const parser::OmpDeclareTargetWithClause &x) {
14721472
}
14731473
}
14741474

1475+
void OmpStructureChecker::Enter(const parser::OpenMPDeclareMapperConstruct &x) {
1476+
const auto &dir{std::get<parser::Verbatim>(x.t)};
1477+
PushContextAndClauseSets(
1478+
dir.source, llvm::omp::Directive::OMPD_declare_mapper);
1479+
const auto &spec{std::get<parser::OmpDeclareMapperSpecifier>(x.t)};
1480+
const auto &type = std::get<parser::TypeSpec>(spec.t);
1481+
if (!std::get_if<parser::DerivedTypeSpec>(&type.u)) {
1482+
context_.Say(dir.source, "Type is not a derived type"_err_en_US);
1483+
}
1484+
}
1485+
1486+
void OmpStructureChecker::Leave(const parser::OpenMPDeclareMapperConstruct &) {
1487+
dirContext_.pop_back();
1488+
}
1489+
14751490
void OmpStructureChecker::Enter(const parser::OpenMPDeclareTargetConstruct &x) {
14761491
const auto &dir{std::get<parser::Verbatim>(x.t)};
14771492
PushContext(dir.source, llvm::omp::Directive::OMPD_declare_target);

‎flang/lib/Semantics/check-omp-structure.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -93,6 +93,8 @@ class OmpStructureChecker
9393
void Leave(const parser::OpenMPDeclareSimdConstruct &);
9494
void Enter(const parser::OpenMPDeclarativeAllocate &);
9595
void Leave(const parser::OpenMPDeclarativeAllocate &);
96+
void Enter(const parser::OpenMPDeclareMapperConstruct &);
97+
void Leave(const parser::OpenMPDeclareMapperConstruct &);
9698
void Enter(const parser::OpenMPDeclareTargetConstruct &);
9799
void Leave(const parser::OpenMPDeclareTargetConstruct &);
98100
void Enter(const parser::OpenMPDepobjConstruct &);

‎flang/lib/Semantics/resolve-directives.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -431,6 +431,9 @@ class OmpAttributeVisitor : DirectiveAttributeVisitor<llvm::omp::Directive> {
431431
bool Pre(const parser::OpenMPDeclareTargetConstruct &);
432432
void Post(const parser::OpenMPDeclareTargetConstruct &) { PopContext(); }
433433

434+
bool Pre(const parser::OpenMPDeclareMapperConstruct &);
435+
void Post(const parser::OpenMPDeclareMapperConstruct &) { PopContext(); }
436+
434437
bool Pre(const parser::OpenMPThreadprivate &);
435438
void Post(const parser::OpenMPThreadprivate &) { PopContext(); }
436439

@@ -1953,6 +1956,11 @@ bool OmpAttributeVisitor::Pre(const parser::OpenMPDeclareTargetConstruct &x) {
19531956
return true;
19541957
}
19551958

1959+
bool OmpAttributeVisitor::Pre(const parser::OpenMPDeclareMapperConstruct &x) {
1960+
PushContext(x.source, llvm::omp::Directive::OMPD_declare_mapper);
1961+
return true;
1962+
}
1963+
19561964
bool OmpAttributeVisitor::Pre(const parser::OpenMPThreadprivate &x) {
19571965
PushContext(x.source, llvm::omp::Directive::OMPD_threadprivate);
19581966
const auto &list{std::get<parser::OmpObjectList>(x.t)};

‎flang/lib/Semantics/resolve-names.cpp

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1468,6 +1468,9 @@ class OmpVisitor : public virtual DeclarationVisitor {
14681468
AddOmpSourceRange(x.source);
14691469
return true;
14701470
}
1471+
1472+
bool Pre(const parser::OpenMPDeclareMapperConstruct &);
1473+
14711474
void Post(const parser::OmpBeginLoopDirective &) {
14721475
messageHandler().set_currStmtSource(std::nullopt);
14731476
}
@@ -1605,6 +1608,37 @@ void OmpVisitor::Post(const parser::OpenMPBlockConstruct &x) {
16051608
}
16061609
}
16071610

1611+
// This "manually" walks the tree of the construct, because we need
1612+
// to resolve the type before the map clauses are processed - when
1613+
// just following the natural flow, the map clauses gets processed before
1614+
// the type has been fully processed.
1615+
bool OmpVisitor::Pre(const parser::OpenMPDeclareMapperConstruct &x) {
1616+
AddOmpSourceRange(x.source);
1617+
BeginDeclTypeSpec();
1618+
const auto &spec{std::get<parser::OmpDeclareMapperSpecifier>(x.t)};
1619+
Symbol *mapperSym{nullptr};
1620+
if (const auto &mapperName{std::get<std::optional<parser::Name>>(spec.t)}) {
1621+
mapperSym =
1622+
&MakeSymbol(*mapperName, MiscDetails{MiscDetails::Kind::ConstructName});
1623+
mapperName->symbol = mapperSym;
1624+
} else {
1625+
const parser::CharBlock defaultName{"default", 7};
1626+
mapperSym = &MakeSymbol(
1627+
defaultName, Attrs{}, MiscDetails{MiscDetails::Kind::ConstructName});
1628+
}
1629+
1630+
PushScope(Scope::Kind::OtherConstruct, nullptr);
1631+
Walk(std::get<parser::TypeSpec>(spec.t));
1632+
const auto &varName{std::get<parser::ObjectName>(spec.t)};
1633+
DeclareObjectEntity(varName);
1634+
1635+
Walk(std::get<parser::OmpClauseList>(x.t));
1636+
1637+
EndDeclTypeSpec();
1638+
PopScope();
1639+
return false;
1640+
}
1641+
16081642
// Walk the parse tree and resolve names to symbols.
16091643
class ResolveNamesVisitor : public virtual ScopeHandler,
16101644
public ModuleVisitor,

‎flang/lib/Semantics/unparse-with-symbols.cpp

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -53,6 +53,14 @@ class SymbolDumpVisitor {
5353
void Post(const parser::OpenMPThreadprivate &) { currStmt_ = std::nullopt; }
5454
void Post(const parser::Name &name);
5555

56+
bool Pre(const parser::OpenMPDeclareMapperConstruct &x) {
57+
currStmt_ = x.source;
58+
return true;
59+
}
60+
void Post(const parser::OpenMPDeclareMapperConstruct &) {
61+
currStmt_ = std::nullopt;
62+
}
63+
5664
private:
5765
std::optional<SourceName> currStmt_; // current statement we are processing
5866
std::multimap<const char *, const Symbol *> symbols_; // location to symbol

0 commit comments

Comments
 (0)