Skip to content

Commit d81a161

Browse files
TIFitisTIFitis
committed
[Flang] Add parser support for AUTOMAP modifier
Add parser support for the new AUTOMAP modifier for OpenMP Declare Target Enter clause introduced in OpenMP 6.0 section 7.9.7.
1 parent 1381ad4 commit d81a161

File tree

14 files changed

+122
-24
lines changed

14 files changed

+122
-24
lines changed

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

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -581,13 +581,17 @@ class ParseTreeDumper {
581581
NODE(parser, OmpDependClause)
582582
NODE(OmpDependClause, TaskDep)
583583
NODE(OmpDependClause::TaskDep, Modifier)
584+
NODE(parser, OmpAutomapModifier)
585+
NODE_ENUM(OmpAutomapModifier, Value)
584586
NODE(parser, OmpDetachClause)
585587
NODE(parser, OmpDoacrossClause)
586588
NODE(parser, OmpDestroyClause)
587589
NODE(parser, OmpEndBlockDirective)
588590
NODE(parser, OmpEndCriticalDirective)
589591
NODE(parser, OmpEndLoopDirective)
590592
NODE(parser, OmpEndSectionsDirective)
593+
NODE(parser, OmpEnterClause)
594+
NODE(OmpEnterClause, Modifier)
591595
NODE(parser, OmpFailClause)
592596
NODE(parser, OmpFromClause)
593597
NODE(OmpFromClause, Modifier)

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

Lines changed: 21 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3769,6 +3769,16 @@ struct OmpAlwaysModifier {
37693769
WRAPPER_CLASS_BOILERPLATE(OmpAlwaysModifier, Value);
37703770
};
37713771

3772+
// Ref: [6.0:289-290]
3773+
//
3774+
// automap-modifier ->
3775+
// automap // since 6.0
3776+
//
3777+
struct OmpAutomapModifier {
3778+
ENUM_CLASS(Value, Automap);
3779+
WRAPPER_CLASS_BOILERPLATE(OmpAutomapModifier, Value);
3780+
};
3781+
37723782
// Ref: [5.2:252-254]
37733783
//
37743784
// chunk-modifier ->
@@ -4350,6 +4360,17 @@ struct OmpDeviceTypeClause {
43504360
WRAPPER_CLASS_BOILERPLATE(OmpDeviceTypeClause, DeviceTypeDescription);
43514361
};
43524362

4363+
// Ref: [5.2:158-159], [6.0:289-290]
4364+
//
4365+
// enter-clause ->
4366+
// ENTER(locator-list) |
4367+
// ENTER(automap-modifier: locator-list) | // since 6.0
4368+
struct OmpEnterClause {
4369+
TUPLE_CLASS_BOILERPLATE(OmpEnterClause);
4370+
MODIFIER_BOILERPLATE(OmpAutomapModifier);
4371+
std::tuple<MODIFIERS(), OmpObjectList> t;
4372+
};
4373+
43534374
// OMP 5.2 15.8.3 extended-atomic, fail-clause ->
43544375
// FAIL(memory-order)
43554376
struct OmpFailClause {

‎flang/include/flang/Semantics/openmp-modifiers.h‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -72,6 +72,7 @@ DECLARE_DESCRIPTOR(parser::OmpAlignModifier);
7272
DECLARE_DESCRIPTOR(parser::OmpAllocatorComplexModifier);
7373
DECLARE_DESCRIPTOR(parser::OmpAllocatorSimpleModifier);
7474
DECLARE_DESCRIPTOR(parser::OmpAlwaysModifier);
75+
DECLARE_DESCRIPTOR(parser::OmpAutomapModifier);
7576
DECLARE_DESCRIPTOR(parser::OmpChunkModifier);
7677
DECLARE_DESCRIPTOR(parser::OmpCloseModifier);
7778
DECLARE_DESCRIPTOR(parser::OmpContextSelector);

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

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -1519,10 +1519,14 @@ bool ClauseProcessor::processTo(
15191519
bool ClauseProcessor::processEnter(
15201520
llvm::SmallVectorImpl<DeclareTargetCapturePair> &result) const {
15211521
return findRepeatableClause<omp::clause::Enter>(
1522-
[&](const omp::clause::Enter &clause, const parser::CharBlock &) {
1522+
[&](const omp::clause::Enter &clause, const parser::CharBlock &source) {
1523+
mlir::Location currentLocation = converter.genLocation(source);
1524+
if (std::get<std::optional<omp::clause::Enter::Modifier>>(clause.t))
1525+
TODO(currentLocation, "Declare target enter AUTOMAP modifier");
15231526
// Case: declare target enter(func, var1, var2)...
1524-
gatherFuncAndVarSyms(
1525-
clause.v, mlir::omp::DeclareTargetCaptureClause::enter, result);
1527+
gatherFuncAndVarSyms(std::get<ObjectList>(clause.t),
1528+
mlir::omp::DeclareTargetCaptureClause::enter,
1529+
result);
15261530
});
15271531
}
15281532

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

Lines changed: 13 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -772,8 +772,19 @@ Doacross make(const parser::OmpClause::Doacross &inp,
772772

773773
Enter make(const parser::OmpClause::Enter &inp,
774774
semantics::SemanticsContext &semaCtx) {
775-
// inp.v -> parser::OmpObjectList
776-
return Enter{makeObjects(/*List=*/inp.v, semaCtx)};
775+
// inp.v -> parser::OmpEnterClause
776+
CLAUSET_ENUM_CONVERT(
777+
convert, parser::OmpAutomapModifier::Value, Enter::Modifier,
778+
// clang-format off
779+
MS(Automap, Automap)
780+
// clang-format on);
781+
);
782+
auto &mods = semantics::OmpGetModifiers(inp.v);
783+
auto *mod = semantics::OmpGetUniqueModifier<parser::OmpAutomapModifier>(mods);
784+
auto &objList = std::get<parser::OmpObjectList>(inp.v.t);
785+
786+
return Enter{{/*Modifier=*/maybeApplyToV(convert, mod),
787+
/*List=*/makeObjects(objList, semaCtx)}};
777788
}
778789

779790
Exclusive make(const parser::OmpClause::Exclusive &inp,

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

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -449,6 +449,9 @@ TYPE_PARSER(construct<OmpAllocatorSimpleModifier>(scalarIntExpr))
449449
TYPE_PARSER(construct<OmpAlwaysModifier>( //
450450
"ALWAYS" >> pure(OmpAlwaysModifier::Value::Always)))
451451

452+
TYPE_PARSER(construct<OmpAutomapModifier>(
453+
"AUTOMAP" >> pure(OmpAutomapModifier::Value::Automap)))
454+
452455
TYPE_PARSER(construct<OmpChunkModifier>( //
453456
"SIMD" >> pure(OmpChunkModifier::Value::Simd)))
454457

@@ -601,6 +604,13 @@ TYPE_PARSER(sourced(construct<OmpDependClause::TaskDep::Modifier>(sourced(
601604
TYPE_PARSER(
602605
sourced(construct<OmpDeviceClause::Modifier>(Parser<OmpDeviceModifier>{})))
603606

607+
TYPE_PARSER(
608+
sourced(construct<OmpEnterClause::Modifier>(Parser<OmpAutomapModifier>{})))
609+
610+
TYPE_PARSER(construct<OmpEnterClause>(
611+
maybe(nonemptyList(Parser<OmpEnterClause::Modifier>{}) / ":"),
612+
Parser<OmpObjectList>{}))
613+
604614
TYPE_PARSER(sourced(construct<OmpFromClause::Modifier>(
605615
sourced(construct<OmpFromClause::Modifier>(Parser<OmpExpectation>{}) ||
606616
construct<OmpFromClause::Modifier>(Parser<OmpMapper>{}) ||
@@ -1023,7 +1033,7 @@ TYPE_PARSER( //
10231033
"DYNAMIC_ALLOCATORS" >>
10241034
construct<OmpClause>(construct<OmpClause::DynamicAllocators>()) ||
10251035
"ENTER" >> construct<OmpClause>(construct<OmpClause::Enter>(
1026-
parenthesized(Parser<OmpObjectList>{}))) ||
1036+
parenthesized(Parser<OmpEnterClause>{}))) ||
10271037
"EXCLUSIVE" >> construct<OmpClause>(construct<OmpClause::Exclusive>(
10281038
parenthesized(Parser<OmpObjectList>{}))) ||
10291039
"FAIL" >> construct<OmpClause>(construct<OmpClause::Fail>(

‎flang/lib/Parser/unparse.cpp‎

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2250,6 +2250,11 @@ class UnparseVisitor {
22502250
Walk(std::get<OmpObjectList>(x.t));
22512251
Walk(": ", std::get<std::optional<std::list<Modifier>>>(x.t));
22522252
}
2253+
void Unparse(const OmpEnterClause &x) {
2254+
using Modifier = OmpEnterClause::Modifier;
2255+
Walk(std::get<std::optional<std::list<Modifier>>>(x.t), ": ");
2256+
Walk(std::get<OmpObjectList>(x.t));
2257+
}
22532258
void Unparse(const OmpFromClause &x) {
22542259
using Modifier = OmpFromClause::Modifier;
22552260
Walk(std::get<std::optional<std::list<Modifier>>>(x.t), ": ");
@@ -2986,6 +2991,7 @@ class UnparseVisitor {
29862991
WALK_NESTED_ENUM(UseStmt, ModuleNature) // R1410
29872992
WALK_NESTED_ENUM(OmpAdjustArgsClause::OmpAdjustOp, Value) // OMP adjustop
29882993
WALK_NESTED_ENUM(OmpAtClause, ActionTime) // OMP at
2994+
WALK_NESTED_ENUM(OmpAutomapModifier, Value) // OMP automap-modifier
29892995
WALK_NESTED_ENUM(OmpBindClause, Binding) // OMP bind
29902996
WALK_NESTED_ENUM(OmpProcBindClause, AffinityPolicy) // OMP proc_bind
29912997
WALK_NESTED_ENUM(OmpDefaultClause, DataSharingAttribute) // OMP default

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

Lines changed: 19 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -1568,9 +1568,10 @@ void OmpStructureChecker::Leave(const parser::OpenMPDeclareTargetConstruct &x) {
15681568
},
15691569
[&](const parser::OmpClause::Enter &enterClause) {
15701570
enterClauseFound = true;
1571-
CheckSymbolNames(dir.source, enterClause.v);
1572-
CheckVarIsNotPartOfAnotherVar(dir.source, enterClause.v);
1573-
CheckThreadprivateOrDeclareTargetVar(enterClause.v);
1571+
auto &objList{std::get<parser::OmpObjectList>(enterClause.v.t)};
1572+
CheckSymbolNames(dir.source, objList);
1573+
CheckVarIsNotPartOfAnotherVar(dir.source, objList);
1574+
CheckThreadprivateOrDeclareTargetVar(objList);
15741575
},
15751576
[&](const parser::OmpClause::DeviceType &deviceTypeClause) {
15761577
deviceTypeClauseFound = true;
@@ -4028,7 +4029,11 @@ void OmpStructureChecker::Enter(const parser::OmpClause::HasDeviceAddr &x) {
40284029

40294030
void OmpStructureChecker::Enter(const parser::OmpClause::Enter &x) {
40304031
CheckAllowedClause(llvm::omp::Clause::OMPC_enter);
4031-
const parser::OmpObjectList &objList{x.v};
4032+
if (!OmpVerifyModifiers(
4033+
x.v, llvm::omp::OMPC_enter, GetContext().clauseSource, context_)) {
4034+
return;
4035+
}
4036+
const parser::OmpObjectList &objList{std::get<parser::OmpObjectList>(x.v.t)};
40324037
SymbolSourceMap symbols;
40334038
GetSymbolsInObjectList(objList, symbols);
40344039
for (const auto &[symbol, source] : symbols) {
@@ -4488,17 +4493,18 @@ const parser::OmpObjectList *OmpStructureChecker::GetOmpObjectList(
44884493
const parser::OmpClause &clause) {
44894494

44904495
// Clauses with OmpObjectList as its data member
4491-
using MemberObjectListClauses = std::tuple<parser::OmpClause::Copyprivate,
4492-
parser::OmpClause::Copyin, parser::OmpClause::Enter,
4493-
parser::OmpClause::Firstprivate, parser::OmpClause::Link,
4494-
parser::OmpClause::Private, parser::OmpClause::Shared,
4495-
parser::OmpClause::UseDevicePtr, parser::OmpClause::UseDeviceAddr>;
4496+
using MemberObjectListClauses =
4497+
std::tuple<parser::OmpClause::Copyprivate, parser::OmpClause::Copyin,
4498+
parser::OmpClause::Firstprivate, parser::OmpClause::Link,
4499+
parser::OmpClause::Private, parser::OmpClause::Shared,
4500+
parser::OmpClause::UseDevicePtr, parser::OmpClause::UseDeviceAddr>;
44964501

44974502
// Clauses with OmpObjectList in the tuple
4498-
using TupleObjectListClauses = std::tuple<parser::OmpClause::Aligned,
4499-
parser::OmpClause::Allocate, parser::OmpClause::From,
4500-
parser::OmpClause::Lastprivate, parser::OmpClause::Map,
4501-
parser::OmpClause::Reduction, parser::OmpClause::To>;
4503+
using TupleObjectListClauses =
4504+
std::tuple<parser::OmpClause::Aligned, parser::OmpClause::Allocate,
4505+
parser::OmpClause::From, parser::OmpClause::Lastprivate,
4506+
parser::OmpClause::Map, parser::OmpClause::Reduction,
4507+
parser::OmpClause::To, parser::OmpClause::Enter>;
45024508

45034509
// TODO:: Generate the tuples using TableGen.
45044510
// Handle other constructs with OmpObjectList such as OpenMPThreadprivate.

‎flang/lib/Semantics/openmp-modifiers.cpp‎

Lines changed: 16 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -156,6 +156,22 @@ const OmpModifierDescriptor &OmpGetDescriptor<parser::OmpAlwaysModifier>() {
156156
return desc;
157157
}
158158

159+
template <>
160+
const OmpModifierDescriptor &OmpGetDescriptor<parser::OmpAutomapModifier>() {
161+
static const OmpModifierDescriptor desc{
162+
/*name=*/"automap-modifier",
163+
/*props=*/
164+
{
165+
{60, {OmpProperty::Unique}},
166+
},
167+
/*clauses=*/
168+
{
169+
{60, {Clause::OMPC_enter}},
170+
},
171+
};
172+
return desc;
173+
}
174+
159175
template <>
160176
const OmpModifierDescriptor &OmpGetDescriptor<parser::OmpChunkModifier>() {
161177
static const OmpModifierDescriptor desc{

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

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2152,7 +2152,8 @@ bool OmpAttributeVisitor::Pre(const parser::OpenMPDeclareTargetConstruct &x) {
21522152
ResolveOmpObjectList(linkClause->v, Symbol::Flag::OmpDeclareTarget);
21532153
} else if (const auto *enterClause{
21542154
std::get_if<parser::OmpClause::Enter>(&clause.u)}) {
2155-
ResolveOmpObjectList(enterClause->v, Symbol::Flag::OmpDeclareTarget);
2155+
ResolveOmpObjectList(std::get<parser::OmpObjectList>(enterClause->v.t),
2156+
Symbol::Flag::OmpDeclareTarget);
21562157
}
21572158
}
21582159
}

0 commit comments

Comments
 (0)