1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889909192939495969798991001011021031041051061071081091101111121131141151161171181191201211221231241251261271281291301311321331341351361371381391401411421431441451461471481491501511521531541551561571581591601611621631641651661671681691701711721731741751761771781791801811821831841851861871881891901911921931941951961971981992002012022032042052062072082092102112122132142152162172182192202212222232242252262272282292302312322332342352362372382392402412422432442452462472482492502512522532542552562572582592602612622632642652662672682692702712722732742752762772782792802812822832842852862872882892902912922932942952962972982993003013023033043053063073083093103113123133143153163173183193203213223233243253263273283293303313323333343353363373383393403413423433443453463473483493503513523533543553563573583593603613623633643653663673683693703713723733743753763773783793803813823833843853863873883893903913923933943953963973983994004014024034044054064074084094104114124134144154164174184194204214224234244254264274284294304314324334344354364374384394404414424434444454464474484494504514524534544554564574584594604614624634644654664674684694704714724734744754764774784794804814824834844854864874884894904914924934944954964974984995005015025035045055065075085095105115125135145155165175185195205215225235245255265275285295305315325335345355365375385395405415425435445455465475485495505515525535545555565575585595605615625635645655665675685695705715725735745755765775785795805815825835845855865875885895905915925935945955965975985996006016026036046056066076086096106116126136146156166176186196206216226236246256266276286296306316326336346356366376386396406416426436446456466476486496506516526536546556566576586596606616626636646656666676686696706716726736746756766776786796806816826836846856866876886896906916926936946956966976986997007017027037047057067077087097107117127137147157167177187197207217227237247257267277287297307317327337347357367377387397407417427437447457467477487497507517527537547557567577587597607617627637647657667677687697707717727737747757767777787797807817827837847857867877887897907917927937947957967977987998008018028038048058068078088098108118128138148158168178188198208218228238248258268278288298308318328338348358368378388398408418428438448458468478488498508518528538548558568578588598608618628638648658668678688698708718728738748758768778788798808818828838848858868878888898908918928938948958968978988999009019029039049059069079089099109119129139149159169179189199209219229239249259269279289299309319329339349359369379389399409419429439449459469479489499509519529539549559569579589599609619629639649659669679689699709719729739749759769779789799809819829839849859869879889899909919929939949959969979989991000100110021003100410051006100710081009101010111012101310141015101610171018101910201021102210231024102510261027102810291030103110321033103410351036103710381039104010411042104310441045104610471048104910501051105210531054105510561057105810591060106110621063106410651066106710681069107010711072107310741075107610771078107910801081108210831084108510861087108810891090109110921093109410951096109710981099110011011102110311041105110611071108110911101111111211131114111511161117111811191120112111221123112411251126112711281129113011311132113311341135113611371138113911401141114211431144114511461147114811491150115111521153115411551156115711581159116011611162116311641165116611671168116911701171117211731174117511761177117811791180118111821183118411851186118711881189119011911192119311941195119611971198119912001201120212031204120512061207120812091210121112121213121412151216121712181219122012211222122312241225122612271228122912301231123212331234123512361237123812391240124112421243124412451246124712481249125012511252125312541255125612571258125912601261126212631264126512661267126812691270127112721273127412751276127712781279128012811282128312841285128612871288128912901291129212931294129512961297129812991300130113021303130413051306130713081309131013111312131313141315131613171318131913201321132213231324132513261327132813291330133113321333133413351336133713381339134013411342134313441345134613471348134913501351135213531354135513561357135813591360136113621363136413651366136713681369137013711372137313741375137613771378137913801381138213831384138513861387138813891390139113921393139413951396139713981399140014011402140314041405140614071408140914101411141214131414141514161417141814191420142114221423142414251426142714281429143014311432143314341435143614371438143914401441144214431444144514461447144814491450145114521453145414551456145714581459146014611462146314641465146614671468146914701471147214731474147514761477147814791480148114821483148414851486148714881489149014911492149314941495149614971498149915001501150215031504150515061507150815091510151115121513151415151516151715181519152015211522152315241525152615271528152915301531153215331534153515361537153815391540154115421543154415451546154715481549155015511552155315541555155615571558155915601561156215631564156515661567156815691570157115721573157415751576157715781579158015811582158315841585158615871588158915901591159215931594159515961597159815991600160116021603160416051606160716081609161016111612161316141615161616171618161916201621162216231624162516261627162816291630163116321633163416351636163716381639164016411642164316441645164616471648164916501651165216531654165516561657165816591660166116621663166416651666166716681669167016711672167316741675167616771678167916801681168216831684168516861687168816891690169116921693169416951696169716981699170017011702170317041705170617071708170917101711171217131714171517161717171817191720172117221723172417251726172717281729173017311732173317341735173617371738173917401741174217431744174517461747174817491750175117521753175417551756175717581759176017611762176317641765176617671768176917701771177217731774177517761777177817791780178117821783178417851786178717881789179017911792179317941795179617971798179918001801180218031804180518061807180818091810181118121813181418151816181718181819182018211822182318241825182618271828182918301831183218331834183518361837183818391840184118421843184418451846184718481849185018511852185318541855185618571858185918601861186218631864186518661867186818691870187118721873187418751876187718781879188018811882188318841885188618871888188918901891189218931894189518961897189818991900190119021903190419051906190719081909191019111912191319141915191619171918191919201921192219231924192519261927192819291930193119321933193419351936193719381939194019411942194319441945194619471948194919501951195219531954195519561957195819591960196119621963196419651966196719681969197019711972197319741975197619771978197919801981198219831984198519861987198819891990199119921993199419951996199719981999200020012002200320042005200620072008200920102011201220132014201520162017201820192020202120222023202420252026202720282029203020312032203320342035203620372038203920402041204220432044204520462047204820492050205120522053205420552056205720582059206020612062206320642065206620672068206920702071207220732074207520762077207820792080208120822083208420852086208720882089209020912092209320942095209620972098209921002101210221032104210521062107210821092110211121122113211421152116211721182119212021212122212321242125212621272128212921302131213221332134 |
- //===--- StmtOpenMP.cpp - Classes for OpenMP directives -------------------===//
- //
- // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
- // See https://llvm.org/LICENSE.txt for license information.
- // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
- //
- //===----------------------------------------------------------------------===//
- //
- // This file implements the subclesses of Stmt class declared in StmtOpenMP.h
- //
- //===----------------------------------------------------------------------===//
- #include "clang/AST/ASTContext.h"
- #include "clang/AST/StmtOpenMP.h"
- using namespace clang;
- using namespace llvm::omp;
- size_t OMPChildren::size(unsigned NumClauses, bool HasAssociatedStmt,
- unsigned NumChildren) {
- return llvm::alignTo(
- totalSizeToAlloc<OMPClause *, Stmt *>(
- NumClauses, NumChildren + (HasAssociatedStmt ? 1 : 0)),
- alignof(OMPChildren));
- }
- void OMPChildren::setClauses(ArrayRef<OMPClause *> Clauses) {
- assert(Clauses.size() == NumClauses &&
- "Number of clauses is not the same as the preallocated buffer");
- llvm::copy(Clauses, getTrailingObjects<OMPClause *>());
- }
- MutableArrayRef<Stmt *> OMPChildren::getChildren() {
- return llvm::makeMutableArrayRef(getTrailingObjects<Stmt *>(), NumChildren);
- }
- OMPChildren *OMPChildren::Create(void *Mem, ArrayRef<OMPClause *> Clauses) {
- auto *Data = CreateEmpty(Mem, Clauses.size());
- Data->setClauses(Clauses);
- return Data;
- }
- OMPChildren *OMPChildren::Create(void *Mem, ArrayRef<OMPClause *> Clauses,
- Stmt *S, unsigned NumChildren) {
- auto *Data = CreateEmpty(Mem, Clauses.size(), S, NumChildren);
- Data->setClauses(Clauses);
- if (S)
- Data->setAssociatedStmt(S);
- return Data;
- }
- OMPChildren *OMPChildren::CreateEmpty(void *Mem, unsigned NumClauses,
- bool HasAssociatedStmt,
- unsigned NumChildren) {
- return new (Mem) OMPChildren(NumClauses, NumChildren, HasAssociatedStmt);
- }
- bool OMPExecutableDirective::isStandaloneDirective() const {
- // Special case: 'omp target enter data', 'omp target exit data',
- // 'omp target update' are stand-alone directives, but for implementation
- // reasons they have empty synthetic structured block, to simplify codegen.
- if (isa<OMPTargetEnterDataDirective>(this) ||
- isa<OMPTargetExitDataDirective>(this) ||
- isa<OMPTargetUpdateDirective>(this))
- return true;
- return !hasAssociatedStmt();
- }
- Stmt *OMPExecutableDirective::getStructuredBlock() {
- assert(!isStandaloneDirective() &&
- "Standalone Executable Directives don't have Structured Blocks.");
- if (auto *LD = dyn_cast<OMPLoopDirective>(this))
- return LD->getBody();
- return getRawStmt();
- }
- Stmt *
- OMPLoopBasedDirective::tryToFindNextInnerLoop(Stmt *CurStmt,
- bool TryImperfectlyNestedLoops) {
- Stmt *OrigStmt = CurStmt;
- CurStmt = CurStmt->IgnoreContainers();
- // Additional work for imperfectly nested loops, introduced in OpenMP 5.0.
- if (TryImperfectlyNestedLoops) {
- if (auto *CS = dyn_cast<CompoundStmt>(CurStmt)) {
- CurStmt = nullptr;
- SmallVector<CompoundStmt *, 4> Statements(1, CS);
- SmallVector<CompoundStmt *, 4> NextStatements;
- while (!Statements.empty()) {
- CS = Statements.pop_back_val();
- if (!CS)
- continue;
- for (Stmt *S : CS->body()) {
- if (!S)
- continue;
- if (auto *CanonLoop = dyn_cast<OMPCanonicalLoop>(S))
- S = CanonLoop->getLoopStmt();
- if (isa<ForStmt>(S) || isa<CXXForRangeStmt>(S) ||
- (isa<OMPLoopBasedDirective>(S) && !isa<OMPLoopDirective>(S))) {
- // Only single loop construct is allowed.
- if (CurStmt) {
- CurStmt = OrigStmt;
- break;
- }
- CurStmt = S;
- continue;
- }
- S = S->IgnoreContainers();
- if (auto *InnerCS = dyn_cast_or_null<CompoundStmt>(S))
- NextStatements.push_back(InnerCS);
- }
- if (Statements.empty()) {
- // Found single inner loop or multiple loops - exit.
- if (CurStmt)
- break;
- Statements.swap(NextStatements);
- }
- }
- if (!CurStmt)
- CurStmt = OrigStmt;
- }
- }
- return CurStmt;
- }
- bool OMPLoopBasedDirective::doForAllLoops(
- Stmt *CurStmt, bool TryImperfectlyNestedLoops, unsigned NumLoops,
- llvm::function_ref<bool(unsigned, Stmt *)> Callback,
- llvm::function_ref<void(OMPLoopTransformationDirective *)>
- OnTransformationCallback) {
- CurStmt = CurStmt->IgnoreContainers();
- for (unsigned Cnt = 0; Cnt < NumLoops; ++Cnt) {
- while (true) {
- auto *Dir = dyn_cast<OMPLoopTransformationDirective>(CurStmt);
- if (!Dir)
- break;
- OnTransformationCallback(Dir);
- Stmt *TransformedStmt = Dir->getTransformedStmt();
- if (!TransformedStmt) {
- unsigned NumGeneratedLoops = Dir->getNumGeneratedLoops();
- if (NumGeneratedLoops == 0) {
- // May happen if the loop transformation does not result in a
- // generated loop (such as full unrolling).
- break;
- }
- if (NumGeneratedLoops > 0) {
- // The loop transformation construct has generated loops, but these
- // may not have been generated yet due to being in a dependent
- // context.
- return true;
- }
- }
- CurStmt = TransformedStmt;
- }
- if (auto *CanonLoop = dyn_cast<OMPCanonicalLoop>(CurStmt))
- CurStmt = CanonLoop->getLoopStmt();
- if (Callback(Cnt, CurStmt))
- return false;
- // Move on to the next nested for loop, or to the loop body.
- // OpenMP [2.8.1, simd construct, Restrictions]
- // All loops associated with the construct must be perfectly nested; that
- // is, there must be no intervening code nor any OpenMP directive between
- // any two loops.
- if (auto *For = dyn_cast<ForStmt>(CurStmt)) {
- CurStmt = For->getBody();
- } else {
- assert(isa<CXXForRangeStmt>(CurStmt) &&
- "Expected canonical for or range-based for loops.");
- CurStmt = cast<CXXForRangeStmt>(CurStmt)->getBody();
- }
- CurStmt = OMPLoopBasedDirective::tryToFindNextInnerLoop(
- CurStmt, TryImperfectlyNestedLoops);
- }
- return true;
- }
- void OMPLoopBasedDirective::doForAllLoopsBodies(
- Stmt *CurStmt, bool TryImperfectlyNestedLoops, unsigned NumLoops,
- llvm::function_ref<void(unsigned, Stmt *, Stmt *)> Callback) {
- bool Res = OMPLoopBasedDirective::doForAllLoops(
- CurStmt, TryImperfectlyNestedLoops, NumLoops,
- [Callback](unsigned Cnt, Stmt *Loop) {
- Stmt *Body = nullptr;
- if (auto *For = dyn_cast<ForStmt>(Loop)) {
- Body = For->getBody();
- } else {
- assert(isa<CXXForRangeStmt>(Loop) &&
- "Expected canonical for or range-based for loops.");
- Body = cast<CXXForRangeStmt>(Loop)->getBody();
- }
- if (auto *CanonLoop = dyn_cast<OMPCanonicalLoop>(Body))
- Body = CanonLoop->getLoopStmt();
- Callback(Cnt, Loop, Body);
- return false;
- });
- assert(Res && "Expected only loops");
- (void)Res;
- }
- Stmt *OMPLoopDirective::getBody() {
- // This relies on the loop form is already checked by Sema.
- Stmt *Body = nullptr;
- OMPLoopBasedDirective::doForAllLoopsBodies(
- Data->getRawStmt(), /*TryImperfectlyNestedLoops=*/true,
- NumAssociatedLoops,
- [&Body](unsigned, Stmt *, Stmt *BodyStmt) { Body = BodyStmt; });
- return Body;
- }
- void OMPLoopDirective::setCounters(ArrayRef<Expr *> A) {
- assert(A.size() == getLoopsNumber() &&
- "Number of loop counters is not the same as the collapsed number");
- llvm::copy(A, getCounters().begin());
- }
- void OMPLoopDirective::setPrivateCounters(ArrayRef<Expr *> A) {
- assert(A.size() == getLoopsNumber() && "Number of loop private counters "
- "is not the same as the collapsed "
- "number");
- llvm::copy(A, getPrivateCounters().begin());
- }
- void OMPLoopDirective::setInits(ArrayRef<Expr *> A) {
- assert(A.size() == getLoopsNumber() &&
- "Number of counter inits is not the same as the collapsed number");
- llvm::copy(A, getInits().begin());
- }
- void OMPLoopDirective::setUpdates(ArrayRef<Expr *> A) {
- assert(A.size() == getLoopsNumber() &&
- "Number of counter updates is not the same as the collapsed number");
- llvm::copy(A, getUpdates().begin());
- }
- void OMPLoopDirective::setFinals(ArrayRef<Expr *> A) {
- assert(A.size() == getLoopsNumber() &&
- "Number of counter finals is not the same as the collapsed number");
- llvm::copy(A, getFinals().begin());
- }
- void OMPLoopDirective::setDependentCounters(ArrayRef<Expr *> A) {
- assert(
- A.size() == getLoopsNumber() &&
- "Number of dependent counters is not the same as the collapsed number");
- llvm::copy(A, getDependentCounters().begin());
- }
- void OMPLoopDirective::setDependentInits(ArrayRef<Expr *> A) {
- assert(A.size() == getLoopsNumber() &&
- "Number of dependent inits is not the same as the collapsed number");
- llvm::copy(A, getDependentInits().begin());
- }
- void OMPLoopDirective::setFinalsConditions(ArrayRef<Expr *> A) {
- assert(A.size() == getLoopsNumber() &&
- "Number of finals conditions is not the same as the collapsed number");
- llvm::copy(A, getFinalsConditions().begin());
- }
- OMPMetaDirective *OMPMetaDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt, Stmt *IfStmt) {
- auto *Dir = createDirective<OMPMetaDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/1, StartLoc, EndLoc);
- Dir->setIfStmt(IfStmt);
- return Dir;
- }
- OMPMetaDirective *OMPMetaDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPMetaDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true,
- /*NumChildren=*/1);
- }
- OMPParallelDirective *OMPParallelDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt, Expr *TaskRedRef,
- bool HasCancel) {
- auto *Dir = createDirective<OMPParallelDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/1, StartLoc, EndLoc);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPParallelDirective *OMPParallelDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPParallelDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true,
- /*NumChildren=*/1);
- }
- OMPSimdDirective *
- OMPSimdDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc, unsigned CollapsedNum,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPSimdDirective>(
- C, Clauses, AssociatedStmt, numLoopChildren(CollapsedNum, OMPD_simd),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPSimdDirective *OMPSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_simd), CollapsedNum);
- }
- OMPForDirective *OMPForDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, Expr *TaskRedRef, bool HasCancel) {
- auto *Dir = createDirective<OMPForDirective>(
- C, Clauses, AssociatedStmt, numLoopChildren(CollapsedNum, OMPD_for) + 1,
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- Stmt *OMPLoopTransformationDirective::getTransformedStmt() const {
- switch (getStmtClass()) {
- #define STMT(CLASS, PARENT)
- #define ABSTRACT_STMT(CLASS)
- #define OMPLOOPTRANSFORMATIONDIRECTIVE(CLASS, PARENT) \
- case Stmt::CLASS##Class: \
- return static_cast<const CLASS *>(this)->getTransformedStmt();
- #include "clang/AST/StmtNodes.inc"
- default:
- llvm_unreachable("Not a loop transformation");
- }
- }
- Stmt *OMPLoopTransformationDirective::getPreInits() const {
- switch (getStmtClass()) {
- #define STMT(CLASS, PARENT)
- #define ABSTRACT_STMT(CLASS)
- #define OMPLOOPTRANSFORMATIONDIRECTIVE(CLASS, PARENT) \
- case Stmt::CLASS##Class: \
- return static_cast<const CLASS *>(this)->getPreInits();
- #include "clang/AST/StmtNodes.inc"
- default:
- llvm_unreachable("Not a loop transformation");
- }
- }
- OMPForDirective *OMPForDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPForDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_for) + 1, CollapsedNum);
- }
- OMPTileDirective *
- OMPTileDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc, ArrayRef<OMPClause *> Clauses,
- unsigned NumLoops, Stmt *AssociatedStmt,
- Stmt *TransformedStmt, Stmt *PreInits) {
- OMPTileDirective *Dir = createDirective<OMPTileDirective>(
- C, Clauses, AssociatedStmt, TransformedStmtOffset + 1, StartLoc, EndLoc,
- NumLoops);
- Dir->setTransformedStmt(TransformedStmt);
- Dir->setPreInits(PreInits);
- return Dir;
- }
- OMPTileDirective *OMPTileDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned NumLoops) {
- return createEmptyDirective<OMPTileDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true, TransformedStmtOffset + 1,
- SourceLocation(), SourceLocation(), NumLoops);
- }
- OMPUnrollDirective *
- OMPUnrollDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc, ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt, unsigned NumGeneratedLoops,
- Stmt *TransformedStmt, Stmt *PreInits) {
- assert(NumGeneratedLoops <= 1 && "Unrolling generates at most one loop");
- auto *Dir = createDirective<OMPUnrollDirective>(
- C, Clauses, AssociatedStmt, TransformedStmtOffset + 1, StartLoc, EndLoc);
- Dir->setNumGeneratedLoops(NumGeneratedLoops);
- Dir->setTransformedStmt(TransformedStmt);
- Dir->setPreInits(PreInits);
- return Dir;
- }
- OMPUnrollDirective *OMPUnrollDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses) {
- return createEmptyDirective<OMPUnrollDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true, TransformedStmtOffset + 1,
- SourceLocation(), SourceLocation());
- }
- OMPForSimdDirective *
- OMPForSimdDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc, unsigned CollapsedNum,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPForSimdDirective>(
- C, Clauses, AssociatedStmt, numLoopChildren(CollapsedNum, OMPD_for_simd),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPForSimdDirective *OMPForSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPForSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_for_simd), CollapsedNum);
- }
- OMPSectionsDirective *OMPSectionsDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt, Expr *TaskRedRef,
- bool HasCancel) {
- auto *Dir = createDirective<OMPSectionsDirective>(C, Clauses, AssociatedStmt,
- /*NumChildren=*/1, StartLoc,
- EndLoc);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPSectionsDirective *OMPSectionsDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPSectionsDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true,
- /*NumChildren=*/1);
- }
- OMPSectionDirective *OMPSectionDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- Stmt *AssociatedStmt,
- bool HasCancel) {
- auto *Dir =
- createDirective<OMPSectionDirective>(C, llvm::None, AssociatedStmt,
- /*NumChildren=*/0, StartLoc, EndLoc);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPSectionDirective *OMPSectionDirective::CreateEmpty(const ASTContext &C,
- EmptyShell) {
- return createEmptyDirective<OMPSectionDirective>(C, /*NumClauses=*/0,
- /*HasAssociatedStmt=*/true);
- }
- OMPSingleDirective *OMPSingleDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt) {
- return createDirective<OMPSingleDirective>(C, Clauses, AssociatedStmt,
- /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPSingleDirective *OMPSingleDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPSingleDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true);
- }
- OMPMasterDirective *OMPMasterDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- Stmt *AssociatedStmt) {
- return createDirective<OMPMasterDirective>(C, llvm::None, AssociatedStmt,
- /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPMasterDirective *OMPMasterDirective::CreateEmpty(const ASTContext &C,
- EmptyShell) {
- return createEmptyDirective<OMPMasterDirective>(C, /*NumClauses=*/0,
- /*HasAssociatedStmt=*/true);
- }
- OMPCriticalDirective *OMPCriticalDirective::Create(
- const ASTContext &C, const DeclarationNameInfo &Name,
- SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
- return createDirective<OMPCriticalDirective>(C, Clauses, AssociatedStmt,
- /*NumChildren=*/0, Name,
- StartLoc, EndLoc);
- }
- OMPCriticalDirective *OMPCriticalDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPCriticalDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true);
- }
- OMPParallelForDirective *OMPParallelForDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, Expr *TaskRedRef, bool HasCancel) {
- auto *Dir = createDirective<OMPParallelForDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_parallel_for) + 1, StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPParallelForDirective *
- OMPParallelForDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPParallelForDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_parallel_for) + 1, CollapsedNum);
- }
- OMPParallelForSimdDirective *OMPParallelForSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPParallelForSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_parallel_for_simd), StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPParallelForSimdDirective *
- OMPParallelForSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPParallelForSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_parallel_for_simd), CollapsedNum);
- }
- OMPParallelMasterDirective *OMPParallelMasterDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt, Expr *TaskRedRef) {
- auto *Dir = createDirective<OMPParallelMasterDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/1, StartLoc, EndLoc);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- return Dir;
- }
- OMPParallelMasterDirective *
- OMPParallelMasterDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses, EmptyShell) {
- return createEmptyDirective<OMPParallelMasterDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true, /*NumChildren=*/1);
- }
- OMPParallelSectionsDirective *OMPParallelSectionsDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt, Expr *TaskRedRef,
- bool HasCancel) {
- auto *Dir = createDirective<OMPParallelSectionsDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/1, StartLoc, EndLoc);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPParallelSectionsDirective *
- OMPParallelSectionsDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses, EmptyShell) {
- return createEmptyDirective<OMPParallelSectionsDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true, /*NumChildren=*/1);
- }
- OMPTaskDirective *
- OMPTaskDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc, ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt, bool HasCancel) {
- auto *Dir = createDirective<OMPTaskDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/0, StartLoc, EndLoc);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPTaskDirective *OMPTaskDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPTaskDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true);
- }
- OMPTaskyieldDirective *OMPTaskyieldDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (C) OMPTaskyieldDirective(StartLoc, EndLoc);
- }
- OMPTaskyieldDirective *OMPTaskyieldDirective::CreateEmpty(const ASTContext &C,
- EmptyShell) {
- return new (C) OMPTaskyieldDirective();
- }
- OMPBarrierDirective *OMPBarrierDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc) {
- return new (C) OMPBarrierDirective(StartLoc, EndLoc);
- }
- OMPBarrierDirective *OMPBarrierDirective::CreateEmpty(const ASTContext &C,
- EmptyShell) {
- return new (C) OMPBarrierDirective();
- }
- OMPTaskwaitDirective *
- OMPTaskwaitDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses) {
- return createDirective<OMPTaskwaitDirective>(
- C, Clauses, /*AssociatedStmt=*/nullptr, /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPTaskwaitDirective *OMPTaskwaitDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPTaskwaitDirective>(C, NumClauses);
- }
- OMPTaskgroupDirective *OMPTaskgroupDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt, Expr *ReductionRef) {
- auto *Dir = createDirective<OMPTaskgroupDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/1, StartLoc, EndLoc);
- Dir->setReductionRef(ReductionRef);
- return Dir;
- }
- OMPTaskgroupDirective *OMPTaskgroupDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPTaskgroupDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true, /*NumChildren=*/1);
- }
- OMPCancellationPointDirective *OMPCancellationPointDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- OpenMPDirectiveKind CancelRegion) {
- auto *Dir = new (C) OMPCancellationPointDirective(StartLoc, EndLoc);
- Dir->setCancelRegion(CancelRegion);
- return Dir;
- }
- OMPCancellationPointDirective *
- OMPCancellationPointDirective::CreateEmpty(const ASTContext &C, EmptyShell) {
- return new (C) OMPCancellationPointDirective();
- }
- OMPCancelDirective *
- OMPCancelDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc, ArrayRef<OMPClause *> Clauses,
- OpenMPDirectiveKind CancelRegion) {
- auto *Dir = createDirective<OMPCancelDirective>(
- C, Clauses, /*AssociatedStmt=*/nullptr, /*NumChildren=*/0, StartLoc,
- EndLoc);
- Dir->setCancelRegion(CancelRegion);
- return Dir;
- }
- OMPCancelDirective *OMPCancelDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPCancelDirective>(C, NumClauses);
- }
- OMPFlushDirective *OMPFlushDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses) {
- return createDirective<OMPFlushDirective>(
- C, Clauses, /*AssociatedStmt=*/nullptr, /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPFlushDirective *OMPFlushDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPFlushDirective>(C, NumClauses);
- }
- OMPDepobjDirective *OMPDepobjDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses) {
- return createDirective<OMPDepobjDirective>(
- C, Clauses, /*AssociatedStmt=*/nullptr,
- /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPDepobjDirective *OMPDepobjDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPDepobjDirective>(C, NumClauses);
- }
- OMPScanDirective *OMPScanDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses) {
- return createDirective<OMPScanDirective>(C, Clauses,
- /*AssociatedStmt=*/nullptr,
- /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPScanDirective *OMPScanDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPScanDirective>(C, NumClauses);
- }
- OMPOrderedDirective *OMPOrderedDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt) {
- return createDirective<OMPOrderedDirective>(
- C, Clauses, cast_or_null<CapturedStmt>(AssociatedStmt),
- /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPOrderedDirective *OMPOrderedDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- bool IsStandalone,
- EmptyShell) {
- return createEmptyDirective<OMPOrderedDirective>(C, NumClauses,
- !IsStandalone);
- }
- OMPAtomicDirective *OMPAtomicDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt, Expr *X, Expr *V,
- Expr *E, Expr *UE, bool IsXLHSInRHSPart, bool IsPostfixUpdate) {
- auto *Dir = createDirective<OMPAtomicDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/4, StartLoc, EndLoc);
- Dir->setX(X);
- Dir->setV(V);
- Dir->setExpr(E);
- Dir->setUpdateExpr(UE);
- Dir->IsXLHSInRHSPart = IsXLHSInRHSPart;
- Dir->IsPostfixUpdate = IsPostfixUpdate;
- return Dir;
- }
- OMPAtomicDirective *OMPAtomicDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPAtomicDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true, /*NumChildren=*/4);
- }
- OMPTargetDirective *OMPTargetDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt) {
- return createDirective<OMPTargetDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPTargetDirective *OMPTargetDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPTargetDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true);
- }
- OMPTargetParallelDirective *OMPTargetParallelDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt, Expr *TaskRedRef,
- bool HasCancel) {
- auto *Dir = createDirective<OMPTargetParallelDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/1, StartLoc, EndLoc);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPTargetParallelDirective *
- OMPTargetParallelDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses, EmptyShell) {
- return createEmptyDirective<OMPTargetParallelDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true, /*NumChildren=*/1);
- }
- OMPTargetParallelForDirective *OMPTargetParallelForDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, Expr *TaskRedRef, bool HasCancel) {
- auto *Dir = createDirective<OMPTargetParallelForDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_target_parallel_for) + 1, StartLoc,
- EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPTargetParallelForDirective *
- OMPTargetParallelForDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPTargetParallelForDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_target_parallel_for) + 1,
- CollapsedNum);
- }
- OMPTargetDataDirective *OMPTargetDataDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
- return createDirective<OMPTargetDataDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPTargetDataDirective *OMPTargetDataDirective::CreateEmpty(const ASTContext &C,
- unsigned N,
- EmptyShell) {
- return createEmptyDirective<OMPTargetDataDirective>(
- C, N, /*HasAssociatedStmt=*/true);
- }
- OMPTargetEnterDataDirective *OMPTargetEnterDataDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
- return createDirective<OMPTargetEnterDataDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPTargetEnterDataDirective *
- OMPTargetEnterDataDirective::CreateEmpty(const ASTContext &C, unsigned N,
- EmptyShell) {
- return createEmptyDirective<OMPTargetEnterDataDirective>(
- C, N, /*HasAssociatedStmt=*/true);
- }
- OMPTargetExitDataDirective *OMPTargetExitDataDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
- return createDirective<OMPTargetExitDataDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPTargetExitDataDirective *
- OMPTargetExitDataDirective::CreateEmpty(const ASTContext &C, unsigned N,
- EmptyShell) {
- return createEmptyDirective<OMPTargetExitDataDirective>(
- C, N, /*HasAssociatedStmt=*/true);
- }
- OMPTeamsDirective *OMPTeamsDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt) {
- return createDirective<OMPTeamsDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/0, StartLoc, EndLoc);
- }
- OMPTeamsDirective *OMPTeamsDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPTeamsDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true);
- }
- OMPTaskLoopDirective *OMPTaskLoopDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, bool HasCancel) {
- auto *Dir = createDirective<OMPTaskLoopDirective>(
- C, Clauses, AssociatedStmt, numLoopChildren(CollapsedNum, OMPD_taskloop),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPTaskLoopDirective *OMPTaskLoopDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTaskLoopDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_taskloop), CollapsedNum);
- }
- OMPTaskLoopSimdDirective *OMPTaskLoopSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTaskLoopSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_taskloop_simd), StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPTaskLoopSimdDirective *
- OMPTaskLoopSimdDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPTaskLoopSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_taskloop_simd), CollapsedNum);
- }
- OMPMasterTaskLoopDirective *OMPMasterTaskLoopDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, bool HasCancel) {
- auto *Dir = createDirective<OMPMasterTaskLoopDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_master_taskloop), StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPMasterTaskLoopDirective *
- OMPMasterTaskLoopDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPMasterTaskLoopDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_master_taskloop), CollapsedNum);
- }
- OMPMasterTaskLoopSimdDirective *OMPMasterTaskLoopSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPMasterTaskLoopSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_master_taskloop_simd), StartLoc,
- EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPMasterTaskLoopSimdDirective *
- OMPMasterTaskLoopSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPMasterTaskLoopSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_master_taskloop_simd), CollapsedNum);
- }
- OMPParallelMasterTaskLoopDirective *OMPParallelMasterTaskLoopDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, bool HasCancel) {
- auto *Dir = createDirective<OMPParallelMasterTaskLoopDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_parallel_master_taskloop), StartLoc,
- EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setHasCancel(HasCancel);
- return Dir;
- }
- OMPParallelMasterTaskLoopDirective *
- OMPParallelMasterTaskLoopDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPParallelMasterTaskLoopDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_parallel_master_taskloop),
- CollapsedNum);
- }
- OMPParallelMasterTaskLoopSimdDirective *
- OMPParallelMasterTaskLoopSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPParallelMasterTaskLoopSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_parallel_master_taskloop_simd),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPParallelMasterTaskLoopSimdDirective *
- OMPParallelMasterTaskLoopSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPParallelMasterTaskLoopSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_parallel_master_taskloop_simd),
- CollapsedNum);
- }
- OMPDistributeDirective *OMPDistributeDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPDistributeDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_distribute), StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPDistributeDirective *
- OMPDistributeDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPDistributeDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_distribute), CollapsedNum);
- }
- OMPTargetUpdateDirective *OMPTargetUpdateDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
- return createDirective<OMPTargetUpdateDirective>(C, Clauses, AssociatedStmt,
- /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPTargetUpdateDirective *
- OMPTargetUpdateDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPTargetUpdateDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true);
- }
- OMPDistributeParallelForDirective *OMPDistributeParallelForDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, Expr *TaskRedRef, bool HasCancel) {
- auto *Dir = createDirective<OMPDistributeParallelForDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_distribute_parallel_for) + 1, StartLoc,
- EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setPrevLowerBoundVariable(Exprs.PrevLB);
- Dir->setPrevUpperBoundVariable(Exprs.PrevUB);
- Dir->setDistInc(Exprs.DistInc);
- Dir->setPrevEnsureUpperBound(Exprs.PrevEUB);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setCombinedLowerBoundVariable(Exprs.DistCombinedFields.LB);
- Dir->setCombinedUpperBoundVariable(Exprs.DistCombinedFields.UB);
- Dir->setCombinedEnsureUpperBound(Exprs.DistCombinedFields.EUB);
- Dir->setCombinedInit(Exprs.DistCombinedFields.Init);
- Dir->setCombinedCond(Exprs.DistCombinedFields.Cond);
- Dir->setCombinedNextLowerBound(Exprs.DistCombinedFields.NLB);
- Dir->setCombinedNextUpperBound(Exprs.DistCombinedFields.NUB);
- Dir->setCombinedDistCond(Exprs.DistCombinedFields.DistCond);
- Dir->setCombinedParForInDistCond(Exprs.DistCombinedFields.ParForInDistCond);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->HasCancel = HasCancel;
- return Dir;
- }
- OMPDistributeParallelForDirective *
- OMPDistributeParallelForDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPDistributeParallelForDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_distribute_parallel_for) + 1,
- CollapsedNum);
- }
- OMPDistributeParallelForSimdDirective *
- OMPDistributeParallelForSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPDistributeParallelForSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_distribute_parallel_for_simd),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setPrevLowerBoundVariable(Exprs.PrevLB);
- Dir->setPrevUpperBoundVariable(Exprs.PrevUB);
- Dir->setDistInc(Exprs.DistInc);
- Dir->setPrevEnsureUpperBound(Exprs.PrevEUB);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setCombinedLowerBoundVariable(Exprs.DistCombinedFields.LB);
- Dir->setCombinedUpperBoundVariable(Exprs.DistCombinedFields.UB);
- Dir->setCombinedEnsureUpperBound(Exprs.DistCombinedFields.EUB);
- Dir->setCombinedInit(Exprs.DistCombinedFields.Init);
- Dir->setCombinedCond(Exprs.DistCombinedFields.Cond);
- Dir->setCombinedNextLowerBound(Exprs.DistCombinedFields.NLB);
- Dir->setCombinedNextUpperBound(Exprs.DistCombinedFields.NUB);
- Dir->setCombinedDistCond(Exprs.DistCombinedFields.DistCond);
- Dir->setCombinedParForInDistCond(Exprs.DistCombinedFields.ParForInDistCond);
- return Dir;
- }
- OMPDistributeParallelForSimdDirective *
- OMPDistributeParallelForSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPDistributeParallelForSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_distribute_parallel_for_simd),
- CollapsedNum);
- }
- OMPDistributeSimdDirective *OMPDistributeSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPDistributeSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_distribute_simd), StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPDistributeSimdDirective *
- OMPDistributeSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPDistributeSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_distribute_simd), CollapsedNum);
- }
- OMPTargetParallelForSimdDirective *OMPTargetParallelForSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTargetParallelForSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_target_parallel_for_simd), StartLoc,
- EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPTargetParallelForSimdDirective *
- OMPTargetParallelForSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTargetParallelForSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_target_parallel_for_simd),
- CollapsedNum);
- }
- OMPTargetSimdDirective *
- OMPTargetSimdDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc, unsigned CollapsedNum,
- ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt, const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTargetSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_target_simd), StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPTargetSimdDirective *
- OMPTargetSimdDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPTargetSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_target_simd), CollapsedNum);
- }
- OMPTeamsDistributeDirective *OMPTeamsDistributeDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTeamsDistributeDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute), StartLoc, EndLoc,
- CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPTeamsDistributeDirective *
- OMPTeamsDistributeDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPTeamsDistributeDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute), CollapsedNum);
- }
- OMPTeamsDistributeSimdDirective *OMPTeamsDistributeSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTeamsDistributeSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute_simd), StartLoc,
- EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPTeamsDistributeSimdDirective *OMPTeamsDistributeSimdDirective::CreateEmpty(
- const ASTContext &C, unsigned NumClauses, unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTeamsDistributeSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute_simd), CollapsedNum);
- }
- OMPTeamsDistributeParallelForSimdDirective *
- OMPTeamsDistributeParallelForSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTeamsDistributeParallelForSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute_parallel_for_simd),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setPrevLowerBoundVariable(Exprs.PrevLB);
- Dir->setPrevUpperBoundVariable(Exprs.PrevUB);
- Dir->setDistInc(Exprs.DistInc);
- Dir->setPrevEnsureUpperBound(Exprs.PrevEUB);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setCombinedLowerBoundVariable(Exprs.DistCombinedFields.LB);
- Dir->setCombinedUpperBoundVariable(Exprs.DistCombinedFields.UB);
- Dir->setCombinedEnsureUpperBound(Exprs.DistCombinedFields.EUB);
- Dir->setCombinedInit(Exprs.DistCombinedFields.Init);
- Dir->setCombinedCond(Exprs.DistCombinedFields.Cond);
- Dir->setCombinedNextLowerBound(Exprs.DistCombinedFields.NLB);
- Dir->setCombinedNextUpperBound(Exprs.DistCombinedFields.NUB);
- Dir->setCombinedDistCond(Exprs.DistCombinedFields.DistCond);
- Dir->setCombinedParForInDistCond(Exprs.DistCombinedFields.ParForInDistCond);
- return Dir;
- }
- OMPTeamsDistributeParallelForSimdDirective *
- OMPTeamsDistributeParallelForSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTeamsDistributeParallelForSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute_parallel_for_simd),
- CollapsedNum);
- }
- OMPTeamsDistributeParallelForDirective *
- OMPTeamsDistributeParallelForDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, Expr *TaskRedRef, bool HasCancel) {
- auto *Dir = createDirective<OMPTeamsDistributeParallelForDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute_parallel_for) + 1,
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setPrevLowerBoundVariable(Exprs.PrevLB);
- Dir->setPrevUpperBoundVariable(Exprs.PrevUB);
- Dir->setDistInc(Exprs.DistInc);
- Dir->setPrevEnsureUpperBound(Exprs.PrevEUB);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setCombinedLowerBoundVariable(Exprs.DistCombinedFields.LB);
- Dir->setCombinedUpperBoundVariable(Exprs.DistCombinedFields.UB);
- Dir->setCombinedEnsureUpperBound(Exprs.DistCombinedFields.EUB);
- Dir->setCombinedInit(Exprs.DistCombinedFields.Init);
- Dir->setCombinedCond(Exprs.DistCombinedFields.Cond);
- Dir->setCombinedNextLowerBound(Exprs.DistCombinedFields.NLB);
- Dir->setCombinedNextUpperBound(Exprs.DistCombinedFields.NUB);
- Dir->setCombinedDistCond(Exprs.DistCombinedFields.DistCond);
- Dir->setCombinedParForInDistCond(Exprs.DistCombinedFields.ParForInDistCond);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->HasCancel = HasCancel;
- return Dir;
- }
- OMPTeamsDistributeParallelForDirective *
- OMPTeamsDistributeParallelForDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTeamsDistributeParallelForDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_teams_distribute_parallel_for) + 1,
- CollapsedNum);
- }
- OMPTargetTeamsDirective *OMPTargetTeamsDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt) {
- return createDirective<OMPTargetTeamsDirective>(C, Clauses, AssociatedStmt,
- /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPTargetTeamsDirective *
- OMPTargetTeamsDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPTargetTeamsDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true);
- }
- OMPTargetTeamsDistributeDirective *OMPTargetTeamsDistributeDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTargetTeamsDistributeDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_target_teams_distribute), StartLoc,
- EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPTargetTeamsDistributeDirective *
- OMPTargetTeamsDistributeDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTargetTeamsDistributeDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_target_teams_distribute),
- CollapsedNum);
- }
- OMPTargetTeamsDistributeParallelForDirective *
- OMPTargetTeamsDistributeParallelForDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs, Expr *TaskRedRef, bool HasCancel) {
- auto *Dir = createDirective<OMPTargetTeamsDistributeParallelForDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_target_teams_distribute_parallel_for) +
- 1,
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setPrevLowerBoundVariable(Exprs.PrevLB);
- Dir->setPrevUpperBoundVariable(Exprs.PrevUB);
- Dir->setDistInc(Exprs.DistInc);
- Dir->setPrevEnsureUpperBound(Exprs.PrevEUB);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setCombinedLowerBoundVariable(Exprs.DistCombinedFields.LB);
- Dir->setCombinedUpperBoundVariable(Exprs.DistCombinedFields.UB);
- Dir->setCombinedEnsureUpperBound(Exprs.DistCombinedFields.EUB);
- Dir->setCombinedInit(Exprs.DistCombinedFields.Init);
- Dir->setCombinedCond(Exprs.DistCombinedFields.Cond);
- Dir->setCombinedNextLowerBound(Exprs.DistCombinedFields.NLB);
- Dir->setCombinedNextUpperBound(Exprs.DistCombinedFields.NUB);
- Dir->setCombinedDistCond(Exprs.DistCombinedFields.DistCond);
- Dir->setCombinedParForInDistCond(Exprs.DistCombinedFields.ParForInDistCond);
- Dir->setTaskReductionRefExpr(TaskRedRef);
- Dir->HasCancel = HasCancel;
- return Dir;
- }
- OMPTargetTeamsDistributeParallelForDirective *
- OMPTargetTeamsDistributeParallelForDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTargetTeamsDistributeParallelForDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_target_teams_distribute_parallel_for) +
- 1,
- CollapsedNum);
- }
- OMPTargetTeamsDistributeParallelForSimdDirective *
- OMPTargetTeamsDistributeParallelForSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTargetTeamsDistributeParallelForSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum,
- OMPD_target_teams_distribute_parallel_for_simd),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setPrevLowerBoundVariable(Exprs.PrevLB);
- Dir->setPrevUpperBoundVariable(Exprs.PrevUB);
- Dir->setDistInc(Exprs.DistInc);
- Dir->setPrevEnsureUpperBound(Exprs.PrevEUB);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- Dir->setCombinedLowerBoundVariable(Exprs.DistCombinedFields.LB);
- Dir->setCombinedUpperBoundVariable(Exprs.DistCombinedFields.UB);
- Dir->setCombinedEnsureUpperBound(Exprs.DistCombinedFields.EUB);
- Dir->setCombinedInit(Exprs.DistCombinedFields.Init);
- Dir->setCombinedCond(Exprs.DistCombinedFields.Cond);
- Dir->setCombinedNextLowerBound(Exprs.DistCombinedFields.NLB);
- Dir->setCombinedNextUpperBound(Exprs.DistCombinedFields.NUB);
- Dir->setCombinedDistCond(Exprs.DistCombinedFields.DistCond);
- Dir->setCombinedParForInDistCond(Exprs.DistCombinedFields.ParForInDistCond);
- return Dir;
- }
- OMPTargetTeamsDistributeParallelForSimdDirective *
- OMPTargetTeamsDistributeParallelForSimdDirective::CreateEmpty(
- const ASTContext &C, unsigned NumClauses, unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTargetTeamsDistributeParallelForSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum,
- OMPD_target_teams_distribute_parallel_for_simd),
- CollapsedNum);
- }
- OMPTargetTeamsDistributeSimdDirective *
- OMPTargetTeamsDistributeSimdDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPTargetTeamsDistributeSimdDirective>(
- C, Clauses, AssociatedStmt,
- numLoopChildren(CollapsedNum, OMPD_target_teams_distribute_simd),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPTargetTeamsDistributeSimdDirective *
- OMPTargetTeamsDistributeSimdDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- unsigned CollapsedNum,
- EmptyShell) {
- return createEmptyDirective<OMPTargetTeamsDistributeSimdDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_target_teams_distribute_simd),
- CollapsedNum);
- }
- OMPInteropDirective *
- OMPInteropDirective::Create(const ASTContext &C, SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses) {
- return createDirective<OMPInteropDirective>(
- C, Clauses, /*AssociatedStmt=*/nullptr, /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPInteropDirective *OMPInteropDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPInteropDirective>(C, NumClauses);
- }
- OMPDispatchDirective *OMPDispatchDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- SourceLocation TargetCallLoc) {
- auto *Dir = createDirective<OMPDispatchDirective>(
- C, Clauses, AssociatedStmt, /*NumChildren=*/0, StartLoc, EndLoc);
- Dir->setTargetCallLoc(TargetCallLoc);
- return Dir;
- }
- OMPDispatchDirective *OMPDispatchDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPDispatchDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true,
- /*NumChildren=*/0);
- }
- OMPMaskedDirective *OMPMaskedDirective::Create(const ASTContext &C,
- SourceLocation StartLoc,
- SourceLocation EndLoc,
- ArrayRef<OMPClause *> Clauses,
- Stmt *AssociatedStmt) {
- return createDirective<OMPMaskedDirective>(C, Clauses, AssociatedStmt,
- /*NumChildren=*/0, StartLoc,
- EndLoc);
- }
- OMPMaskedDirective *OMPMaskedDirective::CreateEmpty(const ASTContext &C,
- unsigned NumClauses,
- EmptyShell) {
- return createEmptyDirective<OMPMaskedDirective>(C, NumClauses,
- /*HasAssociatedStmt=*/true);
- }
- OMPGenericLoopDirective *OMPGenericLoopDirective::Create(
- const ASTContext &C, SourceLocation StartLoc, SourceLocation EndLoc,
- unsigned CollapsedNum, ArrayRef<OMPClause *> Clauses, Stmt *AssociatedStmt,
- const HelperExprs &Exprs) {
- auto *Dir = createDirective<OMPGenericLoopDirective>(
- C, Clauses, AssociatedStmt, numLoopChildren(CollapsedNum, OMPD_loop),
- StartLoc, EndLoc, CollapsedNum);
- Dir->setIterationVariable(Exprs.IterationVarRef);
- Dir->setLastIteration(Exprs.LastIteration);
- Dir->setCalcLastIteration(Exprs.CalcLastIteration);
- Dir->setPreCond(Exprs.PreCond);
- Dir->setCond(Exprs.Cond);
- Dir->setInit(Exprs.Init);
- Dir->setInc(Exprs.Inc);
- Dir->setIsLastIterVariable(Exprs.IL);
- Dir->setLowerBoundVariable(Exprs.LB);
- Dir->setUpperBoundVariable(Exprs.UB);
- Dir->setStrideVariable(Exprs.ST);
- Dir->setEnsureUpperBound(Exprs.EUB);
- Dir->setNextLowerBound(Exprs.NLB);
- Dir->setNextUpperBound(Exprs.NUB);
- Dir->setNumIterations(Exprs.NumIterations);
- Dir->setCounters(Exprs.Counters);
- Dir->setPrivateCounters(Exprs.PrivateCounters);
- Dir->setInits(Exprs.Inits);
- Dir->setUpdates(Exprs.Updates);
- Dir->setFinals(Exprs.Finals);
- Dir->setDependentCounters(Exprs.DependentCounters);
- Dir->setDependentInits(Exprs.DependentInits);
- Dir->setFinalsConditions(Exprs.FinalsConditions);
- Dir->setPreInits(Exprs.PreInits);
- return Dir;
- }
- OMPGenericLoopDirective *
- OMPGenericLoopDirective::CreateEmpty(const ASTContext &C, unsigned NumClauses,
- unsigned CollapsedNum, EmptyShell) {
- return createEmptyDirective<OMPGenericLoopDirective>(
- C, NumClauses, /*HasAssociatedStmt=*/true,
- numLoopChildren(CollapsedNum, OMPD_loop), CollapsedNum);
- }
|