blob: b380143fe52f7ce7cf68bcf5b0e460d2c65e2b3d [file] [log] [blame]
Alexey Bataev9959db52014-05-06 10:08:46 +00001//===--- CGStmtOpenMP.cpp - Emit LLVM Code from Statements ----------------===//
2//
Chandler Carruth2946cd72019-01-19 08:50:56 +00003// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
Alexey Bataev9959db52014-05-06 10:08:46 +00006//
7//===----------------------------------------------------------------------===//
8//
9// This contains code to emit OpenMP nodes as LLVM code.
10//
11//===----------------------------------------------------------------------===//
12
Alexey Bataev3392d762016-02-16 11:18:12 +000013#include "CGCleanup.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000014#include "CGOpenMPRuntime.h"
15#include "CodeGenFunction.h"
16#include "CodeGenModule.h"
Chandler Carruth0d9593d2015-01-14 11:29:14 +000017#include "TargetInfo.h"
Alexey Bataev61205822019-12-04 09:50:21 -050018#include "clang/AST/ASTContext.h"
Reid Kleckner98031782019-12-09 16:11:56 -080019#include "clang/AST/Attr.h"
20#include "clang/AST/DeclOpenMP.h"
Alexey Bataeve6d25832020-01-27 14:10:17 -050021#include "clang/AST/OpenMPClause.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000022#include "clang/AST/Stmt.h"
23#include "clang/AST/StmtOpenMP.h"
Alexey Bataevea9166b2020-02-06 16:30:23 -050024#include "clang/Basic/OpenMPKinds.h"
Alexey Bataev8bbf2e32019-11-04 09:59:11 -050025#include "clang/Basic/PrettyStackTrace.h"
Johannes Doerfert10fedd92019-12-26 11:23:38 -060026#include "llvm/Frontend/OpenMP/OMPIRBuilder.h"
Alexey Bataev0fca7662020-03-27 11:15:17 -040027#include "llvm/IR/Constants.h"
Alexey Bataeve8e05de2020-02-07 12:22:23 -050028#include "llvm/IR/Instructions.h"
Alexey Bataev2d4f80f2020-02-11 15:15:21 -050029#include "llvm/Support/AtomicOrdering.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000030using namespace clang;
31using namespace CodeGen;
Johannes Doerferteb3e81f2019-11-04 22:00:49 -060032using namespace llvm::omp;
Alexey Bataev9959db52014-05-06 10:08:46 +000033
Alexey Bataev90b54fa2020-05-25 16:06:31 -040034static const VarDecl *getBaseDecl(const Expr *Ref);
35
Alexey Bataev3392d762016-02-16 11:18:12 +000036namespace {
37/// Lexical scope for OpenMP executable constructs, that handles correct codegen
38/// for captured expressions.
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +000039class OMPLexicalScope : public CodeGenFunction::LexicalScope {
Alexey Bataev3392d762016-02-16 11:18:12 +000040 void emitPreInitStmt(CodeGenFunction &CGF, const OMPExecutableDirective &S) {
41 for (const auto *C : S.clauses()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +000042 if (const auto *CPI = OMPClauseWithPreInit::get(C)) {
43 if (const auto *PreInit =
44 cast_or_null<DeclStmt>(CPI->getPreInitStmt())) {
Alexey Bataev2bbf7212016-03-03 03:52:24 +000045 for (const auto *I : PreInit->decls()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +000046 if (!I->hasAttr<OMPCaptureNoInitAttr>()) {
Alexey Bataev2bbf7212016-03-03 03:52:24 +000047 CGF.EmitVarDecl(cast<VarDecl>(*I));
Alexey Bataevddf3db92018-04-13 17:31:06 +000048 } else {
Alexey Bataev2bbf7212016-03-03 03:52:24 +000049 CodeGenFunction::AutoVarEmission Emission =
50 CGF.EmitAutoVarAlloca(cast<VarDecl>(*I));
51 CGF.EmitAutoVarCleanups(Emission);
52 }
53 }
Alexey Bataev3392d762016-02-16 11:18:12 +000054 }
55 }
56 }
57 }
Alexey Bataev4ba78a42016-04-27 07:56:03 +000058 CodeGenFunction::OMPPrivateScope InlinedShareds;
59
60 static bool isCapturedVar(CodeGenFunction &CGF, const VarDecl *VD) {
61 return CGF.LambdaCaptureFields.lookup(VD) ||
62 (CGF.CapturedStmtInfo && CGF.CapturedStmtInfo->lookup(VD)) ||
Alexey Bataev172f1462020-03-12 12:52:02 -040063 (CGF.CurCodeDecl && isa<BlockDecl>(CGF.CurCodeDecl) &&
64 cast<BlockDecl>(CGF.CurCodeDecl)->capturesVariable(VD));
Alexey Bataev4ba78a42016-04-27 07:56:03 +000065 }
Alexey Bataev3392d762016-02-16 11:18:12 +000066
Alexey Bataev3392d762016-02-16 11:18:12 +000067public:
Alexey Bataev475a7442018-01-12 19:39:11 +000068 OMPLexicalScope(
69 CodeGenFunction &CGF, const OMPExecutableDirective &S,
70 const llvm::Optional<OpenMPDirectiveKind> CapturedRegion = llvm::None,
71 const bool EmitPreInitStmt = true)
Alexey Bataev4ba78a42016-04-27 07:56:03 +000072 : CodeGenFunction::LexicalScope(CGF, S.getSourceRange()),
73 InlinedShareds(CGF) {
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +000074 if (EmitPreInitStmt)
75 emitPreInitStmt(CGF, S);
Alexey Bataev475a7442018-01-12 19:39:11 +000076 if (!CapturedRegion.hasValue())
77 return;
78 assert(S.hasAssociatedStmt() &&
79 "Expected associated statement for inlined directive.");
80 const CapturedStmt *CS = S.getCapturedStmt(*CapturedRegion);
Alexey Bataevddf3db92018-04-13 17:31:06 +000081 for (const auto &C : CS->captures()) {
Alexey Bataev475a7442018-01-12 19:39:11 +000082 if (C.capturesVariable() || C.capturesVariableByCopy()) {
83 auto *VD = C.getCapturedVar();
84 assert(VD == VD->getCanonicalDecl() &&
85 "Canonical decl must be captured.");
86 DeclRefExpr DRE(
Bruno Ricci5fc4db72018-12-21 14:10:18 +000087 CGF.getContext(), const_cast<VarDecl *>(VD),
Alexey Bataev475a7442018-01-12 19:39:11 +000088 isCapturedVar(CGF, VD) || (CGF.CapturedStmtInfo &&
89 InlinedShareds.isGlobalVarCaptured(VD)),
Alexey Bataeva9b9cc02018-01-23 18:12:38 +000090 VD->getType().getNonReferenceType(), VK_LValue, C.getLocation());
Alexey Bataev475a7442018-01-12 19:39:11 +000091 InlinedShareds.addPrivate(VD, [&CGF, &DRE]() -> Address {
Akira Hatanakaf139ae32019-12-03 15:17:01 -080092 return CGF.EmitLValue(&DRE).getAddress(CGF);
Alexey Bataev475a7442018-01-12 19:39:11 +000093 });
Alexey Bataev4ba78a42016-04-27 07:56:03 +000094 }
95 }
Alexey Bataev475a7442018-01-12 19:39:11 +000096 (void)InlinedShareds.Privatize();
Alexey Bataev3392d762016-02-16 11:18:12 +000097 }
98};
Alexey Bataev14fa1c62016-03-29 05:34:15 +000099
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +0000100/// Lexical scope for OpenMP parallel construct, that handles correct codegen
101/// for captured expressions.
102class OMPParallelScope final : public OMPLexicalScope {
103 bool EmitPreInitStmt(const OMPExecutableDirective &S) {
104 OpenMPDirectiveKind Kind = S.getDirectiveKind();
Carlo Bertollib0ff0a62017-04-25 17:52:12 +0000105 return !(isOpenMPTargetExecutionDirective(Kind) ||
106 isOpenMPLoopBoundSharingDirective(Kind)) &&
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +0000107 isOpenMPParallelDirective(Kind);
108 }
109
110public:
111 OMPParallelScope(CodeGenFunction &CGF, const OMPExecutableDirective &S)
Alexey Bataev475a7442018-01-12 19:39:11 +0000112 : OMPLexicalScope(CGF, S, /*CapturedRegion=*/llvm::None,
113 EmitPreInitStmt(S)) {}
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +0000114};
115
Arpith Chacko Jacob99a1e0e2017-01-25 02:18:43 +0000116/// Lexical scope for OpenMP teams construct, that handles correct codegen
117/// for captured expressions.
118class OMPTeamsScope final : public OMPLexicalScope {
119 bool EmitPreInitStmt(const OMPExecutableDirective &S) {
120 OpenMPDirectiveKind Kind = S.getDirectiveKind();
121 return !isOpenMPTargetExecutionDirective(Kind) &&
122 isOpenMPTeamsDirective(Kind);
123 }
124
125public:
126 OMPTeamsScope(CodeGenFunction &CGF, const OMPExecutableDirective &S)
Alexey Bataev475a7442018-01-12 19:39:11 +0000127 : OMPLexicalScope(CGF, S, /*CapturedRegion=*/llvm::None,
128 EmitPreInitStmt(S)) {}
Arpith Chacko Jacob99a1e0e2017-01-25 02:18:43 +0000129};
130
Alexey Bataev5a3af132016-03-29 08:58:54 +0000131/// Private scope for OpenMP loop-based directives, that supports capturing
132/// of used expression from loop statement.
133class OMPLoopScope : public CodeGenFunction::RunCleanupsScope {
134 void emitPreInitStmt(CodeGenFunction &CGF, const OMPLoopDirective &S) {
Alexey Bataevab4ea222018-03-07 18:17:06 +0000135 CodeGenFunction::OMPMapVars PreCondVars;
Alexey Bataevf71939c2019-09-18 19:24:07 +0000136 llvm::DenseSet<const VarDecl *> EmittedAsPrivate;
Alexey Bataevddf3db92018-04-13 17:31:06 +0000137 for (const auto *E : S.counters()) {
Alexey Bataeve83b3e82017-12-08 20:18:58 +0000138 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
Alexey Bataevf71939c2019-09-18 19:24:07 +0000139 EmittedAsPrivate.insert(VD->getCanonicalDecl());
Alexey Bataevab4ea222018-03-07 18:17:06 +0000140 (void)PreCondVars.setVarAddr(
141 CGF, VD, CGF.CreateMemTemp(VD->getType().getNonReferenceType()));
Alexey Bataeve83b3e82017-12-08 20:18:58 +0000142 }
Alexey Bataevf71939c2019-09-18 19:24:07 +0000143 // Mark private vars as undefs.
144 for (const auto *C : S.getClausesOfKind<OMPPrivateClause>()) {
145 for (const Expr *IRef : C->varlists()) {
146 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(IRef)->getDecl());
147 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
148 (void)PreCondVars.setVarAddr(
149 CGF, OrigVD,
150 Address(llvm::UndefValue::get(
151 CGF.ConvertTypeForMem(CGF.getContext().getPointerType(
152 OrigVD->getType().getNonReferenceType()))),
153 CGF.getContext().getDeclAlign(OrigVD)));
154 }
155 }
156 }
Alexey Bataevab4ea222018-03-07 18:17:06 +0000157 (void)PreCondVars.apply(CGF);
Alexey Bataevbef93a92019-10-07 18:54:57 +0000158 // Emit init, __range and __end variables for C++ range loops.
159 const Stmt *Body =
160 S.getInnermostCapturedStmt()->getCapturedStmt()->IgnoreContainers();
161 for (unsigned Cnt = 0; Cnt < S.getCollapsedNumber(); ++Cnt) {
Alexey Bataev8bbf2e32019-11-04 09:59:11 -0500162 Body = OMPLoopDirective::tryToFindNextInnerLoop(
163 Body, /*TryImperfectlyNestedLoops=*/true);
Alexey Bataevbef93a92019-10-07 18:54:57 +0000164 if (auto *For = dyn_cast<ForStmt>(Body)) {
165 Body = For->getBody();
166 } else {
167 assert(isa<CXXForRangeStmt>(Body) &&
Alexey Bataevd457f7e2019-10-07 19:57:40 +0000168 "Expected canonical for loop or range-based for loop.");
Alexey Bataevbef93a92019-10-07 18:54:57 +0000169 auto *CXXFor = cast<CXXForRangeStmt>(Body);
170 if (const Stmt *Init = CXXFor->getInit())
171 CGF.EmitStmt(Init);
172 CGF.EmitStmt(CXXFor->getRangeStmt());
173 CGF.EmitStmt(CXXFor->getEndStmt());
174 Body = CXXFor->getBody();
175 }
176 }
Alexey Bataevddf3db92018-04-13 17:31:06 +0000177 if (const auto *PreInits = cast_or_null<DeclStmt>(S.getPreInits())) {
George Burgess IV00f70bd2018-03-01 05:43:23 +0000178 for (const auto *I : PreInits->decls())
179 CGF.EmitVarDecl(cast<VarDecl>(*I));
Alexey Bataev5a3af132016-03-29 08:58:54 +0000180 }
Alexey Bataevab4ea222018-03-07 18:17:06 +0000181 PreCondVars.restore(CGF);
Alexey Bataev5a3af132016-03-29 08:58:54 +0000182 }
183
184public:
185 OMPLoopScope(CodeGenFunction &CGF, const OMPLoopDirective &S)
186 : CodeGenFunction::RunCleanupsScope(CGF) {
187 emitPreInitStmt(CGF, S);
188 }
189};
190
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000191class OMPSimdLexicalScope : public CodeGenFunction::LexicalScope {
192 CodeGenFunction::OMPPrivateScope InlinedShareds;
193
194 static bool isCapturedVar(CodeGenFunction &CGF, const VarDecl *VD) {
195 return CGF.LambdaCaptureFields.lookup(VD) ||
196 (CGF.CapturedStmtInfo && CGF.CapturedStmtInfo->lookup(VD)) ||
197 (CGF.CurCodeDecl && isa<BlockDecl>(CGF.CurCodeDecl) &&
198 cast<BlockDecl>(CGF.CurCodeDecl)->capturesVariable(VD));
199 }
200
201public:
202 OMPSimdLexicalScope(CodeGenFunction &CGF, const OMPExecutableDirective &S)
203 : CodeGenFunction::LexicalScope(CGF, S.getSourceRange()),
204 InlinedShareds(CGF) {
205 for (const auto *C : S.clauses()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000206 if (const auto *CPI = OMPClauseWithPreInit::get(C)) {
207 if (const auto *PreInit =
208 cast_or_null<DeclStmt>(CPI->getPreInitStmt())) {
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000209 for (const auto *I : PreInit->decls()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000210 if (!I->hasAttr<OMPCaptureNoInitAttr>()) {
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000211 CGF.EmitVarDecl(cast<VarDecl>(*I));
Alexey Bataevddf3db92018-04-13 17:31:06 +0000212 } else {
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000213 CodeGenFunction::AutoVarEmission Emission =
214 CGF.EmitAutoVarAlloca(cast<VarDecl>(*I));
215 CGF.EmitAutoVarCleanups(Emission);
216 }
217 }
218 }
219 } else if (const auto *UDP = dyn_cast<OMPUseDevicePtrClause>(C)) {
220 for (const Expr *E : UDP->varlists()) {
221 const Decl *D = cast<DeclRefExpr>(E)->getDecl();
222 if (const auto *OED = dyn_cast<OMPCapturedExprDecl>(D))
223 CGF.EmitVarDecl(*OED);
224 }
Alexey Bataev90b54fa2020-05-25 16:06:31 -0400225 } else if (const auto *UDP = dyn_cast<OMPUseDeviceAddrClause>(C)) {
226 for (const Expr *E : UDP->varlists()) {
227 const Decl *D = getBaseDecl(E);
228 if (const auto *OED = dyn_cast<OMPCapturedExprDecl>(D))
229 CGF.EmitVarDecl(*OED);
230 }
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000231 }
232 }
233 if (!isOpenMPSimdDirective(S.getDirectiveKind()))
234 CGF.EmitOMPPrivateClause(S, InlinedShareds);
235 if (const auto *TG = dyn_cast<OMPTaskgroupDirective>(&S)) {
236 if (const Expr *E = TG->getReductionRef())
237 CGF.EmitVarDecl(*cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl()));
238 }
239 const auto *CS = cast_or_null<CapturedStmt>(S.getAssociatedStmt());
240 while (CS) {
241 for (auto &C : CS->captures()) {
242 if (C.capturesVariable() || C.capturesVariableByCopy()) {
243 auto *VD = C.getCapturedVar();
244 assert(VD == VD->getCanonicalDecl() &&
245 "Canonical decl must be captured.");
Bruno Ricci5fc4db72018-12-21 14:10:18 +0000246 DeclRefExpr DRE(CGF.getContext(), const_cast<VarDecl *>(VD),
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000247 isCapturedVar(CGF, VD) ||
248 (CGF.CapturedStmtInfo &&
249 InlinedShareds.isGlobalVarCaptured(VD)),
250 VD->getType().getNonReferenceType(), VK_LValue,
Alexey Bataeva9b9cc02018-01-23 18:12:38 +0000251 C.getLocation());
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000252 InlinedShareds.addPrivate(VD, [&CGF, &DRE]() -> Address {
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800253 return CGF.EmitLValue(&DRE).getAddress(CGF);
Alexey Bataeva8a9153a2017-12-29 18:07:07 +0000254 });
255 }
256 }
257 CS = dyn_cast<CapturedStmt>(CS->getCapturedStmt());
258 }
259 (void)InlinedShareds.Privatize();
260 }
261};
262
Alexey Bataev3392d762016-02-16 11:18:12 +0000263} // namespace
264
Alexey Bataevf8365372017-11-17 17:57:25 +0000265static void emitCommonOMPTargetDirective(CodeGenFunction &CGF,
266 const OMPExecutableDirective &S,
267 const RegionCodeGenTy &CodeGen);
268
Alexey Bataevf47c4b42017-09-26 13:47:31 +0000269LValue CodeGenFunction::EmitOMPSharedLValue(const Expr *E) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000270 if (const auto *OrigDRE = dyn_cast<DeclRefExpr>(E)) {
271 if (const auto *OrigVD = dyn_cast<VarDecl>(OrigDRE->getDecl())) {
Alexey Bataevf47c4b42017-09-26 13:47:31 +0000272 OrigVD = OrigVD->getCanonicalDecl();
273 bool IsCaptured =
274 LambdaCaptureFields.lookup(OrigVD) ||
275 (CapturedStmtInfo && CapturedStmtInfo->lookup(OrigVD)) ||
276 (CurCodeDecl && isa<BlockDecl>(CurCodeDecl));
Bruno Ricci5fc4db72018-12-21 14:10:18 +0000277 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(OrigVD), IsCaptured,
Alexey Bataevf47c4b42017-09-26 13:47:31 +0000278 OrigDRE->getType(), VK_LValue, OrigDRE->getExprLoc());
279 return EmitLValue(&DRE);
280 }
281 }
282 return EmitLValue(E);
283}
284
Alexey Bataev1189bd02016-01-26 12:20:39 +0000285llvm::Value *CodeGenFunction::getTypeSize(QualType Ty) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000286 ASTContext &C = getContext();
Alexey Bataev1189bd02016-01-26 12:20:39 +0000287 llvm::Value *Size = nullptr;
288 auto SizeInChars = C.getTypeSizeInChars(Ty);
289 if (SizeInChars.isZero()) {
290 // getTypeSizeInChars() returns 0 for a VLA.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000291 while (const VariableArrayType *VAT = C.getAsVariableArrayType(Ty)) {
292 VlaSizePair VlaSize = getVLASize(VAT);
Sander de Smalen891af03a2018-02-03 13:55:59 +0000293 Ty = VlaSize.Type;
294 Size = Size ? Builder.CreateNUWMul(Size, VlaSize.NumElts)
295 : VlaSize.NumElts;
Alexey Bataev1189bd02016-01-26 12:20:39 +0000296 }
297 SizeInChars = C.getTypeSizeInChars(Ty);
298 if (SizeInChars.isZero())
299 return llvm::ConstantInt::get(SizeTy, /*V=*/0);
Alexey Bataevddf3db92018-04-13 17:31:06 +0000300 return Builder.CreateNUWMul(Size, CGM.getSize(SizeInChars));
301 }
302 return CGM.getSize(SizeInChars);
Alexey Bataev1189bd02016-01-26 12:20:39 +0000303}
304
Alexey Bataev2377fe92015-09-10 08:12:02 +0000305void CodeGenFunction::GenerateOpenMPCapturedVars(
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000306 const CapturedStmt &S, SmallVectorImpl<llvm::Value *> &CapturedVars) {
Alexey Bataev2377fe92015-09-10 08:12:02 +0000307 const RecordDecl *RD = S.getCapturedRecordDecl();
308 auto CurField = RD->field_begin();
309 auto CurCap = S.captures().begin();
310 for (CapturedStmt::const_capture_init_iterator I = S.capture_init_begin(),
311 E = S.capture_init_end();
312 I != E; ++I, ++CurField, ++CurCap) {
313 if (CurField->hasCapturedVLAType()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000314 const VariableArrayType *VAT = CurField->getCapturedVLAType();
315 llvm::Value *Val = VLASizeMap[VAT->getSizeExpr()];
Samuel Antaobed3c462015-10-02 16:14:20 +0000316 CapturedVars.push_back(Val);
Alexey Bataevddf3db92018-04-13 17:31:06 +0000317 } else if (CurCap->capturesThis()) {
Alexey Bataev2377fe92015-09-10 08:12:02 +0000318 CapturedVars.push_back(CXXThisValue);
Alexey Bataevddf3db92018-04-13 17:31:06 +0000319 } else if (CurCap->capturesVariableByCopy()) {
Alexey Bataev1e491372018-01-23 18:44:14 +0000320 llvm::Value *CV = EmitLoadOfScalar(EmitLValue(*I), CurCap->getLocation());
Samuel Antao6d004262016-06-16 18:39:34 +0000321
322 // If the field is not a pointer, we need to save the actual value
323 // and load it as a void pointer.
324 if (!CurField->getType()->isAnyPointerType()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000325 ASTContext &Ctx = getContext();
326 Address DstAddr = CreateMemTemp(
Samuel Antao6d004262016-06-16 18:39:34 +0000327 Ctx.getUIntPtrType(),
Alexey Bataevddf3db92018-04-13 17:31:06 +0000328 Twine(CurCap->getCapturedVar()->getName(), ".casted"));
Samuel Antao6d004262016-06-16 18:39:34 +0000329 LValue DstLV = MakeAddrLValue(DstAddr, Ctx.getUIntPtrType());
330
Alexey Bataevddf3db92018-04-13 17:31:06 +0000331 llvm::Value *SrcAddrVal = EmitScalarConversion(
Samuel Antao6d004262016-06-16 18:39:34 +0000332 DstAddr.getPointer(), Ctx.getPointerType(Ctx.getUIntPtrType()),
Alexey Bataeva9b9cc02018-01-23 18:12:38 +0000333 Ctx.getPointerType(CurField->getType()), CurCap->getLocation());
Samuel Antao6d004262016-06-16 18:39:34 +0000334 LValue SrcLV =
335 MakeNaturalAlignAddrLValue(SrcAddrVal, CurField->getType());
336
337 // Store the value using the source type pointer.
338 EmitStoreThroughLValue(RValue::get(CV), SrcLV);
339
340 // Load the value using the destination type pointer.
Alexey Bataev1e491372018-01-23 18:44:14 +0000341 CV = EmitLoadOfScalar(DstLV, CurCap->getLocation());
Samuel Antao6d004262016-06-16 18:39:34 +0000342 }
343 CapturedVars.push_back(CV);
344 } else {
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000345 assert(CurCap->capturesVariable() && "Expected capture by reference.");
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800346 CapturedVars.push_back(EmitLValue(*I).getAddress(*this).getPointer());
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000347 }
Alexey Bataev2377fe92015-09-10 08:12:02 +0000348 }
349}
350
Alexey Bataeva9b9cc02018-01-23 18:12:38 +0000351static Address castValueFromUintptr(CodeGenFunction &CGF, SourceLocation Loc,
352 QualType DstType, StringRef Name,
Alexey Bataev06e80f62019-05-23 18:19:54 +0000353 LValue AddrLV) {
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000354 ASTContext &Ctx = CGF.getContext();
355
Alexey Bataevddf3db92018-04-13 17:31:06 +0000356 llvm::Value *CastedPtr = CGF.EmitScalarConversion(
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800357 AddrLV.getAddress(CGF).getPointer(), Ctx.getUIntPtrType(),
Alexey Bataevddf3db92018-04-13 17:31:06 +0000358 Ctx.getPointerType(DstType), Loc);
359 Address TmpAddr =
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000360 CGF.MakeNaturalAlignAddrLValue(CastedPtr, Ctx.getPointerType(DstType))
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800361 .getAddress(CGF);
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000362 return TmpAddr;
363}
364
Alexey Bataevf7ce1662017-04-10 19:16:45 +0000365static QualType getCanonicalParamType(ASTContext &C, QualType T) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000366 if (T->isLValueReferenceType())
Alexey Bataevf7ce1662017-04-10 19:16:45 +0000367 return C.getLValueReferenceType(
368 getCanonicalParamType(C, T.getNonReferenceType()),
369 /*SpelledAsLValue=*/false);
Alexey Bataevf7ce1662017-04-10 19:16:45 +0000370 if (T->isPointerType())
371 return C.getPointerType(getCanonicalParamType(C, T->getPointeeType()));
Alexey Bataevddf3db92018-04-13 17:31:06 +0000372 if (const ArrayType *A = T->getAsArrayTypeUnsafe()) {
373 if (const auto *VLA = dyn_cast<VariableArrayType>(A))
Alexey Bataev1b48c5e2017-10-24 19:52:31 +0000374 return getCanonicalParamType(C, VLA->getElementType());
Alexey Bataevddf3db92018-04-13 17:31:06 +0000375 if (!A->isVariablyModifiedType())
Alexey Bataev1b48c5e2017-10-24 19:52:31 +0000376 return C.getCanonicalType(T);
377 }
Alexey Bataevf7ce1662017-04-10 19:16:45 +0000378 return C.getCanonicalParamType(T);
379}
380
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000381namespace {
Alexey Bataevc33ba8c2020-01-17 14:05:40 -0500382/// Contains required data for proper outlined function codegen.
383struct FunctionOptions {
384 /// Captured statement for which the function is generated.
385 const CapturedStmt *S = nullptr;
386 /// true if cast to/from UIntPtr is required for variables captured by
387 /// value.
388 const bool UIntPtrCastRequired = true;
389 /// true if only casted arguments must be registered as local args or VLA
390 /// sizes.
391 const bool RegisterCastedArgsOnly = false;
392 /// Name of the generated function.
393 const StringRef FunctionName;
394 /// Location of the non-debug version of the outlined function.
395 SourceLocation Loc;
396 explicit FunctionOptions(const CapturedStmt *S, bool UIntPtrCastRequired,
397 bool RegisterCastedArgsOnly, StringRef FunctionName,
398 SourceLocation Loc)
399 : S(S), UIntPtrCastRequired(UIntPtrCastRequired),
400 RegisterCastedArgsOnly(UIntPtrCastRequired && RegisterCastedArgsOnly),
401 FunctionName(FunctionName), Loc(Loc) {}
402};
403} // namespace
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000404
Alexey Bataeve754b182017-08-09 19:38:53 +0000405static llvm::Function *emitOutlinedFunctionPrologue(
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000406 CodeGenFunction &CGF, FunctionArgList &Args,
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000407 llvm::MapVector<const Decl *, std::pair<const VarDecl *, Address>>
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000408 &LocalAddrs,
409 llvm::DenseMap<const Decl *, std::pair<const Expr *, llvm::Value *>>
410 &VLASizes,
411 llvm::Value *&CXXThisValue, const FunctionOptions &FO) {
412 const CapturedDecl *CD = FO.S->getCapturedDecl();
413 const RecordDecl *RD = FO.S->getCapturedRecordDecl();
Alexey Bataev2377fe92015-09-10 08:12:02 +0000414 assert(CD->hasBody() && "missing CapturedDecl body");
415
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000416 CXXThisValue = nullptr;
Alexey Bataev2377fe92015-09-10 08:12:02 +0000417 // Build the argument list.
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000418 CodeGenModule &CGM = CGF.CGM;
Alexey Bataev2377fe92015-09-10 08:12:02 +0000419 ASTContext &Ctx = CGM.getContext();
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000420 FunctionArgList TargetArgs;
Alexey Bataev2377fe92015-09-10 08:12:02 +0000421 Args.append(CD->param_begin(),
422 std::next(CD->param_begin(), CD->getContextParamPosition()));
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000423 TargetArgs.append(
424 CD->param_begin(),
425 std::next(CD->param_begin(), CD->getContextParamPosition()));
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000426 auto I = FO.S->captures().begin();
Alexey Bataevb45d43c2017-11-22 16:02:03 +0000427 FunctionDecl *DebugFunctionDecl = nullptr;
428 if (!FO.UIntPtrCastRequired) {
429 FunctionProtoType::ExtProtoInfo EPI;
Jonas Devlieghere64a26302018-11-11 00:56:15 +0000430 QualType FunctionTy = Ctx.getFunctionType(Ctx.VoidTy, llvm::None, EPI);
Alexey Bataevb45d43c2017-11-22 16:02:03 +0000431 DebugFunctionDecl = FunctionDecl::Create(
Stephen Kellyf2ceec42018-08-09 21:08:08 +0000432 Ctx, Ctx.getTranslationUnitDecl(), FO.S->getBeginLoc(),
Jonas Devlieghere64a26302018-11-11 00:56:15 +0000433 SourceLocation(), DeclarationName(), FunctionTy,
434 Ctx.getTrivialTypeSourceInfo(FunctionTy), SC_Static,
435 /*isInlineSpecified=*/false, /*hasWrittenPrototype=*/false);
Alexey Bataevb45d43c2017-11-22 16:02:03 +0000436 }
Alexey Bataevddf3db92018-04-13 17:31:06 +0000437 for (const FieldDecl *FD : RD->fields()) {
Alexey Bataev2377fe92015-09-10 08:12:02 +0000438 QualType ArgType = FD->getType();
439 IdentifierInfo *II = nullptr;
440 VarDecl *CapVar = nullptr;
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000441
442 // If this is a capture by copy and the type is not a pointer, the outlined
443 // function argument type should be uintptr and the value properly casted to
444 // uintptr. This is necessary given that the runtime library is only able to
445 // deal with pointers. We can pass in the same way the VLA type sizes to the
446 // outlined function.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000447 if (FO.UIntPtrCastRequired &&
448 ((I->capturesVariableByCopy() && !ArgType->isAnyPointerType()) ||
449 I->capturesVariableArrayType()))
450 ArgType = Ctx.getUIntPtrType();
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000451
452 if (I->capturesVariable() || I->capturesVariableByCopy()) {
Alexey Bataev2377fe92015-09-10 08:12:02 +0000453 CapVar = I->getCapturedVar();
454 II = CapVar->getIdentifier();
Alexey Bataevddf3db92018-04-13 17:31:06 +0000455 } else if (I->capturesThis()) {
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000456 II = &Ctx.Idents.get("this");
Alexey Bataevddf3db92018-04-13 17:31:06 +0000457 } else {
Alexey Bataev2377fe92015-09-10 08:12:02 +0000458 assert(I->capturesVariableArrayType());
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000459 II = &Ctx.Idents.get("vla");
Alexey Bataev2377fe92015-09-10 08:12:02 +0000460 }
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000461 if (ArgType->isVariablyModifiedType())
Alexey Bataev1b48c5e2017-10-24 19:52:31 +0000462 ArgType = getCanonicalParamType(Ctx, ArgType);
Alexey Bataevb45d43c2017-11-22 16:02:03 +0000463 VarDecl *Arg;
464 if (DebugFunctionDecl && (CapVar || I->capturesThis())) {
465 Arg = ParmVarDecl::Create(
466 Ctx, DebugFunctionDecl,
Stephen Kellyf2ceec42018-08-09 21:08:08 +0000467 CapVar ? CapVar->getBeginLoc() : FD->getBeginLoc(),
Alexey Bataevb45d43c2017-11-22 16:02:03 +0000468 CapVar ? CapVar->getLocation() : FD->getLocation(), II, ArgType,
469 /*TInfo=*/nullptr, SC_None, /*DefArg=*/nullptr);
470 } else {
471 Arg = ImplicitParamDecl::Create(Ctx, /*DC=*/nullptr, FD->getLocation(),
472 II, ArgType, ImplicitParamDecl::Other);
473 }
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000474 Args.emplace_back(Arg);
475 // Do not cast arguments if we emit function with non-original types.
476 TargetArgs.emplace_back(
477 FO.UIntPtrCastRequired
478 ? Arg
479 : CGM.getOpenMPRuntime().translateParameter(FD, Arg));
Alexey Bataev2377fe92015-09-10 08:12:02 +0000480 ++I;
481 }
482 Args.append(
483 std::next(CD->param_begin(), CD->getContextParamPosition() + 1),
484 CD->param_end());
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000485 TargetArgs.append(
486 std::next(CD->param_begin(), CD->getContextParamPosition() + 1),
487 CD->param_end());
Alexey Bataev2377fe92015-09-10 08:12:02 +0000488
489 // Create the function declaration.
Alexey Bataev2377fe92015-09-10 08:12:02 +0000490 const CGFunctionInfo &FuncInfo =
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000491 CGM.getTypes().arrangeBuiltinFunctionDeclaration(Ctx.VoidTy, TargetArgs);
Alexey Bataev2377fe92015-09-10 08:12:02 +0000492 llvm::FunctionType *FuncLLVMTy = CGM.getTypes().GetFunctionType(FuncInfo);
493
Alexey Bataevddf3db92018-04-13 17:31:06 +0000494 auto *F =
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000495 llvm::Function::Create(FuncLLVMTy, llvm::GlobalValue::InternalLinkage,
496 FO.FunctionName, &CGM.getModule());
Alexey Bataev2377fe92015-09-10 08:12:02 +0000497 CGM.SetInternalFunctionAttributes(CD, F, FuncInfo);
498 if (CD->isNothrow())
Alexey Bataev2c7eee52017-08-04 19:10:54 +0000499 F->setDoesNotThrow();
Alexey Bataevc0f879b2018-04-10 20:10:53 +0000500 F->setDoesNotRecurse();
Alexey Bataev2377fe92015-09-10 08:12:02 +0000501
502 // Generate the function.
Alexey Bataev6e01dc12017-08-14 16:03:47 +0000503 CGF.StartFunction(CD, Ctx.VoidTy, F, FuncInfo, TargetArgs,
Alexey Bataevc33ba8c2020-01-17 14:05:40 -0500504 FO.UIntPtrCastRequired ? FO.Loc : FO.S->getBeginLoc(),
505 FO.UIntPtrCastRequired ? FO.Loc
506 : CD->getBody()->getBeginLoc());
Alexey Bataev2377fe92015-09-10 08:12:02 +0000507 unsigned Cnt = CD->getContextParamPosition();
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000508 I = FO.S->captures().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +0000509 for (const FieldDecl *FD : RD->fields()) {
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000510 // Do not map arguments if we emit function with non-original types.
511 Address LocalAddr(Address::invalid());
512 if (!FO.UIntPtrCastRequired && Args[Cnt] != TargetArgs[Cnt]) {
513 LocalAddr = CGM.getOpenMPRuntime().getParameterAddress(CGF, Args[Cnt],
514 TargetArgs[Cnt]);
515 } else {
516 LocalAddr = CGF.GetAddrOfLocalVar(Args[Cnt]);
517 }
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000518 // If we are capturing a pointer by copy we don't need to do anything, just
519 // use the value that we get from the arguments.
520 if (I->capturesVariableByCopy() && FD->getType()->isAnyPointerType()) {
Samuel Antao403ffd42016-07-27 22:49:49 +0000521 const VarDecl *CurVD = I->getCapturedVar();
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000522 if (!FO.RegisterCastedArgsOnly)
523 LocalAddrs.insert({Args[Cnt], {CurVD, LocalAddr}});
Richard Trieucc3949d2016-02-18 22:34:54 +0000524 ++Cnt;
525 ++I;
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000526 continue;
527 }
528
Ivan A. Kosarev5f8c0ca2017-10-10 09:39:32 +0000529 LValue ArgLVal = CGF.MakeAddrLValue(LocalAddr, Args[Cnt]->getType(),
530 AlignmentSource::Decl);
Alexey Bataev2377fe92015-09-10 08:12:02 +0000531 if (FD->hasCapturedVLAType()) {
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000532 if (FO.UIntPtrCastRequired) {
Alexey Bataeva9b9cc02018-01-23 18:12:38 +0000533 ArgLVal = CGF.MakeAddrLValue(
534 castValueFromUintptr(CGF, I->getLocation(), FD->getType(),
535 Args[Cnt]->getName(), ArgLVal),
536 FD->getType(), AlignmentSource::Decl);
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000537 }
Alexey Bataevddf3db92018-04-13 17:31:06 +0000538 llvm::Value *ExprArg = CGF.EmitLoadOfScalar(ArgLVal, I->getLocation());
539 const VariableArrayType *VAT = FD->getCapturedVLAType();
540 VLASizes.try_emplace(Args[Cnt], VAT->getSizeExpr(), ExprArg);
Alexey Bataev2377fe92015-09-10 08:12:02 +0000541 } else if (I->capturesVariable()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000542 const VarDecl *Var = I->getCapturedVar();
Alexey Bataev2377fe92015-09-10 08:12:02 +0000543 QualType VarTy = Var->getType();
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800544 Address ArgAddr = ArgLVal.getAddress(CGF);
Alexey Bataev06e80f62019-05-23 18:19:54 +0000545 if (ArgLVal.getType()->isLValueReferenceType()) {
546 ArgAddr = CGF.EmitLoadOfReference(ArgLVal);
547 } else if (!VarTy->isVariablyModifiedType() || !VarTy->isPointerType()) {
548 assert(ArgLVal.getType()->isPointerType());
549 ArgAddr = CGF.EmitLoadOfPointer(
550 ArgAddr, ArgLVal.getType()->castAs<PointerType>());
Alexey Bataev2377fe92015-09-10 08:12:02 +0000551 }
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000552 if (!FO.RegisterCastedArgsOnly) {
553 LocalAddrs.insert(
554 {Args[Cnt],
555 {Var, Address(ArgAddr.getPointer(), Ctx.getDeclAlign(Var))}});
556 }
Samuel Antao4af1b7b2015-12-02 17:44:43 +0000557 } else if (I->capturesVariableByCopy()) {
558 assert(!FD->getType()->isAnyPointerType() &&
559 "Not expecting a captured pointer.");
Alexey Bataevddf3db92018-04-13 17:31:06 +0000560 const VarDecl *Var = I->getCapturedVar();
Alexey Bataev06e80f62019-05-23 18:19:54 +0000561 LocalAddrs.insert({Args[Cnt],
562 {Var, FO.UIntPtrCastRequired
563 ? castValueFromUintptr(
564 CGF, I->getLocation(), FD->getType(),
565 Args[Cnt]->getName(), ArgLVal)
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800566 : ArgLVal.getAddress(CGF)}});
Alexey Bataev2377fe92015-09-10 08:12:02 +0000567 } else {
568 // If 'this' is captured, load it into CXXThisValue.
569 assert(I->capturesThis());
Alexey Bataev1e491372018-01-23 18:44:14 +0000570 CXXThisValue = CGF.EmitLoadOfScalar(ArgLVal, I->getLocation());
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800571 LocalAddrs.insert({Args[Cnt], {nullptr, ArgLVal.getAddress(CGF)}});
Alexey Bataev2377fe92015-09-10 08:12:02 +0000572 }
Richard Trieucc3949d2016-02-18 22:34:54 +0000573 ++Cnt;
574 ++I;
Alexey Bataev2377fe92015-09-10 08:12:02 +0000575 }
576
Alexey Bataeve754b182017-08-09 19:38:53 +0000577 return F;
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000578}
579
580llvm::Function *
Alexey Bataevc33ba8c2020-01-17 14:05:40 -0500581CodeGenFunction::GenerateOpenMPCapturedStmtFunction(const CapturedStmt &S,
582 SourceLocation Loc) {
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000583 assert(
584 CapturedStmtInfo &&
585 "CapturedStmtInfo should be set when generating the captured function");
586 const CapturedDecl *CD = S.getCapturedDecl();
587 // Build the argument list.
588 bool NeedWrapperFunction =
Amy Huang53539bb2020-01-13 15:54:54 -0800589 getDebugInfo() && CGM.getCodeGenOpts().hasReducedDebugInfo();
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000590 FunctionArgList Args;
Alexey Bataev3b8d5582017-08-08 18:04:06 +0000591 llvm::MapVector<const Decl *, std::pair<const VarDecl *, Address>> LocalAddrs;
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000592 llvm::DenseMap<const Decl *, std::pair<const Expr *, llvm::Value *>> VLASizes;
Alexey Bataeve754b182017-08-09 19:38:53 +0000593 SmallString<256> Buffer;
594 llvm::raw_svector_ostream Out(Buffer);
595 Out << CapturedStmtInfo->getHelperName();
596 if (NeedWrapperFunction)
597 Out << "_debug__";
Alexey Bataev4aa19052017-08-08 16:45:36 +0000598 FunctionOptions FO(&S, !NeedWrapperFunction, /*RegisterCastedArgsOnly=*/false,
Alexey Bataevc33ba8c2020-01-17 14:05:40 -0500599 Out.str(), Loc);
Alexey Bataeve754b182017-08-09 19:38:53 +0000600 llvm::Function *F = emitOutlinedFunctionPrologue(*this, Args, LocalAddrs,
601 VLASizes, CXXThisValue, FO);
Alexey Bataev06e80f62019-05-23 18:19:54 +0000602 CodeGenFunction::OMPPrivateScope LocalScope(*this);
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000603 for (const auto &LocalAddrPair : LocalAddrs) {
604 if (LocalAddrPair.second.first) {
Alexey Bataev06e80f62019-05-23 18:19:54 +0000605 LocalScope.addPrivate(LocalAddrPair.second.first, [&LocalAddrPair]() {
606 return LocalAddrPair.second.second;
607 });
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000608 }
609 }
Alexey Bataev06e80f62019-05-23 18:19:54 +0000610 (void)LocalScope.Privatize();
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000611 for (const auto &VLASizePair : VLASizes)
612 VLASizeMap[VLASizePair.second.first] = VLASizePair.second.second;
Serge Pavlov3a561452015-12-06 14:32:39 +0000613 PGO.assignRegionCounters(GlobalDecl(CD), F);
Alexey Bataev2377fe92015-09-10 08:12:02 +0000614 CapturedStmtInfo->EmitBody(*this, CD->getBody());
Alexey Bataev06e80f62019-05-23 18:19:54 +0000615 (void)LocalScope.ForceCleanup();
Alexey Bataev2377fe92015-09-10 08:12:02 +0000616 FinishFunction(CD->getBodyRBrace());
Alexey Bataeve754b182017-08-09 19:38:53 +0000617 if (!NeedWrapperFunction)
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000618 return F;
Alexey Bataev2377fe92015-09-10 08:12:02 +0000619
Alexey Bataevefd884d2017-08-04 21:26:25 +0000620 FunctionOptions WrapperFO(&S, /*UIntPtrCastRequired=*/true,
Alexey Bataeve754b182017-08-09 19:38:53 +0000621 /*RegisterCastedArgsOnly=*/true,
Alexey Bataevc33ba8c2020-01-17 14:05:40 -0500622 CapturedStmtInfo->getHelperName(), Loc);
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000623 CodeGenFunction WrapperCGF(CGM, /*suppressNewContext=*/true);
Gheorghe-Teodor Bercead3dcf2f2018-03-14 14:17:45 +0000624 WrapperCGF.CapturedStmtInfo = CapturedStmtInfo;
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000625 Args.clear();
626 LocalAddrs.clear();
627 VLASizes.clear();
628 llvm::Function *WrapperF =
629 emitOutlinedFunctionPrologue(WrapperCGF, Args, LocalAddrs, VLASizes,
Alexey Bataeve754b182017-08-09 19:38:53 +0000630 WrapperCGF.CXXThisValue, WrapperFO);
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000631 llvm::SmallVector<llvm::Value *, 4> CallArgs;
632 for (const auto *Arg : Args) {
633 llvm::Value *CallArg;
634 auto I = LocalAddrs.find(Arg);
635 if (I != LocalAddrs.end()) {
Alexey Bataev7ba57af2017-10-17 16:47:34 +0000636 LValue LV = WrapperCGF.MakeAddrLValue(
637 I->second.second,
638 I->second.first ? I->second.first->getType() : Arg->getType(),
639 AlignmentSource::Decl);
Stephen Kellyf2ceec42018-08-09 21:08:08 +0000640 CallArg = WrapperCGF.EmitLoadOfScalar(LV, S.getBeginLoc());
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000641 } else {
642 auto EI = VLASizes.find(Arg);
Alexey Bataevddf3db92018-04-13 17:31:06 +0000643 if (EI != VLASizes.end()) {
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000644 CallArg = EI->second.second;
Alexey Bataevddf3db92018-04-13 17:31:06 +0000645 } else {
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000646 LValue LV = WrapperCGF.MakeAddrLValue(WrapperCGF.GetAddrOfLocalVar(Arg),
Ivan A. Kosarev5f8c0ca2017-10-10 09:39:32 +0000647 Arg->getType(),
648 AlignmentSource::Decl);
Stephen Kellyf2ceec42018-08-09 21:08:08 +0000649 CallArg = WrapperCGF.EmitLoadOfScalar(LV, S.getBeginLoc());
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000650 }
651 }
Alexey Bataev7ba57af2017-10-17 16:47:34 +0000652 CallArgs.emplace_back(WrapperCGF.EmitFromMemory(CallArg, Arg->getType()));
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000653 }
Alexey Bataevc33ba8c2020-01-17 14:05:40 -0500654 CGM.getOpenMPRuntime().emitOutlinedFunctionCall(WrapperCGF, Loc, F, CallArgs);
Alexey Bataev1fdfdf72017-06-29 16:43:05 +0000655 WrapperCGF.FinishFunction();
656 return WrapperF;
Alexey Bataev2377fe92015-09-10 08:12:02 +0000657}
658
Alexey Bataev9959db52014-05-06 10:08:46 +0000659//===----------------------------------------------------------------------===//
660// OpenMP Directive Emission
661//===----------------------------------------------------------------------===//
Alexey Bataev420d45b2015-04-14 05:11:24 +0000662void CodeGenFunction::EmitOMPAggregateAssign(
John McCall7f416cc2015-09-08 08:05:57 +0000663 Address DestAddr, Address SrcAddr, QualType OriginalType,
Alexey Bataevddf3db92018-04-13 17:31:06 +0000664 const llvm::function_ref<void(Address, Address)> CopyGen) {
Alexey Bataev420d45b2015-04-14 05:11:24 +0000665 // Perform element-by-element initialization.
666 QualType ElementTy;
John McCall7f416cc2015-09-08 08:05:57 +0000667
668 // Drill down to the base element type on both arrays.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000669 const ArrayType *ArrayTy = OriginalType->getAsArrayTypeUnsafe();
670 llvm::Value *NumElements = emitArrayLength(ArrayTy, ElementTy, DestAddr);
John McCall7f416cc2015-09-08 08:05:57 +0000671 SrcAddr = Builder.CreateElementBitCast(SrcAddr, DestAddr.getElementType());
672
Alexey Bataevddf3db92018-04-13 17:31:06 +0000673 llvm::Value *SrcBegin = SrcAddr.getPointer();
674 llvm::Value *DestBegin = DestAddr.getPointer();
Alexey Bataev420d45b2015-04-14 05:11:24 +0000675 // Cast from pointer to array type to pointer to single element.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000676 llvm::Value *DestEnd = Builder.CreateGEP(DestBegin, NumElements);
Alexey Bataev420d45b2015-04-14 05:11:24 +0000677 // The basic structure here is a while-do loop.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000678 llvm::BasicBlock *BodyBB = createBasicBlock("omp.arraycpy.body");
679 llvm::BasicBlock *DoneBB = createBasicBlock("omp.arraycpy.done");
680 llvm::Value *IsEmpty =
Alexey Bataev420d45b2015-04-14 05:11:24 +0000681 Builder.CreateICmpEQ(DestBegin, DestEnd, "omp.arraycpy.isempty");
682 Builder.CreateCondBr(IsEmpty, DoneBB, BodyBB);
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000683
Alexey Bataev420d45b2015-04-14 05:11:24 +0000684 // Enter the loop body, making that address the current address.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000685 llvm::BasicBlock *EntryBB = Builder.GetInsertBlock();
Alexey Bataev420d45b2015-04-14 05:11:24 +0000686 EmitBlock(BodyBB);
John McCall7f416cc2015-09-08 08:05:57 +0000687
688 CharUnits ElementSize = getContext().getTypeSizeInChars(ElementTy);
689
690 llvm::PHINode *SrcElementPHI =
691 Builder.CreatePHI(SrcBegin->getType(), 2, "omp.arraycpy.srcElementPast");
692 SrcElementPHI->addIncoming(SrcBegin, EntryBB);
693 Address SrcElementCurrent =
694 Address(SrcElementPHI,
695 SrcAddr.getAlignment().alignmentOfArrayElement(ElementSize));
696
697 llvm::PHINode *DestElementPHI =
698 Builder.CreatePHI(DestBegin->getType(), 2, "omp.arraycpy.destElementPast");
699 DestElementPHI->addIncoming(DestBegin, EntryBB);
700 Address DestElementCurrent =
701 Address(DestElementPHI,
702 DestAddr.getAlignment().alignmentOfArrayElement(ElementSize));
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000703
Alexey Bataev420d45b2015-04-14 05:11:24 +0000704 // Emit copy.
705 CopyGen(DestElementCurrent, SrcElementCurrent);
706
707 // Shift the address forward by one element.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000708 llvm::Value *DestElementNext = Builder.CreateConstGEP1_32(
John McCall7f416cc2015-09-08 08:05:57 +0000709 DestElementPHI, /*Idx0=*/1, "omp.arraycpy.dest.element");
Alexey Bataevddf3db92018-04-13 17:31:06 +0000710 llvm::Value *SrcElementNext = Builder.CreateConstGEP1_32(
John McCall7f416cc2015-09-08 08:05:57 +0000711 SrcElementPHI, /*Idx0=*/1, "omp.arraycpy.src.element");
Alexey Bataev420d45b2015-04-14 05:11:24 +0000712 // Check whether we've reached the end.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000713 llvm::Value *Done =
Alexey Bataev420d45b2015-04-14 05:11:24 +0000714 Builder.CreateICmpEQ(DestElementNext, DestEnd, "omp.arraycpy.done");
715 Builder.CreateCondBr(Done, DoneBB, BodyBB);
John McCall7f416cc2015-09-08 08:05:57 +0000716 DestElementPHI->addIncoming(DestElementNext, Builder.GetInsertBlock());
717 SrcElementPHI->addIncoming(SrcElementNext, Builder.GetInsertBlock());
Alexey Bataev420d45b2015-04-14 05:11:24 +0000718
719 // Done.
720 EmitBlock(DoneBB, /*IsFinished=*/true);
721}
722
John McCall7f416cc2015-09-08 08:05:57 +0000723void CodeGenFunction::EmitOMPCopy(QualType OriginalType, Address DestAddr,
724 Address SrcAddr, const VarDecl *DestVD,
Alexey Bataev420d45b2015-04-14 05:11:24 +0000725 const VarDecl *SrcVD, const Expr *Copy) {
726 if (OriginalType->isArrayType()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000727 const auto *BO = dyn_cast<BinaryOperator>(Copy);
Alexey Bataev420d45b2015-04-14 05:11:24 +0000728 if (BO && BO->getOpcode() == BO_Assign) {
729 // Perform simple memcpy for simple copying.
Ivan A. Kosarev1860b522018-01-25 14:21:55 +0000730 LValue Dest = MakeAddrLValue(DestAddr, OriginalType);
731 LValue Src = MakeAddrLValue(SrcAddr, OriginalType);
732 EmitAggregateAssign(Dest, Src, OriginalType);
Alexey Bataev420d45b2015-04-14 05:11:24 +0000733 } else {
734 // For arrays with complex element types perform element by element
735 // copying.
John McCall7f416cc2015-09-08 08:05:57 +0000736 EmitOMPAggregateAssign(
Alexey Bataev420d45b2015-04-14 05:11:24 +0000737 DestAddr, SrcAddr, OriginalType,
John McCall7f416cc2015-09-08 08:05:57 +0000738 [this, Copy, SrcVD, DestVD](Address DestElement, Address SrcElement) {
Alexey Bataev420d45b2015-04-14 05:11:24 +0000739 // Working with the single array element, so have to remap
740 // destination and source variables to corresponding array
741 // elements.
John McCall7f416cc2015-09-08 08:05:57 +0000742 CodeGenFunction::OMPPrivateScope Remap(*this);
Alexey Bataevddf3db92018-04-13 17:31:06 +0000743 Remap.addPrivate(DestVD, [DestElement]() { return DestElement; });
744 Remap.addPrivate(SrcVD, [SrcElement]() { return SrcElement; });
Alexey Bataev420d45b2015-04-14 05:11:24 +0000745 (void)Remap.Privatize();
John McCall7f416cc2015-09-08 08:05:57 +0000746 EmitIgnoredExpr(Copy);
Alexey Bataev420d45b2015-04-14 05:11:24 +0000747 });
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000748 }
Alexey Bataev420d45b2015-04-14 05:11:24 +0000749 } else {
750 // Remap pseudo source variable to private copy.
John McCall7f416cc2015-09-08 08:05:57 +0000751 CodeGenFunction::OMPPrivateScope Remap(*this);
Alexey Bataevddf3db92018-04-13 17:31:06 +0000752 Remap.addPrivate(SrcVD, [SrcAddr]() { return SrcAddr; });
753 Remap.addPrivate(DestVD, [DestAddr]() { return DestAddr; });
Alexey Bataev420d45b2015-04-14 05:11:24 +0000754 (void)Remap.Privatize();
755 // Emit copying of the whole variable.
John McCall7f416cc2015-09-08 08:05:57 +0000756 EmitIgnoredExpr(Copy);
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000757 }
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000758}
759
Alexey Bataev69c62a92015-04-15 04:52:20 +0000760bool CodeGenFunction::EmitOMPFirstprivateClause(const OMPExecutableDirective &D,
761 OMPPrivateScope &PrivateScope) {
Alexey Bataev8ef31412015-12-18 07:58:25 +0000762 if (!HaveInsertPoint())
763 return false;
Alexey Bataev1af5bd52019-03-05 17:47:18 +0000764 bool DeviceConstTarget =
765 getLangOpts().OpenMPIsDevice &&
766 isOpenMPTargetExecutionDirective(D.getDirectiveKind());
Alexey Bataevcd8b6a22016-02-15 08:07:17 +0000767 bool FirstprivateIsLastprivate = false;
Alexey Bataev46978742020-01-30 10:46:11 -0500768 llvm::DenseMap<const VarDecl *, OpenMPLastprivateModifier> Lastprivates;
Alexey Bataevcd8b6a22016-02-15 08:07:17 +0000769 for (const auto *C : D.getClausesOfKind<OMPLastprivateClause>()) {
770 for (const auto *D : C->varlists())
Alexey Bataev46978742020-01-30 10:46:11 -0500771 Lastprivates.try_emplace(
772 cast<VarDecl>(cast<DeclRefExpr>(D)->getDecl())->getCanonicalDecl(),
773 C->getKind());
Alexey Bataevcd8b6a22016-02-15 08:07:17 +0000774 }
Alexey Bataev69c62a92015-04-15 04:52:20 +0000775 llvm::DenseSet<const VarDecl *> EmittedAsFirstprivate;
Alexey Bataev475a7442018-01-12 19:39:11 +0000776 llvm::SmallVector<OpenMPDirectiveKind, 4> CaptureRegions;
777 getOpenMPCaptureRegions(CaptureRegions, D.getDirectiveKind());
778 // Force emission of the firstprivate copy if the directive does not emit
779 // outlined function, like omp for, omp simd, omp distribute etc.
780 bool MustEmitFirstprivateCopy =
781 CaptureRegions.size() == 1 && CaptureRegions.back() == OMPD_unknown;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +0000782 for (const auto *C : D.getClausesOfKind<OMPFirstprivateClause>()) {
Alexey Bataev46978742020-01-30 10:46:11 -0500783 const auto *IRef = C->varlist_begin();
784 const auto *InitsRef = C->inits().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +0000785 for (const Expr *IInit : C->private_copies()) {
786 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev7ace49d2016-05-17 08:55:33 +0000787 bool ThisFirstprivateIsLastprivate =
788 Lastprivates.count(OrigVD->getCanonicalDecl()) > 0;
Alexey Bataevddf3db92018-04-13 17:31:06 +0000789 const FieldDecl *FD = CapturedStmtInfo->lookup(OrigVD);
Alexey Bataev9c397812019-04-03 17:57:06 +0000790 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(IInit)->getDecl());
Alexey Bataev475a7442018-01-12 19:39:11 +0000791 if (!MustEmitFirstprivateCopy && !ThisFirstprivateIsLastprivate && FD &&
Alexey Bataev9c397812019-04-03 17:57:06 +0000792 !FD->getType()->isReferenceType() &&
793 (!VD || !VD->hasAttr<OMPAllocateDeclAttr>())) {
Alexey Bataev7ace49d2016-05-17 08:55:33 +0000794 EmittedAsFirstprivate.insert(OrigVD->getCanonicalDecl());
795 ++IRef;
796 ++InitsRef;
797 continue;
798 }
Alexey Bataev1af5bd52019-03-05 17:47:18 +0000799 // Do not emit copy for firstprivate constant variables in target regions,
800 // captured by reference.
801 if (DeviceConstTarget && OrigVD->getType().isConstant(getContext()) &&
Alexey Bataev9c397812019-04-03 17:57:06 +0000802 FD && FD->getType()->isReferenceType() &&
803 (!VD || !VD->hasAttr<OMPAllocateDeclAttr>())) {
Alexey Bataev1af5bd52019-03-05 17:47:18 +0000804 (void)CGM.getOpenMPRuntime().registerTargetFirstprivateCopy(*this,
805 OrigVD);
806 ++IRef;
807 ++InitsRef;
808 continue;
809 }
Alexey Bataevcd8b6a22016-02-15 08:07:17 +0000810 FirstprivateIsLastprivate =
Alexey Bataev7ace49d2016-05-17 08:55:33 +0000811 FirstprivateIsLastprivate || ThisFirstprivateIsLastprivate;
Alexey Bataevcd8b6a22016-02-15 08:07:17 +0000812 if (EmittedAsFirstprivate.insert(OrigVD->getCanonicalDecl()).second) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000813 const auto *VDInit =
814 cast<VarDecl>(cast<DeclRefExpr>(*InitsRef)->getDecl());
Alexey Bataev69c62a92015-04-15 04:52:20 +0000815 bool IsRegistered;
Bruno Ricci5fc4db72018-12-21 14:10:18 +0000816 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(OrigVD),
Alexey Bataev7ace49d2016-05-17 08:55:33 +0000817 /*RefersToEnclosingVariableOrCapture=*/FD != nullptr,
818 (*IRef)->getType(), VK_LValue, (*IRef)->getExprLoc());
Alexey Bataeve0ef04f2019-05-23 22:30:43 +0000819 LValue OriginalLVal;
820 if (!FD) {
821 // Check if the firstprivate variable is just a constant value.
822 ConstantEmission CE = tryEmitAsConstant(&DRE);
823 if (CE && !CE.isReference()) {
824 // Constant value, no need to create a copy.
825 ++IRef;
826 ++InitsRef;
827 continue;
828 }
829 if (CE && CE.isReference()) {
830 OriginalLVal = CE.getReferenceLValue(*this, &DRE);
831 } else {
832 assert(!CE && "Expected non-constant firstprivate.");
833 OriginalLVal = EmitLValue(&DRE);
834 }
835 } else {
836 OriginalLVal = EmitLValue(&DRE);
837 }
Alexey Bataevfeddd642016-04-22 09:05:03 +0000838 QualType Type = VD->getType();
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000839 if (Type->isArrayType()) {
Alexey Bataev69c62a92015-04-15 04:52:20 +0000840 // Emit VarDecl with copy init for arrays.
841 // Get the address of the original variable captured in current
842 // captured region.
Alexey Bataevddf3db92018-04-13 17:31:06 +0000843 IsRegistered = PrivateScope.addPrivate(
844 OrigVD, [this, VD, Type, OriginalLVal, VDInit]() {
845 AutoVarEmission Emission = EmitAutoVarAlloca(*VD);
846 const Expr *Init = VD->getInit();
847 if (!isa<CXXConstructExpr>(Init) ||
848 isTrivialInitializer(Init)) {
849 // Perform simple memcpy.
850 LValue Dest =
851 MakeAddrLValue(Emission.getAllocatedAddress(), Type);
852 EmitAggregateAssign(Dest, OriginalLVal, Type);
853 } else {
854 EmitOMPAggregateAssign(
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800855 Emission.getAllocatedAddress(),
856 OriginalLVal.getAddress(*this), Type,
Alexey Bataevddf3db92018-04-13 17:31:06 +0000857 [this, VDInit, Init](Address DestElement,
858 Address SrcElement) {
859 // Clean up any temporaries needed by the
860 // initialization.
861 RunCleanupsScope InitScope(*this);
862 // Emit initialization for single element.
863 setAddrOfLocalVar(VDInit, SrcElement);
864 EmitAnyExprToMem(Init, DestElement,
865 Init->getType().getQualifiers(),
866 /*IsInitializer*/ false);
867 LocalDeclMap.erase(VDInit);
868 });
869 }
870 EmitAutoVarCleanups(Emission);
871 return Emission.getAllocatedAddress();
872 });
Alexey Bataev69c62a92015-04-15 04:52:20 +0000873 } else {
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800874 Address OriginalAddr = OriginalLVal.getAddress(*this);
Alexey Bataev46978742020-01-30 10:46:11 -0500875 IsRegistered =
876 PrivateScope.addPrivate(OrigVD, [this, VDInit, OriginalAddr, VD,
877 ThisFirstprivateIsLastprivate,
878 OrigVD, &Lastprivates, IRef]() {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000879 // Emit private VarDecl with copy init.
880 // Remap temp VDInit variable to the address of the original
881 // variable (for proper handling of captured global variables).
882 setAddrOfLocalVar(VDInit, OriginalAddr);
883 EmitDecl(*VD);
884 LocalDeclMap.erase(VDInit);
Alexey Bataev46978742020-01-30 10:46:11 -0500885 if (ThisFirstprivateIsLastprivate &&
886 Lastprivates[OrigVD->getCanonicalDecl()] ==
887 OMPC_LASTPRIVATE_conditional) {
888 // Create/init special variable for lastprivate conditionals.
889 Address VDAddr =
890 CGM.getOpenMPRuntime().emitLastprivateConditionalInit(
891 *this, OrigVD);
892 llvm::Value *V = EmitLoadOfScalar(
893 MakeAddrLValue(GetAddrOfLocalVar(VD), (*IRef)->getType(),
894 AlignmentSource::Decl),
895 (*IRef)->getExprLoc());
896 EmitStoreOfScalar(V,
897 MakeAddrLValue(VDAddr, (*IRef)->getType(),
898 AlignmentSource::Decl));
899 LocalDeclMap.erase(VD);
900 setAddrOfLocalVar(VD, VDAddr);
901 return VDAddr;
902 }
Alexey Bataevddf3db92018-04-13 17:31:06 +0000903 return GetAddrOfLocalVar(VD);
904 });
Alexey Bataev69c62a92015-04-15 04:52:20 +0000905 }
906 assert(IsRegistered &&
907 "firstprivate var already registered as private");
908 // Silence the warning about unused variable.
909 (void)IsRegistered;
910 }
Richard Trieucc3949d2016-02-18 22:34:54 +0000911 ++IRef;
912 ++InitsRef;
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000913 }
914 }
Alexey Bataevcd8b6a22016-02-15 08:07:17 +0000915 return FirstprivateIsLastprivate && !EmittedAsFirstprivate.empty();
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000916}
917
Alexey Bataev03b340a2014-10-21 03:16:40 +0000918void CodeGenFunction::EmitOMPPrivateClause(
919 const OMPExecutableDirective &D,
920 CodeGenFunction::OMPPrivateScope &PrivateScope) {
Alexey Bataev8ef31412015-12-18 07:58:25 +0000921 if (!HaveInsertPoint())
922 return;
Alexey Bataev50a64582015-04-22 12:24:45 +0000923 llvm::DenseSet<const VarDecl *> EmittedAsPrivate;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +0000924 for (const auto *C : D.getClausesOfKind<OMPPrivateClause>()) {
Alexey Bataev03b340a2014-10-21 03:16:40 +0000925 auto IRef = C->varlist_begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +0000926 for (const Expr *IInit : C->private_copies()) {
927 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev50a64582015-04-22 12:24:45 +0000928 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
Alexey Bataevddf3db92018-04-13 17:31:06 +0000929 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(IInit)->getDecl());
930 bool IsRegistered = PrivateScope.addPrivate(OrigVD, [this, VD]() {
931 // Emit private VarDecl with copy init.
932 EmitDecl(*VD);
933 return GetAddrOfLocalVar(VD);
934 });
Alexey Bataev50a64582015-04-22 12:24:45 +0000935 assert(IsRegistered && "private var already registered as private");
936 // Silence the warning about unused variable.
937 (void)IsRegistered;
938 }
Alexey Bataev03b340a2014-10-21 03:16:40 +0000939 ++IRef;
940 }
941 }
942}
943
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000944bool CodeGenFunction::EmitOMPCopyinClause(const OMPExecutableDirective &D) {
Alexey Bataev8ef31412015-12-18 07:58:25 +0000945 if (!HaveInsertPoint())
946 return false;
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000947 // threadprivate_var1 = master_threadprivate_var1;
948 // operator=(threadprivate_var2, master_threadprivate_var2);
949 // ...
950 // __kmpc_barrier(&loc, global_tid);
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000951 llvm::DenseSet<const VarDecl *> CopiedVars;
952 llvm::BasicBlock *CopyBegin = nullptr, *CopyEnd = nullptr;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +0000953 for (const auto *C : D.getClausesOfKind<OMPCopyinClause>()) {
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000954 auto IRef = C->varlist_begin();
955 auto ISrcRef = C->source_exprs().begin();
956 auto IDestRef = C->destination_exprs().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +0000957 for (const Expr *AssignOp : C->assignment_ops()) {
958 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000959 QualType Type = VD->getType();
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000960 if (CopiedVars.insert(VD->getCanonicalDecl()).second) {
Samuel Antao9c75cfe2015-07-27 16:38:06 +0000961 // Get the address of the master variable. If we are emitting code with
962 // TLS support, the address is passed from the master as field in the
963 // captured declaration.
John McCall7f416cc2015-09-08 08:05:57 +0000964 Address MasterAddr = Address::invalid();
Samuel Antao9c75cfe2015-07-27 16:38:06 +0000965 if (getLangOpts().OpenMPUseTLS &&
966 getContext().getTargetInfo().isTLSSupported()) {
967 assert(CapturedStmtInfo->lookup(VD) &&
968 "Copyin threadprivates should have been captured!");
Bruno Ricci5fc4db72018-12-21 14:10:18 +0000969 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(VD), true,
970 (*IRef)->getType(), VK_LValue, (*IRef)->getExprLoc());
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800971 MasterAddr = EmitLValue(&DRE).getAddress(*this);
Alexey Bataev2377fe92015-09-10 08:12:02 +0000972 LocalDeclMap.erase(VD);
Samuel Antao9c75cfe2015-07-27 16:38:06 +0000973 } else {
John McCall7f416cc2015-09-08 08:05:57 +0000974 MasterAddr =
975 Address(VD->isStaticLocal() ? CGM.getStaticLocalDeclAddress(VD)
976 : CGM.GetAddrOfGlobal(VD),
977 getContext().getDeclAlign(VD));
Samuel Antao9c75cfe2015-07-27 16:38:06 +0000978 }
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000979 // Get the address of the threadprivate variable.
Akira Hatanakaf139ae32019-12-03 15:17:01 -0800980 Address PrivateAddr = EmitLValue(*IRef).getAddress(*this);
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000981 if (CopiedVars.size() == 1) {
982 // At first check if current thread is a master thread. If it is, no
983 // need to copy data.
984 CopyBegin = createBasicBlock("copyin.not.master");
985 CopyEnd = createBasicBlock("copyin.not.master.end");
986 Builder.CreateCondBr(
987 Builder.CreateICmpNE(
John McCall7f416cc2015-09-08 08:05:57 +0000988 Builder.CreatePtrToInt(MasterAddr.getPointer(), CGM.IntPtrTy),
Alexey Bataevddf3db92018-04-13 17:31:06 +0000989 Builder.CreatePtrToInt(PrivateAddr.getPointer(),
990 CGM.IntPtrTy)),
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000991 CopyBegin, CopyEnd);
992 EmitBlock(CopyBegin);
993 }
Alexey Bataevddf3db92018-04-13 17:31:06 +0000994 const auto *SrcVD =
995 cast<VarDecl>(cast<DeclRefExpr>(*ISrcRef)->getDecl());
996 const auto *DestVD =
997 cast<VarDecl>(cast<DeclRefExpr>(*IDestRef)->getDecl());
John McCall7f416cc2015-09-08 08:05:57 +0000998 EmitOMPCopy(Type, PrivateAddr, MasterAddr, DestVD, SrcVD, AssignOp);
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000999 }
1000 ++IRef;
1001 ++ISrcRef;
1002 ++IDestRef;
1003 }
1004 }
1005 if (CopyEnd) {
1006 // Exit out of copying procedure for non-master thread.
1007 EmitBlock(CopyEnd, /*IsFinished=*/true);
1008 return true;
1009 }
1010 return false;
1011}
1012
Alexey Bataev38e89532015-04-16 04:54:05 +00001013bool CodeGenFunction::EmitOMPLastprivateClauseInit(
1014 const OMPExecutableDirective &D, OMPPrivateScope &PrivateScope) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001015 if (!HaveInsertPoint())
1016 return false;
Alexey Bataev38e89532015-04-16 04:54:05 +00001017 bool HasAtLeastOneLastprivate = false;
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001018 llvm::DenseSet<const VarDecl *> SIMDLCVs;
1019 if (isOpenMPSimdDirective(D.getDirectiveKind())) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001020 const auto *LoopDirective = cast<OMPLoopDirective>(&D);
1021 for (const Expr *C : LoopDirective->counters()) {
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001022 SIMDLCVs.insert(
1023 cast<VarDecl>(cast<DeclRefExpr>(C)->getDecl())->getCanonicalDecl());
1024 }
1025 }
Alexey Bataev38e89532015-04-16 04:54:05 +00001026 llvm::DenseSet<const VarDecl *> AlreadyEmittedVars;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001027 for (const auto *C : D.getClausesOfKind<OMPLastprivateClause>()) {
Alexey Bataevd130fd12015-05-13 10:23:02 +00001028 HasAtLeastOneLastprivate = true;
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00001029 if (isOpenMPTaskLoopDirective(D.getDirectiveKind()) &&
1030 !getLangOpts().OpenMPSimd)
Alexey Bataevf93095a2016-05-05 08:46:22 +00001031 break;
Alexey Bataev46978742020-01-30 10:46:11 -05001032 const auto *IRef = C->varlist_begin();
1033 const auto *IDestRef = C->destination_exprs().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00001034 for (const Expr *IInit : C->private_copies()) {
Alexey Bataev38e89532015-04-16 04:54:05 +00001035 // Keep the address of the original variable for future update at the end
1036 // of the loop.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001037 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataevf93095a2016-05-05 08:46:22 +00001038 // Taskloops do not require additional initialization, it is done in
1039 // runtime support library.
Alexey Bataev38e89532015-04-16 04:54:05 +00001040 if (AlreadyEmittedVars.insert(OrigVD->getCanonicalDecl()).second) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001041 const auto *DestVD =
1042 cast<VarDecl>(cast<DeclRefExpr>(*IDestRef)->getDecl());
1043 PrivateScope.addPrivate(DestVD, [this, OrigVD, IRef]() {
Bruno Ricci5fc4db72018-12-21 14:10:18 +00001044 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(OrigVD),
1045 /*RefersToEnclosingVariableOrCapture=*/
1046 CapturedStmtInfo->lookup(OrigVD) != nullptr,
1047 (*IRef)->getType(), VK_LValue, (*IRef)->getExprLoc());
Akira Hatanakaf139ae32019-12-03 15:17:01 -08001048 return EmitLValue(&DRE).getAddress(*this);
Alexey Bataev38e89532015-04-16 04:54:05 +00001049 });
1050 // Check if the variable is also a firstprivate: in this case IInit is
1051 // not generated. Initialization of this variable will happen in codegen
1052 // for 'firstprivate' clause.
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001053 if (IInit && !SIMDLCVs.count(OrigVD->getCanonicalDecl())) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001054 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(IInit)->getDecl());
Alexey Bataev46978742020-01-30 10:46:11 -05001055 bool IsRegistered = PrivateScope.addPrivate(OrigVD, [this, VD, C,
1056 OrigVD]() {
1057 if (C->getKind() == OMPC_LASTPRIVATE_conditional) {
1058 Address VDAddr =
1059 CGM.getOpenMPRuntime().emitLastprivateConditionalInit(*this,
1060 OrigVD);
1061 setAddrOfLocalVar(VD, VDAddr);
1062 return VDAddr;
1063 }
Alexey Bataevf93095a2016-05-05 08:46:22 +00001064 // Emit private VarDecl with copy init.
1065 EmitDecl(*VD);
1066 return GetAddrOfLocalVar(VD);
1067 });
Alexey Bataevd130fd12015-05-13 10:23:02 +00001068 assert(IsRegistered &&
1069 "lastprivate var already registered as private");
1070 (void)IsRegistered;
1071 }
Alexey Bataev38e89532015-04-16 04:54:05 +00001072 }
Richard Trieucc3949d2016-02-18 22:34:54 +00001073 ++IRef;
1074 ++IDestRef;
Alexey Bataev38e89532015-04-16 04:54:05 +00001075 }
1076 }
1077 return HasAtLeastOneLastprivate;
1078}
1079
1080void CodeGenFunction::EmitOMPLastprivateClauseFinal(
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001081 const OMPExecutableDirective &D, bool NoFinals,
1082 llvm::Value *IsLastIterCond) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001083 if (!HaveInsertPoint())
1084 return;
Alexey Bataev38e89532015-04-16 04:54:05 +00001085 // Emit following code:
1086 // if (<IsLastIterCond>) {
1087 // orig_var1 = private_orig_var1;
1088 // ...
1089 // orig_varn = private_orig_varn;
1090 // }
Alexey Bataevfc087ec2015-06-16 13:14:42 +00001091 llvm::BasicBlock *ThenBB = nullptr;
1092 llvm::BasicBlock *DoneBB = nullptr;
1093 if (IsLastIterCond) {
Alexey Bataeva58da1a2019-12-27 09:44:43 -05001094 // Emit implicit barrier if at least one lastprivate conditional is found
1095 // and this is not a simd mode.
1096 if (!getLangOpts().OpenMPSimd &&
1097 llvm::any_of(D.getClausesOfKind<OMPLastprivateClause>(),
1098 [](const OMPLastprivateClause *C) {
1099 return C->getKind() == OMPC_LASTPRIVATE_conditional;
1100 })) {
1101 CGM.getOpenMPRuntime().emitBarrierCall(*this, D.getBeginLoc(),
1102 OMPD_unknown,
1103 /*EmitChecks=*/false,
1104 /*ForceSimpleCall=*/true);
1105 }
Alexey Bataevfc087ec2015-06-16 13:14:42 +00001106 ThenBB = createBasicBlock(".omp.lastprivate.then");
1107 DoneBB = createBasicBlock(".omp.lastprivate.done");
1108 Builder.CreateCondBr(IsLastIterCond, ThenBB, DoneBB);
1109 EmitBlock(ThenBB);
1110 }
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001111 llvm::DenseSet<const VarDecl *> AlreadyEmittedVars;
1112 llvm::DenseMap<const VarDecl *, const Expr *> LoopCountersAndUpdates;
Alexey Bataevddf3db92018-04-13 17:31:06 +00001113 if (const auto *LoopDirective = dyn_cast<OMPLoopDirective>(&D)) {
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001114 auto IC = LoopDirective->counters().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00001115 for (const Expr *F : LoopDirective->finals()) {
1116 const auto *D =
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001117 cast<VarDecl>(cast<DeclRefExpr>(*IC)->getDecl())->getCanonicalDecl();
1118 if (NoFinals)
1119 AlreadyEmittedVars.insert(D);
1120 else
1121 LoopCountersAndUpdates[D] = F;
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001122 ++IC;
Alexey Bataev7a228ff2015-05-21 07:59:51 +00001123 }
1124 }
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001125 for (const auto *C : D.getClausesOfKind<OMPLastprivateClause>()) {
1126 auto IRef = C->varlist_begin();
1127 auto ISrcRef = C->source_exprs().begin();
1128 auto IDestRef = C->destination_exprs().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00001129 for (const Expr *AssignOp : C->assignment_ops()) {
1130 const auto *PrivateVD =
1131 cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001132 QualType Type = PrivateVD->getType();
Alexey Bataevddf3db92018-04-13 17:31:06 +00001133 const auto *CanonicalVD = PrivateVD->getCanonicalDecl();
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001134 if (AlreadyEmittedVars.insert(CanonicalVD).second) {
1135 // If lastprivate variable is a loop control variable for loop-based
1136 // directive, update its value before copyin back to original
1137 // variable.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001138 if (const Expr *FinalExpr = LoopCountersAndUpdates.lookup(CanonicalVD))
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001139 EmitIgnoredExpr(FinalExpr);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001140 const auto *SrcVD =
1141 cast<VarDecl>(cast<DeclRefExpr>(*ISrcRef)->getDecl());
1142 const auto *DestVD =
1143 cast<VarDecl>(cast<DeclRefExpr>(*IDestRef)->getDecl());
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001144 // Get the address of the private variable.
1145 Address PrivateAddr = GetAddrOfLocalVar(PrivateVD);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001146 if (const auto *RefTy = PrivateVD->getType()->getAs<ReferenceType>())
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001147 PrivateAddr =
John McCall7f416cc2015-09-08 08:05:57 +00001148 Address(Builder.CreateLoad(PrivateAddr),
Eli Friedman62f3ef22020-05-18 11:29:11 -07001149 CGM.getNaturalTypeAlignment(RefTy->getPointeeType()));
Alexey Bataeva58da1a2019-12-27 09:44:43 -05001150 // Store the last value to the private copy in the last iteration.
1151 if (C->getKind() == OMPC_LASTPRIVATE_conditional)
1152 CGM.getOpenMPRuntime().emitLastprivateConditionalFinalUpdate(
1153 *this, MakeAddrLValue(PrivateAddr, (*IRef)->getType()), PrivateVD,
1154 (*IRef)->getExprLoc());
1155 // Get the address of the original variable.
1156 Address OriginalAddr = GetAddrOfLocalVar(DestVD);
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001157 EmitOMPCopy(Type, OriginalAddr, PrivateAddr, DestVD, SrcVD, AssignOp);
Alexey Bataev38e89532015-04-16 04:54:05 +00001158 }
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001159 ++IRef;
1160 ++ISrcRef;
1161 ++IDestRef;
Alexey Bataev38e89532015-04-16 04:54:05 +00001162 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00001163 if (const Expr *PostUpdate = C->getPostUpdateExpr())
Alexey Bataev005248a2016-02-25 05:25:57 +00001164 EmitIgnoredExpr(PostUpdate);
Alexey Bataev38e89532015-04-16 04:54:05 +00001165 }
Alexey Bataev8ffcc942016-02-18 13:48:15 +00001166 if (IsLastIterCond)
Alexey Bataevfc087ec2015-06-16 13:14:42 +00001167 EmitBlock(DoneBB, /*IsFinished=*/true);
Alexey Bataev38e89532015-04-16 04:54:05 +00001168}
1169
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001170void CodeGenFunction::EmitOMPReductionClauseInit(
1171 const OMPExecutableDirective &D,
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04001172 CodeGenFunction::OMPPrivateScope &PrivateScope, bool ForInscan) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001173 if (!HaveInsertPoint())
1174 return;
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001175 SmallVector<const Expr *, 4> Shareds;
1176 SmallVector<const Expr *, 4> Privates;
1177 SmallVector<const Expr *, 4> ReductionOps;
1178 SmallVector<const Expr *, 4> LHSs;
1179 SmallVector<const Expr *, 4> RHSs;
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04001180 OMPTaskDataTy Data;
1181 SmallVector<const Expr *, 4> TaskLHSs;
1182 SmallVector<const Expr *, 4> TaskRHSs;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001183 for (const auto *C : D.getClausesOfKind<OMPReductionClause>()) {
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04001184 if (ForInscan != (C->getModifier() == OMPC_REDUCTION_inscan))
1185 continue;
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04001186 Shareds.append(C->varlist_begin(), C->varlist_end());
1187 Privates.append(C->privates().begin(), C->privates().end());
1188 ReductionOps.append(C->reduction_ops().begin(), C->reduction_ops().end());
1189 LHSs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
1190 RHSs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
1191 if (C->getModifier() == OMPC_REDUCTION_task) {
1192 Data.ReductionVars.append(C->privates().begin(), C->privates().end());
1193 Data.ReductionOrigs.append(C->varlist_begin(), C->varlist_end());
1194 Data.ReductionCopies.append(C->privates().begin(), C->privates().end());
1195 Data.ReductionOps.append(C->reduction_ops().begin(),
1196 C->reduction_ops().end());
1197 TaskLHSs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
1198 TaskRHSs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001199 }
1200 }
Alexey Bataeve9bfa1d2020-04-23 13:27:03 -04001201 ReductionCodeGen RedCG(Shareds, Shareds, Privates, ReductionOps);
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001202 unsigned Count = 0;
Alexey Bataeve9bfa1d2020-04-23 13:27:03 -04001203 auto *ILHS = LHSs.begin();
1204 auto *IRHS = RHSs.begin();
1205 auto *IPriv = Privates.begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00001206 for (const Expr *IRef : Shareds) {
1207 const auto *PrivateVD = cast<VarDecl>(cast<DeclRefExpr>(*IPriv)->getDecl());
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001208 // Emit private VarDecl with reduction init.
Alexey Bataeve9bfa1d2020-04-23 13:27:03 -04001209 RedCG.emitSharedOrigLValue(*this, Count);
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001210 RedCG.emitAggregateType(*this, Count);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001211 AutoVarEmission Emission = EmitAutoVarAlloca(*PrivateVD);
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001212 RedCG.emitInitialization(*this, Count, Emission.getAllocatedAddress(),
1213 RedCG.getSharedLValue(Count),
1214 [&Emission](CodeGenFunction &CGF) {
1215 CGF.EmitAutoVarInit(Emission);
1216 return true;
1217 });
1218 EmitAutoVarCleanups(Emission);
1219 Address BaseAddr = RedCG.adjustPrivateAddress(
1220 *this, Count, Emission.getAllocatedAddress());
1221 bool IsRegistered = PrivateScope.addPrivate(
Alexey Bataevddf3db92018-04-13 17:31:06 +00001222 RedCG.getBaseDecl(Count), [BaseAddr]() { return BaseAddr; });
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001223 assert(IsRegistered && "private var already registered as private");
1224 // Silence the warning about unused variable.
1225 (void)IsRegistered;
1226
Alexey Bataevddf3db92018-04-13 17:31:06 +00001227 const auto *LHSVD = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
1228 const auto *RHSVD = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
Jonas Hahnfeld4525c822017-10-23 19:01:35 +00001229 QualType Type = PrivateVD->getType();
1230 bool isaOMPArraySectionExpr = isa<OMPArraySectionExpr>(IRef);
1231 if (isaOMPArraySectionExpr && Type->isVariablyModifiedType()) {
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001232 // Store the address of the original variable associated with the LHS
1233 // implicit variable.
Akira Hatanakaf139ae32019-12-03 15:17:01 -08001234 PrivateScope.addPrivate(LHSVD, [&RedCG, Count, this]() {
1235 return RedCG.getSharedLValue(Count).getAddress(*this);
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001236 });
Alexey Bataevddf3db92018-04-13 17:31:06 +00001237 PrivateScope.addPrivate(
1238 RHSVD, [this, PrivateVD]() { return GetAddrOfLocalVar(PrivateVD); });
Jonas Hahnfeld4525c822017-10-23 19:01:35 +00001239 } else if ((isaOMPArraySectionExpr && Type->isScalarType()) ||
1240 isa<ArraySubscriptExpr>(IRef)) {
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001241 // Store the address of the original variable associated with the LHS
1242 // implicit variable.
Akira Hatanakaf139ae32019-12-03 15:17:01 -08001243 PrivateScope.addPrivate(LHSVD, [&RedCG, Count, this]() {
1244 return RedCG.getSharedLValue(Count).getAddress(*this);
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001245 });
Alexey Bataevddf3db92018-04-13 17:31:06 +00001246 PrivateScope.addPrivate(RHSVD, [this, PrivateVD, RHSVD]() {
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001247 return Builder.CreateElementBitCast(GetAddrOfLocalVar(PrivateVD),
1248 ConvertTypeForMem(RHSVD->getType()),
1249 "rhs.begin");
1250 });
1251 } else {
1252 QualType Type = PrivateVD->getType();
1253 bool IsArray = getContext().getAsArrayType(Type) != nullptr;
Akira Hatanakaf139ae32019-12-03 15:17:01 -08001254 Address OriginalAddr = RedCG.getSharedLValue(Count).getAddress(*this);
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001255 // Store the address of the original variable associated with the LHS
1256 // implicit variable.
1257 if (IsArray) {
1258 OriginalAddr = Builder.CreateElementBitCast(
1259 OriginalAddr, ConvertTypeForMem(LHSVD->getType()), "lhs.begin");
1260 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00001261 PrivateScope.addPrivate(LHSVD, [OriginalAddr]() { return OriginalAddr; });
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001262 PrivateScope.addPrivate(
Alexey Bataevddf3db92018-04-13 17:31:06 +00001263 RHSVD, [this, PrivateVD, RHSVD, IsArray]() {
Alexey Bataev5c40bec2017-07-13 13:36:14 +00001264 return IsArray
1265 ? Builder.CreateElementBitCast(
1266 GetAddrOfLocalVar(PrivateVD),
1267 ConvertTypeForMem(RHSVD->getType()), "rhs.begin")
1268 : GetAddrOfLocalVar(PrivateVD);
1269 });
1270 }
1271 ++ILHS;
1272 ++IRHS;
1273 ++IPriv;
1274 ++Count;
1275 }
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04001276 if (!Data.ReductionVars.empty()) {
1277 Data.IsReductionWithTaskMod = true;
1278 Data.IsWorksharingReduction =
1279 isOpenMPWorksharingDirective(D.getDirectiveKind());
1280 llvm::Value *ReductionDesc = CGM.getOpenMPRuntime().emitTaskReductionInit(
1281 *this, D.getBeginLoc(), TaskLHSs, TaskRHSs, Data);
1282 const Expr *TaskRedRef = nullptr;
1283 switch (D.getDirectiveKind()) {
1284 case OMPD_parallel:
1285 TaskRedRef = cast<OMPParallelDirective>(D).getTaskReductionRefExpr();
1286 break;
1287 case OMPD_for:
1288 TaskRedRef = cast<OMPForDirective>(D).getTaskReductionRefExpr();
1289 break;
1290 case OMPD_sections:
1291 TaskRedRef = cast<OMPSectionsDirective>(D).getTaskReductionRefExpr();
1292 break;
1293 case OMPD_parallel_for:
1294 TaskRedRef = cast<OMPParallelForDirective>(D).getTaskReductionRefExpr();
1295 break;
1296 case OMPD_parallel_master:
1297 TaskRedRef =
1298 cast<OMPParallelMasterDirective>(D).getTaskReductionRefExpr();
1299 break;
1300 case OMPD_parallel_sections:
1301 TaskRedRef =
1302 cast<OMPParallelSectionsDirective>(D).getTaskReductionRefExpr();
1303 break;
1304 case OMPD_target_parallel:
1305 TaskRedRef =
1306 cast<OMPTargetParallelDirective>(D).getTaskReductionRefExpr();
1307 break;
1308 case OMPD_target_parallel_for:
1309 TaskRedRef =
1310 cast<OMPTargetParallelForDirective>(D).getTaskReductionRefExpr();
1311 break;
1312 case OMPD_distribute_parallel_for:
1313 TaskRedRef =
1314 cast<OMPDistributeParallelForDirective>(D).getTaskReductionRefExpr();
1315 break;
1316 case OMPD_teams_distribute_parallel_for:
1317 TaskRedRef = cast<OMPTeamsDistributeParallelForDirective>(D)
1318 .getTaskReductionRefExpr();
1319 break;
1320 case OMPD_target_teams_distribute_parallel_for:
1321 TaskRedRef = cast<OMPTargetTeamsDistributeParallelForDirective>(D)
1322 .getTaskReductionRefExpr();
1323 break;
1324 case OMPD_simd:
1325 case OMPD_for_simd:
1326 case OMPD_section:
1327 case OMPD_single:
1328 case OMPD_master:
1329 case OMPD_critical:
1330 case OMPD_parallel_for_simd:
1331 case OMPD_task:
1332 case OMPD_taskyield:
1333 case OMPD_barrier:
1334 case OMPD_taskwait:
1335 case OMPD_taskgroup:
1336 case OMPD_flush:
1337 case OMPD_depobj:
1338 case OMPD_scan:
1339 case OMPD_ordered:
1340 case OMPD_atomic:
1341 case OMPD_teams:
1342 case OMPD_target:
1343 case OMPD_cancellation_point:
1344 case OMPD_cancel:
1345 case OMPD_target_data:
1346 case OMPD_target_enter_data:
1347 case OMPD_target_exit_data:
1348 case OMPD_taskloop:
1349 case OMPD_taskloop_simd:
1350 case OMPD_master_taskloop:
1351 case OMPD_master_taskloop_simd:
1352 case OMPD_parallel_master_taskloop:
1353 case OMPD_parallel_master_taskloop_simd:
1354 case OMPD_distribute:
1355 case OMPD_target_update:
1356 case OMPD_distribute_parallel_for_simd:
1357 case OMPD_distribute_simd:
1358 case OMPD_target_parallel_for_simd:
1359 case OMPD_target_simd:
1360 case OMPD_teams_distribute:
1361 case OMPD_teams_distribute_simd:
1362 case OMPD_teams_distribute_parallel_for_simd:
1363 case OMPD_target_teams:
1364 case OMPD_target_teams_distribute:
1365 case OMPD_target_teams_distribute_parallel_for_simd:
1366 case OMPD_target_teams_distribute_simd:
1367 case OMPD_declare_target:
1368 case OMPD_end_declare_target:
1369 case OMPD_threadprivate:
1370 case OMPD_allocate:
1371 case OMPD_declare_reduction:
1372 case OMPD_declare_mapper:
1373 case OMPD_declare_simd:
1374 case OMPD_requires:
1375 case OMPD_declare_variant:
1376 case OMPD_begin_declare_variant:
1377 case OMPD_end_declare_variant:
1378 case OMPD_unknown:
1379 llvm_unreachable("Enexpected directive with task reductions.");
1380 }
1381
1382 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(TaskRedRef)->getDecl());
1383 EmitVarDecl(*VD);
1384 EmitStoreOfScalar(ReductionDesc, GetAddrOfLocalVar(VD),
1385 /*Volatile=*/false, TaskRedRef->getType());
1386 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001387}
1388
1389void CodeGenFunction::EmitOMPReductionClauseFinal(
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00001390 const OMPExecutableDirective &D, const OpenMPDirectiveKind ReductionKind) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001391 if (!HaveInsertPoint())
1392 return;
Alexey Bataevf24e7b12015-10-08 09:10:53 +00001393 llvm::SmallVector<const Expr *, 8> Privates;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001394 llvm::SmallVector<const Expr *, 8> LHSExprs;
1395 llvm::SmallVector<const Expr *, 8> RHSExprs;
1396 llvm::SmallVector<const Expr *, 8> ReductionOps;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001397 bool HasAtLeastOneReduction = false;
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04001398 bool IsReductionWithTaskMod = false;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001399 for (const auto *C : D.getClausesOfKind<OMPReductionClause>()) {
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04001400 // Do not emit for inscan reductions.
1401 if (C->getModifier() == OMPC_REDUCTION_inscan)
1402 continue;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001403 HasAtLeastOneReduction = true;
Alexey Bataevf24e7b12015-10-08 09:10:53 +00001404 Privates.append(C->privates().begin(), C->privates().end());
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001405 LHSExprs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
1406 RHSExprs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
1407 ReductionOps.append(C->reduction_ops().begin(), C->reduction_ops().end());
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04001408 IsReductionWithTaskMod =
1409 IsReductionWithTaskMod || C->getModifier() == OMPC_REDUCTION_task;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001410 }
1411 if (HasAtLeastOneReduction) {
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04001412 if (IsReductionWithTaskMod) {
1413 CGM.getOpenMPRuntime().emitTaskReductionFini(
1414 *this, D.getBeginLoc(),
1415 isOpenMPWorksharingDirective(D.getDirectiveKind()));
1416 }
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00001417 bool WithNowait = D.getSingleClause<OMPNowaitClause>() ||
1418 isOpenMPParallelDirective(D.getDirectiveKind()) ||
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00001419 ReductionKind == OMPD_simd;
1420 bool SimpleReduction = ReductionKind == OMPD_simd;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001421 // Emit nowait reduction if nowait clause is present or directive is a
1422 // parallel directive (it always has implicit barrier).
1423 CGM.getOpenMPRuntime().emitReduction(
Stephen Kelly1c301dc2018-08-09 21:09:38 +00001424 *this, D.getEndLoc(), Privates, LHSExprs, RHSExprs, ReductionOps,
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00001425 {WithNowait, SimpleReduction, ReductionKind});
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001426 }
1427}
1428
Alexey Bataev61205072016-03-02 04:57:40 +00001429static void emitPostUpdateForReductionClause(
1430 CodeGenFunction &CGF, const OMPExecutableDirective &D,
Alexey Bataevddf3db92018-04-13 17:31:06 +00001431 const llvm::function_ref<llvm::Value *(CodeGenFunction &)> CondGen) {
Alexey Bataev61205072016-03-02 04:57:40 +00001432 if (!CGF.HaveInsertPoint())
1433 return;
1434 llvm::BasicBlock *DoneBB = nullptr;
1435 for (const auto *C : D.getClausesOfKind<OMPReductionClause>()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001436 if (const Expr *PostUpdate = C->getPostUpdateExpr()) {
Alexey Bataev61205072016-03-02 04:57:40 +00001437 if (!DoneBB) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001438 if (llvm::Value *Cond = CondGen(CGF)) {
Alexey Bataev61205072016-03-02 04:57:40 +00001439 // If the first post-update expression is found, emit conditional
1440 // block if it was requested.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001441 llvm::BasicBlock *ThenBB = CGF.createBasicBlock(".omp.reduction.pu");
Alexey Bataev61205072016-03-02 04:57:40 +00001442 DoneBB = CGF.createBasicBlock(".omp.reduction.pu.done");
1443 CGF.Builder.CreateCondBr(Cond, ThenBB, DoneBB);
1444 CGF.EmitBlock(ThenBB);
1445 }
1446 }
1447 CGF.EmitIgnoredExpr(PostUpdate);
1448 }
1449 }
1450 if (DoneBB)
1451 CGF.EmitBlock(DoneBB, /*IsFinished=*/true);
1452}
1453
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00001454namespace {
1455/// Codegen lambda for appending distribute lower and upper bounds to outlined
1456/// parallel function. This is necessary for combined constructs such as
1457/// 'distribute parallel for'
1458typedef llvm::function_ref<void(CodeGenFunction &,
1459 const OMPExecutableDirective &,
1460 llvm::SmallVectorImpl<llvm::Value *> &)>
1461 CodeGenBoundParametersTy;
1462} // anonymous namespace
1463
Alexey Bataev46978742020-01-30 10:46:11 -05001464static void
1465checkForLastprivateConditionalUpdate(CodeGenFunction &CGF,
1466 const OMPExecutableDirective &S) {
1467 if (CGF.getLangOpts().OpenMP < 50)
1468 return;
1469 llvm::DenseSet<CanonicalDeclPtr<const VarDecl>> PrivateDecls;
1470 for (const auto *C : S.getClausesOfKind<OMPReductionClause>()) {
1471 for (const Expr *Ref : C->varlists()) {
1472 if (!Ref->getType()->isScalarType())
1473 continue;
1474 const auto *DRE = dyn_cast<DeclRefExpr>(Ref->IgnoreParenImpCasts());
1475 if (!DRE)
1476 continue;
1477 PrivateDecls.insert(cast<VarDecl>(DRE->getDecl()));
1478 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, Ref);
1479 }
1480 }
1481 for (const auto *C : S.getClausesOfKind<OMPLastprivateClause>()) {
1482 for (const Expr *Ref : C->varlists()) {
1483 if (!Ref->getType()->isScalarType())
1484 continue;
1485 const auto *DRE = dyn_cast<DeclRefExpr>(Ref->IgnoreParenImpCasts());
1486 if (!DRE)
1487 continue;
1488 PrivateDecls.insert(cast<VarDecl>(DRE->getDecl()));
1489 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, Ref);
1490 }
1491 }
1492 for (const auto *C : S.getClausesOfKind<OMPLinearClause>()) {
1493 for (const Expr *Ref : C->varlists()) {
1494 if (!Ref->getType()->isScalarType())
1495 continue;
1496 const auto *DRE = dyn_cast<DeclRefExpr>(Ref->IgnoreParenImpCasts());
1497 if (!DRE)
1498 continue;
1499 PrivateDecls.insert(cast<VarDecl>(DRE->getDecl()));
1500 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, Ref);
1501 }
1502 }
1503 // Privates should ne analyzed since they are not captured at all.
1504 // Task reductions may be skipped - tasks are ignored.
1505 // Firstprivates do not return value but may be passed by reference - no need
1506 // to check for updated lastprivate conditional.
1507 for (const auto *C : S.getClausesOfKind<OMPFirstprivateClause>()) {
1508 for (const Expr *Ref : C->varlists()) {
1509 if (!Ref->getType()->isScalarType())
1510 continue;
1511 const auto *DRE = dyn_cast<DeclRefExpr>(Ref->IgnoreParenImpCasts());
1512 if (!DRE)
1513 continue;
1514 PrivateDecls.insert(cast<VarDecl>(DRE->getDecl()));
1515 }
1516 }
1517 CGF.CGM.getOpenMPRuntime().checkAndEmitSharedLastprivateConditional(
1518 CGF, S, PrivateDecls);
1519}
1520
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00001521static void emitCommonOMPParallelDirective(
1522 CodeGenFunction &CGF, const OMPExecutableDirective &S,
1523 OpenMPDirectiveKind InnermostKind, const RegionCodeGenTy &CodeGen,
1524 const CodeGenBoundParametersTy &CodeGenBoundParameters) {
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00001525 const CapturedStmt *CS = S.getCapturedStmt(OMPD_parallel);
James Y Knight9871db02019-02-05 16:42:33 +00001526 llvm::Function *OutlinedFn =
Alexey Bataevddf3db92018-04-13 17:31:06 +00001527 CGF.CGM.getOpenMPRuntime().emitParallelOutlinedFunction(
1528 S, *CS->getCapturedDecl()->param_begin(), InnermostKind, CodeGen);
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001529 if (const auto *NumThreadsClause = S.getSingleClause<OMPNumThreadsClause>()) {
Alexey Bataev1d677132015-04-22 13:57:31 +00001530 CodeGenFunction::RunCleanupsScope NumThreadsScope(CGF);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001531 llvm::Value *NumThreads =
1532 CGF.EmitScalarExpr(NumThreadsClause->getNumThreads(),
1533 /*IgnoreResultAssign=*/true);
Alexey Bataev1d677132015-04-22 13:57:31 +00001534 CGF.CGM.getOpenMPRuntime().emitNumThreadsClause(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00001535 CGF, NumThreads, NumThreadsClause->getBeginLoc());
Alexey Bataev1d677132015-04-22 13:57:31 +00001536 }
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001537 if (const auto *ProcBindClause = S.getSingleClause<OMPProcBindClause>()) {
Alexey Bataev14fa1c62016-03-29 05:34:15 +00001538 CodeGenFunction::RunCleanupsScope ProcBindScope(CGF);
Alexey Bataev7f210c62015-06-18 13:40:03 +00001539 CGF.CGM.getOpenMPRuntime().emitProcBindClause(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00001540 CGF, ProcBindClause->getProcBindKind(), ProcBindClause->getBeginLoc());
Alexey Bataev7f210c62015-06-18 13:40:03 +00001541 }
Alexey Bataev1d677132015-04-22 13:57:31 +00001542 const Expr *IfCond = nullptr;
Alexey Bataev7371aa32015-09-03 08:45:56 +00001543 for (const auto *C : S.getClausesOfKind<OMPIfClause>()) {
1544 if (C->getNameModifier() == OMPD_unknown ||
1545 C->getNameModifier() == OMPD_parallel) {
1546 IfCond = C->getCondition();
1547 break;
1548 }
Alexey Bataev1d677132015-04-22 13:57:31 +00001549 }
Alexey Bataev14fa1c62016-03-29 05:34:15 +00001550
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +00001551 OMPParallelScope Scope(CGF, S);
Alexey Bataev14fa1c62016-03-29 05:34:15 +00001552 llvm::SmallVector<llvm::Value *, 16> CapturedVars;
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00001553 // Combining 'distribute' with 'for' requires sharing each 'distribute' chunk
1554 // lower and upper bounds with the pragma 'for' chunking mechanism.
1555 // The following lambda takes care of appending the lower and upper bound
1556 // parameters when necessary
1557 CodeGenBoundParameters(CGF, S, CapturedVars);
Alexey Bataev14fa1c62016-03-29 05:34:15 +00001558 CGF.GenerateOpenMPCapturedVars(*CS, CapturedVars);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00001559 CGF.CGM.getOpenMPRuntime().emitParallelCall(CGF, S.getBeginLoc(), OutlinedFn,
Alexey Bataev2377fe92015-09-10 08:12:02 +00001560 CapturedVars, IfCond);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001561}
1562
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00001563static void emitEmptyBoundParameters(CodeGenFunction &,
1564 const OMPExecutableDirective &,
1565 llvm::SmallVectorImpl<llvm::Value *> &) {}
1566
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001567void CodeGenFunction::EmitOMPParallelDirective(const OMPParallelDirective &S) {
Johannes Doerfert10fedd92019-12-26 11:23:38 -06001568 if (llvm::OpenMPIRBuilder *OMPBuilder = CGM.getOpenMPIRBuilder()) {
1569 // Check if we have any if clause associated with the directive.
1570 llvm::Value *IfCond = nullptr;
1571 if (const auto *C = S.getSingleClause<OMPIfClause>())
1572 IfCond = EmitScalarExpr(C->getCondition(),
1573 /*IgnoreResultAssign=*/true);
1574
1575 llvm::Value *NumThreads = nullptr;
1576 if (const auto *NumThreadsClause = S.getSingleClause<OMPNumThreadsClause>())
1577 NumThreads = EmitScalarExpr(NumThreadsClause->getNumThreads(),
1578 /*IgnoreResultAssign=*/true);
1579
1580 ProcBindKind ProcBind = OMP_PROC_BIND_default;
1581 if (const auto *ProcBindClause = S.getSingleClause<OMPProcBindClause>())
1582 ProcBind = ProcBindClause->getProcBindKind();
1583
1584 using InsertPointTy = llvm::OpenMPIRBuilder::InsertPointTy;
1585
1586 // The cleanup callback that finalizes all variabels at the given location,
1587 // thus calls destructors etc.
1588 auto FiniCB = [this](InsertPointTy IP) {
Fady Ghanimba3f8632020-02-19 13:50:26 -06001589 OMPBuilderCBHelpers::FinalizeOMPRegion(*this, IP);
Johannes Doerfert10fedd92019-12-26 11:23:38 -06001590 };
1591
1592 // Privatization callback that performs appropriate action for
1593 // shared/private/firstprivate/lastprivate/copyin/... variables.
1594 //
1595 // TODO: This defaults to shared right now.
1596 auto PrivCB = [](InsertPointTy AllocaIP, InsertPointTy CodeGenIP,
1597 llvm::Value &Val, llvm::Value *&ReplVal) {
1598 // The next line is appropriate only for variables (Val) with the
1599 // data-sharing attribute "shared".
1600 ReplVal = &Val;
1601
1602 return CodeGenIP;
1603 };
1604
1605 const CapturedStmt *CS = S.getCapturedStmt(OMPD_parallel);
1606 const Stmt *ParallelRegionBodyStmt = CS->getCapturedStmt();
1607
1608 auto BodyGenCB = [ParallelRegionBodyStmt,
1609 this](InsertPointTy AllocaIP, InsertPointTy CodeGenIP,
1610 llvm::BasicBlock &ContinuationBB) {
Fady Ghanimba3f8632020-02-19 13:50:26 -06001611 OMPBuilderCBHelpers::OutlinedRegionBodyRAII ORB(*this, AllocaIP,
1612 ContinuationBB);
1613 OMPBuilderCBHelpers::EmitOMPRegionBody(*this, ParallelRegionBodyStmt,
1614 CodeGenIP, ContinuationBB);
Johannes Doerfert10fedd92019-12-26 11:23:38 -06001615 };
1616
1617 CGCapturedStmtInfo CGSI(*CS, CR_OpenMP);
1618 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(*this, &CGSI);
1619 Builder.restoreIP(OMPBuilder->CreateParallel(Builder, BodyGenCB, PrivCB,
1620 FiniCB, IfCond, NumThreads,
1621 ProcBind, S.hasCancel()));
1622 return;
1623 }
1624
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001625 // Emit parallel region as a standalone region.
Alexey Bataevc99042b2018-03-15 18:10:54 +00001626 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00001627 Action.Enter(CGF);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001628 OMPPrivateScope PrivateScope(CGF);
Alexey Bataevf56f98c2015-04-16 05:39:01 +00001629 bool Copyins = CGF.EmitOMPCopyinClause(S);
Alexey Bataevcd8b6a22016-02-15 08:07:17 +00001630 (void)CGF.EmitOMPFirstprivateClause(S, PrivateScope);
1631 if (Copyins) {
Alexey Bataev69c62a92015-04-15 04:52:20 +00001632 // Emit implicit barrier to synchronize threads and avoid data races on
Alexey Bataevcd8b6a22016-02-15 08:07:17 +00001633 // propagation master's thread values of threadprivate variables to local
1634 // instances of that variables of all other implicit threads.
Alexey Bataev25e5b442015-09-15 12:52:43 +00001635 CGF.CGM.getOpenMPRuntime().emitBarrierCall(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00001636 CGF, S.getBeginLoc(), OMPD_unknown, /*EmitChecks=*/false,
Alexey Bataev25e5b442015-09-15 12:52:43 +00001637 /*ForceSimpleCall=*/true);
Alexey Bataev69c62a92015-04-15 04:52:20 +00001638 }
1639 CGF.EmitOMPPrivateClause(S, PrivateScope);
1640 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
1641 (void)PrivateScope.Privatize();
Alexey Bataev475a7442018-01-12 19:39:11 +00001642 CGF.EmitStmt(S.getCapturedStmt(OMPD_parallel)->getCapturedStmt());
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00001643 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_parallel);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001644 };
Alexey Bataev46978742020-01-30 10:46:11 -05001645 {
1646 auto LPCRegion =
1647 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
1648 emitCommonOMPParallelDirective(*this, S, OMPD_parallel, CodeGen,
1649 emitEmptyBoundParameters);
1650 emitPostUpdateForReductionClause(*this, S,
1651 [](CodeGenFunction &) { return nullptr; });
1652 }
1653 // Check for outer lastprivate conditional update.
1654 checkForLastprivateConditionalUpdate(*this, S);
Alexey Bataev9959db52014-05-06 10:08:46 +00001655}
Alexander Musman515ad8c2014-05-22 08:54:05 +00001656
Alexey Bataev8bbf2e32019-11-04 09:59:11 -05001657static void emitBody(CodeGenFunction &CGF, const Stmt *S, const Stmt *NextLoop,
1658 int MaxLevel, int Level = 0) {
1659 assert(Level < MaxLevel && "Too deep lookup during loop body codegen.");
1660 const Stmt *SimplifiedS = S->IgnoreContainers();
1661 if (const auto *CS = dyn_cast<CompoundStmt>(SimplifiedS)) {
1662 PrettyStackTraceLoc CrashInfo(
1663 CGF.getContext().getSourceManager(), CS->getLBracLoc(),
1664 "LLVM IR generation of compound statement ('{}')");
1665
1666 // Keep track of the current cleanup stack depth, including debug scopes.
1667 CodeGenFunction::LexicalScope Scope(CGF, S->getSourceRange());
1668 for (const Stmt *CurStmt : CS->body())
1669 emitBody(CGF, CurStmt, NextLoop, MaxLevel, Level);
1670 return;
1671 }
1672 if (SimplifiedS == NextLoop) {
1673 if (const auto *For = dyn_cast<ForStmt>(SimplifiedS)) {
1674 S = For->getBody();
1675 } else {
1676 assert(isa<CXXForRangeStmt>(SimplifiedS) &&
1677 "Expected canonical for loop or range-based for loop.");
1678 const auto *CXXFor = cast<CXXForRangeStmt>(SimplifiedS);
1679 CGF.EmitStmt(CXXFor->getLoopVarStmt());
1680 S = CXXFor->getBody();
1681 }
1682 if (Level + 1 < MaxLevel) {
1683 NextLoop = OMPLoopDirective::tryToFindNextInnerLoop(
1684 S, /*TryImperfectlyNestedLoops=*/true);
1685 emitBody(CGF, S, NextLoop, MaxLevel, Level + 1);
1686 return;
1687 }
1688 }
1689 CGF.EmitStmt(S);
1690}
1691
Alexey Bataev0f34da12015-07-02 04:17:07 +00001692void CodeGenFunction::EmitOMPLoopBody(const OMPLoopDirective &D,
1693 JumpDest LoopExit) {
Alexander Musmana5f070a2014-10-01 06:03:56 +00001694 RunCleanupsScope BodyScope(*this);
1695 // Update counters values on current iteration.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001696 for (const Expr *UE : D.updates())
1697 EmitIgnoredExpr(UE);
Alexander Musman3276a272015-03-21 10:12:56 +00001698 // Update the linear variables.
Alexey Bataev617db5f2017-12-04 15:38:33 +00001699 // In distribute directives only loop counters may be marked as linear, no
1700 // need to generate the code for them.
1701 if (!isOpenMPDistributeDirective(D.getDirectiveKind())) {
1702 for (const auto *C : D.getClausesOfKind<OMPLinearClause>()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001703 for (const Expr *UE : C->updates())
1704 EmitIgnoredExpr(UE);
Alexey Bataev617db5f2017-12-04 15:38:33 +00001705 }
Alexander Musman3276a272015-03-21 10:12:56 +00001706 }
1707
Alexander Musmana5f070a2014-10-01 06:03:56 +00001708 // On a continue in the body, jump to the end.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001709 JumpDest Continue = getJumpDestInCurrentScope("omp.body.continue");
Alexey Bataev0f34da12015-07-02 04:17:07 +00001710 BreakContinueStack.push_back(BreakContinue(LoopExit, Continue));
Alexey Bataevf8be4762019-08-14 19:30:06 +00001711 for (const Expr *E : D.finals_conditions()) {
1712 if (!E)
1713 continue;
1714 // Check that loop counter in non-rectangular nest fits into the iteration
1715 // space.
1716 llvm::BasicBlock *NextBB = createBasicBlock("omp.body.next");
1717 EmitBranchOnBoolExpr(E, NextBB, Continue.getBlock(),
1718 getProfileCount(D.getBody()));
1719 EmitBlock(NextBB);
1720 }
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04001721
1722 OMPPrivateScope InscanScope(*this);
1723 EmitOMPReductionClauseInit(D, InscanScope, /*ForInscan=*/true);
1724 bool IsInscanRegion = InscanScope.Privatize();
1725 if (IsInscanRegion) {
1726 // Need to remember the block before and after scan directive
1727 // to dispatch them correctly depending on the clause used in
1728 // this directive, inclusive or exclusive. For inclusive scan the natural
1729 // order of the blocks is used, for exclusive clause the blocks must be
1730 // executed in reverse order.
1731 OMPBeforeScanBlock = createBasicBlock("omp.before.scan.bb");
1732 OMPAfterScanBlock = createBasicBlock("omp.after.scan.bb");
Alexey Bataev43101d12020-06-11 11:28:59 -04001733 // No need to allocate inscan exit block, in simd mode it is selected in the
1734 // codegen for the scan directive.
Alexey Bataev34ee2542020-06-16 17:06:49 -04001735 if (D.getDirectiveKind() != OMPD_simd && !getLangOpts().OpenMPSimd)
Alexey Bataev43101d12020-06-11 11:28:59 -04001736 OMPScanExitBlock = createBasicBlock("omp.exit.inscan.bb");
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04001737 OMPScanDispatch = createBasicBlock("omp.inscan.dispatch");
1738 EmitBranch(OMPScanDispatch);
1739 EmitBlock(OMPBeforeScanBlock);
1740 }
1741
Alexey Bataevbef93a92019-10-07 18:54:57 +00001742 // Emit loop variables for C++ range loops.
1743 const Stmt *Body =
1744 D.getInnermostCapturedStmt()->getCapturedStmt()->IgnoreContainers();
Alexander Musmana5f070a2014-10-01 06:03:56 +00001745 // Emit loop body.
Alexey Bataev8bbf2e32019-11-04 09:59:11 -05001746 emitBody(*this, Body,
1747 OMPLoopDirective::tryToFindNextInnerLoop(
1748 Body, /*TryImperfectlyNestedLoops=*/true),
1749 D.getCollapsedNumber());
1750
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04001751 // Jump to the dispatcher at the end of the loop body.
1752 if (IsInscanRegion)
1753 EmitBranch(OMPScanExitBlock);
1754
Alexander Musmana5f070a2014-10-01 06:03:56 +00001755 // The end (updates/cleanups).
1756 EmitBlock(Continue.getBlock());
1757 BreakContinueStack.pop_back();
Alexander Musmana5f070a2014-10-01 06:03:56 +00001758}
1759
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001760void CodeGenFunction::EmitOMPInnerLoop(
Alexey Bataevcb9191c2020-06-09 08:04:48 -04001761 const OMPExecutableDirective &S, bool RequiresCleanup, const Expr *LoopCond,
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001762 const Expr *IncExpr,
Alexey Bataevddf3db92018-04-13 17:31:06 +00001763 const llvm::function_ref<void(CodeGenFunction &)> BodyGen,
1764 const llvm::function_ref<void(CodeGenFunction &)> PostIncGen) {
Alexander Musmand196ef22014-10-07 08:57:09 +00001765 auto LoopExit = getJumpDestInCurrentScope("omp.inner.for.end");
Alexander Musmana5f070a2014-10-01 06:03:56 +00001766
1767 // Start the loop with a block that tests the condition.
Alexander Musmand196ef22014-10-07 08:57:09 +00001768 auto CondBlock = createBasicBlock("omp.inner.for.cond");
Alexander Musmana5f070a2014-10-01 06:03:56 +00001769 EmitBlock(CondBlock);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001770 const SourceRange R = S.getSourceRange();
ISHIGURO, Hiroshiac2c5af2020-05-22 11:34:49 +09001771
1772 // If attributes are attached, push to the basic block with them.
1773 const auto &OMPED = cast<OMPExecutableDirective>(S);
1774 const CapturedStmt *ICS = OMPED.getInnermostCapturedStmt();
1775 const Stmt *SS = ICS->getCapturedStmt();
1776 const AttributedStmt *AS = dyn_cast_or_null<AttributedStmt>(SS);
1777 if (AS)
1778 LoopStack.push(CondBlock, CGM.getContext(), CGM.getCodeGenOpts(),
1779 AS->getAttrs(), SourceLocToDebugLoc(R.getBegin()),
1780 SourceLocToDebugLoc(R.getEnd()));
1781 else
1782 LoopStack.push(CondBlock, SourceLocToDebugLoc(R.getBegin()),
1783 SourceLocToDebugLoc(R.getEnd()));
Alexander Musmana5f070a2014-10-01 06:03:56 +00001784
1785 // If there are any cleanups between here and the loop-exit scope,
1786 // create a block to stage a loop exit along.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001787 llvm::BasicBlock *ExitBlock = LoopExit.getBlock();
Alexey Bataev2df54a02015-03-12 08:53:29 +00001788 if (RequiresCleanup)
Alexander Musmand196ef22014-10-07 08:57:09 +00001789 ExitBlock = createBasicBlock("omp.inner.for.cond.cleanup");
Alexander Musmana5f070a2014-10-01 06:03:56 +00001790
Alexey Bataevddf3db92018-04-13 17:31:06 +00001791 llvm::BasicBlock *LoopBody = createBasicBlock("omp.inner.for.body");
Alexander Musmana5f070a2014-10-01 06:03:56 +00001792
Alexey Bataev2df54a02015-03-12 08:53:29 +00001793 // Emit condition.
Justin Bogner66242d62015-04-23 23:06:47 +00001794 EmitBranchOnBoolExpr(LoopCond, LoopBody, ExitBlock, getProfileCount(&S));
Alexander Musmana5f070a2014-10-01 06:03:56 +00001795 if (ExitBlock != LoopExit.getBlock()) {
1796 EmitBlock(ExitBlock);
1797 EmitBranchThroughCleanup(LoopExit);
1798 }
1799
1800 EmitBlock(LoopBody);
Justin Bogner66242d62015-04-23 23:06:47 +00001801 incrementProfileCounter(&S);
Alexander Musmana5f070a2014-10-01 06:03:56 +00001802
1803 // Create a block for the increment.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001804 JumpDest Continue = getJumpDestInCurrentScope("omp.inner.for.inc");
Alexander Musmana5f070a2014-10-01 06:03:56 +00001805 BreakContinueStack.push_back(BreakContinue(LoopExit, Continue));
1806
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001807 BodyGen(*this);
Alexander Musmana5f070a2014-10-01 06:03:56 +00001808
1809 // Emit "IV = IV + 1" and a back-edge to the condition block.
1810 EmitBlock(Continue.getBlock());
Alexey Bataev2df54a02015-03-12 08:53:29 +00001811 EmitIgnoredExpr(IncExpr);
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001812 PostIncGen(*this);
Alexander Musmana5f070a2014-10-01 06:03:56 +00001813 BreakContinueStack.pop_back();
1814 EmitBranch(CondBlock);
1815 LoopStack.pop();
1816 // Emit the fall-through block.
1817 EmitBlock(LoopExit.getBlock());
1818}
1819
Alexey Bataev8c3edfe2017-08-16 15:58:46 +00001820bool CodeGenFunction::EmitOMPLinearClauseInit(const OMPLoopDirective &D) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001821 if (!HaveInsertPoint())
Alexey Bataev8c3edfe2017-08-16 15:58:46 +00001822 return false;
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001823 // Emit inits for the linear variables.
Alexey Bataev8c3edfe2017-08-16 15:58:46 +00001824 bool HasLinears = false;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001825 for (const auto *C : D.getClausesOfKind<OMPLinearClause>()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001826 for (const Expr *Init : C->inits()) {
Alexey Bataev8c3edfe2017-08-16 15:58:46 +00001827 HasLinears = true;
Alexey Bataevddf3db92018-04-13 17:31:06 +00001828 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(Init)->getDecl());
1829 if (const auto *Ref =
1830 dyn_cast<DeclRefExpr>(VD->getInit()->IgnoreImpCasts())) {
Alexey Bataevef549a82016-03-09 09:49:09 +00001831 AutoVarEmission Emission = EmitAutoVarAlloca(*VD);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001832 const auto *OrigVD = cast<VarDecl>(Ref->getDecl());
Bruno Ricci5fc4db72018-12-21 14:10:18 +00001833 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(OrigVD),
Alexey Bataevef549a82016-03-09 09:49:09 +00001834 CapturedStmtInfo->lookup(OrigVD) != nullptr,
1835 VD->getInit()->getType(), VK_LValue,
1836 VD->getInit()->getExprLoc());
1837 EmitExprAsInit(&DRE, VD, MakeAddrLValue(Emission.getAllocatedAddress(),
1838 VD->getType()),
1839 /*capturedByInit=*/false);
1840 EmitAutoVarCleanups(Emission);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001841 } else {
Alexey Bataevef549a82016-03-09 09:49:09 +00001842 EmitVarDecl(*VD);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001843 }
Alexander Musmana5f070a2014-10-01 06:03:56 +00001844 }
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001845 // Emit the linear steps for the linear clauses.
1846 // If a step is not constant, it is pre-calculated before the loop.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001847 if (const auto *CS = cast_or_null<BinaryOperator>(C->getCalcStep()))
1848 if (const auto *SaveRef = cast<DeclRefExpr>(CS->getLHS())) {
Alexey Bataev3b5b5c42015-06-18 10:10:12 +00001849 EmitVarDecl(*cast<VarDecl>(SaveRef->getDecl()));
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001850 // Emit calculation of the linear step.
Alexey Bataev3b5b5c42015-06-18 10:10:12 +00001851 EmitIgnoredExpr(CS);
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001852 }
Alexander Musmana5f070a2014-10-01 06:03:56 +00001853 }
Alexey Bataev8c3edfe2017-08-16 15:58:46 +00001854 return HasLinears;
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001855}
1856
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001857void CodeGenFunction::EmitOMPLinearClauseFinal(
1858 const OMPLoopDirective &D,
Alexey Bataevddf3db92018-04-13 17:31:06 +00001859 const llvm::function_ref<llvm::Value *(CodeGenFunction &)> CondGen) {
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001860 if (!HaveInsertPoint())
Alexey Bataev8ef31412015-12-18 07:58:25 +00001861 return;
Alexey Bataevef549a82016-03-09 09:49:09 +00001862 llvm::BasicBlock *DoneBB = nullptr;
Alexander Musman3276a272015-03-21 10:12:56 +00001863 // Emit the final values of the linear variables.
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001864 for (const auto *C : D.getClausesOfKind<OMPLinearClause>()) {
Alexey Bataev39f915b82015-05-08 10:41:21 +00001865 auto IC = C->varlist_begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00001866 for (const Expr *F : C->finals()) {
Alexey Bataevef549a82016-03-09 09:49:09 +00001867 if (!DoneBB) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00001868 if (llvm::Value *Cond = CondGen(*this)) {
Alexey Bataevef549a82016-03-09 09:49:09 +00001869 // If the first post-update expression is found, emit conditional
1870 // block if it was requested.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001871 llvm::BasicBlock *ThenBB = createBasicBlock(".omp.linear.pu");
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001872 DoneBB = createBasicBlock(".omp.linear.pu.done");
1873 Builder.CreateCondBr(Cond, ThenBB, DoneBB);
1874 EmitBlock(ThenBB);
Alexey Bataevef549a82016-03-09 09:49:09 +00001875 }
1876 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00001877 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IC)->getDecl());
Bruno Ricci5fc4db72018-12-21 14:10:18 +00001878 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(OrigVD),
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001879 CapturedStmtInfo->lookup(OrigVD) != nullptr,
Alexey Bataev39f915b82015-05-08 10:41:21 +00001880 (*IC)->getType(), VK_LValue, (*IC)->getExprLoc());
Akira Hatanakaf139ae32019-12-03 15:17:01 -08001881 Address OrigAddr = EmitLValue(&DRE).getAddress(*this);
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001882 CodeGenFunction::OMPPrivateScope VarScope(*this);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001883 VarScope.addPrivate(OrigVD, [OrigAddr]() { return OrigAddr; });
Alexey Bataev39f915b82015-05-08 10:41:21 +00001884 (void)VarScope.Privatize();
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001885 EmitIgnoredExpr(F);
Alexey Bataev39f915b82015-05-08 10:41:21 +00001886 ++IC;
Alexander Musman3276a272015-03-21 10:12:56 +00001887 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00001888 if (const Expr *PostUpdate = C->getPostUpdateExpr())
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001889 EmitIgnoredExpr(PostUpdate);
Alexander Musman3276a272015-03-21 10:12:56 +00001890 }
Alexey Bataevef549a82016-03-09 09:49:09 +00001891 if (DoneBB)
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001892 EmitBlock(DoneBB, /*IsFinished=*/true);
Alexander Musmana5f070a2014-10-01 06:03:56 +00001893}
1894
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001895static void emitAlignedClause(CodeGenFunction &CGF,
1896 const OMPExecutableDirective &D) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001897 if (!CGF.HaveInsertPoint())
1898 return;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00001899 for (const auto *Clause : D.getClausesOfKind<OMPAlignedClause>()) {
Erich Keanef7593952019-10-11 14:59:44 +00001900 llvm::APInt ClauseAlignment(64, 0);
Alexey Bataevddf3db92018-04-13 17:31:06 +00001901 if (const Expr *AlignmentExpr = Clause->getAlignment()) {
1902 auto *AlignmentCI =
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001903 cast<llvm::ConstantInt>(CGF.EmitScalarExpr(AlignmentExpr));
Erich Keanef7593952019-10-11 14:59:44 +00001904 ClauseAlignment = AlignmentCI->getValue();
Alexander Musman09184fe2014-09-30 05:29:28 +00001905 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00001906 for (const Expr *E : Clause->varlists()) {
Erich Keanef7593952019-10-11 14:59:44 +00001907 llvm::APInt Alignment(ClauseAlignment);
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001908 if (Alignment == 0) {
1909 // OpenMP [2.8.1, Description]
1910 // If no optional parameter is specified, implementation-defined default
1911 // alignments for SIMD instructions on the target platforms are assumed.
1912 Alignment =
Alexey Bataev00396512015-07-02 03:40:19 +00001913 CGF.getContext()
1914 .toCharUnitsFromBits(CGF.getContext().getOpenMPDefaultSimdAlign(
1915 E->getType()->getPointeeType()))
1916 .getQuantity();
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001917 }
Erich Keanef7593952019-10-11 14:59:44 +00001918 assert((Alignment == 0 || Alignment.isPowerOf2()) &&
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001919 "alignment is not power of 2");
1920 if (Alignment != 0) {
1921 llvm::Value *PtrValue = CGF.EmitScalarExpr(E);
Fangrui Song1d49eb02020-02-13 16:36:27 -08001922 CGF.emitAlignmentAssumption(
Erich Keanef7593952019-10-11 14:59:44 +00001923 PtrValue, E, /*No second loc needed*/ SourceLocation(),
1924 llvm::ConstantInt::get(CGF.getLLVMContext(), Alignment));
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001925 }
Alexander Musman09184fe2014-09-30 05:29:28 +00001926 }
1927 }
1928}
1929
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001930void CodeGenFunction::EmitOMPPrivateLoopCounters(
1931 const OMPLoopDirective &S, CodeGenFunction::OMPPrivateScope &LoopScope) {
1932 if (!HaveInsertPoint())
Alexey Bataev8ef31412015-12-18 07:58:25 +00001933 return;
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001934 auto I = S.private_counters().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00001935 for (const Expr *E : S.counters()) {
1936 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
1937 const auto *PrivateVD = cast<VarDecl>(cast<DeclRefExpr>(*I)->getDecl());
Alexey Bataevab4ea222018-03-07 18:17:06 +00001938 // Emit var without initialization.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001939 AutoVarEmission VarEmission = EmitAutoVarAlloca(*PrivateVD);
Alexey Bataevab4ea222018-03-07 18:17:06 +00001940 EmitAutoVarCleanups(VarEmission);
1941 LocalDeclMap.erase(PrivateVD);
1942 (void)LoopScope.addPrivate(VD, [&VarEmission]() {
1943 return VarEmission.getAllocatedAddress();
Alexey Bataev435ad7b2014-10-10 09:48:26 +00001944 });
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001945 if (LocalDeclMap.count(VD) || CapturedStmtInfo->lookup(VD) ||
1946 VD->hasGlobalStorage()) {
Alexey Bataevab4ea222018-03-07 18:17:06 +00001947 (void)LoopScope.addPrivate(PrivateVD, [this, VD, E]() {
Bruno Ricci5fc4db72018-12-21 14:10:18 +00001948 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(VD),
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001949 LocalDeclMap.count(VD) || CapturedStmtInfo->lookup(VD),
1950 E->getType(), VK_LValue, E->getExprLoc());
Akira Hatanakaf139ae32019-12-03 15:17:01 -08001951 return EmitLValue(&DRE).getAddress(*this);
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001952 });
Alexey Bataevab4ea222018-03-07 18:17:06 +00001953 } else {
1954 (void)LoopScope.addPrivate(PrivateVD, [&VarEmission]() {
1955 return VarEmission.getAllocatedAddress();
1956 });
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001957 }
Alexey Bataeva8899172015-08-06 12:30:57 +00001958 ++I;
Alexey Bataev435ad7b2014-10-10 09:48:26 +00001959 }
Alexey Bataevf138fda2018-08-13 19:04:24 +00001960 // Privatize extra loop counters used in loops for ordered(n) clauses.
1961 for (const auto *C : S.getClausesOfKind<OMPOrderedClause>()) {
1962 if (!C->getNumForLoops())
1963 continue;
1964 for (unsigned I = S.getCollapsedNumber(),
1965 E = C->getLoopNumIterations().size();
1966 I < E; ++I) {
Mike Rice0ed46662018-09-20 17:19:41 +00001967 const auto *DRE = cast<DeclRefExpr>(C->getLoopCounter(I));
Alexey Bataevf138fda2018-08-13 19:04:24 +00001968 const auto *VD = cast<VarDecl>(DRE->getDecl());
Alexey Bataev0d8fcdf2019-03-14 20:36:00 +00001969 // Override only those variables that can be captured to avoid re-emission
1970 // of the variables declared within the loops.
1971 if (DRE->refersToEnclosingVariableOrCapture()) {
Alexey Bataevf138fda2018-08-13 19:04:24 +00001972 (void)LoopScope.addPrivate(VD, [this, DRE, VD]() {
1973 return CreateMemTemp(DRE->getType(), VD->getName());
1974 });
1975 }
1976 }
1977 }
Alexey Bataev435ad7b2014-10-10 09:48:26 +00001978}
1979
Alexey Bataev62dbb972015-04-22 11:59:37 +00001980static void emitPreCond(CodeGenFunction &CGF, const OMPLoopDirective &S,
1981 const Expr *Cond, llvm::BasicBlock *TrueBlock,
1982 llvm::BasicBlock *FalseBlock, uint64_t TrueCount) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001983 if (!CGF.HaveInsertPoint())
1984 return;
Alexey Bataev6e8248f2015-06-11 10:53:56 +00001985 {
1986 CodeGenFunction::OMPPrivateScope PreCondScope(CGF);
Alexey Bataev5dff95c2016-04-22 03:56:56 +00001987 CGF.EmitOMPPrivateLoopCounters(S, PreCondScope);
Alexey Bataev6e8248f2015-06-11 10:53:56 +00001988 (void)PreCondScope.Privatize();
Alexey Bataev6e8248f2015-06-11 10:53:56 +00001989 // Get initial values of real counters.
Alexey Bataevddf3db92018-04-13 17:31:06 +00001990 for (const Expr *I : S.inits()) {
Alexey Bataev6e8248f2015-06-11 10:53:56 +00001991 CGF.EmitIgnoredExpr(I);
1992 }
Alexey Bataev62dbb972015-04-22 11:59:37 +00001993 }
Alexey Bataevf8be4762019-08-14 19:30:06 +00001994 // Create temp loop control variables with their init values to support
1995 // non-rectangular loops.
1996 CodeGenFunction::OMPMapVars PreCondVars;
1997 for (const Expr * E: S.dependent_counters()) {
1998 if (!E)
1999 continue;
2000 assert(!E->getType().getNonReferenceType()->isRecordType() &&
2001 "dependent counter must not be an iterator.");
2002 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
2003 Address CounterAddr =
2004 CGF.CreateMemTemp(VD->getType().getNonReferenceType());
2005 (void)PreCondVars.setVarAddr(CGF, VD, CounterAddr);
2006 }
2007 (void)PreCondVars.apply(CGF);
2008 for (const Expr *E : S.dependent_inits()) {
2009 if (!E)
2010 continue;
2011 CGF.EmitIgnoredExpr(E);
2012 }
Alexey Bataev62dbb972015-04-22 11:59:37 +00002013 // Check that loop is executed at least one time.
2014 CGF.EmitBranchOnBoolExpr(Cond, TrueBlock, FalseBlock, TrueCount);
Alexey Bataevf8be4762019-08-14 19:30:06 +00002015 PreCondVars.restore(CGF);
Alexey Bataev62dbb972015-04-22 11:59:37 +00002016}
2017
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002018void CodeGenFunction::EmitOMPLinearClause(
2019 const OMPLoopDirective &D, CodeGenFunction::OMPPrivateScope &PrivateScope) {
2020 if (!HaveInsertPoint())
Alexey Bataev8ef31412015-12-18 07:58:25 +00002021 return;
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002022 llvm::DenseSet<const VarDecl *> SIMDLCVs;
2023 if (isOpenMPSimdDirective(D.getDirectiveKind())) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002024 const auto *LoopDirective = cast<OMPLoopDirective>(&D);
2025 for (const Expr *C : LoopDirective->counters()) {
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002026 SIMDLCVs.insert(
2027 cast<VarDecl>(cast<DeclRefExpr>(C)->getDecl())->getCanonicalDecl());
2028 }
2029 }
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00002030 for (const auto *C : D.getClausesOfKind<OMPLinearClause>()) {
Alexey Bataevbd9fec12015-08-18 06:47:21 +00002031 auto CurPrivate = C->privates().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00002032 for (const Expr *E : C->varlists()) {
2033 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
2034 const auto *PrivateVD =
Alexey Bataevbd9fec12015-08-18 06:47:21 +00002035 cast<VarDecl>(cast<DeclRefExpr>(*CurPrivate)->getDecl());
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002036 if (!SIMDLCVs.count(VD->getCanonicalDecl())) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002037 bool IsRegistered = PrivateScope.addPrivate(VD, [this, PrivateVD]() {
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002038 // Emit private VarDecl with copy init.
2039 EmitVarDecl(*PrivateVD);
2040 return GetAddrOfLocalVar(PrivateVD);
2041 });
2042 assert(IsRegistered && "linear var already registered as private");
2043 // Silence the warning about unused variable.
2044 (void)IsRegistered;
Alexey Bataevddf3db92018-04-13 17:31:06 +00002045 } else {
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002046 EmitVarDecl(*PrivateVD);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002047 }
Alexey Bataevbd9fec12015-08-18 06:47:21 +00002048 ++CurPrivate;
Alexander Musman3276a272015-03-21 10:12:56 +00002049 }
2050 }
2051}
2052
Alexey Bataev45bfad52015-08-21 12:19:04 +00002053static void emitSimdlenSafelenClause(CodeGenFunction &CGF,
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002054 const OMPExecutableDirective &D,
2055 bool IsMonotonic) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00002056 if (!CGF.HaveInsertPoint())
2057 return;
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00002058 if (const auto *C = D.getSingleClause<OMPSimdlenClause>()) {
Alexey Bataev45bfad52015-08-21 12:19:04 +00002059 RValue Len = CGF.EmitAnyExpr(C->getSimdlen(), AggValueSlot::ignored(),
2060 /*ignoreResult=*/true);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002061 auto *Val = cast<llvm::ConstantInt>(Len.getScalarVal());
Alexey Bataev45bfad52015-08-21 12:19:04 +00002062 CGF.LoopStack.setVectorizeWidth(Val->getZExtValue());
2063 // In presence of finite 'safelen', it may be unsafe to mark all
2064 // the memory instructions parallel, because loop-carried
2065 // dependences of 'safelen' iterations are possible.
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002066 if (!IsMonotonic)
2067 CGF.LoopStack.setParallel(!D.getSingleClause<OMPSafelenClause>());
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00002068 } else if (const auto *C = D.getSingleClause<OMPSafelenClause>()) {
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00002069 RValue Len = CGF.EmitAnyExpr(C->getSafelen(), AggValueSlot::ignored(),
2070 /*ignoreResult=*/true);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002071 auto *Val = cast<llvm::ConstantInt>(Len.getScalarVal());
Tyler Nowickida46d0e2015-07-14 23:03:09 +00002072 CGF.LoopStack.setVectorizeWidth(Val->getZExtValue());
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00002073 // In presence of finite 'safelen', it may be unsafe to mark all
2074 // the memory instructions parallel, because loop-carried
2075 // dependences of 'safelen' iterations are possible.
Alexey Bataevddf3db92018-04-13 17:31:06 +00002076 CGF.LoopStack.setParallel(/*Enable=*/false);
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00002077 }
2078}
2079
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002080void CodeGenFunction::EmitOMPSimdInit(const OMPLoopDirective &D,
2081 bool IsMonotonic) {
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002082 // Walk clauses and process safelen/lastprivate.
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002083 LoopStack.setParallel(!IsMonotonic);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002084 LoopStack.setVectorizeEnable();
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002085 emitSimdlenSafelenClause(*this, D, IsMonotonic);
Alexey Bataeva7815212020-02-03 12:08:16 -05002086 if (const auto *C = D.getSingleClause<OMPOrderClause>())
2087 if (C->getKind() == OMPC_ORDER_concurrent)
2088 LoopStack.setParallel(/*Enable=*/true);
Alexey Bataev43101d12020-06-11 11:28:59 -04002089 if ((D.getDirectiveKind() == OMPD_simd ||
2090 (getLangOpts().OpenMPSimd &&
2091 isOpenMPSimdDirective(D.getDirectiveKind()))) &&
2092 llvm::any_of(D.getClausesOfKind<OMPReductionClause>(),
2093 [](const OMPReductionClause *C) {
2094 return C->getModifier() == OMPC_REDUCTION_inscan;
2095 }))
2096 // Disable parallel access in case of prefix sum.
2097 LoopStack.setParallel(/*Enable=*/false);
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002098}
2099
Alexey Bataevef549a82016-03-09 09:49:09 +00002100void CodeGenFunction::EmitOMPSimdFinal(
2101 const OMPLoopDirective &D,
Alexey Bataevddf3db92018-04-13 17:31:06 +00002102 const llvm::function_ref<llvm::Value *(CodeGenFunction &)> CondGen) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00002103 if (!HaveInsertPoint())
2104 return;
Alexey Bataevef549a82016-03-09 09:49:09 +00002105 llvm::BasicBlock *DoneBB = nullptr;
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002106 auto IC = D.counters().begin();
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002107 auto IPC = D.private_counters().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00002108 for (const Expr *F : D.finals()) {
2109 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>((*IC))->getDecl());
2110 const auto *PrivateVD = cast<VarDecl>(cast<DeclRefExpr>((*IPC))->getDecl());
2111 const auto *CED = dyn_cast<OMPCapturedExprDecl>(OrigVD);
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002112 if (LocalDeclMap.count(OrigVD) || CapturedStmtInfo->lookup(OrigVD) ||
2113 OrigVD->hasGlobalStorage() || CED) {
Alexey Bataevef549a82016-03-09 09:49:09 +00002114 if (!DoneBB) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002115 if (llvm::Value *Cond = CondGen(*this)) {
Alexey Bataevef549a82016-03-09 09:49:09 +00002116 // If the first post-update expression is found, emit conditional
2117 // block if it was requested.
Alexey Bataevddf3db92018-04-13 17:31:06 +00002118 llvm::BasicBlock *ThenBB = createBasicBlock(".omp.final.then");
Alexey Bataevef549a82016-03-09 09:49:09 +00002119 DoneBB = createBasicBlock(".omp.final.done");
2120 Builder.CreateCondBr(Cond, ThenBB, DoneBB);
2121 EmitBlock(ThenBB);
2122 }
2123 }
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002124 Address OrigAddr = Address::invalid();
Alexey Bataevab4ea222018-03-07 18:17:06 +00002125 if (CED) {
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002126 OrigAddr =
2127 EmitLValue(CED->getInit()->IgnoreImpCasts()).getAddress(*this);
Alexey Bataevab4ea222018-03-07 18:17:06 +00002128 } else {
Bruno Ricci5fc4db72018-12-21 14:10:18 +00002129 DeclRefExpr DRE(getContext(), const_cast<VarDecl *>(PrivateVD),
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002130 /*RefersToEnclosingVariableOrCapture=*/false,
2131 (*IPC)->getType(), VK_LValue, (*IPC)->getExprLoc());
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002132 OrigAddr = EmitLValue(&DRE).getAddress(*this);
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002133 }
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002134 OMPPrivateScope VarScope(*this);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002135 VarScope.addPrivate(OrigVD, [OrigAddr]() { return OrigAddr; });
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002136 (void)VarScope.Privatize();
2137 EmitIgnoredExpr(F);
2138 }
2139 ++IC;
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002140 ++IPC;
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002141 }
Alexey Bataevef549a82016-03-09 09:49:09 +00002142 if (DoneBB)
2143 EmitBlock(DoneBB, /*IsFinished=*/true);
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002144}
2145
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002146static void emitOMPLoopBodyWithStopPoint(CodeGenFunction &CGF,
2147 const OMPLoopDirective &S,
2148 CodeGenFunction::JumpDest LoopExit) {
2149 CGF.EmitOMPLoopBody(S, LoopExit);
2150 CGF.EmitStopPoint(&S);
Hans Wennborged129ae2017-04-27 17:02:25 +00002151}
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002152
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00002153/// Emit a helper variable and return corresponding lvalue.
2154static LValue EmitOMPHelperVar(CodeGenFunction &CGF,
2155 const DeclRefExpr *Helper) {
2156 auto VDecl = cast<VarDecl>(Helper->getDecl());
2157 CGF.EmitVarDecl(*VDecl);
2158 return CGF.EmitLValue(Helper);
2159}
2160
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002161static void emitCommonSimdLoop(CodeGenFunction &CGF, const OMPLoopDirective &S,
2162 const RegionCodeGenTy &SimdInitGen,
2163 const RegionCodeGenTy &BodyCodeGen) {
Alexey Bataev0860db92019-12-19 10:01:10 -05002164 auto &&ThenGen = [&S, &SimdInitGen, &BodyCodeGen](CodeGenFunction &CGF,
2165 PrePostActionTy &) {
2166 CGOpenMPRuntime::NontemporalDeclsRAII NontemporalsRegion(CGF.CGM, S);
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002167 CodeGenFunction::OMPLocalDeclMapRAII Scope(CGF);
2168 SimdInitGen(CGF);
2169
2170 BodyCodeGen(CGF);
2171 };
2172 auto &&ElseGen = [&BodyCodeGen](CodeGenFunction &CGF, PrePostActionTy &) {
2173 CodeGenFunction::OMPLocalDeclMapRAII Scope(CGF);
2174 CGF.LoopStack.setVectorizeEnable(/*Enable=*/false);
2175
2176 BodyCodeGen(CGF);
2177 };
2178 const Expr *IfCond = nullptr;
Alexey Bataev18789bf2020-02-13 09:21:15 -05002179 if (isOpenMPSimdDirective(S.getDirectiveKind())) {
2180 for (const auto *C : S.getClausesOfKind<OMPIfClause>()) {
2181 if (CGF.getLangOpts().OpenMP >= 50 &&
2182 (C->getNameModifier() == OMPD_unknown ||
2183 C->getNameModifier() == OMPD_simd)) {
2184 IfCond = C->getCondition();
2185 break;
2186 }
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002187 }
2188 }
2189 if (IfCond) {
2190 CGF.CGM.getOpenMPRuntime().emitIfClause(CGF, IfCond, ThenGen, ElseGen);
2191 } else {
2192 RegionCodeGenTy ThenRCG(ThenGen);
2193 ThenRCG(CGF);
2194 }
2195}
2196
Alexey Bataevf8365372017-11-17 17:57:25 +00002197static void emitOMPSimdRegion(CodeGenFunction &CGF, const OMPLoopDirective &S,
2198 PrePostActionTy &Action) {
2199 Action.Enter(CGF);
2200 assert(isOpenMPSimdDirective(S.getDirectiveKind()) &&
2201 "Expected simd directive");
2202 OMPLoopScope PreInitScope(CGF, S);
2203 // if (PreCond) {
2204 // for (IV in 0..LastIteration) BODY;
2205 // <Final counter/linear vars updates>;
2206 // }
2207 //
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00002208 if (isOpenMPDistributeDirective(S.getDirectiveKind()) ||
2209 isOpenMPWorksharingDirective(S.getDirectiveKind()) ||
2210 isOpenMPTaskLoopDirective(S.getDirectiveKind())) {
2211 (void)EmitOMPHelperVar(CGF, cast<DeclRefExpr>(S.getLowerBoundVariable()));
2212 (void)EmitOMPHelperVar(CGF, cast<DeclRefExpr>(S.getUpperBoundVariable()));
2213 }
Alexander Musmana5f070a2014-10-01 06:03:56 +00002214
Alexey Bataevf8365372017-11-17 17:57:25 +00002215 // Emit: if (PreCond) - begin.
2216 // If the condition constant folds and can be elided, avoid emitting the
2217 // whole loop.
2218 bool CondConstant;
2219 llvm::BasicBlock *ContBlock = nullptr;
2220 if (CGF.ConstantFoldsToSimpleInteger(S.getPreCond(), CondConstant)) {
2221 if (!CondConstant)
2222 return;
2223 } else {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002224 llvm::BasicBlock *ThenBlock = CGF.createBasicBlock("simd.if.then");
Alexey Bataevf8365372017-11-17 17:57:25 +00002225 ContBlock = CGF.createBasicBlock("simd.if.end");
2226 emitPreCond(CGF, S, S.getPreCond(), ThenBlock, ContBlock,
2227 CGF.getProfileCount(&S));
2228 CGF.EmitBlock(ThenBlock);
2229 CGF.incrementProfileCounter(&S);
2230 }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00002231
Alexey Bataevf8365372017-11-17 17:57:25 +00002232 // Emit the loop iteration variable.
2233 const Expr *IVExpr = S.getIterationVariable();
Alexey Bataevddf3db92018-04-13 17:31:06 +00002234 const auto *IVDecl = cast<VarDecl>(cast<DeclRefExpr>(IVExpr)->getDecl());
Alexey Bataevf8365372017-11-17 17:57:25 +00002235 CGF.EmitVarDecl(*IVDecl);
2236 CGF.EmitIgnoredExpr(S.getInit());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00002237
Alexey Bataevf8365372017-11-17 17:57:25 +00002238 // Emit the iterations count variable.
2239 // If it is not a variable, Sema decided to calculate iterations count on
2240 // each iteration (e.g., it is foldable into a constant).
Alexey Bataevddf3db92018-04-13 17:31:06 +00002241 if (const auto *LIExpr = dyn_cast<DeclRefExpr>(S.getLastIteration())) {
Alexey Bataevf8365372017-11-17 17:57:25 +00002242 CGF.EmitVarDecl(*cast<VarDecl>(LIExpr->getDecl()));
2243 // Emit calculation of the iterations count.
2244 CGF.EmitIgnoredExpr(S.getCalcLastIteration());
2245 }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00002246
Alexey Bataevf8365372017-11-17 17:57:25 +00002247 emitAlignedClause(CGF, S);
2248 (void)CGF.EmitOMPLinearClauseInit(S);
2249 {
2250 CodeGenFunction::OMPPrivateScope LoopScope(CGF);
2251 CGF.EmitOMPPrivateLoopCounters(S, LoopScope);
2252 CGF.EmitOMPLinearClause(S, LoopScope);
2253 CGF.EmitOMPPrivateClause(S, LoopScope);
2254 CGF.EmitOMPReductionClauseInit(S, LoopScope);
Alexey Bataeva58da1a2019-12-27 09:44:43 -05002255 CGOpenMPRuntime::LastprivateConditionalRAII LPCRegion(
2256 CGF, S, CGF.EmitLValue(S.getIterationVariable()));
Alexey Bataevf8365372017-11-17 17:57:25 +00002257 bool HasLastprivateClause = CGF.EmitOMPLastprivateClauseInit(S, LoopScope);
2258 (void)LoopScope.Privatize();
Alexey Bataev60705422018-10-30 15:50:12 +00002259 if (isOpenMPTargetExecutionDirective(S.getDirectiveKind()))
2260 CGF.CGM.getOpenMPRuntime().adjustTargetSpecificDataForLambdas(CGF, S);
Alexey Bataevd08c0562019-11-19 12:07:54 -05002261
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002262 emitCommonSimdLoop(
2263 CGF, S,
2264 [&S](CodeGenFunction &CGF, PrePostActionTy &) {
2265 CGF.EmitOMPSimdInit(S);
2266 },
2267 [&S, &LoopScope](CodeGenFunction &CGF, PrePostActionTy &) {
2268 CGF.EmitOMPInnerLoop(
2269 S, LoopScope.requiresCleanups(), S.getCond(), S.getInc(),
2270 [&S](CodeGenFunction &CGF) {
Alexey Bataevcb9191c2020-06-09 08:04:48 -04002271 emitOMPLoopBodyWithStopPoint(CGF, S,
2272 CodeGenFunction::JumpDest());
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002273 },
2274 [](CodeGenFunction &) {});
2275 });
Alexey Bataevddf3db92018-04-13 17:31:06 +00002276 CGF.EmitOMPSimdFinal(S, [](CodeGenFunction &) { return nullptr; });
Alexey Bataevf8365372017-11-17 17:57:25 +00002277 // Emit final copy of the lastprivate variables at the end of loops.
2278 if (HasLastprivateClause)
2279 CGF.EmitOMPLastprivateClauseFinal(S, /*NoFinals=*/true);
2280 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_simd);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002281 emitPostUpdateForReductionClause(CGF, S,
2282 [](CodeGenFunction &) { return nullptr; });
Alexey Bataevf8365372017-11-17 17:57:25 +00002283 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00002284 CGF.EmitOMPLinearClauseFinal(S, [](CodeGenFunction &) { return nullptr; });
Alexey Bataevf8365372017-11-17 17:57:25 +00002285 // Emit: if (PreCond) - end.
2286 if (ContBlock) {
2287 CGF.EmitBranch(ContBlock);
2288 CGF.EmitBlock(ContBlock, true);
2289 }
2290}
2291
2292void CodeGenFunction::EmitOMPSimdDirective(const OMPSimdDirective &S) {
Alexey Bataev43101d12020-06-11 11:28:59 -04002293 ParentLoopDirectiveForScanRegion ScanRegion(*this, S);
2294 OMPFirstScanLoop = true;
Alexey Bataevf8365372017-11-17 17:57:25 +00002295 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
2296 emitOMPSimdRegion(CGF, S, Action);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00002297 };
Alexey Bataev46978742020-01-30 10:46:11 -05002298 {
2299 auto LPCRegion =
2300 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
2301 OMPLexicalScope Scope(*this, S, OMPD_unknown);
2302 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_simd, CodeGen);
2303 }
2304 // Check for outer lastprivate conditional update.
2305 checkForLastprivateConditionalUpdate(*this, S);
Alexander Musman515ad8c2014-05-22 08:54:05 +00002306}
2307
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002308void CodeGenFunction::EmitOMPOuterLoop(
2309 bool DynamicOrOrdered, bool IsMonotonic, const OMPLoopDirective &S,
2310 CodeGenFunction::OMPPrivateScope &LoopScope,
2311 const CodeGenFunction::OMPLoopArguments &LoopArgs,
2312 const CodeGenFunction::CodeGenLoopTy &CodeGenLoop,
2313 const CodeGenFunction::CodeGenOrderedTy &CodeGenOrdered) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002314 CGOpenMPRuntime &RT = CGM.getOpenMPRuntime();
Alexander Musman92bdaab2015-03-12 13:37:50 +00002315
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002316 const Expr *IVExpr = S.getIterationVariable();
2317 const unsigned IVSize = getContext().getTypeSize(IVExpr->getType());
2318 const bool IVSigned = IVExpr->getType()->hasSignedIntegerRepresentation();
2319
Alexey Bataevddf3db92018-04-13 17:31:06 +00002320 JumpDest LoopExit = getJumpDestInCurrentScope("omp.dispatch.end");
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002321
2322 // Start the loop with a block that tests the condition.
Alexey Bataevddf3db92018-04-13 17:31:06 +00002323 llvm::BasicBlock *CondBlock = createBasicBlock("omp.dispatch.cond");
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002324 EmitBlock(CondBlock);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002325 const SourceRange R = S.getSourceRange();
Amara Emerson652795d2016-11-10 14:44:30 +00002326 LoopStack.push(CondBlock, SourceLocToDebugLoc(R.getBegin()),
2327 SourceLocToDebugLoc(R.getEnd()));
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002328
2329 llvm::Value *BoolCondVal = nullptr;
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00002330 if (!DynamicOrOrdered) {
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002331 // UB = min(UB, GlobalUB) or
2332 // UB = min(UB, PrevUB) for combined loop sharing constructs (e.g.
2333 // 'distribute parallel for')
2334 EmitIgnoredExpr(LoopArgs.EUB);
Alexander Musman92bdaab2015-03-12 13:37:50 +00002335 // IV = LB
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002336 EmitIgnoredExpr(LoopArgs.Init);
Alexander Musman92bdaab2015-03-12 13:37:50 +00002337 // IV < UB
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002338 BoolCondVal = EvaluateExprAsBool(LoopArgs.Cond);
Alexander Musman92bdaab2015-03-12 13:37:50 +00002339 } else {
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002340 BoolCondVal =
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002341 RT.emitForNext(*this, S.getBeginLoc(), IVSize, IVSigned, LoopArgs.IL,
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002342 LoopArgs.LB, LoopArgs.UB, LoopArgs.ST);
Alexander Musman92bdaab2015-03-12 13:37:50 +00002343 }
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002344
2345 // If there are any cleanups between here and the loop-exit scope,
2346 // create a block to stage a loop exit along.
Alexey Bataevddf3db92018-04-13 17:31:06 +00002347 llvm::BasicBlock *ExitBlock = LoopExit.getBlock();
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002348 if (LoopScope.requiresCleanups())
2349 ExitBlock = createBasicBlock("omp.dispatch.cleanup");
2350
Alexey Bataevddf3db92018-04-13 17:31:06 +00002351 llvm::BasicBlock *LoopBody = createBasicBlock("omp.dispatch.body");
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002352 Builder.CreateCondBr(BoolCondVal, LoopBody, ExitBlock);
2353 if (ExitBlock != LoopExit.getBlock()) {
2354 EmitBlock(ExitBlock);
2355 EmitBranchThroughCleanup(LoopExit);
2356 }
2357 EmitBlock(LoopBody);
2358
Alexander Musman92bdaab2015-03-12 13:37:50 +00002359 // Emit "IV = LB" (in case of static schedule, we have already calculated new
2360 // LB for loop condition and emitted it above).
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00002361 if (DynamicOrOrdered)
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002362 EmitIgnoredExpr(LoopArgs.Init);
Alexander Musman92bdaab2015-03-12 13:37:50 +00002363
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002364 // Create a block for the increment.
Alexey Bataevddf3db92018-04-13 17:31:06 +00002365 JumpDest Continue = getJumpDestInCurrentScope("omp.dispatch.inc");
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002366 BreakContinueStack.push_back(BreakContinue(LoopExit, Continue));
2367
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002368 emitCommonSimdLoop(
2369 *this, S,
2370 [&S, IsMonotonic](CodeGenFunction &CGF, PrePostActionTy &) {
2371 // Generate !llvm.loop.parallel metadata for loads and stores for loops
2372 // with dynamic/guided scheduling and without ordered clause.
Alexey Bataeva7815212020-02-03 12:08:16 -05002373 if (!isOpenMPSimdDirective(S.getDirectiveKind())) {
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002374 CGF.LoopStack.setParallel(!IsMonotonic);
Alexey Bataeva7815212020-02-03 12:08:16 -05002375 if (const auto *C = S.getSingleClause<OMPOrderClause>())
2376 if (C->getKind() == OMPC_ORDER_concurrent)
2377 CGF.LoopStack.setParallel(/*Enable=*/true);
2378 } else {
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002379 CGF.EmitOMPSimdInit(S, IsMonotonic);
Alexey Bataeva7815212020-02-03 12:08:16 -05002380 }
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002381 },
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002382 [&S, &LoopArgs, LoopExit, &CodeGenLoop, IVSize, IVSigned, &CodeGenOrdered,
2383 &LoopScope](CodeGenFunction &CGF, PrePostActionTy &) {
2384 SourceLocation Loc = S.getBeginLoc();
2385 // when 'distribute' is not combined with a 'for':
2386 // while (idx <= UB) { BODY; ++idx; }
2387 // when 'distribute' is combined with a 'for'
2388 // (e.g. 'distribute parallel for')
2389 // while (idx <= UB) { <CodeGen rest of pragma>; idx += ST; }
2390 CGF.EmitOMPInnerLoop(
2391 S, LoopScope.requiresCleanups(), LoopArgs.Cond, LoopArgs.IncExpr,
2392 [&S, LoopExit, &CodeGenLoop](CodeGenFunction &CGF) {
2393 CodeGenLoop(CGF, S, LoopExit);
2394 },
2395 [IVSize, IVSigned, Loc, &CodeGenOrdered](CodeGenFunction &CGF) {
2396 CodeGenOrdered(CGF, Loc, IVSize, IVSigned);
2397 });
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002398 });
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002399
2400 EmitBlock(Continue.getBlock());
2401 BreakContinueStack.pop_back();
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00002402 if (!DynamicOrOrdered) {
Alexander Musman92bdaab2015-03-12 13:37:50 +00002403 // Emit "LB = LB + Stride", "UB = UB + Stride".
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002404 EmitIgnoredExpr(LoopArgs.NextLB);
2405 EmitIgnoredExpr(LoopArgs.NextUB);
Alexander Musman92bdaab2015-03-12 13:37:50 +00002406 }
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002407
2408 EmitBranch(CondBlock);
2409 LoopStack.pop();
2410 // Emit the fall-through block.
2411 EmitBlock(LoopExit.getBlock());
2412
2413 // Tell the runtime we are done.
Alexey Bataev957d8562016-11-17 15:12:05 +00002414 auto &&CodeGen = [DynamicOrOrdered, &S](CodeGenFunction &CGF) {
2415 if (!DynamicOrOrdered)
Stephen Kelly1c301dc2018-08-09 21:09:38 +00002416 CGF.CGM.getOpenMPRuntime().emitForStaticFinish(CGF, S.getEndLoc(),
Alexey Bataevf43f7142017-09-06 16:17:35 +00002417 S.getDirectiveKind());
Alexey Bataev957d8562016-11-17 15:12:05 +00002418 };
2419 OMPCancelStack.emitExit(*this, S.getDirectiveKind(), CodeGen);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002420}
2421
2422void CodeGenFunction::EmitOMPForOuterLoop(
Alexey Bataev9ebd7422016-05-10 09:57:36 +00002423 const OpenMPScheduleTy &ScheduleKind, bool IsMonotonic,
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002424 const OMPLoopDirective &S, OMPPrivateScope &LoopScope, bool Ordered,
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002425 const OMPLoopArguments &LoopArgs,
2426 const CodeGenDispatchBoundsTy &CGDispatchBounds) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002427 CGOpenMPRuntime &RT = CGM.getOpenMPRuntime();
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002428
2429 // Dynamic scheduling of the outer loop (dynamic, guided, auto, runtime).
Alexey Bataev9ebd7422016-05-10 09:57:36 +00002430 const bool DynamicOrOrdered =
2431 Ordered || RT.isDynamic(ScheduleKind.Schedule);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002432
2433 assert((Ordered ||
Alexey Bataev9ebd7422016-05-10 09:57:36 +00002434 !RT.isStaticNonchunked(ScheduleKind.Schedule,
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002435 LoopArgs.Chunk != nullptr)) &&
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002436 "static non-chunked schedule does not need outer loop");
2437
2438 // Emit outer loop.
2439 //
2440 // OpenMP [2.7.1, Loop Construct, Description, table 2-1]
2441 // When schedule(dynamic,chunk_size) is specified, the iterations are
2442 // distributed to threads in the team in chunks as the threads request them.
2443 // Each thread executes a chunk of iterations, then requests another chunk,
2444 // until no chunks remain to be distributed. Each chunk contains chunk_size
2445 // iterations, except for the last chunk to be distributed, which may have
2446 // fewer iterations. When no chunk_size is specified, it defaults to 1.
2447 //
2448 // When schedule(guided,chunk_size) is specified, the iterations are assigned
2449 // to threads in the team in chunks as the executing threads request them.
2450 // Each thread executes a chunk of iterations, then requests another chunk,
2451 // until no chunks remain to be assigned. For a chunk_size of 1, the size of
2452 // each chunk is proportional to the number of unassigned iterations divided
2453 // by the number of threads in the team, decreasing to 1. For a chunk_size
2454 // with value k (greater than 1), the size of each chunk is determined in the
2455 // same way, with the restriction that the chunks do not contain fewer than k
2456 // iterations (except for the last chunk to be assigned, which may have fewer
2457 // than k iterations).
2458 //
2459 // When schedule(auto) is specified, the decision regarding scheduling is
2460 // delegated to the compiler and/or runtime system. The programmer gives the
2461 // implementation the freedom to choose any possible mapping of iterations to
2462 // threads in the team.
2463 //
2464 // When schedule(runtime) is specified, the decision regarding scheduling is
2465 // deferred until run time, and the schedule and chunk size are taken from the
2466 // run-sched-var ICV. If the ICV is set to auto, the schedule is
2467 // implementation defined
2468 //
2469 // while(__kmpc_dispatch_next(&LB, &UB)) {
2470 // idx = LB;
2471 // while (idx <= UB) { BODY; ++idx;
2472 // __kmpc_dispatch_fini_(4|8)[u](); // For ordered loops only.
2473 // } // inner loop
2474 // }
2475 //
2476 // OpenMP [2.7.1, Loop Construct, Description, table 2-1]
2477 // When schedule(static, chunk_size) is specified, iterations are divided into
2478 // chunks of size chunk_size, and the chunks are assigned to the threads in
2479 // the team in a round-robin fashion in the order of the thread number.
2480 //
2481 // while(UB = min(UB, GlobalUB), idx = LB, idx < UB) {
2482 // while (idx <= UB) { BODY; ++idx; } // inner loop
2483 // LB = LB + ST;
2484 // UB = UB + ST;
2485 // }
2486 //
2487
2488 const Expr *IVExpr = S.getIterationVariable();
2489 const unsigned IVSize = getContext().getTypeSize(IVExpr->getType());
2490 const bool IVSigned = IVExpr->getType()->hasSignedIntegerRepresentation();
2491
2492 if (DynamicOrOrdered) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002493 const std::pair<llvm::Value *, llvm::Value *> DispatchBounds =
2494 CGDispatchBounds(*this, S, LoopArgs.LB, LoopArgs.UB);
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002495 llvm::Value *LBVal = DispatchBounds.first;
2496 llvm::Value *UBVal = DispatchBounds.second;
2497 CGOpenMPRuntime::DispatchRTInput DipatchRTInputValues = {LBVal, UBVal,
2498 LoopArgs.Chunk};
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002499 RT.emitForDispatchInit(*this, S.getBeginLoc(), ScheduleKind, IVSize,
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002500 IVSigned, Ordered, DipatchRTInputValues);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002501 } else {
Alexey Bataev0f87dbe2017-08-14 17:56:13 +00002502 CGOpenMPRuntime::StaticRTInput StaticInit(
2503 IVSize, IVSigned, Ordered, LoopArgs.IL, LoopArgs.LB, LoopArgs.UB,
2504 LoopArgs.ST, LoopArgs.Chunk);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002505 RT.emitForStaticInit(*this, S.getBeginLoc(), S.getDirectiveKind(),
Alexey Bataev0f87dbe2017-08-14 17:56:13 +00002506 ScheduleKind, StaticInit);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002507 }
2508
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002509 auto &&CodeGenOrdered = [Ordered](CodeGenFunction &CGF, SourceLocation Loc,
2510 const unsigned IVSize,
2511 const bool IVSigned) {
2512 if (Ordered) {
2513 CGF.CGM.getOpenMPRuntime().emitForOrderedIterationEnd(CGF, Loc, IVSize,
2514 IVSigned);
2515 }
2516 };
2517
2518 OMPLoopArguments OuterLoopArgs(LoopArgs.LB, LoopArgs.UB, LoopArgs.ST,
2519 LoopArgs.IL, LoopArgs.Chunk, LoopArgs.EUB);
2520 OuterLoopArgs.IncExpr = S.getInc();
2521 OuterLoopArgs.Init = S.getInit();
2522 OuterLoopArgs.Cond = S.getCond();
2523 OuterLoopArgs.NextLB = S.getNextLowerBound();
2524 OuterLoopArgs.NextUB = S.getNextUpperBound();
2525 EmitOMPOuterLoop(DynamicOrOrdered, IsMonotonic, S, LoopScope, OuterLoopArgs,
2526 emitOMPLoopBodyWithStopPoint, CodeGenOrdered);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002527}
2528
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002529static void emitEmptyOrdered(CodeGenFunction &, SourceLocation Loc,
2530 const unsigned IVSize, const bool IVSigned) {}
2531
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002532void CodeGenFunction::EmitOMPDistributeOuterLoop(
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002533 OpenMPDistScheduleClauseKind ScheduleKind, const OMPLoopDirective &S,
2534 OMPPrivateScope &LoopScope, const OMPLoopArguments &LoopArgs,
2535 const CodeGenLoopTy &CodeGenLoopContent) {
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002536
Alexey Bataevddf3db92018-04-13 17:31:06 +00002537 CGOpenMPRuntime &RT = CGM.getOpenMPRuntime();
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002538
2539 // Emit outer loop.
2540 // Same behavior as a OMPForOuterLoop, except that schedule cannot be
2541 // dynamic
2542 //
2543
2544 const Expr *IVExpr = S.getIterationVariable();
2545 const unsigned IVSize = getContext().getTypeSize(IVExpr->getType());
2546 const bool IVSigned = IVExpr->getType()->hasSignedIntegerRepresentation();
2547
Alexey Bataev0f87dbe2017-08-14 17:56:13 +00002548 CGOpenMPRuntime::StaticRTInput StaticInit(
2549 IVSize, IVSigned, /* Ordered = */ false, LoopArgs.IL, LoopArgs.LB,
2550 LoopArgs.UB, LoopArgs.ST, LoopArgs.Chunk);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002551 RT.emitDistributeStaticInit(*this, S.getBeginLoc(), ScheduleKind, StaticInit);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00002552
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002553 // for combined 'distribute' and 'for' the increment expression of distribute
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002554 // is stored in DistInc. For 'distribute' alone, it is in Inc.
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002555 Expr *IncExpr;
2556 if (isOpenMPLoopBoundSharingDirective(S.getDirectiveKind()))
2557 IncExpr = S.getDistInc();
2558 else
2559 IncExpr = S.getInc();
2560
2561 // this routine is shared by 'omp distribute parallel for' and
2562 // 'omp distribute': select the right EUB expression depending on the
2563 // directive
2564 OMPLoopArguments OuterLoopArgs;
2565 OuterLoopArgs.LB = LoopArgs.LB;
2566 OuterLoopArgs.UB = LoopArgs.UB;
2567 OuterLoopArgs.ST = LoopArgs.ST;
2568 OuterLoopArgs.IL = LoopArgs.IL;
2569 OuterLoopArgs.Chunk = LoopArgs.Chunk;
2570 OuterLoopArgs.EUB = isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
2571 ? S.getCombinedEnsureUpperBound()
2572 : S.getEnsureUpperBound();
2573 OuterLoopArgs.IncExpr = IncExpr;
2574 OuterLoopArgs.Init = isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
2575 ? S.getCombinedInit()
2576 : S.getInit();
2577 OuterLoopArgs.Cond = isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
2578 ? S.getCombinedCond()
2579 : S.getCond();
2580 OuterLoopArgs.NextLB = isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
2581 ? S.getCombinedNextLowerBound()
2582 : S.getNextLowerBound();
2583 OuterLoopArgs.NextUB = isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
2584 ? S.getCombinedNextUpperBound()
2585 : S.getNextUpperBound();
2586
2587 EmitOMPOuterLoop(/* DynamicOrOrdered = */ false, /* IsMonotonic = */ false, S,
2588 LoopScope, OuterLoopArgs, CodeGenLoopContent,
2589 emitEmptyOrdered);
2590}
2591
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002592static std::pair<LValue, LValue>
2593emitDistributeParallelForInnerBounds(CodeGenFunction &CGF,
2594 const OMPExecutableDirective &S) {
2595 const OMPLoopDirective &LS = cast<OMPLoopDirective>(S);
2596 LValue LB =
2597 EmitOMPHelperVar(CGF, cast<DeclRefExpr>(LS.getLowerBoundVariable()));
2598 LValue UB =
2599 EmitOMPHelperVar(CGF, cast<DeclRefExpr>(LS.getUpperBoundVariable()));
2600
2601 // When composing 'distribute' with 'for' (e.g. as in 'distribute
2602 // parallel for') we need to use the 'distribute'
2603 // chunk lower and upper bounds rather than the whole loop iteration
2604 // space. These are parameters to the outlined function for 'parallel'
2605 // and we copy the bounds of the previous schedule into the
2606 // the current ones.
2607 LValue PrevLB = CGF.EmitLValue(LS.getPrevLowerBoundVariable());
2608 LValue PrevUB = CGF.EmitLValue(LS.getPrevUpperBoundVariable());
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00002609 llvm::Value *PrevLBVal = CGF.EmitLoadOfScalar(
2610 PrevLB, LS.getPrevLowerBoundVariable()->getExprLoc());
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002611 PrevLBVal = CGF.EmitScalarConversion(
2612 PrevLBVal, LS.getPrevLowerBoundVariable()->getType(),
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00002613 LS.getIterationVariable()->getType(),
2614 LS.getPrevLowerBoundVariable()->getExprLoc());
2615 llvm::Value *PrevUBVal = CGF.EmitLoadOfScalar(
2616 PrevUB, LS.getPrevUpperBoundVariable()->getExprLoc());
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002617 PrevUBVal = CGF.EmitScalarConversion(
2618 PrevUBVal, LS.getPrevUpperBoundVariable()->getType(),
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00002619 LS.getIterationVariable()->getType(),
2620 LS.getPrevUpperBoundVariable()->getExprLoc());
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002621
2622 CGF.EmitStoreOfScalar(PrevLBVal, LB);
2623 CGF.EmitStoreOfScalar(PrevUBVal, UB);
2624
2625 return {LB, UB};
2626}
2627
2628/// if the 'for' loop has a dispatch schedule (e.g. dynamic, guided) then
2629/// we need to use the LB and UB expressions generated by the worksharing
2630/// code generation support, whereas in non combined situations we would
2631/// just emit 0 and the LastIteration expression
2632/// This function is necessary due to the difference of the LB and UB
2633/// types for the RT emission routines for 'for_static_init' and
2634/// 'for_dispatch_init'
2635static std::pair<llvm::Value *, llvm::Value *>
2636emitDistributeParallelForDispatchBounds(CodeGenFunction &CGF,
2637 const OMPExecutableDirective &S,
2638 Address LB, Address UB) {
2639 const OMPLoopDirective &LS = cast<OMPLoopDirective>(S);
2640 const Expr *IVExpr = LS.getIterationVariable();
2641 // when implementing a dynamic schedule for a 'for' combined with a
2642 // 'distribute' (e.g. 'distribute parallel for'), the 'for' loop
2643 // is not normalized as each team only executes its own assigned
2644 // distribute chunk
2645 QualType IteratorTy = IVExpr->getType();
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00002646 llvm::Value *LBVal =
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002647 CGF.EmitLoadOfScalar(LB, /*Volatile=*/false, IteratorTy, S.getBeginLoc());
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00002648 llvm::Value *UBVal =
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002649 CGF.EmitLoadOfScalar(UB, /*Volatile=*/false, IteratorTy, S.getBeginLoc());
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002650 return {LBVal, UBVal};
Hans Wennborged129ae2017-04-27 17:02:25 +00002651}
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002652
2653static void emitDistributeParallelForDistributeInnerBoundParams(
2654 CodeGenFunction &CGF, const OMPExecutableDirective &S,
2655 llvm::SmallVectorImpl<llvm::Value *> &CapturedVars) {
2656 const auto &Dir = cast<OMPLoopDirective>(S);
2657 LValue LB =
2658 CGF.EmitLValue(cast<DeclRefExpr>(Dir.getCombinedLowerBoundVariable()));
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002659 llvm::Value *LBCast =
2660 CGF.Builder.CreateIntCast(CGF.Builder.CreateLoad(LB.getAddress(CGF)),
2661 CGF.SizeTy, /*isSigned=*/false);
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002662 CapturedVars.push_back(LBCast);
2663 LValue UB =
2664 CGF.EmitLValue(cast<DeclRefExpr>(Dir.getCombinedUpperBoundVariable()));
2665
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002666 llvm::Value *UBCast =
2667 CGF.Builder.CreateIntCast(CGF.Builder.CreateLoad(UB.getAddress(CGF)),
2668 CGF.SizeTy, /*isSigned=*/false);
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002669 CapturedVars.push_back(UBCast);
Hans Wennborged129ae2017-04-27 17:02:25 +00002670}
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002671
2672static void
2673emitInnerParallelForWhenCombined(CodeGenFunction &CGF,
2674 const OMPLoopDirective &S,
2675 CodeGenFunction::JumpDest LoopExit) {
2676 auto &&CGInlinedWorksharingLoop = [&S](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00002677 PrePostActionTy &Action) {
2678 Action.Enter(CGF);
Alexey Bataevdcb4b8fb2017-11-22 20:19:50 +00002679 bool HasCancel = false;
2680 if (!isOpenMPSimdDirective(S.getDirectiveKind())) {
2681 if (const auto *D = dyn_cast<OMPTeamsDistributeParallelForDirective>(&S))
2682 HasCancel = D->hasCancel();
2683 else if (const auto *D = dyn_cast<OMPDistributeParallelForDirective>(&S))
2684 HasCancel = D->hasCancel();
Alexey Bataev16e79882017-11-22 21:12:03 +00002685 else if (const auto *D =
2686 dyn_cast<OMPTargetTeamsDistributeParallelForDirective>(&S))
2687 HasCancel = D->hasCancel();
Alexey Bataevdcb4b8fb2017-11-22 20:19:50 +00002688 }
2689 CodeGenFunction::OMPCancelStackRAII CancelRegion(CGF, S.getDirectiveKind(),
2690 HasCancel);
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002691 CGF.EmitOMPWorksharingLoop(S, S.getPrevEnsureUpperBound(),
2692 emitDistributeParallelForInnerBounds,
2693 emitDistributeParallelForDispatchBounds);
2694 };
2695
2696 emitCommonOMPParallelDirective(
Alexey Bataevdcb4b8fb2017-11-22 20:19:50 +00002697 CGF, S,
2698 isOpenMPSimdDirective(S.getDirectiveKind()) ? OMPD_for_simd : OMPD_for,
2699 CGInlinedWorksharingLoop,
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002700 emitDistributeParallelForDistributeInnerBoundParams);
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002701}
2702
Carlo Bertolli9925f152016-06-27 14:55:37 +00002703void CodeGenFunction::EmitOMPDistributeParallelForDirective(
2704 const OMPDistributeParallelForDirective &S) {
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002705 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
2706 CGF.EmitOMPDistributeLoop(S, emitInnerParallelForWhenCombined,
2707 S.getDistInc());
2708 };
Alexey Bataev475a7442018-01-12 19:39:11 +00002709 OMPLexicalScope Scope(*this, S, OMPD_parallel);
Alexey Bataev10a54312017-11-27 16:54:08 +00002710 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_distribute, CodeGen);
Carlo Bertolli9925f152016-06-27 14:55:37 +00002711}
2712
Kelvin Li4a39add2016-07-05 05:00:15 +00002713void CodeGenFunction::EmitOMPDistributeParallelForSimdDirective(
2714 const OMPDistributeParallelForSimdDirective &S) {
Alexey Bataev0b49f9e2017-11-27 19:38:58 +00002715 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
2716 CGF.EmitOMPDistributeLoop(S, emitInnerParallelForWhenCombined,
2717 S.getDistInc());
2718 };
Alexey Bataev475a7442018-01-12 19:39:11 +00002719 OMPLexicalScope Scope(*this, S, OMPD_parallel);
Alexey Bataev0b49f9e2017-11-27 19:38:58 +00002720 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_distribute, CodeGen);
Kelvin Li4a39add2016-07-05 05:00:15 +00002721}
Kelvin Li787f3fc2016-07-06 04:45:38 +00002722
2723void CodeGenFunction::EmitOMPDistributeSimdDirective(
2724 const OMPDistributeSimdDirective &S) {
Alexey Bataev617db5f2017-12-04 15:38:33 +00002725 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
2726 CGF.EmitOMPDistributeLoop(S, emitOMPLoopBodyWithStopPoint, S.getInc());
2727 };
Alexey Bataev475a7442018-01-12 19:39:11 +00002728 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataev617db5f2017-12-04 15:38:33 +00002729 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_simd, CodeGen);
Kelvin Li787f3fc2016-07-06 04:45:38 +00002730}
2731
Alexey Bataevf8365372017-11-17 17:57:25 +00002732void CodeGenFunction::EmitOMPTargetSimdDeviceFunction(
2733 CodeGenModule &CGM, StringRef ParentName, const OMPTargetSimdDirective &S) {
2734 // Emit SPMD target parallel for region as a standalone region.
2735 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
2736 emitOMPSimdRegion(CGF, S, Action);
2737 };
2738 llvm::Function *Fn;
2739 llvm::Constant *Addr;
2740 // Emit target region as a standalone region.
2741 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
2742 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
2743 assert(Fn && Addr && "Target device function emission failed.");
2744}
2745
Kelvin Li986330c2016-07-20 22:57:10 +00002746void CodeGenFunction::EmitOMPTargetSimdDirective(
2747 const OMPTargetSimdDirective &S) {
Alexey Bataevf8365372017-11-17 17:57:25 +00002748 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
2749 emitOMPSimdRegion(CGF, S, Action);
2750 };
2751 emitCommonOMPTargetDirective(*this, S, CodeGen);
Kelvin Li986330c2016-07-20 22:57:10 +00002752}
2753
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002754namespace {
2755 struct ScheduleKindModifiersTy {
2756 OpenMPScheduleClauseKind Kind;
2757 OpenMPScheduleClauseModifier M1;
2758 OpenMPScheduleClauseModifier M2;
2759 ScheduleKindModifiersTy(OpenMPScheduleClauseKind Kind,
2760 OpenMPScheduleClauseModifier M1,
2761 OpenMPScheduleClauseModifier M2)
2762 : Kind(Kind), M1(M1), M2(M2) {}
2763 };
2764} // namespace
2765
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002766bool CodeGenFunction::EmitOMPWorksharingLoop(
2767 const OMPLoopDirective &S, Expr *EUB,
2768 const CodeGenLoopBoundsTy &CodeGenLoopBounds,
2769 const CodeGenDispatchBoundsTy &CGDispatchBounds) {
Alexander Musmanc6388682014-12-15 07:07:06 +00002770 // Emit the loop iteration variable.
Alexey Bataevddf3db92018-04-13 17:31:06 +00002771 const auto *IVExpr = cast<DeclRefExpr>(S.getIterationVariable());
2772 const auto *IVDecl = cast<VarDecl>(IVExpr->getDecl());
Alexander Musmanc6388682014-12-15 07:07:06 +00002773 EmitVarDecl(*IVDecl);
2774
2775 // Emit the iterations count variable.
2776 // If it is not a variable, Sema decided to calculate iterations count on each
2777 // iteration (e.g., it is foldable into a constant).
Alexey Bataevddf3db92018-04-13 17:31:06 +00002778 if (const auto *LIExpr = dyn_cast<DeclRefExpr>(S.getLastIteration())) {
Alexander Musmanc6388682014-12-15 07:07:06 +00002779 EmitVarDecl(*cast<VarDecl>(LIExpr->getDecl()));
2780 // Emit calculation of the iterations count.
2781 EmitIgnoredExpr(S.getCalcLastIteration());
2782 }
2783
Alexey Bataevddf3db92018-04-13 17:31:06 +00002784 CGOpenMPRuntime &RT = CGM.getOpenMPRuntime();
Alexander Musmanc6388682014-12-15 07:07:06 +00002785
Alexey Bataev38e89532015-04-16 04:54:05 +00002786 bool HasLastprivateClause;
Alexander Musmanc6388682014-12-15 07:07:06 +00002787 // Check pre-condition.
2788 {
Alexey Bataev5a3af132016-03-29 08:58:54 +00002789 OMPLoopScope PreInitScope(*this, S);
Alexander Musmanc6388682014-12-15 07:07:06 +00002790 // Skip the entire loop if we don't meet the precondition.
Alexey Bataev62dbb972015-04-22 11:59:37 +00002791 // If the condition constant folds and can be elided, avoid emitting the
2792 // whole loop.
2793 bool CondConstant;
2794 llvm::BasicBlock *ContBlock = nullptr;
2795 if (ConstantFoldsToSimpleInteger(S.getPreCond(), CondConstant)) {
2796 if (!CondConstant)
2797 return false;
2798 } else {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002799 llvm::BasicBlock *ThenBlock = createBasicBlock("omp.precond.then");
Alexey Bataev62dbb972015-04-22 11:59:37 +00002800 ContBlock = createBasicBlock("omp.precond.end");
2801 emitPreCond(*this, S, S.getPreCond(), ThenBlock, ContBlock,
Justin Bogner66242d62015-04-23 23:06:47 +00002802 getProfileCount(&S));
Alexey Bataev62dbb972015-04-22 11:59:37 +00002803 EmitBlock(ThenBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00002804 incrementProfileCounter(&S);
Alexey Bataev62dbb972015-04-22 11:59:37 +00002805 }
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002806
Alexey Bataevea33dee2018-02-15 23:39:43 +00002807 RunCleanupsScope DoacrossCleanupScope(*this);
Alexey Bataev8b427062016-05-25 12:36:08 +00002808 bool Ordered = false;
Alexey Bataevddf3db92018-04-13 17:31:06 +00002809 if (const auto *OrderedClause = S.getSingleClause<OMPOrderedClause>()) {
Alexey Bataev8b427062016-05-25 12:36:08 +00002810 if (OrderedClause->getNumForLoops())
Alexey Bataevf138fda2018-08-13 19:04:24 +00002811 RT.emitDoacrossInit(*this, S, OrderedClause->getLoopNumIterations());
Alexey Bataev8b427062016-05-25 12:36:08 +00002812 else
2813 Ordered = true;
2814 }
2815
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002816 llvm::DenseSet<const Expr *> EmittedFinals;
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00002817 emitAlignedClause(*this, S);
Alexey Bataev8c3edfe2017-08-16 15:58:46 +00002818 bool HasLinears = EmitOMPLinearClauseInit(S);
Alexey Bataevef549a82016-03-09 09:49:09 +00002819 // Emit helper vars inits.
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002820
2821 std::pair<LValue, LValue> Bounds = CodeGenLoopBounds(*this, S);
2822 LValue LB = Bounds.first;
2823 LValue UB = Bounds.second;
Alexey Bataevef549a82016-03-09 09:49:09 +00002824 LValue ST =
2825 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getStrideVariable()));
2826 LValue IL =
2827 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getIsLastIterVariable()));
2828
Alexander Musmanc6388682014-12-15 07:07:06 +00002829 // Emit 'then' code.
2830 {
Alexander Musmanc6388682014-12-15 07:07:06 +00002831 OMPPrivateScope LoopScope(*this);
Alexey Bataev8c3edfe2017-08-16 15:58:46 +00002832 if (EmitOMPFirstprivateClause(S, LoopScope) || HasLinears) {
Alexey Bataev69c62a92015-04-15 04:52:20 +00002833 // Emit implicit barrier to synchronize threads and avoid data races on
Alexey Bataevcd8b6a22016-02-15 08:07:17 +00002834 // initialization of firstprivate variables and post-update of
2835 // lastprivate variables.
Alexey Bataev25e5b442015-09-15 12:52:43 +00002836 CGM.getOpenMPRuntime().emitBarrierCall(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002837 *this, S.getBeginLoc(), OMPD_unknown, /*EmitChecks=*/false,
Alexey Bataev25e5b442015-09-15 12:52:43 +00002838 /*ForceSimpleCall=*/true);
Alexey Bataev69c62a92015-04-15 04:52:20 +00002839 }
Alexey Bataev50a64582015-04-22 12:24:45 +00002840 EmitOMPPrivateClause(S, LoopScope);
Alexey Bataeva58da1a2019-12-27 09:44:43 -05002841 CGOpenMPRuntime::LastprivateConditionalRAII LPCRegion(
2842 *this, S, EmitLValue(S.getIterationVariable()));
Alexey Bataev38e89532015-04-16 04:54:05 +00002843 HasLastprivateClause = EmitOMPLastprivateClauseInit(S, LoopScope);
Alexey Bataev7ebe5fd2015-04-22 13:43:03 +00002844 EmitOMPReductionClauseInit(S, LoopScope);
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002845 EmitOMPPrivateLoopCounters(S, LoopScope);
2846 EmitOMPLinearClause(S, LoopScope);
Alexander Musman7931b982015-03-16 07:14:41 +00002847 (void)LoopScope.Privatize();
Alexey Bataev60705422018-10-30 15:50:12 +00002848 if (isOpenMPTargetExecutionDirective(S.getDirectiveKind()))
2849 CGM.getOpenMPRuntime().adjustTargetSpecificDataForLambdas(*this, S);
Alexander Musmanc6388682014-12-15 07:07:06 +00002850
2851 // Detect the loop schedule kind and chunk.
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002852 const Expr *ChunkExpr = nullptr;
Alexey Bataev9ebd7422016-05-10 09:57:36 +00002853 OpenMPScheduleTy ScheduleKind;
Alexey Bataevddf3db92018-04-13 17:31:06 +00002854 if (const auto *C = S.getSingleClause<OMPScheduleClause>()) {
Alexey Bataev9ebd7422016-05-10 09:57:36 +00002855 ScheduleKind.Schedule = C->getScheduleKind();
2856 ScheduleKind.M1 = C->getFirstScheduleModifier();
2857 ScheduleKind.M2 = C->getSecondScheduleModifier();
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002858 ChunkExpr = C->getChunkSize();
Gheorghe-Teodor Bercea8233af92018-09-27 20:29:00 +00002859 } else {
2860 // Default behaviour for schedule clause.
2861 CGM.getOpenMPRuntime().getDefaultScheduleAndChunk(
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002862 *this, S, ScheduleKind.Schedule, ChunkExpr);
2863 }
2864 bool HasChunkSizeOne = false;
2865 llvm::Value *Chunk = nullptr;
2866 if (ChunkExpr) {
2867 Chunk = EmitScalarExpr(ChunkExpr);
2868 Chunk = EmitScalarConversion(Chunk, ChunkExpr->getType(),
2869 S.getIterationVariable()->getType(),
2870 S.getBeginLoc());
Fangrui Song407659a2018-11-30 23:41:18 +00002871 Expr::EvalResult Result;
2872 if (ChunkExpr->EvaluateAsInt(Result, getContext())) {
2873 llvm::APSInt EvaluatedChunk = Result.Val.getInt();
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002874 HasChunkSizeOne = (EvaluatedChunk.getLimitedValue() == 1);
Fangrui Song407659a2018-11-30 23:41:18 +00002875 }
Alexey Bataev3392d762016-02-16 11:18:12 +00002876 }
Alexander Musmanc6388682014-12-15 07:07:06 +00002877 const unsigned IVSize = getContext().getTypeSize(IVExpr->getType());
2878 const bool IVSigned = IVExpr->getType()->hasSignedIntegerRepresentation();
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002879 // OpenMP 4.5, 2.7.1 Loop Construct, Description.
2880 // If the static schedule kind is specified or if the ordered clause is
2881 // specified, and if no monotonic modifier is specified, the effect will
2882 // be as if the monotonic modifier was specified.
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002883 bool StaticChunkedOne = RT.isStaticChunked(ScheduleKind.Schedule,
2884 /* Chunked */ Chunk != nullptr) && HasChunkSizeOne &&
2885 isOpenMPLoopBoundSharingDirective(S.getDirectiveKind());
Alexey Bataev4971d0b2020-06-17 12:52:37 -04002886 bool IsMonotonic =
2887 Ordered ||
2888 ((ScheduleKind.Schedule == OMPC_SCHEDULE_static ||
2889 ScheduleKind.Schedule == OMPC_SCHEDULE_unknown) &&
2890 !(ScheduleKind.M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic ||
2891 ScheduleKind.M1 == OMPC_SCHEDULE_MODIFIER_nonmonotonic)) ||
2892 ScheduleKind.M1 == OMPC_SCHEDULE_MODIFIER_monotonic ||
2893 ScheduleKind.M2 == OMPC_SCHEDULE_MODIFIER_monotonic;
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002894 if ((RT.isStaticNonchunked(ScheduleKind.Schedule,
2895 /* Chunked */ Chunk != nullptr) ||
2896 StaticChunkedOne) &&
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00002897 !Ordered) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00002898 JumpDest LoopExit =
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002899 getJumpDestInCurrentScope(createBasicBlock("omp.loop.exit"));
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002900 emitCommonSimdLoop(
2901 *this, S,
Alexey Bataev4971d0b2020-06-17 12:52:37 -04002902 [&S, IsMonotonic](CodeGenFunction &CGF, PrePostActionTy &) {
Alexey Bataeva7815212020-02-03 12:08:16 -05002903 if (isOpenMPSimdDirective(S.getDirectiveKind())) {
Alexey Bataev4971d0b2020-06-17 12:52:37 -04002904 CGF.EmitOMPSimdInit(S, IsMonotonic);
Alexey Bataeva7815212020-02-03 12:08:16 -05002905 } else if (const auto *C = S.getSingleClause<OMPOrderClause>()) {
2906 if (C->getKind() == OMPC_ORDER_concurrent)
2907 CGF.LoopStack.setParallel(/*Enable=*/true);
2908 }
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00002909 },
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002910 [IVSize, IVSigned, Ordered, IL, LB, UB, ST, StaticChunkedOne, Chunk,
2911 &S, ScheduleKind, LoopExit,
2912 &LoopScope](CodeGenFunction &CGF, PrePostActionTy &) {
2913 // OpenMP [2.7.1, Loop Construct, Description, table 2-1]
2914 // When no chunk_size is specified, the iteration space is divided
2915 // into chunks that are approximately equal in size, and at most
2916 // one chunk is distributed to each thread. Note that the size of
2917 // the chunks is unspecified in this case.
2918 CGOpenMPRuntime::StaticRTInput StaticInit(
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002919 IVSize, IVSigned, Ordered, IL.getAddress(CGF),
2920 LB.getAddress(CGF), UB.getAddress(CGF), ST.getAddress(CGF),
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002921 StaticChunkedOne ? Chunk : nullptr);
2922 CGF.CGM.getOpenMPRuntime().emitForStaticInit(
2923 CGF, S.getBeginLoc(), S.getDirectiveKind(), ScheduleKind,
2924 StaticInit);
2925 // UB = min(UB, GlobalUB);
2926 if (!StaticChunkedOne)
2927 CGF.EmitIgnoredExpr(S.getEnsureUpperBound());
2928 // IV = LB;
2929 CGF.EmitIgnoredExpr(S.getInit());
2930 // For unchunked static schedule generate:
2931 //
2932 // while (idx <= UB) {
2933 // BODY;
2934 // ++idx;
2935 // }
2936 //
2937 // For static schedule with chunk one:
2938 //
2939 // while (IV <= PrevUB) {
2940 // BODY;
2941 // IV += ST;
2942 // }
2943 CGF.EmitOMPInnerLoop(
2944 S, LoopScope.requiresCleanups(),
2945 StaticChunkedOne ? S.getCombinedParForInDistCond()
2946 : S.getCond(),
2947 StaticChunkedOne ? S.getDistInc() : S.getInc(),
2948 [&S, LoopExit](CodeGenFunction &CGF) {
Alexey Bataevcb9191c2020-06-09 08:04:48 -04002949 emitOMPLoopBodyWithStopPoint(CGF, S, LoopExit);
Alexey Bataev103f3c9e2019-11-20 15:59:03 -05002950 },
2951 [](CodeGenFunction &) {});
2952 });
Alexey Bataev0f34da12015-07-02 04:17:07 +00002953 EmitBlock(LoopExit.getBlock());
Alexander Musmanc6388682014-12-15 07:07:06 +00002954 // Tell the runtime we are done.
Alexey Bataev957d8562016-11-17 15:12:05 +00002955 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
Stephen Kelly1c301dc2018-08-09 21:09:38 +00002956 CGF.CGM.getOpenMPRuntime().emitForStaticFinish(CGF, S.getEndLoc(),
Alexey Bataevf43f7142017-09-06 16:17:35 +00002957 S.getDirectiveKind());
Alexey Bataev957d8562016-11-17 15:12:05 +00002958 };
2959 OMPCancelStack.emitExit(*this, S.getDirectiveKind(), CodeGen);
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002960 } else {
2961 // Emit the outer loop, which requests its work chunk [LB..UB] from
2962 // runtime and runs the inner loop to process it.
Akira Hatanakaf139ae32019-12-03 15:17:01 -08002963 const OMPLoopArguments LoopArguments(
2964 LB.getAddress(*this), UB.getAddress(*this), ST.getAddress(*this),
2965 IL.getAddress(*this), Chunk, EUB);
Alexey Bataeva6f2a142015-12-31 06:52:34 +00002966 EmitOMPForOuterLoop(ScheduleKind, IsMonotonic, S, LoopScope, Ordered,
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00002967 LoopArguments, CGDispatchBounds);
Alexander Musmandf7a8e22015-01-22 08:49:35 +00002968 }
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002969 if (isOpenMPSimdDirective(S.getDirectiveKind())) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002970 EmitOMPSimdFinal(S, [IL, &S](CodeGenFunction &CGF) {
2971 return CGF.Builder.CreateIsNotNull(
2972 CGF.EmitLoadOfScalar(IL, S.getBeginLoc()));
2973 });
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002974 }
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00002975 EmitOMPReductionClauseFinal(
2976 S, /*ReductionKind=*/isOpenMPSimdDirective(S.getDirectiveKind())
2977 ? /*Parallel and Simd*/ OMPD_parallel_for_simd
2978 : /*Parallel only*/ OMPD_parallel);
Alexey Bataev61205072016-03-02 04:57:40 +00002979 // Emit post-update of the reduction variables if IsLastIter != 0.
2980 emitPostUpdateForReductionClause(
Alexey Bataevddf3db92018-04-13 17:31:06 +00002981 *this, S, [IL, &S](CodeGenFunction &CGF) {
Alexey Bataev61205072016-03-02 04:57:40 +00002982 return CGF.Builder.CreateIsNotNull(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002983 CGF.EmitLoadOfScalar(IL, S.getBeginLoc()));
Alexey Bataev61205072016-03-02 04:57:40 +00002984 });
Alexey Bataev38e89532015-04-16 04:54:05 +00002985 // Emit final copy of the lastprivate variables if IsLastIter != 0.
2986 if (HasLastprivateClause)
2987 EmitOMPLastprivateClauseFinal(
Alexey Bataev5dff95c2016-04-22 03:56:56 +00002988 S, isOpenMPSimdDirective(S.getDirectiveKind()),
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002989 Builder.CreateIsNotNull(EmitLoadOfScalar(IL, S.getBeginLoc())));
Alexander Musmanc6388682014-12-15 07:07:06 +00002990 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00002991 EmitOMPLinearClauseFinal(S, [IL, &S](CodeGenFunction &CGF) {
Alexey Bataevef549a82016-03-09 09:49:09 +00002992 return CGF.Builder.CreateIsNotNull(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00002993 CGF.EmitLoadOfScalar(IL, S.getBeginLoc()));
Alexey Bataevef549a82016-03-09 09:49:09 +00002994 });
Alexey Bataevea33dee2018-02-15 23:39:43 +00002995 DoacrossCleanupScope.ForceCleanup();
Alexander Musmanc6388682014-12-15 07:07:06 +00002996 // We're now done with the loop, so jump to the continuation block.
Alexey Bataev62dbb972015-04-22 11:59:37 +00002997 if (ContBlock) {
2998 EmitBranch(ContBlock);
Alexey Bataevddf3db92018-04-13 17:31:06 +00002999 EmitBlock(ContBlock, /*IsFinished=*/true);
Alexey Bataev62dbb972015-04-22 11:59:37 +00003000 }
Alexander Musmanc6388682014-12-15 07:07:06 +00003001 }
Alexey Bataev38e89532015-04-16 04:54:05 +00003002 return HasLastprivateClause;
Alexander Musmanc6388682014-12-15 07:07:06 +00003003}
3004
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00003005/// The following two functions generate expressions for the loop lower
3006/// and upper bounds in case of static and dynamic (dispatch) schedule
3007/// of the associated 'for' or 'distribute' loop.
3008static std::pair<LValue, LValue>
3009emitForLoopBounds(CodeGenFunction &CGF, const OMPExecutableDirective &S) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00003010 const auto &LS = cast<OMPLoopDirective>(S);
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00003011 LValue LB =
3012 EmitOMPHelperVar(CGF, cast<DeclRefExpr>(LS.getLowerBoundVariable()));
3013 LValue UB =
3014 EmitOMPHelperVar(CGF, cast<DeclRefExpr>(LS.getUpperBoundVariable()));
3015 return {LB, UB};
3016}
3017
3018/// When dealing with dispatch schedules (e.g. dynamic, guided) we do not
3019/// consider the lower and upper bound expressions generated by the
3020/// worksharing loop support, but we use 0 and the iteration space size as
3021/// constants
3022static std::pair<llvm::Value *, llvm::Value *>
3023emitDispatchForLoopBounds(CodeGenFunction &CGF, const OMPExecutableDirective &S,
3024 Address LB, Address UB) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00003025 const auto &LS = cast<OMPLoopDirective>(S);
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00003026 const Expr *IVExpr = LS.getIterationVariable();
3027 const unsigned IVSize = CGF.getContext().getTypeSize(IVExpr->getType());
3028 llvm::Value *LBVal = CGF.Builder.getIntN(IVSize, 0);
3029 llvm::Value *UBVal = CGF.EmitScalarExpr(LS.getLastIteration());
3030 return {LBVal, UBVal};
3031}
3032
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04003033/// Emits the code for the directive with inscan reductions.
3034/// The code is the following:
3035/// \code
3036/// size num_iters = <num_iters>;
3037/// <type> buffer[num_iters];
3038/// #pragma omp ...
3039/// for (i: 0..<num_iters>) {
3040/// <input phase>;
3041/// buffer[i] = red;
3042/// }
3043/// for (int k = 0; k != ceil(log2(num_iters)); ++k)
3044/// for (size cnt = last_iter; cnt >= pow(2, k); --k)
3045/// buffer[i] op= buffer[i-pow(2,k)];
3046/// #pragma omp ...
3047/// for (0..<num_iters>) {
3048/// red = InclusiveScan ? buffer[i] : buffer[i-1];
3049/// <scan phase>;
3050/// }
3051/// \endcode
3052static void emitScanBasedDirective(
3053 CodeGenFunction &CGF, const OMPLoopDirective &S,
3054 llvm::function_ref<llvm::Value *(CodeGenFunction &)> NumIteratorsGen,
3055 llvm::function_ref<void(CodeGenFunction &)> FirstGen,
3056 llvm::function_ref<void(CodeGenFunction &)> SecondGen) {
3057 llvm::Value *OMPScanNumIterations = CGF.Builder.CreateIntCast(
3058 NumIteratorsGen(CGF), CGF.SizeTy, /*isSigned=*/false);
3059 SmallVector<const Expr *, 4> Shareds;
3060 SmallVector<const Expr *, 4> Privates;
3061 SmallVector<const Expr *, 4> ReductionOps;
3062 SmallVector<const Expr *, 4> LHSs;
3063 SmallVector<const Expr *, 4> RHSs;
3064 SmallVector<const Expr *, 4> CopyOps;
3065 SmallVector<const Expr *, 4> CopyArrayTemps;
3066 SmallVector<const Expr *, 4> CopyArrayElems;
3067 for (const auto *C : S.getClausesOfKind<OMPReductionClause>()) {
3068 assert(C->getModifier() == OMPC_REDUCTION_inscan &&
3069 "Only inscan reductions are expected.");
3070 Shareds.append(C->varlist_begin(), C->varlist_end());
3071 Privates.append(C->privates().begin(), C->privates().end());
3072 ReductionOps.append(C->reduction_ops().begin(), C->reduction_ops().end());
3073 LHSs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
3074 RHSs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
3075 CopyOps.append(C->copy_ops().begin(), C->copy_ops().end());
3076 CopyArrayTemps.append(C->copy_array_temps().begin(),
3077 C->copy_array_temps().end());
3078 CopyArrayElems.append(C->copy_array_elems().begin(),
3079 C->copy_array_elems().end());
3080 }
3081 {
3082 // Emit buffers for each reduction variables.
3083 // ReductionCodeGen is required to emit correctly the code for array
3084 // reductions.
3085 ReductionCodeGen RedCG(Shareds, Shareds, Privates, ReductionOps);
3086 unsigned Count = 0;
3087 auto *ITA = CopyArrayTemps.begin();
3088 for (const Expr *IRef : Privates) {
3089 const auto *PrivateVD = cast<VarDecl>(cast<DeclRefExpr>(IRef)->getDecl());
3090 // Emit variably modified arrays, used for arrays/array sections
3091 // reductions.
3092 if (PrivateVD->getType()->isVariablyModifiedType()) {
3093 RedCG.emitSharedOrigLValue(CGF, Count);
3094 RedCG.emitAggregateType(CGF, Count);
3095 }
3096 CodeGenFunction::OpaqueValueMapping DimMapping(
3097 CGF,
3098 cast<OpaqueValueExpr>(
3099 cast<VariableArrayType>((*ITA)->getType()->getAsArrayTypeUnsafe())
3100 ->getSizeExpr()),
3101 RValue::get(OMPScanNumIterations));
3102 // Emit temp buffer.
3103 CGF.EmitVarDecl(*cast<VarDecl>(cast<DeclRefExpr>(*ITA)->getDecl()));
3104 ++ITA;
3105 ++Count;
3106 }
3107 }
3108 CodeGenFunction::ParentLoopDirectiveForScanRegion ScanRegion(CGF, S);
3109 {
3110 // Emit loop with input phase:
3111 // #pragma omp ...
3112 // for (i: 0..<num_iters>) {
3113 // <input phase>;
3114 // buffer[i] = red;
3115 // }
3116 CGF.OMPFirstScanLoop = true;
3117 CodeGenFunction::OMPLocalDeclMapRAII Scope(CGF);
3118 FirstGen(CGF);
3119 }
3120 // Emit prefix reduction:
3121 // for (int k = 0; k <= ceil(log2(n)); ++k)
3122 llvm::BasicBlock *InputBB = CGF.Builder.GetInsertBlock();
3123 llvm::BasicBlock *LoopBB = CGF.createBasicBlock("omp.outer.log.scan.body");
3124 llvm::BasicBlock *ExitBB = CGF.createBasicBlock("omp.outer.log.scan.exit");
3125 llvm::Function *F = CGF.CGM.getIntrinsic(llvm::Intrinsic::log2, CGF.DoubleTy);
3126 llvm::Value *Arg =
3127 CGF.Builder.CreateUIToFP(OMPScanNumIterations, CGF.DoubleTy);
3128 llvm::Value *LogVal = CGF.EmitNounwindRuntimeCall(F, Arg);
3129 F = CGF.CGM.getIntrinsic(llvm::Intrinsic::ceil, CGF.DoubleTy);
3130 LogVal = CGF.EmitNounwindRuntimeCall(F, LogVal);
3131 LogVal = CGF.Builder.CreateFPToUI(LogVal, CGF.IntTy);
3132 llvm::Value *NMin1 = CGF.Builder.CreateNUWSub(
3133 OMPScanNumIterations, llvm::ConstantInt::get(CGF.SizeTy, 1));
3134 auto DL = ApplyDebugLocation::CreateDefaultArtificial(CGF, S.getBeginLoc());
3135 CGF.EmitBlock(LoopBB);
3136 auto *Counter = CGF.Builder.CreatePHI(CGF.IntTy, 2);
3137 // size pow2k = 1;
3138 auto *Pow2K = CGF.Builder.CreatePHI(CGF.SizeTy, 2);
3139 Counter->addIncoming(llvm::ConstantInt::get(CGF.IntTy, 0), InputBB);
3140 Pow2K->addIncoming(llvm::ConstantInt::get(CGF.SizeTy, 1), InputBB);
3141 // for (size i = n - 1; i >= 2 ^ k; --i)
3142 // tmp[i] op= tmp[i-pow2k];
3143 llvm::BasicBlock *InnerLoopBB =
3144 CGF.createBasicBlock("omp.inner.log.scan.body");
3145 llvm::BasicBlock *InnerExitBB =
3146 CGF.createBasicBlock("omp.inner.log.scan.exit");
3147 llvm::Value *CmpI = CGF.Builder.CreateICmpUGE(NMin1, Pow2K);
3148 CGF.Builder.CreateCondBr(CmpI, InnerLoopBB, InnerExitBB);
3149 CGF.EmitBlock(InnerLoopBB);
3150 auto *IVal = CGF.Builder.CreatePHI(CGF.SizeTy, 2);
3151 IVal->addIncoming(NMin1, LoopBB);
3152 {
3153 CodeGenFunction::OMPPrivateScope PrivScope(CGF);
3154 auto *ILHS = LHSs.begin();
3155 auto *IRHS = RHSs.begin();
3156 for (const Expr *CopyArrayElem : CopyArrayElems) {
3157 const auto *LHSVD = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
3158 const auto *RHSVD = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
3159 Address LHSAddr = Address::invalid();
3160 {
3161 CodeGenFunction::OpaqueValueMapping IdxMapping(
3162 CGF,
3163 cast<OpaqueValueExpr>(
3164 cast<ArraySubscriptExpr>(CopyArrayElem)->getIdx()),
3165 RValue::get(IVal));
3166 LHSAddr = CGF.EmitLValue(CopyArrayElem).getAddress(CGF);
3167 }
3168 PrivScope.addPrivate(LHSVD, [LHSAddr]() { return LHSAddr; });
3169 Address RHSAddr = Address::invalid();
3170 {
3171 llvm::Value *OffsetIVal = CGF.Builder.CreateNUWSub(IVal, Pow2K);
3172 CodeGenFunction::OpaqueValueMapping IdxMapping(
3173 CGF,
3174 cast<OpaqueValueExpr>(
3175 cast<ArraySubscriptExpr>(CopyArrayElem)->getIdx()),
3176 RValue::get(OffsetIVal));
3177 RHSAddr = CGF.EmitLValue(CopyArrayElem).getAddress(CGF);
3178 }
3179 PrivScope.addPrivate(RHSVD, [RHSAddr]() { return RHSAddr; });
3180 ++ILHS;
3181 ++IRHS;
3182 }
3183 PrivScope.Privatize();
3184 CGF.CGM.getOpenMPRuntime().emitReduction(
3185 CGF, S.getEndLoc(), Privates, LHSs, RHSs, ReductionOps,
3186 {/*WithNowait=*/true, /*SimpleReduction=*/true, OMPD_unknown});
3187 }
3188 llvm::Value *NextIVal =
3189 CGF.Builder.CreateNUWSub(IVal, llvm::ConstantInt::get(CGF.SizeTy, 1));
3190 IVal->addIncoming(NextIVal, CGF.Builder.GetInsertBlock());
3191 CmpI = CGF.Builder.CreateICmpUGE(NextIVal, Pow2K);
3192 CGF.Builder.CreateCondBr(CmpI, InnerLoopBB, InnerExitBB);
3193 CGF.EmitBlock(InnerExitBB);
3194 llvm::Value *Next =
3195 CGF.Builder.CreateNUWAdd(Counter, llvm::ConstantInt::get(CGF.IntTy, 1));
3196 Counter->addIncoming(Next, CGF.Builder.GetInsertBlock());
3197 // pow2k <<= 1;
3198 llvm::Value *NextPow2K = CGF.Builder.CreateShl(Pow2K, 1, "", /*HasNUW=*/true);
3199 Pow2K->addIncoming(NextPow2K, CGF.Builder.GetInsertBlock());
3200 llvm::Value *Cmp = CGF.Builder.CreateICmpNE(Next, LogVal);
3201 CGF.Builder.CreateCondBr(Cmp, LoopBB, ExitBB);
3202 auto DL1 = ApplyDebugLocation::CreateDefaultArtificial(CGF, S.getEndLoc());
3203 CGF.EmitBlock(ExitBB);
3204
3205 CGF.OMPFirstScanLoop = false;
3206 SecondGen(CGF);
3207}
3208
Alexey Bataev1ec469c2020-06-09 11:55:25 -04003209static bool emitWorksharingDirective(CodeGenFunction &CGF,
3210 const OMPLoopDirective &S,
3211 bool HasCancel) {
3212 bool HasLastprivates;
3213 if (llvm::any_of(S.getClausesOfKind<OMPReductionClause>(),
3214 [](const OMPReductionClause *C) {
3215 return C->getModifier() == OMPC_REDUCTION_inscan;
3216 })) {
3217 const auto &&NumIteratorsGen = [&S](CodeGenFunction &CGF) {
3218 CodeGenFunction::OMPLocalDeclMapRAII Scope(CGF);
3219 OMPLoopScope LoopScope(CGF, S);
3220 return CGF.EmitScalarExpr(S.getNumIterations());
3221 };
3222 const auto &&FirstGen = [&S, HasCancel](CodeGenFunction &CGF) {
3223 CodeGenFunction::OMPCancelStackRAII CancelRegion(
3224 CGF, S.getDirectiveKind(), HasCancel);
3225 (void)CGF.EmitOMPWorksharingLoop(S, S.getEnsureUpperBound(),
3226 emitForLoopBounds,
3227 emitDispatchForLoopBounds);
3228 // Emit an implicit barrier at the end.
3229 CGF.CGM.getOpenMPRuntime().emitBarrierCall(CGF, S.getBeginLoc(),
3230 OMPD_for);
3231 };
3232 const auto &&SecondGen = [&S, HasCancel,
3233 &HasLastprivates](CodeGenFunction &CGF) {
3234 CodeGenFunction::OMPCancelStackRAII CancelRegion(
3235 CGF, S.getDirectiveKind(), HasCancel);
3236 HasLastprivates = CGF.EmitOMPWorksharingLoop(S, S.getEnsureUpperBound(),
3237 emitForLoopBounds,
3238 emitDispatchForLoopBounds);
3239 };
3240 emitScanBasedDirective(CGF, S, NumIteratorsGen, FirstGen, SecondGen);
3241 } else {
3242 CodeGenFunction::OMPCancelStackRAII CancelRegion(CGF, S.getDirectiveKind(),
3243 HasCancel);
3244 HasLastprivates = CGF.EmitOMPWorksharingLoop(S, S.getEnsureUpperBound(),
3245 emitForLoopBounds,
3246 emitDispatchForLoopBounds);
3247 }
3248 return HasLastprivates;
3249}
3250
Alexander Musmanc6388682014-12-15 07:07:06 +00003251void CodeGenFunction::EmitOMPForDirective(const OMPForDirective &S) {
Alexey Bataev38e89532015-04-16 04:54:05 +00003252 bool HasLastprivates = false;
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003253 auto &&CodeGen = [&S, &HasLastprivates](CodeGenFunction &CGF,
3254 PrePostActionTy &) {
Alexey Bataev1ec469c2020-06-09 11:55:25 -04003255 HasLastprivates = emitWorksharingDirective(CGF, S, S.hasCancel());
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003256 };
Alexey Bataev3392d762016-02-16 11:18:12 +00003257 {
Alexey Bataev46978742020-01-30 10:46:11 -05003258 auto LPCRegion =
3259 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev475a7442018-01-12 19:39:11 +00003260 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataev3392d762016-02-16 11:18:12 +00003261 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_for, CodeGen,
3262 S.hasCancel());
3263 }
Alexander Musmanc6388682014-12-15 07:07:06 +00003264
3265 // Emit an implicit barrier at the end.
Alexey Bataevddf3db92018-04-13 17:31:06 +00003266 if (!S.getSingleClause<OMPNowaitClause>() || HasLastprivates)
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003267 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getBeginLoc(), OMPD_for);
Alexey Bataev46978742020-01-30 10:46:11 -05003268 // Check for outer lastprivate conditional update.
3269 checkForLastprivateConditionalUpdate(*this, S);
Alexey Bataevf29276e2014-06-18 04:14:57 +00003270}
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00003271
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00003272void CodeGenFunction::EmitOMPForSimdDirective(const OMPForSimdDirective &S) {
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00003273 bool HasLastprivates = false;
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003274 auto &&CodeGen = [&S, &HasLastprivates](CodeGenFunction &CGF,
3275 PrePostActionTy &) {
Alexey Bataev1ec469c2020-06-09 11:55:25 -04003276 HasLastprivates = emitWorksharingDirective(CGF, S, /*HasCancel=*/false);
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003277 };
Alexey Bataev3392d762016-02-16 11:18:12 +00003278 {
Alexey Bataev46978742020-01-30 10:46:11 -05003279 auto LPCRegion =
3280 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev475a7442018-01-12 19:39:11 +00003281 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataev3392d762016-02-16 11:18:12 +00003282 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_simd, CodeGen);
3283 }
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00003284
3285 // Emit an implicit barrier at the end.
Alexey Bataevddf3db92018-04-13 17:31:06 +00003286 if (!S.getSingleClause<OMPNowaitClause>() || HasLastprivates)
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003287 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getBeginLoc(), OMPD_for);
Alexey Bataev46978742020-01-30 10:46:11 -05003288 // Check for outer lastprivate conditional update.
3289 checkForLastprivateConditionalUpdate(*this, S);
Alexander Musmanf82886e2014-09-18 05:12:34 +00003290}
3291
Alexey Bataev2df54a02015-03-12 08:53:29 +00003292static LValue createSectionLVal(CodeGenFunction &CGF, QualType Ty,
3293 const Twine &Name,
3294 llvm::Value *Init = nullptr) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00003295 LValue LVal = CGF.MakeAddrLValue(CGF.CreateMemTemp(Ty, Name), Ty);
Alexey Bataev2df54a02015-03-12 08:53:29 +00003296 if (Init)
Akira Hatanaka642f7992016-10-18 19:05:41 +00003297 CGF.EmitStoreThroughLValue(RValue::get(Init), LVal, /*isInit*/ true);
Alexey Bataev2df54a02015-03-12 08:53:29 +00003298 return LVal;
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00003299}
3300
Alexey Bataev3392d762016-02-16 11:18:12 +00003301void CodeGenFunction::EmitSections(const OMPExecutableDirective &S) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00003302 const Stmt *CapturedStmt = S.getInnermostCapturedStmt()->getCapturedStmt();
3303 const auto *CS = dyn_cast<CompoundStmt>(CapturedStmt);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003304 bool HasLastprivates = false;
Alexey Bataevddf3db92018-04-13 17:31:06 +00003305 auto &&CodeGen = [&S, CapturedStmt, CS,
3306 &HasLastprivates](CodeGenFunction &CGF, PrePostActionTy &) {
Melanie Blower2ba4e3a2020-04-10 13:34:46 -07003307 const ASTContext &C = CGF.getContext();
Alexey Bataevddf3db92018-04-13 17:31:06 +00003308 QualType KmpInt32Ty =
3309 C.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/1);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003310 // Emit helper vars inits.
3311 LValue LB = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.lb.",
3312 CGF.Builder.getInt32(0));
Alexey Bataevddf3db92018-04-13 17:31:06 +00003313 llvm::ConstantInt *GlobalUBVal = CS != nullptr
3314 ? CGF.Builder.getInt32(CS->size() - 1)
3315 : CGF.Builder.getInt32(0);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003316 LValue UB =
3317 createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.ub.", GlobalUBVal);
3318 LValue ST = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.st.",
3319 CGF.Builder.getInt32(1));
3320 LValue IL = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.il.",
3321 CGF.Builder.getInt32(0));
3322 // Loop counter.
3323 LValue IV = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.iv.");
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003324 OpaqueValueExpr IVRefExpr(S.getBeginLoc(), KmpInt32Ty, VK_LValue);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003325 CodeGenFunction::OpaqueValueMapping OpaqueIV(CGF, &IVRefExpr, IV);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003326 OpaqueValueExpr UBRefExpr(S.getBeginLoc(), KmpInt32Ty, VK_LValue);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003327 CodeGenFunction::OpaqueValueMapping OpaqueUB(CGF, &UBRefExpr, UB);
3328 // Generate condition for loop.
Melanie Blower2ba4e3a2020-04-10 13:34:46 -07003329 BinaryOperator *Cond = BinaryOperator::Create(
3330 C, &IVRefExpr, &UBRefExpr, BO_LE, C.BoolTy, VK_RValue, OK_Ordinary,
Melanie Blowerdefd43a2020-06-26 08:45:12 -07003331 S.getBeginLoc(), FPOptions(C.getLangOpts()));
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003332 // Increment for loop counter.
Melanie Blowerf5360d42020-05-01 10:32:06 -07003333 UnaryOperator *Inc = UnaryOperator::Create(
3334 C, &IVRefExpr, UO_PreInc, KmpInt32Ty, VK_RValue, OK_Ordinary,
Melanie Blowerdefd43a2020-06-26 08:45:12 -07003335 S.getBeginLoc(), true, FPOptions(C.getLangOpts()));
Alexey Bataevddf3db92018-04-13 17:31:06 +00003336 auto &&BodyGen = [CapturedStmt, CS, &S, &IV](CodeGenFunction &CGF) {
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003337 // Iterate through all sections and emit a switch construct:
3338 // switch (IV) {
3339 // case 0:
3340 // <SectionStmt[0]>;
3341 // break;
3342 // ...
3343 // case <NumSection> - 1:
3344 // <SectionStmt[<NumSection> - 1]>;
3345 // break;
3346 // }
3347 // .omp.sections.exit:
Alexey Bataevddf3db92018-04-13 17:31:06 +00003348 llvm::BasicBlock *ExitBB = CGF.createBasicBlock(".omp.sections.exit");
3349 llvm::SwitchInst *SwitchStmt =
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003350 CGF.Builder.CreateSwitch(CGF.EmitLoadOfScalar(IV, S.getBeginLoc()),
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00003351 ExitBB, CS == nullptr ? 1 : CS->size());
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003352 if (CS) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003353 unsigned CaseNumber = 0;
Alexey Bataevddf3db92018-04-13 17:31:06 +00003354 for (const Stmt *SubStmt : CS->children()) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003355 auto CaseBB = CGF.createBasicBlock(".omp.sections.case");
3356 CGF.EmitBlock(CaseBB);
3357 SwitchStmt->addCase(CGF.Builder.getInt32(CaseNumber), CaseBB);
Benjamin Kramer642f1732015-07-02 21:03:14 +00003358 CGF.EmitStmt(SubStmt);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003359 CGF.EmitBranch(ExitBB);
Benjamin Kramer642f1732015-07-02 21:03:14 +00003360 ++CaseNumber;
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003361 }
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003362 } else {
Alexey Bataevddf3db92018-04-13 17:31:06 +00003363 llvm::BasicBlock *CaseBB = CGF.createBasicBlock(".omp.sections.case");
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003364 CGF.EmitBlock(CaseBB);
3365 SwitchStmt->addCase(CGF.Builder.getInt32(0), CaseBB);
Alexey Bataevddf3db92018-04-13 17:31:06 +00003366 CGF.EmitStmt(CapturedStmt);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003367 CGF.EmitBranch(ExitBB);
Alexey Bataev2cb9b952015-04-24 03:37:03 +00003368 }
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003369 CGF.EmitBlock(ExitBB, /*IsFinished=*/true);
Alexey Bataev2df54a02015-03-12 08:53:29 +00003370 };
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003371
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003372 CodeGenFunction::OMPPrivateScope LoopScope(CGF);
3373 if (CGF.EmitOMPFirstprivateClause(S, LoopScope)) {
Alexey Bataev9efc03b2015-04-27 04:34:03 +00003374 // Emit implicit barrier to synchronize threads and avoid data races on
Alexey Bataevcd8b6a22016-02-15 08:07:17 +00003375 // initialization of firstprivate variables and post-update of lastprivate
3376 // variables.
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003377 CGF.CGM.getOpenMPRuntime().emitBarrierCall(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003378 CGF, S.getBeginLoc(), OMPD_unknown, /*EmitChecks=*/false,
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003379 /*ForceSimpleCall=*/true);
Alexey Bataev9efc03b2015-04-27 04:34:03 +00003380 }
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003381 CGF.EmitOMPPrivateClause(S, LoopScope);
Alexey Bataeva58da1a2019-12-27 09:44:43 -05003382 CGOpenMPRuntime::LastprivateConditionalRAII LPCRegion(CGF, S, IV);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003383 HasLastprivates = CGF.EmitOMPLastprivateClauseInit(S, LoopScope);
3384 CGF.EmitOMPReductionClauseInit(S, LoopScope);
3385 (void)LoopScope.Privatize();
Alexey Bataev60705422018-10-30 15:50:12 +00003386 if (isOpenMPTargetExecutionDirective(S.getDirectiveKind()))
3387 CGF.CGM.getOpenMPRuntime().adjustTargetSpecificDataForLambdas(CGF, S);
Alexey Bataev2cb9b952015-04-24 03:37:03 +00003388
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003389 // Emit static non-chunked loop.
Alexey Bataev9ebd7422016-05-10 09:57:36 +00003390 OpenMPScheduleTy ScheduleKind;
3391 ScheduleKind.Schedule = OMPC_SCHEDULE_static;
Alexey Bataev0f87dbe2017-08-14 17:56:13 +00003392 CGOpenMPRuntime::StaticRTInput StaticInit(
Akira Hatanakaf139ae32019-12-03 15:17:01 -08003393 /*IVSize=*/32, /*IVSigned=*/true, /*Ordered=*/false, IL.getAddress(CGF),
3394 LB.getAddress(CGF), UB.getAddress(CGF), ST.getAddress(CGF));
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003395 CGF.CGM.getOpenMPRuntime().emitForStaticInit(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003396 CGF, S.getBeginLoc(), S.getDirectiveKind(), ScheduleKind, StaticInit);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003397 // UB = min(UB, GlobalUB);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003398 llvm::Value *UBVal = CGF.EmitLoadOfScalar(UB, S.getBeginLoc());
Alexey Bataevddf3db92018-04-13 17:31:06 +00003399 llvm::Value *MinUBGlobalUB = CGF.Builder.CreateSelect(
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003400 CGF.Builder.CreateICmpSLT(UBVal, GlobalUBVal), UBVal, GlobalUBVal);
3401 CGF.EmitStoreOfScalar(MinUBGlobalUB, UB);
3402 // IV = LB;
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003403 CGF.EmitStoreOfScalar(CGF.EmitLoadOfScalar(LB, S.getBeginLoc()), IV);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003404 // while (idx <= UB) { BODY; ++idx; }
Melanie Blowerf5360d42020-05-01 10:32:06 -07003405 CGF.EmitOMPInnerLoop(S, /*RequiresCleanup=*/false, Cond, Inc, BodyGen,
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003406 [](CodeGenFunction &) {});
3407 // Tell the runtime we are done.
Alexey Bataev957d8562016-11-17 15:12:05 +00003408 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
Stephen Kelly1c301dc2018-08-09 21:09:38 +00003409 CGF.CGM.getOpenMPRuntime().emitForStaticFinish(CGF, S.getEndLoc(),
Alexey Bataevf43f7142017-09-06 16:17:35 +00003410 S.getDirectiveKind());
Alexey Bataev957d8562016-11-17 15:12:05 +00003411 };
3412 CGF.OMPCancelStack.emitExit(CGF, S.getDirectiveKind(), CodeGen);
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00003413 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_parallel);
Alexey Bataev61205072016-03-02 04:57:40 +00003414 // Emit post-update of the reduction variables if IsLastIter != 0.
Alexey Bataevddf3db92018-04-13 17:31:06 +00003415 emitPostUpdateForReductionClause(CGF, S, [IL, &S](CodeGenFunction &CGF) {
3416 return CGF.Builder.CreateIsNotNull(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003417 CGF.EmitLoadOfScalar(IL, S.getBeginLoc()));
Alexey Bataevddf3db92018-04-13 17:31:06 +00003418 });
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003419
3420 // Emit final copy of the lastprivate variables if IsLastIter != 0.
3421 if (HasLastprivates)
3422 CGF.EmitOMPLastprivateClauseFinal(
Alexey Bataev5dff95c2016-04-22 03:56:56 +00003423 S, /*NoFinals=*/false,
3424 CGF.Builder.CreateIsNotNull(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003425 CGF.EmitLoadOfScalar(IL, S.getBeginLoc())));
Alexey Bataev68adb7d2015-04-14 03:29:22 +00003426 };
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003427
3428 bool HasCancel = false;
3429 if (auto *OSD = dyn_cast<OMPSectionsDirective>(&S))
3430 HasCancel = OSD->hasCancel();
3431 else if (auto *OPSD = dyn_cast<OMPParallelSectionsDirective>(&S))
3432 HasCancel = OPSD->hasCancel();
Alexey Bataev957d8562016-11-17 15:12:05 +00003433 OMPCancelStackRAII CancelRegion(*this, S.getDirectiveKind(), HasCancel);
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003434 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_sections, CodeGen,
3435 HasCancel);
3436 // Emit barrier for lastprivates only if 'sections' directive has 'nowait'
3437 // clause. Otherwise the barrier will be generated by the codegen for the
3438 // directive.
3439 if (HasLastprivates && S.getSingleClause<OMPNowaitClause>()) {
Alexey Bataev2cb9b952015-04-24 03:37:03 +00003440 // Emit implicit barrier to synchronize threads and avoid data races on
3441 // initialization of firstprivate variables.
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003442 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getBeginLoc(),
Alexey Bataev3015bcc2016-01-22 08:56:50 +00003443 OMPD_unknown);
Alexey Bataev2cb9b952015-04-24 03:37:03 +00003444 }
Alexey Bataev68adb7d2015-04-14 03:29:22 +00003445}
Alexey Bataev2df54a02015-03-12 08:53:29 +00003446
Alexey Bataev68adb7d2015-04-14 03:29:22 +00003447void CodeGenFunction::EmitOMPSectionsDirective(const OMPSectionsDirective &S) {
Alexey Bataev3392d762016-02-16 11:18:12 +00003448 {
Alexey Bataev46978742020-01-30 10:46:11 -05003449 auto LPCRegion =
3450 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev475a7442018-01-12 19:39:11 +00003451 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataev3392d762016-02-16 11:18:12 +00003452 EmitSections(S);
3453 }
Alexey Bataev2df54a02015-03-12 08:53:29 +00003454 // Emit an implicit barrier at the end.
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00003455 if (!S.getSingleClause<OMPNowaitClause>()) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003456 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getBeginLoc(),
Alexey Bataev3392d762016-02-16 11:18:12 +00003457 OMPD_sections);
Alexey Bataevf2685682015-03-30 04:30:22 +00003458 }
Alexey Bataev46978742020-01-30 10:46:11 -05003459 // Check for outer lastprivate conditional update.
3460 checkForLastprivateConditionalUpdate(*this, S);
Alexey Bataev2df54a02015-03-12 08:53:29 +00003461}
3462
3463void CodeGenFunction::EmitOMPSectionDirective(const OMPSectionDirective &S) {
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003464 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
Alexey Bataev475a7442018-01-12 19:39:11 +00003465 CGF.EmitStmt(S.getInnermostCapturedStmt()->getCapturedStmt());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003466 };
Alexey Bataev475a7442018-01-12 19:39:11 +00003467 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataev25e5b442015-09-15 12:52:43 +00003468 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_section, CodeGen,
3469 S.hasCancel());
Alexey Bataev1e0498a2014-06-26 08:21:58 +00003470}
3471
Alexey Bataev6956e2e2015-02-05 06:35:41 +00003472void CodeGenFunction::EmitOMPSingleDirective(const OMPSingleDirective &S) {
Alexey Bataeva63048e2015-03-23 06:18:07 +00003473 llvm::SmallVector<const Expr *, 8> CopyprivateVars;
Alexey Bataev420d45b2015-04-14 05:11:24 +00003474 llvm::SmallVector<const Expr *, 8> DestExprs;
Alexey Bataeva63048e2015-03-23 06:18:07 +00003475 llvm::SmallVector<const Expr *, 8> SrcExprs;
Alexey Bataeva63048e2015-03-23 06:18:07 +00003476 llvm::SmallVector<const Expr *, 8> AssignmentOps;
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003477 // Check if there are any 'copyprivate' clauses associated with this
Alexey Bataevcd8b6a22016-02-15 08:07:17 +00003478 // 'single' construct.
Alexey Bataeva63048e2015-03-23 06:18:07 +00003479 // Build a list of copyprivate variables along with helper expressions
3480 // (<source>, <destination>, <destination>=<source> expressions)
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00003481 for (const auto *C : S.getClausesOfKind<OMPCopyprivateClause>()) {
Alexey Bataeva63048e2015-03-23 06:18:07 +00003482 CopyprivateVars.append(C->varlists().begin(), C->varlists().end());
Alexey Bataev420d45b2015-04-14 05:11:24 +00003483 DestExprs.append(C->destination_exprs().begin(),
3484 C->destination_exprs().end());
Alexey Bataeva63048e2015-03-23 06:18:07 +00003485 SrcExprs.append(C->source_exprs().begin(), C->source_exprs().end());
Alexey Bataeva63048e2015-03-23 06:18:07 +00003486 AssignmentOps.append(C->assignment_ops().begin(),
3487 C->assignment_ops().end());
3488 }
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003489 // Emit code for 'single' region along with 'copyprivate' clauses
3490 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
3491 Action.Enter(CGF);
3492 OMPPrivateScope SingleScope(CGF);
3493 (void)CGF.EmitOMPFirstprivateClause(S, SingleScope);
3494 CGF.EmitOMPPrivateClause(S, SingleScope);
3495 (void)SingleScope.Privatize();
Alexey Bataev475a7442018-01-12 19:39:11 +00003496 CGF.EmitStmt(S.getInnermostCapturedStmt()->getCapturedStmt());
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003497 };
Alexey Bataev3392d762016-02-16 11:18:12 +00003498 {
Alexey Bataev46978742020-01-30 10:46:11 -05003499 auto LPCRegion =
3500 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev475a7442018-01-12 19:39:11 +00003501 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003502 CGM.getOpenMPRuntime().emitSingleRegion(*this, CodeGen, S.getBeginLoc(),
Alexey Bataev3392d762016-02-16 11:18:12 +00003503 CopyprivateVars, DestExprs,
3504 SrcExprs, AssignmentOps);
3505 }
3506 // Emit an implicit barrier at the end (to avoid data race on firstprivate
3507 // init or if no 'nowait' clause was specified and no 'copyprivate' clause).
Alexey Bataev417089f2016-02-17 13:19:37 +00003508 if (!S.getSingleClause<OMPNowaitClause>() && CopyprivateVars.empty()) {
Alexey Bataev5521d782015-04-24 04:21:15 +00003509 CGM.getOpenMPRuntime().emitBarrierCall(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003510 *this, S.getBeginLoc(),
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00003511 S.getSingleClause<OMPNowaitClause>() ? OMPD_unknown : OMPD_single);
Alexey Bataevf2685682015-03-30 04:30:22 +00003512 }
Alexey Bataev46978742020-01-30 10:46:11 -05003513 // Check for outer lastprivate conditional update.
3514 checkForLastprivateConditionalUpdate(*this, S);
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00003515}
3516
cchen47d60942019-12-05 13:43:48 -05003517static void emitMaster(CodeGenFunction &CGF, const OMPExecutableDirective &S) {
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003518 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
3519 Action.Enter(CGF);
Alexey Bataev475a7442018-01-12 19:39:11 +00003520 CGF.EmitStmt(S.getInnermostCapturedStmt()->getCapturedStmt());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003521 };
cchen47d60942019-12-05 13:43:48 -05003522 CGF.CGM.getOpenMPRuntime().emitMasterRegion(CGF, CodeGen, S.getBeginLoc());
3523}
3524
3525void CodeGenFunction::EmitOMPMasterDirective(const OMPMasterDirective &S) {
Fady Ghanim7438059a2020-02-15 00:42:23 -06003526 if (llvm::OpenMPIRBuilder *OMPBuilder = CGM.getOpenMPIRBuilder()) {
3527 using InsertPointTy = llvm::OpenMPIRBuilder::InsertPointTy;
3528
3529 const CapturedStmt *CS = S.getInnermostCapturedStmt();
3530 const Stmt *MasterRegionBodyStmt = CS->getCapturedStmt();
3531
Fady Ghanim7438059a2020-02-15 00:42:23 -06003532 auto FiniCB = [this](InsertPointTy IP) {
Fady Ghanimba3f8632020-02-19 13:50:26 -06003533 OMPBuilderCBHelpers::FinalizeOMPRegion(*this, IP);
Fady Ghanim7438059a2020-02-15 00:42:23 -06003534 };
3535
Fady Ghanim7438059a2020-02-15 00:42:23 -06003536 auto BodyGenCB = [MasterRegionBodyStmt, this](InsertPointTy AllocaIP,
3537 InsertPointTy CodeGenIP,
3538 llvm::BasicBlock &FiniBB) {
Fady Ghanimba3f8632020-02-19 13:50:26 -06003539 OMPBuilderCBHelpers::InlinedRegionBodyRAII IRB(*this, AllocaIP, FiniBB);
3540 OMPBuilderCBHelpers::EmitOMPRegionBody(*this, MasterRegionBodyStmt,
3541 CodeGenIP, FiniBB);
Fady Ghanim7438059a2020-02-15 00:42:23 -06003542 };
Fady Ghanimba3f8632020-02-19 13:50:26 -06003543
Fady Ghanim7438059a2020-02-15 00:42:23 -06003544 CGCapturedStmtInfo CGSI(*CS, CR_OpenMP);
3545 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(*this, &CGSI);
3546 Builder.restoreIP(OMPBuilder->CreateMaster(Builder, BodyGenCB, FiniCB));
3547
3548 return;
3549 }
Alexey Bataev475a7442018-01-12 19:39:11 +00003550 OMPLexicalScope Scope(*this, S, OMPD_unknown);
cchen47d60942019-12-05 13:43:48 -05003551 emitMaster(*this, S);
Alexander Musman80c22892014-07-17 08:54:58 +00003552}
3553
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +00003554void CodeGenFunction::EmitOMPCriticalDirective(const OMPCriticalDirective &S) {
Fady Ghanim7438059a2020-02-15 00:42:23 -06003555 if (llvm::OpenMPIRBuilder *OMPBuilder = CGM.getOpenMPIRBuilder()) {
3556 using InsertPointTy = llvm::OpenMPIRBuilder::InsertPointTy;
3557
3558 const CapturedStmt *CS = S.getInnermostCapturedStmt();
3559 const Stmt *CriticalRegionBodyStmt = CS->getCapturedStmt();
3560 const Expr *Hint = nullptr;
3561 if (const auto *HintClause = S.getSingleClause<OMPHintClause>())
3562 Hint = HintClause->getHint();
3563
3564 // TODO: This is slightly different from what's currently being done in
3565 // clang. Fix the Int32Ty to IntPtrTy (pointer width size) when everything
3566 // about typing is final.
3567 llvm::Value *HintInst = nullptr;
3568 if (Hint)
3569 HintInst =
3570 Builder.CreateIntCast(EmitScalarExpr(Hint), CGM.Int32Ty, false);
3571
Fady Ghanim7438059a2020-02-15 00:42:23 -06003572 auto FiniCB = [this](InsertPointTy IP) {
Fady Ghanimba3f8632020-02-19 13:50:26 -06003573 OMPBuilderCBHelpers::FinalizeOMPRegion(*this, IP);
Fady Ghanim7438059a2020-02-15 00:42:23 -06003574 };
3575
Fady Ghanim7438059a2020-02-15 00:42:23 -06003576 auto BodyGenCB = [CriticalRegionBodyStmt, this](InsertPointTy AllocaIP,
3577 InsertPointTy CodeGenIP,
3578 llvm::BasicBlock &FiniBB) {
Fady Ghanimba3f8632020-02-19 13:50:26 -06003579 OMPBuilderCBHelpers::InlinedRegionBodyRAII IRB(*this, AllocaIP, FiniBB);
3580 OMPBuilderCBHelpers::EmitOMPRegionBody(*this, CriticalRegionBodyStmt,
3581 CodeGenIP, FiniBB);
Fady Ghanim7438059a2020-02-15 00:42:23 -06003582 };
3583
3584 CGCapturedStmtInfo CGSI(*CS, CR_OpenMP);
3585 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(*this, &CGSI);
3586 Builder.restoreIP(OMPBuilder->CreateCritical(
3587 Builder, BodyGenCB, FiniCB, S.getDirectiveName().getAsString(),
3588 HintInst));
3589
3590 return;
3591 }
3592
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003593 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
3594 Action.Enter(CGF);
Alexey Bataev475a7442018-01-12 19:39:11 +00003595 CGF.EmitStmt(S.getInnermostCapturedStmt()->getCapturedStmt());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003596 };
Alexey Bataevddf3db92018-04-13 17:31:06 +00003597 const Expr *Hint = nullptr;
3598 if (const auto *HintClause = S.getSingleClause<OMPHintClause>())
Alexey Bataevfc57d162015-12-15 10:55:09 +00003599 Hint = HintClause->getHint();
Alexey Bataev475a7442018-01-12 19:39:11 +00003600 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataevfc57d162015-12-15 10:55:09 +00003601 CGM.getOpenMPRuntime().emitCriticalRegion(*this,
3602 S.getDirectiveName().getAsString(),
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003603 CodeGen, S.getBeginLoc(), Hint);
Alexander Musmand9ed09f2014-07-21 09:42:05 +00003604}
3605
Alexey Bataev671605e2015-04-13 05:28:11 +00003606void CodeGenFunction::EmitOMPParallelForDirective(
3607 const OMPParallelForDirective &S) {
3608 // Emit directive as a combined directive that consists of two implicit
3609 // directives: 'parallel' with 'for' directive.
Alexey Bataevc99042b2018-03-15 18:10:54 +00003610 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
3611 Action.Enter(CGF);
Alexey Bataev1ec469c2020-06-09 11:55:25 -04003612 (void)emitWorksharingDirective(CGF, S, S.hasCancel());
Alexey Bataev671605e2015-04-13 05:28:11 +00003613 };
Alexey Bataev46978742020-01-30 10:46:11 -05003614 {
3615 auto LPCRegion =
3616 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
3617 emitCommonOMPParallelDirective(*this, S, OMPD_for, CodeGen,
3618 emitEmptyBoundParameters);
3619 }
3620 // Check for outer lastprivate conditional update.
3621 checkForLastprivateConditionalUpdate(*this, S);
Alexey Bataev4acb8592014-07-07 13:01:15 +00003622}
3623
Alexander Musmane4e893b2014-09-23 09:33:00 +00003624void CodeGenFunction::EmitOMPParallelForSimdDirective(
Alexey Bataev3b5b5c42015-06-18 10:10:12 +00003625 const OMPParallelForSimdDirective &S) {
3626 // Emit directive as a combined directive that consists of two implicit
3627 // directives: 'parallel' with 'for' directive.
Alexey Bataevc99042b2018-03-15 18:10:54 +00003628 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
3629 Action.Enter(CGF);
Alexey Bataevcb90e6a2020-06-18 13:34:23 -04003630 (void)emitWorksharingDirective(CGF, S, /*HasCancel=*/false);
Alexey Bataev3b5b5c42015-06-18 10:10:12 +00003631 };
Alexey Bataev46978742020-01-30 10:46:11 -05003632 {
3633 auto LPCRegion =
3634 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataevcb90e6a2020-06-18 13:34:23 -04003635 emitCommonOMPParallelDirective(*this, S, OMPD_for_simd, CodeGen,
Alexey Bataev46978742020-01-30 10:46:11 -05003636 emitEmptyBoundParameters);
3637 }
3638 // Check for outer lastprivate conditional update.
3639 checkForLastprivateConditionalUpdate(*this, S);
Alexander Musmane4e893b2014-09-23 09:33:00 +00003640}
3641
cchen47d60942019-12-05 13:43:48 -05003642void CodeGenFunction::EmitOMPParallelMasterDirective(
3643 const OMPParallelMasterDirective &S) {
3644 // Emit directive as a combined directive that consists of two implicit
3645 // directives: 'parallel' with 'master' directive.
3646 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
3647 Action.Enter(CGF);
3648 OMPPrivateScope PrivateScope(CGF);
3649 bool Copyins = CGF.EmitOMPCopyinClause(S);
3650 (void)CGF.EmitOMPFirstprivateClause(S, PrivateScope);
3651 if (Copyins) {
3652 // Emit implicit barrier to synchronize threads and avoid data races on
3653 // propagation master's thread values of threadprivate variables to local
3654 // instances of that variables of all other implicit threads.
3655 CGF.CGM.getOpenMPRuntime().emitBarrierCall(
3656 CGF, S.getBeginLoc(), OMPD_unknown, /*EmitChecks=*/false,
3657 /*ForceSimpleCall=*/true);
3658 }
3659 CGF.EmitOMPPrivateClause(S, PrivateScope);
3660 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
3661 (void)PrivateScope.Privatize();
3662 emitMaster(CGF, S);
3663 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_parallel);
3664 };
Alexey Bataev46978742020-01-30 10:46:11 -05003665 {
3666 auto LPCRegion =
3667 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
3668 emitCommonOMPParallelDirective(*this, S, OMPD_master, CodeGen,
3669 emitEmptyBoundParameters);
3670 emitPostUpdateForReductionClause(*this, S,
3671 [](CodeGenFunction &) { return nullptr; });
3672 }
3673 // Check for outer lastprivate conditional update.
3674 checkForLastprivateConditionalUpdate(*this, S);
cchen47d60942019-12-05 13:43:48 -05003675}
3676
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00003677void CodeGenFunction::EmitOMPParallelSectionsDirective(
Alexey Bataev68adb7d2015-04-14 03:29:22 +00003678 const OMPParallelSectionsDirective &S) {
3679 // Emit directive as a combined directive that consists of two implicit
3680 // directives: 'parallel' with 'sections' directive.
Alexey Bataevc99042b2018-03-15 18:10:54 +00003681 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
3682 Action.Enter(CGF);
Alexey Bataev14fa1c62016-03-29 05:34:15 +00003683 CGF.EmitSections(S);
3684 };
Alexey Bataev46978742020-01-30 10:46:11 -05003685 {
3686 auto LPCRegion =
3687 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
3688 emitCommonOMPParallelDirective(*this, S, OMPD_sections, CodeGen,
3689 emitEmptyBoundParameters);
3690 }
3691 // Check for outer lastprivate conditional update.
3692 checkForLastprivateConditionalUpdate(*this, S);
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00003693}
3694
Alexey Bataev475a7442018-01-12 19:39:11 +00003695void CodeGenFunction::EmitOMPTaskBasedDirective(
3696 const OMPExecutableDirective &S, const OpenMPDirectiveKind CapturedRegion,
3697 const RegionCodeGenTy &BodyGen, const TaskGenTy &TaskGen,
3698 OMPTaskDataTy &Data) {
Alexey Bataev62b63b12015-03-10 07:28:44 +00003699 // Emit outlined function for task construct.
Alexey Bataev475a7442018-01-12 19:39:11 +00003700 const CapturedStmt *CS = S.getCapturedStmt(CapturedRegion);
Alexey Bataevddf3db92018-04-13 17:31:06 +00003701 auto I = CS->getCapturedDecl()->param_begin();
3702 auto PartId = std::next(I);
3703 auto TaskT = std::next(I, 4);
Alexey Bataev24b5bae2016-04-28 09:23:51 +00003704 // Check if the task is final
3705 if (const auto *Clause = S.getSingleClause<OMPFinalClause>()) {
3706 // If the condition constant folds and can be elided, try to avoid emitting
3707 // the condition and the dead arm of the if/else.
Alexey Bataevddf3db92018-04-13 17:31:06 +00003708 const Expr *Cond = Clause->getCondition();
Alexey Bataev24b5bae2016-04-28 09:23:51 +00003709 bool CondConstant;
3710 if (ConstantFoldsToSimpleInteger(Cond, CondConstant))
3711 Data.Final.setInt(CondConstant);
3712 else
3713 Data.Final.setPointer(EvaluateExprAsBool(Cond));
3714 } else {
3715 // By default the task is not final.
3716 Data.Final.setInt(/*IntVal=*/false);
3717 }
Alexey Bataev1e1e2862016-05-10 12:21:02 +00003718 // Check if the task has 'priority' clause.
3719 if (const auto *Clause = S.getSingleClause<OMPPriorityClause>()) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00003720 const Expr *Prio = Clause->getPriority();
Alexey Bataev5140e742016-07-19 04:21:09 +00003721 Data.Priority.setInt(/*IntVal=*/true);
Alexey Bataevad537bb2016-05-30 09:06:50 +00003722 Data.Priority.setPointer(EmitScalarConversion(
3723 EmitScalarExpr(Prio), Prio->getType(),
3724 getContext().getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/1),
3725 Prio->getExprLoc()));
Alexey Bataev1e1e2862016-05-10 12:21:02 +00003726 }
Alexey Bataev62b63b12015-03-10 07:28:44 +00003727 // The first function argument for tasks is a thread id, the second one is a
3728 // part id (0 for tied tasks, >=0 for untied task).
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003729 llvm::DenseSet<const VarDecl *> EmittedAsPrivate;
3730 // Get list of private variables.
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00003731 for (const auto *C : S.getClausesOfKind<OMPPrivateClause>()) {
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003732 auto IRef = C->varlist_begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00003733 for (const Expr *IInit : C->private_copies()) {
3734 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003735 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
Alexey Bataev7292c292016-04-25 12:22:29 +00003736 Data.PrivateVars.push_back(*IRef);
3737 Data.PrivateCopies.push_back(IInit);
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003738 }
3739 ++IRef;
3740 }
3741 }
3742 EmittedAsPrivate.clear();
3743 // Get list of firstprivate variables.
Benjamin Kramerfc600dc2015-08-30 15:12:28 +00003744 for (const auto *C : S.getClausesOfKind<OMPFirstprivateClause>()) {
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003745 auto IRef = C->varlist_begin();
3746 auto IElemInitRef = C->inits().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00003747 for (const Expr *IInit : C->private_copies()) {
3748 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003749 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
Alexey Bataev7292c292016-04-25 12:22:29 +00003750 Data.FirstprivateVars.push_back(*IRef);
3751 Data.FirstprivateCopies.push_back(IInit);
3752 Data.FirstprivateInits.push_back(*IElemInitRef);
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003753 }
Richard Trieucc3949d2016-02-18 22:34:54 +00003754 ++IRef;
3755 ++IElemInitRef;
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003756 }
3757 }
Alexey Bataevf93095a2016-05-05 08:46:22 +00003758 // Get list of lastprivate variables (for taskloops).
3759 llvm::DenseMap<const VarDecl *, const DeclRefExpr *> LastprivateDstsOrigs;
3760 for (const auto *C : S.getClausesOfKind<OMPLastprivateClause>()) {
3761 auto IRef = C->varlist_begin();
3762 auto ID = C->destination_exprs().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00003763 for (const Expr *IInit : C->private_copies()) {
3764 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataevf93095a2016-05-05 08:46:22 +00003765 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
3766 Data.LastprivateVars.push_back(*IRef);
3767 Data.LastprivateCopies.push_back(IInit);
3768 }
3769 LastprivateDstsOrigs.insert(
3770 {cast<VarDecl>(cast<DeclRefExpr>(*ID)->getDecl()),
3771 cast<DeclRefExpr>(*IRef)});
3772 ++IRef;
3773 ++ID;
3774 }
3775 }
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003776 SmallVector<const Expr *, 4> LHSs;
3777 SmallVector<const Expr *, 4> RHSs;
3778 for (const auto *C : S.getClausesOfKind<OMPReductionClause>()) {
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04003779 Data.ReductionVars.append(C->varlist_begin(), C->varlist_end());
3780 Data.ReductionOrigs.append(C->varlist_begin(), C->varlist_end());
3781 Data.ReductionCopies.append(C->privates().begin(), C->privates().end());
3782 Data.ReductionOps.append(C->reduction_ops().begin(),
3783 C->reduction_ops().end());
3784 LHSs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
3785 RHSs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003786 }
3787 Data.Reductions = CGM.getOpenMPRuntime().emitTaskReductionInit(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003788 *this, S.getBeginLoc(), LHSs, RHSs, Data);
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003789 // Build list of dependences.
Alexey Bataevbe99c612020-04-06 11:21:43 -04003790 for (const auto *C : S.getClausesOfKind<OMPDependClause>()) {
3791 OMPTaskDataTy::DependData &DD =
3792 Data.Dependences.emplace_back(C->getDependencyKind(), C->getModifier());
3793 DD.DepExprs.append(C->varlist_begin(), C->varlist_end());
3794 }
Alexey Bataev475a7442018-01-12 19:39:11 +00003795 auto &&CodeGen = [&Data, &S, CS, &BodyGen, &LastprivateDstsOrigs,
3796 CapturedRegion](CodeGenFunction &CGF,
3797 PrePostActionTy &Action) {
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003798 // Set proper addresses for generated private copies.
Alexey Bataev7292c292016-04-25 12:22:29 +00003799 OMPPrivateScope Scope(CGF);
Alexey Bataevb3998a02020-03-09 08:55:57 -04003800 llvm::SmallVector<std::pair<const VarDecl *, Address>, 16> FirstprivatePtrs;
Alexey Bataevf93095a2016-05-05 08:46:22 +00003801 if (!Data.PrivateVars.empty() || !Data.FirstprivateVars.empty() ||
3802 !Data.LastprivateVars.empty()) {
James Y Knight9871db02019-02-05 16:42:33 +00003803 llvm::FunctionType *CopyFnTy = llvm::FunctionType::get(
3804 CGF.Builder.getVoidTy(), {CGF.Builder.getInt8PtrTy()}, true);
Alexey Bataev3c595a62017-08-14 15:01:03 +00003805 enum { PrivatesParam = 2, CopyFnParam = 3 };
Alexey Bataevddf3db92018-04-13 17:31:06 +00003806 llvm::Value *CopyFn = CGF.Builder.CreateLoad(
3807 CGF.GetAddrOfLocalVar(CS->getCapturedDecl()->getParam(CopyFnParam)));
3808 llvm::Value *PrivatesPtr = CGF.Builder.CreateLoad(CGF.GetAddrOfLocalVar(
3809 CS->getCapturedDecl()->getParam(PrivatesParam)));
Alexey Bataev48591dd2016-04-20 04:01:36 +00003810 // Map privates.
3811 llvm::SmallVector<std::pair<const VarDecl *, Address>, 16> PrivatePtrs;
3812 llvm::SmallVector<llvm::Value *, 16> CallArgs;
3813 CallArgs.push_back(PrivatesPtr);
Alexey Bataevddf3db92018-04-13 17:31:06 +00003814 for (const Expr *E : Data.PrivateVars) {
3815 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
Alexey Bataev48591dd2016-04-20 04:01:36 +00003816 Address PrivatePtr = CGF.CreateMemTemp(
3817 CGF.getContext().getPointerType(E->getType()), ".priv.ptr.addr");
Alexey Bataevddf3db92018-04-13 17:31:06 +00003818 PrivatePtrs.emplace_back(VD, PrivatePtr);
Alexey Bataev48591dd2016-04-20 04:01:36 +00003819 CallArgs.push_back(PrivatePtr.getPointer());
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003820 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00003821 for (const Expr *E : Data.FirstprivateVars) {
3822 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
Alexey Bataev48591dd2016-04-20 04:01:36 +00003823 Address PrivatePtr =
3824 CGF.CreateMemTemp(CGF.getContext().getPointerType(E->getType()),
3825 ".firstpriv.ptr.addr");
Alexey Bataevddf3db92018-04-13 17:31:06 +00003826 PrivatePtrs.emplace_back(VD, PrivatePtr);
Alexey Bataevb3998a02020-03-09 08:55:57 -04003827 FirstprivatePtrs.emplace_back(VD, PrivatePtr);
Alexey Bataev48591dd2016-04-20 04:01:36 +00003828 CallArgs.push_back(PrivatePtr.getPointer());
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003829 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00003830 for (const Expr *E : Data.LastprivateVars) {
3831 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
Alexey Bataevf93095a2016-05-05 08:46:22 +00003832 Address PrivatePtr =
3833 CGF.CreateMemTemp(CGF.getContext().getPointerType(E->getType()),
3834 ".lastpriv.ptr.addr");
Alexey Bataevddf3db92018-04-13 17:31:06 +00003835 PrivatePtrs.emplace_back(VD, PrivatePtr);
Alexey Bataevf93095a2016-05-05 08:46:22 +00003836 CallArgs.push_back(PrivatePtr.getPointer());
3837 }
James Y Knight9871db02019-02-05 16:42:33 +00003838 CGF.CGM.getOpenMPRuntime().emitOutlinedFunctionCall(
3839 CGF, S.getBeginLoc(), {CopyFnTy, CopyFn}, CallArgs);
Alexey Bataevddf3db92018-04-13 17:31:06 +00003840 for (const auto &Pair : LastprivateDstsOrigs) {
3841 const auto *OrigVD = cast<VarDecl>(Pair.second->getDecl());
Bruno Ricci5fc4db72018-12-21 14:10:18 +00003842 DeclRefExpr DRE(CGF.getContext(), const_cast<VarDecl *>(OrigVD),
3843 /*RefersToEnclosingVariableOrCapture=*/
3844 CGF.CapturedStmtInfo->lookup(OrigVD) != nullptr,
3845 Pair.second->getType(), VK_LValue,
3846 Pair.second->getExprLoc());
Alexey Bataevf93095a2016-05-05 08:46:22 +00003847 Scope.addPrivate(Pair.first, [&CGF, &DRE]() {
Akira Hatanakaf139ae32019-12-03 15:17:01 -08003848 return CGF.EmitLValue(&DRE).getAddress(CGF);
Alexey Bataevf93095a2016-05-05 08:46:22 +00003849 });
3850 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00003851 for (const auto &Pair : PrivatePtrs) {
Alexey Bataev48591dd2016-04-20 04:01:36 +00003852 Address Replacement(CGF.Builder.CreateLoad(Pair.second),
3853 CGF.getContext().getDeclAlign(Pair.first));
3854 Scope.addPrivate(Pair.first, [Replacement]() { return Replacement; });
3855 }
Alexey Bataev3ae88e22015-05-22 08:56:35 +00003856 }
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003857 if (Data.Reductions) {
Alexey Bataevb3998a02020-03-09 08:55:57 -04003858 OMPPrivateScope FirstprivateScope(CGF);
3859 for (const auto &Pair : FirstprivatePtrs) {
3860 Address Replacement(CGF.Builder.CreateLoad(Pair.second),
3861 CGF.getContext().getDeclAlign(Pair.first));
3862 FirstprivateScope.addPrivate(Pair.first,
3863 [Replacement]() { return Replacement; });
3864 }
3865 (void)FirstprivateScope.Privatize();
Alexey Bataev475a7442018-01-12 19:39:11 +00003866 OMPLexicalScope LexScope(CGF, S, CapturedRegion);
Alexey Bataeve9bfa1d2020-04-23 13:27:03 -04003867 ReductionCodeGen RedCG(Data.ReductionVars, Data.ReductionVars,
3868 Data.ReductionCopies, Data.ReductionOps);
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003869 llvm::Value *ReductionsPtr = CGF.Builder.CreateLoad(
3870 CGF.GetAddrOfLocalVar(CS->getCapturedDecl()->getParam(9)));
3871 for (unsigned Cnt = 0, E = Data.ReductionVars.size(); Cnt < E; ++Cnt) {
Alexey Bataeve9bfa1d2020-04-23 13:27:03 -04003872 RedCG.emitSharedOrigLValue(CGF, Cnt);
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003873 RedCG.emitAggregateType(CGF, Cnt);
Alexey Bataev2e0cbe502018-03-08 15:24:08 +00003874 // FIXME: This must removed once the runtime library is fixed.
3875 // Emit required threadprivate variables for
Raphael Isemannb23ccec2018-12-10 12:37:46 +00003876 // initializer/combiner/finalizer.
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003877 CGF.CGM.getOpenMPRuntime().emitTaskReductionFixups(CGF, S.getBeginLoc(),
Alexey Bataev2e0cbe502018-03-08 15:24:08 +00003878 RedCG, Cnt);
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003879 Address Replacement = CGF.CGM.getOpenMPRuntime().getTaskReductionItem(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003880 CGF, S.getBeginLoc(), ReductionsPtr, RedCG.getSharedLValue(Cnt));
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003881 Replacement =
3882 Address(CGF.EmitScalarConversion(
3883 Replacement.getPointer(), CGF.getContext().VoidPtrTy,
3884 CGF.getContext().getPointerType(
3885 Data.ReductionCopies[Cnt]->getType()),
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00003886 Data.ReductionCopies[Cnt]->getExprLoc()),
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003887 Replacement.getAlignment());
3888 Replacement = RedCG.adjustPrivateAddress(CGF, Cnt, Replacement);
3889 Scope.addPrivate(RedCG.getBaseDecl(Cnt),
3890 [Replacement]() { return Replacement; });
Alexey Bataevbe5a8b42017-07-17 13:30:36 +00003891 }
3892 }
Alexey Bataev88202be2017-07-27 13:20:36 +00003893 // Privatize all private variables except for in_reduction items.
Alexey Bataev48591dd2016-04-20 04:01:36 +00003894 (void)Scope.Privatize();
Alexey Bataev88202be2017-07-27 13:20:36 +00003895 SmallVector<const Expr *, 4> InRedVars;
3896 SmallVector<const Expr *, 4> InRedPrivs;
3897 SmallVector<const Expr *, 4> InRedOps;
3898 SmallVector<const Expr *, 4> TaskgroupDescriptors;
3899 for (const auto *C : S.getClausesOfKind<OMPInReductionClause>()) {
3900 auto IPriv = C->privates().begin();
3901 auto IRed = C->reduction_ops().begin();
3902 auto ITD = C->taskgroup_descriptors().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00003903 for (const Expr *Ref : C->varlists()) {
Alexey Bataev88202be2017-07-27 13:20:36 +00003904 InRedVars.emplace_back(Ref);
3905 InRedPrivs.emplace_back(*IPriv);
3906 InRedOps.emplace_back(*IRed);
3907 TaskgroupDescriptors.emplace_back(*ITD);
3908 std::advance(IPriv, 1);
3909 std::advance(IRed, 1);
3910 std::advance(ITD, 1);
3911 }
3912 }
3913 // Privatize in_reduction items here, because taskgroup descriptors must be
3914 // privatized earlier.
3915 OMPPrivateScope InRedScope(CGF);
3916 if (!InRedVars.empty()) {
Alexey Bataeve9bfa1d2020-04-23 13:27:03 -04003917 ReductionCodeGen RedCG(InRedVars, InRedVars, InRedPrivs, InRedOps);
Alexey Bataev88202be2017-07-27 13:20:36 +00003918 for (unsigned Cnt = 0, E = InRedVars.size(); Cnt < E; ++Cnt) {
Alexey Bataeve9bfa1d2020-04-23 13:27:03 -04003919 RedCG.emitSharedOrigLValue(CGF, Cnt);
Alexey Bataev88202be2017-07-27 13:20:36 +00003920 RedCG.emitAggregateType(CGF, Cnt);
3921 // The taskgroup descriptor variable is always implicit firstprivate and
Raphael Isemannb23ccec2018-12-10 12:37:46 +00003922 // privatized already during processing of the firstprivates.
Alexey Bataev2e0cbe502018-03-08 15:24:08 +00003923 // FIXME: This must removed once the runtime library is fixed.
3924 // Emit required threadprivate variables for
Raphael Isemannb23ccec2018-12-10 12:37:46 +00003925 // initializer/combiner/finalizer.
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003926 CGF.CGM.getOpenMPRuntime().emitTaskReductionFixups(CGF, S.getBeginLoc(),
Alexey Bataev2e0cbe502018-03-08 15:24:08 +00003927 RedCG, Cnt);
Alexey Bataev0fca7662020-03-27 11:15:17 -04003928 llvm::Value *ReductionsPtr;
3929 if (const Expr *TRExpr = TaskgroupDescriptors[Cnt]) {
3930 ReductionsPtr = CGF.EmitLoadOfScalar(CGF.EmitLValue(TRExpr),
3931 TRExpr->getExprLoc());
3932 } else {
3933 ReductionsPtr = llvm::ConstantPointerNull::get(CGF.VoidPtrTy);
3934 }
Alexey Bataev88202be2017-07-27 13:20:36 +00003935 Address Replacement = CGF.CGM.getOpenMPRuntime().getTaskReductionItem(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00003936 CGF, S.getBeginLoc(), ReductionsPtr, RedCG.getSharedLValue(Cnt));
Alexey Bataev88202be2017-07-27 13:20:36 +00003937 Replacement = Address(
3938 CGF.EmitScalarConversion(
3939 Replacement.getPointer(), CGF.getContext().VoidPtrTy,
3940 CGF.getContext().getPointerType(InRedPrivs[Cnt]->getType()),
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00003941 InRedPrivs[Cnt]->getExprLoc()),
Alexey Bataev88202be2017-07-27 13:20:36 +00003942 Replacement.getAlignment());
3943 Replacement = RedCG.adjustPrivateAddress(CGF, Cnt, Replacement);
3944 InRedScope.addPrivate(RedCG.getBaseDecl(Cnt),
3945 [Replacement]() { return Replacement; });
Alexey Bataev88202be2017-07-27 13:20:36 +00003946 }
3947 }
3948 (void)InRedScope.Privatize();
Alexey Bataev48591dd2016-04-20 04:01:36 +00003949
3950 Action.Enter(CGF);
Alexey Bataev7292c292016-04-25 12:22:29 +00003951 BodyGen(CGF);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003952 };
James Y Knight9871db02019-02-05 16:42:33 +00003953 llvm::Function *OutlinedFn = CGM.getOpenMPRuntime().emitTaskOutlinedFunction(
Alexey Bataev7292c292016-04-25 12:22:29 +00003954 S, *I, *PartId, *TaskT, S.getDirectiveKind(), CodeGen, Data.Tied,
3955 Data.NumberOfParts);
Alexey Bataev5bbcead2019-10-14 17:17:41 +00003956 OMPLexicalScope Scope(*this, S, llvm::None,
Alexey Bataev61205822019-12-04 09:50:21 -05003957 !isOpenMPParallelDirective(S.getDirectiveKind()) &&
3958 !isOpenMPSimdDirective(S.getDirectiveKind()));
Alexey Bataev7292c292016-04-25 12:22:29 +00003959 TaskGen(*this, OutlinedFn, Data);
3960}
3961
Alexey Bataevd2202ca2017-12-27 17:58:32 +00003962static ImplicitParamDecl *
3963createImplicitFirstprivateForType(ASTContext &C, OMPTaskDataTy &Data,
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00003964 QualType Ty, CapturedDecl *CD,
3965 SourceLocation Loc) {
3966 auto *OrigVD = ImplicitParamDecl::Create(C, CD, Loc, /*Id=*/nullptr, Ty,
3967 ImplicitParamDecl::Other);
3968 auto *OrigRef = DeclRefExpr::Create(
3969 C, NestedNameSpecifierLoc(), SourceLocation(), OrigVD,
3970 /*RefersToEnclosingVariableOrCapture=*/false, Loc, Ty, VK_LValue);
3971 auto *PrivateVD = ImplicitParamDecl::Create(C, CD, Loc, /*Id=*/nullptr, Ty,
3972 ImplicitParamDecl::Other);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00003973 auto *PrivateRef = DeclRefExpr::Create(
3974 C, NestedNameSpecifierLoc(), SourceLocation(), PrivateVD,
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00003975 /*RefersToEnclosingVariableOrCapture=*/false, Loc, Ty, VK_LValue);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00003976 QualType ElemType = C.getBaseElementType(Ty);
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00003977 auto *InitVD = ImplicitParamDecl::Create(C, CD, Loc, /*Id=*/nullptr, ElemType,
3978 ImplicitParamDecl::Other);
3979 auto *InitRef = DeclRefExpr::Create(
3980 C, NestedNameSpecifierLoc(), SourceLocation(), InitVD,
3981 /*RefersToEnclosingVariableOrCapture=*/false, Loc, ElemType, VK_LValue);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00003982 PrivateVD->setInitStyle(VarDecl::CInit);
3983 PrivateVD->setInit(ImplicitCastExpr::Create(C, ElemType, CK_LValueToRValue,
3984 InitRef, /*BasePath=*/nullptr,
3985 VK_RValue));
3986 Data.FirstprivateVars.emplace_back(OrigRef);
3987 Data.FirstprivateCopies.emplace_back(PrivateRef);
3988 Data.FirstprivateInits.emplace_back(InitRef);
3989 return OrigVD;
3990}
3991
3992void CodeGenFunction::EmitOMPTargetTaskBasedDirective(
3993 const OMPExecutableDirective &S, const RegionCodeGenTy &BodyGen,
3994 OMPTargetDataInfo &InputInfo) {
3995 // Emit outlined function for task construct.
Alexey Bataevddf3db92018-04-13 17:31:06 +00003996 const CapturedStmt *CS = S.getCapturedStmt(OMPD_task);
3997 Address CapturedStruct = GenerateCapturedStmtArgument(*CS);
3998 QualType SharedsTy = getContext().getRecordType(CS->getCapturedRecordDecl());
3999 auto I = CS->getCapturedDecl()->param_begin();
4000 auto PartId = std::next(I);
4001 auto TaskT = std::next(I, 4);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004002 OMPTaskDataTy Data;
4003 // The task is not final.
4004 Data.Final.setInt(/*IntVal=*/false);
4005 // Get list of firstprivate variables.
4006 for (const auto *C : S.getClausesOfKind<OMPFirstprivateClause>()) {
4007 auto IRef = C->varlist_begin();
4008 auto IElemInitRef = C->inits().begin();
4009 for (auto *IInit : C->private_copies()) {
4010 Data.FirstprivateVars.push_back(*IRef);
4011 Data.FirstprivateCopies.push_back(IInit);
4012 Data.FirstprivateInits.push_back(*IElemInitRef);
4013 ++IRef;
4014 ++IElemInitRef;
4015 }
4016 }
4017 OMPPrivateScope TargetScope(*this);
4018 VarDecl *BPVD = nullptr;
4019 VarDecl *PVD = nullptr;
4020 VarDecl *SVD = nullptr;
4021 if (InputInfo.NumberOfTargetItems > 0) {
4022 auto *CD = CapturedDecl::Create(
4023 getContext(), getContext().getTranslationUnitDecl(), /*NumParams=*/0);
4024 llvm::APInt ArrSize(/*numBits=*/32, InputInfo.NumberOfTargetItems);
4025 QualType BaseAndPointersType = getContext().getConstantArrayType(
Richard Smith772e2662019-10-04 01:25:59 +00004026 getContext().VoidPtrTy, ArrSize, nullptr, ArrayType::Normal,
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004027 /*IndexTypeQuals=*/0);
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00004028 BPVD = createImplicitFirstprivateForType(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004029 getContext(), Data, BaseAndPointersType, CD, S.getBeginLoc());
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00004030 PVD = createImplicitFirstprivateForType(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004031 getContext(), Data, BaseAndPointersType, CD, S.getBeginLoc());
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004032 QualType SizesType = getContext().getConstantArrayType(
Alexey Bataeva90fc662019-06-25 16:00:43 +00004033 getContext().getIntTypeForBitwidth(/*DestWidth=*/64, /*Signed=*/1),
Richard Smith772e2662019-10-04 01:25:59 +00004034 ArrSize, nullptr, ArrayType::Normal,
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004035 /*IndexTypeQuals=*/0);
Alexey Bataeva9b9cc02018-01-23 18:12:38 +00004036 SVD = createImplicitFirstprivateForType(getContext(), Data, SizesType, CD,
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004037 S.getBeginLoc());
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004038 TargetScope.addPrivate(
4039 BPVD, [&InputInfo]() { return InputInfo.BasePointersArray; });
4040 TargetScope.addPrivate(PVD,
4041 [&InputInfo]() { return InputInfo.PointersArray; });
4042 TargetScope.addPrivate(SVD,
4043 [&InputInfo]() { return InputInfo.SizesArray; });
4044 }
4045 (void)TargetScope.Privatize();
4046 // Build list of dependences.
Alexey Bataevbe99c612020-04-06 11:21:43 -04004047 for (const auto *C : S.getClausesOfKind<OMPDependClause>()) {
4048 OMPTaskDataTy::DependData &DD =
4049 Data.Dependences.emplace_back(C->getDependencyKind(), C->getModifier());
4050 DD.DepExprs.append(C->varlist_begin(), C->varlist_end());
4051 }
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004052 auto &&CodeGen = [&Data, &S, CS, &BodyGen, BPVD, PVD, SVD,
4053 &InputInfo](CodeGenFunction &CGF, PrePostActionTy &Action) {
4054 // Set proper addresses for generated private copies.
4055 OMPPrivateScope Scope(CGF);
4056 if (!Data.FirstprivateVars.empty()) {
James Y Knight9871db02019-02-05 16:42:33 +00004057 llvm::FunctionType *CopyFnTy = llvm::FunctionType::get(
4058 CGF.Builder.getVoidTy(), {CGF.Builder.getInt8PtrTy()}, true);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004059 enum { PrivatesParam = 2, CopyFnParam = 3 };
Alexey Bataevddf3db92018-04-13 17:31:06 +00004060 llvm::Value *CopyFn = CGF.Builder.CreateLoad(
4061 CGF.GetAddrOfLocalVar(CS->getCapturedDecl()->getParam(CopyFnParam)));
4062 llvm::Value *PrivatesPtr = CGF.Builder.CreateLoad(CGF.GetAddrOfLocalVar(
4063 CS->getCapturedDecl()->getParam(PrivatesParam)));
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004064 // Map privates.
4065 llvm::SmallVector<std::pair<const VarDecl *, Address>, 16> PrivatePtrs;
4066 llvm::SmallVector<llvm::Value *, 16> CallArgs;
4067 CallArgs.push_back(PrivatesPtr);
Alexey Bataevddf3db92018-04-13 17:31:06 +00004068 for (const Expr *E : Data.FirstprivateVars) {
4069 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004070 Address PrivatePtr =
4071 CGF.CreateMemTemp(CGF.getContext().getPointerType(E->getType()),
4072 ".firstpriv.ptr.addr");
Alexey Bataevddf3db92018-04-13 17:31:06 +00004073 PrivatePtrs.emplace_back(VD, PrivatePtr);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004074 CallArgs.push_back(PrivatePtr.getPointer());
4075 }
James Y Knight9871db02019-02-05 16:42:33 +00004076 CGF.CGM.getOpenMPRuntime().emitOutlinedFunctionCall(
4077 CGF, S.getBeginLoc(), {CopyFnTy, CopyFn}, CallArgs);
Alexey Bataevddf3db92018-04-13 17:31:06 +00004078 for (const auto &Pair : PrivatePtrs) {
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004079 Address Replacement(CGF.Builder.CreateLoad(Pair.second),
4080 CGF.getContext().getDeclAlign(Pair.first));
4081 Scope.addPrivate(Pair.first, [Replacement]() { return Replacement; });
4082 }
4083 }
4084 // Privatize all private variables except for in_reduction items.
4085 (void)Scope.Privatize();
Alexey Bataev8451efa2018-01-15 19:06:12 +00004086 if (InputInfo.NumberOfTargetItems > 0) {
4087 InputInfo.BasePointersArray = CGF.Builder.CreateConstArrayGEP(
James Y Knight751fe282019-02-09 22:22:28 +00004088 CGF.GetAddrOfLocalVar(BPVD), /*Index=*/0);
Alexey Bataev8451efa2018-01-15 19:06:12 +00004089 InputInfo.PointersArray = CGF.Builder.CreateConstArrayGEP(
James Y Knight751fe282019-02-09 22:22:28 +00004090 CGF.GetAddrOfLocalVar(PVD), /*Index=*/0);
Alexey Bataev8451efa2018-01-15 19:06:12 +00004091 InputInfo.SizesArray = CGF.Builder.CreateConstArrayGEP(
James Y Knight751fe282019-02-09 22:22:28 +00004092 CGF.GetAddrOfLocalVar(SVD), /*Index=*/0);
Alexey Bataev8451efa2018-01-15 19:06:12 +00004093 }
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004094
4095 Action.Enter(CGF);
Alexey Bataev475a7442018-01-12 19:39:11 +00004096 OMPLexicalScope LexScope(CGF, S, OMPD_task, /*EmitPreInitStmt=*/false);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004097 BodyGen(CGF);
4098 };
James Y Knight9871db02019-02-05 16:42:33 +00004099 llvm::Function *OutlinedFn = CGM.getOpenMPRuntime().emitTaskOutlinedFunction(
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004100 S, *I, *PartId, *TaskT, S.getDirectiveKind(), CodeGen, /*Tied=*/true,
4101 Data.NumberOfParts);
4102 llvm::APInt TrueOrFalse(32, S.hasClausesOfKind<OMPNowaitClause>() ? 1 : 0);
4103 IntegerLiteral IfCond(getContext(), TrueOrFalse,
4104 getContext().getIntTypeForBitwidth(32, /*Signed=*/0),
4105 SourceLocation());
4106
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004107 CGM.getOpenMPRuntime().emitTaskCall(*this, S.getBeginLoc(), S, OutlinedFn,
Alexey Bataevd2202ca2017-12-27 17:58:32 +00004108 SharedsTy, CapturedStruct, &IfCond, Data);
4109}
4110
Alexey Bataev7292c292016-04-25 12:22:29 +00004111void CodeGenFunction::EmitOMPTaskDirective(const OMPTaskDirective &S) {
4112 // Emit outlined function for task construct.
Alexey Bataev475a7442018-01-12 19:39:11 +00004113 const CapturedStmt *CS = S.getCapturedStmt(OMPD_task);
Alexey Bataevddf3db92018-04-13 17:31:06 +00004114 Address CapturedStruct = GenerateCapturedStmtArgument(*CS);
4115 QualType SharedsTy = getContext().getRecordType(CS->getCapturedRecordDecl());
Alexey Bataev1d677132015-04-22 13:57:31 +00004116 const Expr *IfCond = nullptr;
Alexey Bataev7371aa32015-09-03 08:45:56 +00004117 for (const auto *C : S.getClausesOfKind<OMPIfClause>()) {
4118 if (C->getNameModifier() == OMPD_unknown ||
4119 C->getNameModifier() == OMPD_task) {
4120 IfCond = C->getCondition();
4121 break;
4122 }
Alexey Bataev1d677132015-04-22 13:57:31 +00004123 }
Alexey Bataev7292c292016-04-25 12:22:29 +00004124
Alexey Bataev24b5bae2016-04-28 09:23:51 +00004125 OMPTaskDataTy Data;
4126 // Check if we should emit tied or untied task.
4127 Data.Tied = !S.getSingleClause<OMPUntiedClause>();
Alexey Bataev7292c292016-04-25 12:22:29 +00004128 auto &&BodyGen = [CS](CodeGenFunction &CGF, PrePostActionTy &) {
4129 CGF.EmitStmt(CS->getCapturedStmt());
4130 };
Alexey Bataev24b5bae2016-04-28 09:23:51 +00004131 auto &&TaskGen = [&S, SharedsTy, CapturedStruct,
James Y Knight9871db02019-02-05 16:42:33 +00004132 IfCond](CodeGenFunction &CGF, llvm::Function *OutlinedFn,
Alexey Bataev24b5bae2016-04-28 09:23:51 +00004133 const OMPTaskDataTy &Data) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004134 CGF.CGM.getOpenMPRuntime().emitTaskCall(CGF, S.getBeginLoc(), S, OutlinedFn,
Alexey Bataev24b5bae2016-04-28 09:23:51 +00004135 SharedsTy, CapturedStruct, IfCond,
4136 Data);
Alexey Bataev7292c292016-04-25 12:22:29 +00004137 };
Alexey Bataev46978742020-01-30 10:46:11 -05004138 auto LPCRegion =
4139 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev475a7442018-01-12 19:39:11 +00004140 EmitOMPTaskBasedDirective(S, OMPD_task, BodyGen, TaskGen, Data);
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00004141}
4142
Alexey Bataev9f797f32015-02-05 05:57:51 +00004143void CodeGenFunction::EmitOMPTaskyieldDirective(
4144 const OMPTaskyieldDirective &S) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004145 CGM.getOpenMPRuntime().emitTaskyieldCall(*this, S.getBeginLoc());
Alexey Bataev68446b72014-07-18 07:47:19 +00004146}
4147
Alexey Bataev8f7c1b02014-12-05 04:09:23 +00004148void CodeGenFunction::EmitOMPBarrierDirective(const OMPBarrierDirective &S) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004149 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getBeginLoc(), OMPD_barrier);
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00004150}
4151
Alexey Bataev8b8e2022015-04-27 05:22:09 +00004152void CodeGenFunction::EmitOMPTaskwaitDirective(const OMPTaskwaitDirective &S) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004153 CGM.getOpenMPRuntime().emitTaskwaitCall(*this, S.getBeginLoc());
Alexey Bataev2df347a2014-07-18 10:17:07 +00004154}
4155
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00004156void CodeGenFunction::EmitOMPTaskgroupDirective(
4157 const OMPTaskgroupDirective &S) {
Alexey Bataev14fa1c62016-03-29 05:34:15 +00004158 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
4159 Action.Enter(CGF);
Alexey Bataev3b1b8952017-07-25 15:53:26 +00004160 if (const Expr *E = S.getReductionRef()) {
4161 SmallVector<const Expr *, 4> LHSs;
4162 SmallVector<const Expr *, 4> RHSs;
4163 OMPTaskDataTy Data;
4164 for (const auto *C : S.getClausesOfKind<OMPTaskReductionClause>()) {
Alexey Bataev8c2f4e02020-04-24 09:56:29 -04004165 Data.ReductionVars.append(C->varlist_begin(), C->varlist_end());
4166 Data.ReductionOrigs.append(C->varlist_begin(), C->varlist_end());
4167 Data.ReductionCopies.append(C->privates().begin(), C->privates().end());
4168 Data.ReductionOps.append(C->reduction_ops().begin(),
4169 C->reduction_ops().end());
4170 LHSs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
4171 RHSs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
Alexey Bataev3b1b8952017-07-25 15:53:26 +00004172 }
4173 llvm::Value *ReductionDesc =
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004174 CGF.CGM.getOpenMPRuntime().emitTaskReductionInit(CGF, S.getBeginLoc(),
Alexey Bataev3b1b8952017-07-25 15:53:26 +00004175 LHSs, RHSs, Data);
4176 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
4177 CGF.EmitVarDecl(*VD);
4178 CGF.EmitStoreOfScalar(ReductionDesc, CGF.GetAddrOfLocalVar(VD),
4179 /*Volatile=*/false, E->getType());
4180 }
Alexey Bataev475a7442018-01-12 19:39:11 +00004181 CGF.EmitStmt(S.getInnermostCapturedStmt()->getCapturedStmt());
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00004182 };
Alexey Bataev475a7442018-01-12 19:39:11 +00004183 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004184 CGM.getOpenMPRuntime().emitTaskgroupRegion(*this, CodeGen, S.getBeginLoc());
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00004185}
4186
Alexey Bataevcc37cc12014-11-20 04:34:54 +00004187void CodeGenFunction::EmitOMPFlushDirective(const OMPFlushDirective &S) {
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004188 llvm::AtomicOrdering AO = S.getSingleClause<OMPFlushClause>()
4189 ? llvm::AtomicOrdering::NotAtomic
4190 : llvm::AtomicOrdering::AcquireRelease;
Alexey Bataevddf3db92018-04-13 17:31:06 +00004191 CGM.getOpenMPRuntime().emitFlush(
4192 *this,
4193 [&S]() -> ArrayRef<const Expr *> {
4194 if (const auto *FlushClause = S.getSingleClause<OMPFlushClause>())
4195 return llvm::makeArrayRef(FlushClause->varlist_begin(),
4196 FlushClause->varlist_end());
4197 return llvm::None;
4198 }(),
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004199 S.getBeginLoc(), AO);
Alexey Bataev6125da92014-07-21 11:26:11 +00004200}
4201
Alexey Bataeve46f0fe2020-03-04 14:37:51 -05004202void CodeGenFunction::EmitOMPDepobjDirective(const OMPDepobjDirective &S) {
4203 const auto *DO = S.getSingleClause<OMPDepobjClause>();
4204 LValue DOLVal = EmitLValue(DO->getDepobj());
4205 if (const auto *DC = S.getSingleClause<OMPDependClause>()) {
Alexey Bataevbe99c612020-04-06 11:21:43 -04004206 OMPTaskDataTy::DependData Dependencies(DC->getDependencyKind(),
4207 DC->getModifier());
4208 Dependencies.DepExprs.append(DC->varlist_begin(), DC->varlist_end());
4209 Address DepAddr = CGM.getOpenMPRuntime().emitDepobjDependClause(
4210 *this, Dependencies, DC->getBeginLoc());
Alexey Bataeve46f0fe2020-03-04 14:37:51 -05004211 EmitStoreOfScalar(DepAddr.getPointer(), DOLVal);
Alexey Bataevb27ff4d2020-03-04 16:15:28 -05004212 return;
4213 }
4214 if (const auto *DC = S.getSingleClause<OMPDestroyClause>()) {
4215 CGM.getOpenMPRuntime().emitDestroyClause(*this, DOLVal, DC->getBeginLoc());
4216 return;
Alexey Bataeve46f0fe2020-03-04 14:37:51 -05004217 }
Alexey Bataev8d7b1182020-03-05 13:45:28 -05004218 if (const auto *UC = S.getSingleClause<OMPUpdateClause>()) {
4219 CGM.getOpenMPRuntime().emitUpdateClause(
4220 *this, DOLVal, UC->getDependencyKind(), UC->getBeginLoc());
4221 return;
4222 }
Alexey Bataeve46f0fe2020-03-04 14:37:51 -05004223}
Alexey Bataevc112e942020-02-28 09:52:15 -05004224
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04004225void CodeGenFunction::EmitOMPScanDirective(const OMPScanDirective &S) {
Alexey Bataev43101d12020-06-11 11:28:59 -04004226 if (!OMPParentLoopDirectiveForScan)
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04004227 return;
4228 const OMPExecutableDirective &ParentDir = *OMPParentLoopDirectiveForScan;
Alexey Bataev43101d12020-06-11 11:28:59 -04004229 bool IsInclusive = S.hasClausesOfKind<OMPInclusiveClause>();
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04004230 SmallVector<const Expr *, 4> Shareds;
4231 SmallVector<const Expr *, 4> Privates;
4232 SmallVector<const Expr *, 4> LHSs;
4233 SmallVector<const Expr *, 4> RHSs;
Alexey Bataev43101d12020-06-11 11:28:59 -04004234 SmallVector<const Expr *, 4> ReductionOps;
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04004235 SmallVector<const Expr *, 4> CopyOps;
4236 SmallVector<const Expr *, 4> CopyArrayTemps;
4237 SmallVector<const Expr *, 4> CopyArrayElems;
4238 for (const auto *C : ParentDir.getClausesOfKind<OMPReductionClause>()) {
4239 if (C->getModifier() != OMPC_REDUCTION_inscan)
4240 continue;
4241 Shareds.append(C->varlist_begin(), C->varlist_end());
4242 Privates.append(C->privates().begin(), C->privates().end());
4243 LHSs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
4244 RHSs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
Alexey Bataev43101d12020-06-11 11:28:59 -04004245 ReductionOps.append(C->reduction_ops().begin(), C->reduction_ops().end());
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04004246 CopyOps.append(C->copy_ops().begin(), C->copy_ops().end());
4247 CopyArrayTemps.append(C->copy_array_temps().begin(),
4248 C->copy_array_temps().end());
4249 CopyArrayElems.append(C->copy_array_elems().begin(),
4250 C->copy_array_elems().end());
4251 }
Alexey Bataev43101d12020-06-11 11:28:59 -04004252 if (ParentDir.getDirectiveKind() == OMPD_simd ||
4253 (getLangOpts().OpenMPSimd &&
4254 isOpenMPSimdDirective(ParentDir.getDirectiveKind()))) {
4255 // For simd directive and simd-based directives in simd only mode, use the
4256 // following codegen:
4257 // int x = 0;
4258 // #pragma omp simd reduction(inscan, +: x)
4259 // for (..) {
4260 // <first part>
4261 // #pragma omp scan inclusive(x)
4262 // <second part>
4263 // }
4264 // is transformed to:
4265 // int x = 0;
4266 // for (..) {
4267 // int x_priv = 0;
4268 // <first part>
4269 // x = x_priv + x;
4270 // x_priv = x;
4271 // <second part>
4272 // }
4273 // and
4274 // int x = 0;
4275 // #pragma omp simd reduction(inscan, +: x)
4276 // for (..) {
4277 // <first part>
4278 // #pragma omp scan exclusive(x)
4279 // <second part>
4280 // }
4281 // to
4282 // int x = 0;
4283 // for (..) {
4284 // int x_priv = 0;
4285 // <second part>
4286 // int temp = x;
4287 // x = x_priv + x;
4288 // x_priv = temp;
4289 // <first part>
4290 // }
4291 llvm::BasicBlock *OMPScanReduce = createBasicBlock("omp.inscan.reduce");
4292 EmitBranch(IsInclusive
4293 ? OMPScanReduce
4294 : BreakContinueStack.back().ContinueBlock.getBlock());
4295 EmitBlock(OMPScanDispatch);
4296 {
4297 // New scope for correct construction/destruction of temp variables for
4298 // exclusive scan.
4299 LexicalScope Scope(*this, S.getSourceRange());
4300 EmitBranch(IsInclusive ? OMPBeforeScanBlock : OMPAfterScanBlock);
4301 EmitBlock(OMPScanReduce);
4302 if (!IsInclusive) {
4303 // Create temp var and copy LHS value to this temp value.
4304 // TMP = LHS;
4305 for (unsigned I = 0, E = CopyArrayElems.size(); I < E; ++I) {
4306 const Expr *PrivateExpr = Privates[I];
4307 const Expr *TempExpr = CopyArrayTemps[I];
4308 EmitAutoVarDecl(
4309 *cast<VarDecl>(cast<DeclRefExpr>(TempExpr)->getDecl()));
4310 LValue DestLVal = EmitLValue(TempExpr);
4311 LValue SrcLVal = EmitLValue(LHSs[I]);
4312 EmitOMPCopy(PrivateExpr->getType(), DestLVal.getAddress(*this),
4313 SrcLVal.getAddress(*this),
4314 cast<VarDecl>(cast<DeclRefExpr>(LHSs[I])->getDecl()),
4315 cast<VarDecl>(cast<DeclRefExpr>(RHSs[I])->getDecl()),
4316 CopyOps[I]);
4317 }
4318 }
4319 CGM.getOpenMPRuntime().emitReduction(
4320 *this, ParentDir.getEndLoc(), Privates, LHSs, RHSs, ReductionOps,
4321 {/*WithNowait=*/true, /*SimpleReduction=*/true, OMPD_simd});
4322 for (unsigned I = 0, E = CopyArrayElems.size(); I < E; ++I) {
4323 const Expr *PrivateExpr = Privates[I];
4324 LValue DestLVal;
4325 LValue SrcLVal;
4326 if (IsInclusive) {
4327 DestLVal = EmitLValue(RHSs[I]);
4328 SrcLVal = EmitLValue(LHSs[I]);
4329 } else {
4330 const Expr *TempExpr = CopyArrayTemps[I];
4331 DestLVal = EmitLValue(RHSs[I]);
4332 SrcLVal = EmitLValue(TempExpr);
4333 }
4334 EmitOMPCopy(PrivateExpr->getType(), DestLVal.getAddress(*this),
4335 SrcLVal.getAddress(*this),
4336 cast<VarDecl>(cast<DeclRefExpr>(LHSs[I])->getDecl()),
4337 cast<VarDecl>(cast<DeclRefExpr>(RHSs[I])->getDecl()),
4338 CopyOps[I]);
4339 }
4340 }
4341 EmitBranch(IsInclusive ? OMPAfterScanBlock : OMPBeforeScanBlock);
4342 OMPScanExitBlock = IsInclusive
4343 ? BreakContinueStack.back().ContinueBlock.getBlock()
4344 : OMPScanReduce;
4345 EmitBlock(OMPAfterScanBlock);
4346 return;
4347 }
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04004348 if (!IsInclusive) {
4349 EmitBranch(BreakContinueStack.back().ContinueBlock.getBlock());
4350 EmitBlock(OMPScanExitBlock);
4351 }
4352 if (OMPFirstScanLoop) {
4353 // Emit buffer[i] = red; at the end of the input phase.
4354 const auto *IVExpr = cast<OMPLoopDirective>(ParentDir)
4355 .getIterationVariable()
4356 ->IgnoreParenImpCasts();
4357 LValue IdxLVal = EmitLValue(IVExpr);
4358 llvm::Value *IdxVal = EmitLoadOfScalar(IdxLVal, IVExpr->getExprLoc());
4359 IdxVal = Builder.CreateIntCast(IdxVal, SizeTy, /*isSigned=*/false);
4360 for (unsigned I = 0, E = CopyArrayElems.size(); I < E; ++I) {
4361 const Expr *PrivateExpr = Privates[I];
4362 const Expr *OrigExpr = Shareds[I];
4363 const Expr *CopyArrayElem = CopyArrayElems[I];
4364 OpaqueValueMapping IdxMapping(
4365 *this,
4366 cast<OpaqueValueExpr>(
4367 cast<ArraySubscriptExpr>(CopyArrayElem)->getIdx()),
4368 RValue::get(IdxVal));
4369 LValue DestLVal = EmitLValue(CopyArrayElem);
4370 LValue SrcLVal = EmitLValue(OrigExpr);
4371 EmitOMPCopy(PrivateExpr->getType(), DestLVal.getAddress(*this),
4372 SrcLVal.getAddress(*this),
4373 cast<VarDecl>(cast<DeclRefExpr>(LHSs[I])->getDecl()),
4374 cast<VarDecl>(cast<DeclRefExpr>(RHSs[I])->getDecl()),
4375 CopyOps[I]);
4376 }
4377 }
4378 EmitBranch(BreakContinueStack.back().ContinueBlock.getBlock());
4379 if (IsInclusive) {
4380 EmitBlock(OMPScanExitBlock);
4381 EmitBranch(BreakContinueStack.back().ContinueBlock.getBlock());
4382 }
4383 EmitBlock(OMPScanDispatch);
4384 if (!OMPFirstScanLoop) {
4385 // Emit red = buffer[i]; at the entrance to the scan phase.
4386 const auto *IVExpr = cast<OMPLoopDirective>(ParentDir)
4387 .getIterationVariable()
4388 ->IgnoreParenImpCasts();
4389 LValue IdxLVal = EmitLValue(IVExpr);
4390 llvm::Value *IdxVal = EmitLoadOfScalar(IdxLVal, IVExpr->getExprLoc());
4391 IdxVal = Builder.CreateIntCast(IdxVal, SizeTy, /*isSigned=*/false);
4392 llvm::BasicBlock *ExclusiveExitBB = nullptr;
4393 if (!IsInclusive) {
4394 llvm::BasicBlock *ContBB = createBasicBlock("omp.exclusive.dec");
4395 ExclusiveExitBB = createBasicBlock("omp.exclusive.copy.exit");
4396 llvm::Value *Cmp = Builder.CreateIsNull(IdxVal);
4397 Builder.CreateCondBr(Cmp, ExclusiveExitBB, ContBB);
4398 EmitBlock(ContBB);
4399 // Use idx - 1 iteration for exclusive scan.
4400 IdxVal = Builder.CreateNUWSub(IdxVal, llvm::ConstantInt::get(SizeTy, 1));
4401 }
4402 for (unsigned I = 0, E = CopyArrayElems.size(); I < E; ++I) {
4403 const Expr *PrivateExpr = Privates[I];
4404 const Expr *OrigExpr = Shareds[I];
4405 const Expr *CopyArrayElem = CopyArrayElems[I];
4406 OpaqueValueMapping IdxMapping(
4407 *this,
4408 cast<OpaqueValueExpr>(
4409 cast<ArraySubscriptExpr>(CopyArrayElem)->getIdx()),
4410 RValue::get(IdxVal));
4411 LValue SrcLVal = EmitLValue(CopyArrayElem);
4412 LValue DestLVal = EmitLValue(OrigExpr);
4413 EmitOMPCopy(PrivateExpr->getType(), DestLVal.getAddress(*this),
4414 SrcLVal.getAddress(*this),
4415 cast<VarDecl>(cast<DeclRefExpr>(LHSs[I])->getDecl()),
4416 cast<VarDecl>(cast<DeclRefExpr>(RHSs[I])->getDecl()),
4417 CopyOps[I]);
4418 }
4419 if (!IsInclusive) {
4420 EmitBlock(ExclusiveExitBB);
4421 }
4422 }
4423 EmitBranch((OMPFirstScanLoop == IsInclusive) ? OMPBeforeScanBlock
4424 : OMPAfterScanBlock);
4425 EmitBlock(OMPAfterScanBlock);
4426}
4427
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004428void CodeGenFunction::EmitOMPDistributeLoop(const OMPLoopDirective &S,
4429 const CodeGenLoopTy &CodeGenLoop,
4430 Expr *IncExpr) {
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004431 // Emit the loop iteration variable.
Alexey Bataevddf3db92018-04-13 17:31:06 +00004432 const auto *IVExpr = cast<DeclRefExpr>(S.getIterationVariable());
4433 const auto *IVDecl = cast<VarDecl>(IVExpr->getDecl());
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004434 EmitVarDecl(*IVDecl);
4435
4436 // Emit the iterations count variable.
4437 // If it is not a variable, Sema decided to calculate iterations count on each
4438 // iteration (e.g., it is foldable into a constant).
Alexey Bataevddf3db92018-04-13 17:31:06 +00004439 if (const auto *LIExpr = dyn_cast<DeclRefExpr>(S.getLastIteration())) {
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004440 EmitVarDecl(*cast<VarDecl>(LIExpr->getDecl()));
4441 // Emit calculation of the iterations count.
4442 EmitIgnoredExpr(S.getCalcLastIteration());
4443 }
4444
Alexey Bataevddf3db92018-04-13 17:31:06 +00004445 CGOpenMPRuntime &RT = CGM.getOpenMPRuntime();
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004446
Carlo Bertolli962bb802017-01-03 18:24:42 +00004447 bool HasLastprivateClause = false;
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004448 // Check pre-condition.
4449 {
Alexey Bataev5a3af132016-03-29 08:58:54 +00004450 OMPLoopScope PreInitScope(*this, S);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004451 // Skip the entire loop if we don't meet the precondition.
4452 // If the condition constant folds and can be elided, avoid emitting the
4453 // whole loop.
4454 bool CondConstant;
4455 llvm::BasicBlock *ContBlock = nullptr;
4456 if (ConstantFoldsToSimpleInteger(S.getPreCond(), CondConstant)) {
4457 if (!CondConstant)
4458 return;
4459 } else {
Alexey Bataevddf3db92018-04-13 17:31:06 +00004460 llvm::BasicBlock *ThenBlock = createBasicBlock("omp.precond.then");
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004461 ContBlock = createBasicBlock("omp.precond.end");
4462 emitPreCond(*this, S, S.getPreCond(), ThenBlock, ContBlock,
4463 getProfileCount(&S));
4464 EmitBlock(ThenBlock);
4465 incrementProfileCounter(&S);
4466 }
4467
Alexey Bataev617db5f2017-12-04 15:38:33 +00004468 emitAlignedClause(*this, S);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004469 // Emit 'then' code.
4470 {
4471 // Emit helper vars inits.
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004472
4473 LValue LB = EmitOMPHelperVar(
4474 *this, cast<DeclRefExpr>(
4475 (isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
4476 ? S.getCombinedLowerBoundVariable()
4477 : S.getLowerBoundVariable())));
4478 LValue UB = EmitOMPHelperVar(
4479 *this, cast<DeclRefExpr>(
4480 (isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
4481 ? S.getCombinedUpperBoundVariable()
4482 : S.getUpperBoundVariable())));
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004483 LValue ST =
4484 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getStrideVariable()));
4485 LValue IL =
4486 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getIsLastIterVariable()));
4487
4488 OMPPrivateScope LoopScope(*this);
Carlo Bertolli962bb802017-01-03 18:24:42 +00004489 if (EmitOMPFirstprivateClause(S, LoopScope)) {
Alexey Bataev617db5f2017-12-04 15:38:33 +00004490 // Emit implicit barrier to synchronize threads and avoid data races
4491 // on initialization of firstprivate variables and post-update of
Carlo Bertolli962bb802017-01-03 18:24:42 +00004492 // lastprivate variables.
4493 CGM.getOpenMPRuntime().emitBarrierCall(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004494 *this, S.getBeginLoc(), OMPD_unknown, /*EmitChecks=*/false,
Alexey Bataev617db5f2017-12-04 15:38:33 +00004495 /*ForceSimpleCall=*/true);
Carlo Bertolli962bb802017-01-03 18:24:42 +00004496 }
4497 EmitOMPPrivateClause(S, LoopScope);
Alexey Bataev617db5f2017-12-04 15:38:33 +00004498 if (isOpenMPSimdDirective(S.getDirectiveKind()) &&
Alexey Bataev999277a2017-12-06 14:31:09 +00004499 !isOpenMPParallelDirective(S.getDirectiveKind()) &&
4500 !isOpenMPTeamsDirective(S.getDirectiveKind()))
Alexey Bataev617db5f2017-12-04 15:38:33 +00004501 EmitOMPReductionClauseInit(S, LoopScope);
Carlo Bertolli962bb802017-01-03 18:24:42 +00004502 HasLastprivateClause = EmitOMPLastprivateClauseInit(S, LoopScope);
Alexey Bataev5dff95c2016-04-22 03:56:56 +00004503 EmitOMPPrivateLoopCounters(S, LoopScope);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004504 (void)LoopScope.Privatize();
Alexey Bataev60705422018-10-30 15:50:12 +00004505 if (isOpenMPTargetExecutionDirective(S.getDirectiveKind()))
4506 CGM.getOpenMPRuntime().adjustTargetSpecificDataForLambdas(*this, S);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004507
4508 // Detect the distribute schedule kind and chunk.
4509 llvm::Value *Chunk = nullptr;
4510 OpenMPDistScheduleClauseKind ScheduleKind = OMPC_DIST_SCHEDULE_unknown;
Alexey Bataevddf3db92018-04-13 17:31:06 +00004511 if (const auto *C = S.getSingleClause<OMPDistScheduleClause>()) {
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004512 ScheduleKind = C->getDistScheduleKind();
Alexey Bataevddf3db92018-04-13 17:31:06 +00004513 if (const Expr *Ch = C->getChunkSize()) {
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004514 Chunk = EmitScalarExpr(Ch);
4515 Chunk = EmitScalarConversion(Chunk, Ch->getType(),
Alexey Bataev617db5f2017-12-04 15:38:33 +00004516 S.getIterationVariable()->getType(),
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004517 S.getBeginLoc());
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004518 }
Gheorghe-Teodor Bercea02650d42018-09-27 19:22:56 +00004519 } else {
4520 // Default behaviour for dist_schedule clause.
4521 CGM.getOpenMPRuntime().getDefaultDistScheduleAndChunk(
4522 *this, S, ScheduleKind, Chunk);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004523 }
4524 const unsigned IVSize = getContext().getTypeSize(IVExpr->getType());
4525 const bool IVSigned = IVExpr->getType()->hasSignedIntegerRepresentation();
4526
4527 // OpenMP [2.10.8, distribute Construct, Description]
4528 // If dist_schedule is specified, kind must be static. If specified,
4529 // iterations are divided into chunks of size chunk_size, chunks are
4530 // assigned to the teams of the league in a round-robin fashion in the
4531 // order of the team number. When no chunk_size is specified, the
4532 // iteration space is divided into chunks that are approximately equal
4533 // in size, and at most one chunk is distributed to each team of the
4534 // league. The size of the chunks is unspecified in this case.
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00004535 bool StaticChunked = RT.isStaticChunked(
4536 ScheduleKind, /* Chunked */ Chunk != nullptr) &&
4537 isOpenMPLoopBoundSharingDirective(S.getDirectiveKind());
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004538 if (RT.isStaticNonchunked(ScheduleKind,
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00004539 /* Chunked */ Chunk != nullptr) ||
4540 StaticChunked) {
Alexey Bataev0f87dbe2017-08-14 17:56:13 +00004541 CGOpenMPRuntime::StaticRTInput StaticInit(
Akira Hatanakaf139ae32019-12-03 15:17:01 -08004542 IVSize, IVSigned, /* Ordered = */ false, IL.getAddress(*this),
4543 LB.getAddress(*this), UB.getAddress(*this), ST.getAddress(*this),
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00004544 StaticChunked ? Chunk : nullptr);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004545 RT.emitDistributeStaticInit(*this, S.getBeginLoc(), ScheduleKind,
Alexey Bataev0f87dbe2017-08-14 17:56:13 +00004546 StaticInit);
Alexey Bataevddf3db92018-04-13 17:31:06 +00004547 JumpDest LoopExit =
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004548 getJumpDestInCurrentScope(createBasicBlock("omp.loop.exit"));
4549 // UB = min(UB, GlobalUB);
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004550 EmitIgnoredExpr(isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
4551 ? S.getCombinedEnsureUpperBound()
4552 : S.getEnsureUpperBound());
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004553 // IV = LB;
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004554 EmitIgnoredExpr(isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
4555 ? S.getCombinedInit()
4556 : S.getInit());
4557
Alexey Bataevddf3db92018-04-13 17:31:06 +00004558 const Expr *Cond =
4559 isOpenMPLoopBoundSharingDirective(S.getDirectiveKind())
4560 ? S.getCombinedCond()
4561 : S.getCond();
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004562
Gheorghe-Teodor Berceae9256762018-10-29 15:45:47 +00004563 if (StaticChunked)
4564 Cond = S.getCombinedDistCond();
4565
4566 // For static unchunked schedules generate:
4567 //
4568 // 1. For distribute alone, codegen
4569 // while (idx <= UB) {
4570 // BODY;
4571 // ++idx;
4572 // }
4573 //
4574 // 2. When combined with 'for' (e.g. as in 'distribute parallel for')
4575 // while (idx <= UB) {
4576 // <CodeGen rest of pragma>(LB, UB);
4577 // idx += ST;
4578 // }
4579 //
4580 // For static chunk one schedule generate:
4581 //
4582 // while (IV <= GlobalUB) {
4583 // <CodeGen rest of pragma>(LB, UB);
4584 // LB += ST;
4585 // UB += ST;
4586 // UB = min(UB, GlobalUB);
4587 // IV = LB;
4588 // }
4589 //
Alexey Bataev779a1802019-12-06 12:21:31 -05004590 emitCommonSimdLoop(
4591 *this, S,
4592 [&S](CodeGenFunction &CGF, PrePostActionTy &) {
4593 if (isOpenMPSimdDirective(S.getDirectiveKind()))
4594 CGF.EmitOMPSimdInit(S, /*IsMonotonic=*/true);
4595 },
4596 [&S, &LoopScope, Cond, IncExpr, LoopExit, &CodeGenLoop,
4597 StaticChunked](CodeGenFunction &CGF, PrePostActionTy &) {
4598 CGF.EmitOMPInnerLoop(
4599 S, LoopScope.requiresCleanups(), Cond, IncExpr,
4600 [&S, LoopExit, &CodeGenLoop](CodeGenFunction &CGF) {
4601 CodeGenLoop(CGF, S, LoopExit);
4602 },
4603 [&S, StaticChunked](CodeGenFunction &CGF) {
4604 if (StaticChunked) {
4605 CGF.EmitIgnoredExpr(S.getCombinedNextLowerBound());
4606 CGF.EmitIgnoredExpr(S.getCombinedNextUpperBound());
4607 CGF.EmitIgnoredExpr(S.getCombinedEnsureUpperBound());
4608 CGF.EmitIgnoredExpr(S.getCombinedInit());
4609 }
4610 });
4611 });
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004612 EmitBlock(LoopExit.getBlock());
4613 // Tell the runtime we are done.
Alexey Bataevc33ba8c2020-01-17 14:05:40 -05004614 RT.emitForStaticFinish(*this, S.getEndLoc(), S.getDirectiveKind());
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004615 } else {
4616 // Emit the outer loop, which requests its work chunk [LB..UB] from
4617 // runtime and runs the inner loop to process it.
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004618 const OMPLoopArguments LoopArguments = {
Akira Hatanakaf139ae32019-12-03 15:17:01 -08004619 LB.getAddress(*this), UB.getAddress(*this), ST.getAddress(*this),
4620 IL.getAddress(*this), Chunk};
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004621 EmitOMPDistributeOuterLoop(ScheduleKind, S, LoopScope, LoopArguments,
4622 CodeGenLoop);
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004623 }
Alexey Bataev617db5f2017-12-04 15:38:33 +00004624 if (isOpenMPSimdDirective(S.getDirectiveKind())) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00004625 EmitOMPSimdFinal(S, [IL, &S](CodeGenFunction &CGF) {
Alexey Bataev617db5f2017-12-04 15:38:33 +00004626 return CGF.Builder.CreateIsNotNull(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004627 CGF.EmitLoadOfScalar(IL, S.getBeginLoc()));
Alexey Bataev617db5f2017-12-04 15:38:33 +00004628 });
4629 }
Carlo Bertollibeda2142018-02-22 19:38:14 +00004630 if (isOpenMPSimdDirective(S.getDirectiveKind()) &&
4631 !isOpenMPParallelDirective(S.getDirectiveKind()) &&
4632 !isOpenMPTeamsDirective(S.getDirectiveKind())) {
Jonas Hahnfeld5aaaece2018-10-02 19:12:47 +00004633 EmitOMPReductionClauseFinal(S, OMPD_simd);
Carlo Bertollibeda2142018-02-22 19:38:14 +00004634 // Emit post-update of the reduction variables if IsLastIter != 0.
4635 emitPostUpdateForReductionClause(
Alexey Bataevddf3db92018-04-13 17:31:06 +00004636 *this, S, [IL, &S](CodeGenFunction &CGF) {
Carlo Bertollibeda2142018-02-22 19:38:14 +00004637 return CGF.Builder.CreateIsNotNull(
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004638 CGF.EmitLoadOfScalar(IL, S.getBeginLoc()));
Carlo Bertollibeda2142018-02-22 19:38:14 +00004639 });
Alexey Bataev617db5f2017-12-04 15:38:33 +00004640 }
Carlo Bertolli962bb802017-01-03 18:24:42 +00004641 // Emit final copy of the lastprivate variables if IsLastIter != 0.
Alexey Bataev617db5f2017-12-04 15:38:33 +00004642 if (HasLastprivateClause) {
Carlo Bertolli962bb802017-01-03 18:24:42 +00004643 EmitOMPLastprivateClauseFinal(
4644 S, /*NoFinals=*/false,
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004645 Builder.CreateIsNotNull(EmitLoadOfScalar(IL, S.getBeginLoc())));
Alexey Bataev617db5f2017-12-04 15:38:33 +00004646 }
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004647 }
4648
4649 // We're now done with the loop, so jump to the continuation block.
4650 if (ContBlock) {
4651 EmitBranch(ContBlock);
4652 EmitBlock(ContBlock, true);
4653 }
4654 }
4655}
4656
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00004657void CodeGenFunction::EmitOMPDistributeDirective(
4658 const OMPDistributeDirective &S) {
Alexey Bataev14fa1c62016-03-29 05:34:15 +00004659 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00004660 CGF.EmitOMPDistributeLoop(S, emitOMPLoopBodyWithStopPoint, S.getInc());
Carlo Bertollifc35ad22016-03-07 16:04:49 +00004661 };
Alexey Bataev475a7442018-01-12 19:39:11 +00004662 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataev10a54312017-11-27 16:54:08 +00004663 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_distribute, CodeGen);
Carlo Bertolli6200a3d2015-12-14 14:51:25 +00004664}
4665
Alexey Bataev5f600d62015-09-29 03:48:57 +00004666static llvm::Function *emitOutlinedOrderedFunction(CodeGenModule &CGM,
Alexey Bataevc33ba8c2020-01-17 14:05:40 -05004667 const CapturedStmt *S,
4668 SourceLocation Loc) {
Alexey Bataev5f600d62015-09-29 03:48:57 +00004669 CodeGenFunction CGF(CGM, /*suppressNewContext=*/true);
4670 CodeGenFunction::CGCapturedStmtInfo CapStmtInfo;
4671 CGF.CapturedStmtInfo = &CapStmtInfo;
Alexey Bataevc33ba8c2020-01-17 14:05:40 -05004672 llvm::Function *Fn = CGF.GenerateOpenMPCapturedStmtFunction(*S, Loc);
Alexey Bataevc0f879b2018-04-10 20:10:53 +00004673 Fn->setDoesNotRecurse();
Alexey Bataev5f600d62015-09-29 03:48:57 +00004674 return Fn;
4675}
4676
Alexey Bataev98eb6e32015-04-22 11:15:40 +00004677void CodeGenFunction::EmitOMPOrderedDirective(const OMPOrderedDirective &S) {
Alexey Bataev475a7442018-01-12 19:39:11 +00004678 if (S.hasClausesOfKind<OMPDependClause>()) {
4679 assert(!S.getAssociatedStmt() &&
4680 "No associated statement must be in ordered depend construct.");
Alexey Bataev8b427062016-05-25 12:36:08 +00004681 for (const auto *DC : S.getClausesOfKind<OMPDependClause>())
4682 CGM.getOpenMPRuntime().emitDoacrossOrdered(*this, DC);
Alexey Bataev8ef31412015-12-18 07:58:25 +00004683 return;
Alexey Bataev8b427062016-05-25 12:36:08 +00004684 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00004685 const auto *C = S.getSingleClause<OMPSIMDClause>();
Alexey Bataev14fa1c62016-03-29 05:34:15 +00004686 auto &&CodeGen = [&S, C, this](CodeGenFunction &CGF,
4687 PrePostActionTy &Action) {
Alexey Bataev475a7442018-01-12 19:39:11 +00004688 const CapturedStmt *CS = S.getInnermostCapturedStmt();
Alexey Bataev5f600d62015-09-29 03:48:57 +00004689 if (C) {
Alexey Bataev5f600d62015-09-29 03:48:57 +00004690 llvm::SmallVector<llvm::Value *, 16> CapturedVars;
4691 CGF.GenerateOpenMPCapturedVars(*CS, CapturedVars);
Alexey Bataevc33ba8c2020-01-17 14:05:40 -05004692 llvm::Function *OutlinedFn =
4693 emitOutlinedOrderedFunction(CGM, CS, S.getBeginLoc());
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004694 CGM.getOpenMPRuntime().emitOutlinedFunctionCall(CGF, S.getBeginLoc(),
Alexey Bataev3c595a62017-08-14 15:01:03 +00004695 OutlinedFn, CapturedVars);
Alexey Bataev5f600d62015-09-29 03:48:57 +00004696 } else {
Alexey Bataev14fa1c62016-03-29 05:34:15 +00004697 Action.Enter(CGF);
Alexey Bataev475a7442018-01-12 19:39:11 +00004698 CGF.EmitStmt(CS->getCapturedStmt());
Alexey Bataev5f600d62015-09-29 03:48:57 +00004699 }
Alexey Bataev98eb6e32015-04-22 11:15:40 +00004700 };
Alexey Bataev475a7442018-01-12 19:39:11 +00004701 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00004702 CGM.getOpenMPRuntime().emitOrderedRegion(*this, CodeGen, S.getBeginLoc(), !C);
Alexey Bataev9fb6e642014-07-22 06:45:04 +00004703}
4704
Alexey Bataevb57056f2015-01-22 06:17:56 +00004705static llvm::Value *convertToScalarValue(CodeGenFunction &CGF, RValue Val,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004706 QualType SrcType, QualType DestType,
4707 SourceLocation Loc) {
Alexey Bataevb57056f2015-01-22 06:17:56 +00004708 assert(CGF.hasScalarEvaluationKind(DestType) &&
4709 "DestType must have scalar evaluation kind.");
4710 assert(!Val.isAggregate() && "Must be a scalar or complex.");
Alexey Bataevddf3db92018-04-13 17:31:06 +00004711 return Val.isScalar() ? CGF.EmitScalarConversion(Val.getScalarVal(), SrcType,
4712 DestType, Loc)
4713 : CGF.EmitComplexToScalarConversion(
4714 Val.getComplexVal(), SrcType, DestType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00004715}
4716
4717static CodeGenFunction::ComplexPairTy
4718convertToComplexValue(CodeGenFunction &CGF, RValue Val, QualType SrcType,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004719 QualType DestType, SourceLocation Loc) {
Alexey Bataevb57056f2015-01-22 06:17:56 +00004720 assert(CGF.getEvaluationKind(DestType) == TEK_Complex &&
4721 "DestType must have complex evaluation kind.");
4722 CodeGenFunction::ComplexPairTy ComplexVal;
4723 if (Val.isScalar()) {
4724 // Convert the input element to the element type of the complex.
Alexey Bataevddf3db92018-04-13 17:31:06 +00004725 QualType DestElementType =
4726 DestType->castAs<ComplexType>()->getElementType();
4727 llvm::Value *ScalarVal = CGF.EmitScalarConversion(
4728 Val.getScalarVal(), SrcType, DestElementType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00004729 ComplexVal = CodeGenFunction::ComplexPairTy(
4730 ScalarVal, llvm::Constant::getNullValue(ScalarVal->getType()));
4731 } else {
4732 assert(Val.isComplex() && "Must be a scalar or complex.");
Alexey Bataevddf3db92018-04-13 17:31:06 +00004733 QualType SrcElementType = SrcType->castAs<ComplexType>()->getElementType();
4734 QualType DestElementType =
4735 DestType->castAs<ComplexType>()->getElementType();
Alexey Bataevb57056f2015-01-22 06:17:56 +00004736 ComplexVal.first = CGF.EmitScalarConversion(
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004737 Val.getComplexVal().first, SrcElementType, DestElementType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00004738 ComplexVal.second = CGF.EmitScalarConversion(
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00004739 Val.getComplexVal().second, SrcElementType, DestElementType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00004740 }
4741 return ComplexVal;
4742}
4743
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004744static void emitSimpleAtomicStore(CodeGenFunction &CGF, llvm::AtomicOrdering AO,
Alexey Bataev5e018f92015-04-23 06:35:10 +00004745 LValue LVal, RValue RVal) {
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004746 if (LVal.isGlobalReg())
Alexey Bataev5e018f92015-04-23 06:35:10 +00004747 CGF.EmitStoreThroughGlobalRegLValue(RVal, LVal);
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004748 else
4749 CGF.EmitAtomicStore(RVal, LVal, AO, LVal.isVolatile(), /*isInit=*/false);
4750}
4751
4752static RValue emitSimpleAtomicLoad(CodeGenFunction &CGF,
4753 llvm::AtomicOrdering AO, LValue LVal,
4754 SourceLocation Loc) {
4755 if (LVal.isGlobalReg())
4756 return CGF.EmitLoadOfLValue(LVal, Loc);
4757 return CGF.EmitAtomicLoad(
4758 LVal, Loc, llvm::AtomicCmpXchgInst::getStrongestFailureOrdering(AO),
4759 LVal.isVolatile());
Alexey Bataev5e018f92015-04-23 06:35:10 +00004760}
4761
Alexey Bataev8524d152016-01-21 12:35:58 +00004762void CodeGenFunction::emitOMPSimpleStore(LValue LVal, RValue RVal,
4763 QualType RValTy, SourceLocation Loc) {
4764 switch (getEvaluationKind(LVal.getType())) {
Alexey Bataev5e018f92015-04-23 06:35:10 +00004765 case TEK_Scalar:
Alexey Bataev8524d152016-01-21 12:35:58 +00004766 EmitStoreThroughLValue(RValue::get(convertToScalarValue(
4767 *this, RVal, RValTy, LVal.getType(), Loc)),
4768 LVal);
Alexey Bataev5e018f92015-04-23 06:35:10 +00004769 break;
4770 case TEK_Complex:
Alexey Bataev8524d152016-01-21 12:35:58 +00004771 EmitStoreOfComplex(
4772 convertToComplexValue(*this, RVal, RValTy, LVal.getType(), Loc), LVal,
Alexey Bataev5e018f92015-04-23 06:35:10 +00004773 /*isInit=*/false);
4774 break;
4775 case TEK_Aggregate:
4776 llvm_unreachable("Must be a scalar or complex.");
4777 }
4778}
4779
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004780static void emitOMPAtomicReadExpr(CodeGenFunction &CGF, llvm::AtomicOrdering AO,
Alexey Bataevb57056f2015-01-22 06:17:56 +00004781 const Expr *X, const Expr *V,
4782 SourceLocation Loc) {
4783 // v = x;
4784 assert(V->isLValue() && "V of 'omp atomic read' is not lvalue");
4785 assert(X->isLValue() && "X of 'omp atomic read' is not lvalue");
4786 LValue XLValue = CGF.EmitLValue(X);
4787 LValue VLValue = CGF.EmitLValue(V);
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004788 RValue Res = emitSimpleAtomicLoad(CGF, AO, XLValue, Loc);
4789 // OpenMP, 2.17.7, atomic Construct
4790 // If the read or capture clause is specified and the acquire, acq_rel, or
4791 // seq_cst clause is specified then the strong flush on exit from the atomic
4792 // operation is also an acquire flush.
4793 switch (AO) {
4794 case llvm::AtomicOrdering::Acquire:
4795 case llvm::AtomicOrdering::AcquireRelease:
4796 case llvm::AtomicOrdering::SequentiallyConsistent:
4797 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc,
4798 llvm::AtomicOrdering::Acquire);
4799 break;
4800 case llvm::AtomicOrdering::Monotonic:
4801 case llvm::AtomicOrdering::Release:
4802 break;
4803 case llvm::AtomicOrdering::NotAtomic:
4804 case llvm::AtomicOrdering::Unordered:
4805 llvm_unreachable("Unexpected ordering.");
4806 }
Alexey Bataev8524d152016-01-21 12:35:58 +00004807 CGF.emitOMPSimpleStore(VLValue, Res, X->getType().getNonReferenceType(), Loc);
Alexey Bataevf117f2c2020-01-28 11:19:45 -05004808 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, V);
Alexey Bataevb57056f2015-01-22 06:17:56 +00004809}
4810
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004811static void emitOMPAtomicWriteExpr(CodeGenFunction &CGF,
4812 llvm::AtomicOrdering AO, const Expr *X,
4813 const Expr *E, SourceLocation Loc) {
Alexey Bataevb8329262015-02-27 06:33:30 +00004814 // x = expr;
4815 assert(X->isLValue() && "X of 'omp atomic write' is not lvalue");
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004816 emitSimpleAtomicStore(CGF, AO, CGF.EmitLValue(X), CGF.EmitAnyExpr(E));
Alexey Bataevf117f2c2020-01-28 11:19:45 -05004817 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, X);
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004818 // OpenMP, 2.17.7, atomic Construct
4819 // If the write, update, or capture clause is specified and the release,
4820 // acq_rel, or seq_cst clause is specified then the strong flush on entry to
4821 // the atomic operation is also a release flush.
4822 switch (AO) {
4823 case llvm::AtomicOrdering::Release:
4824 case llvm::AtomicOrdering::AcquireRelease:
4825 case llvm::AtomicOrdering::SequentiallyConsistent:
4826 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc,
4827 llvm::AtomicOrdering::Release);
4828 break;
4829 case llvm::AtomicOrdering::Acquire:
4830 case llvm::AtomicOrdering::Monotonic:
4831 break;
4832 case llvm::AtomicOrdering::NotAtomic:
4833 case llvm::AtomicOrdering::Unordered:
4834 llvm_unreachable("Unexpected ordering.");
4835 }
Alexey Bataevb8329262015-02-27 06:33:30 +00004836}
4837
Benjamin Kramer439ee9d2015-05-01 13:59:53 +00004838static std::pair<bool, RValue> emitOMPAtomicRMW(CodeGenFunction &CGF, LValue X,
4839 RValue Update,
4840 BinaryOperatorKind BO,
4841 llvm::AtomicOrdering AO,
4842 bool IsXLHSInRHSPart) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00004843 ASTContext &Context = CGF.getContext();
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004844 // Allow atomicrmw only if 'x' and 'update' are integer values, lvalue for 'x'
Alexey Bataevb4505a72015-03-30 05:20:59 +00004845 // expression is simple and atomic is allowed for the given type for the
4846 // target platform.
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004847 if (BO == BO_Comma || !Update.isScalar() ||
Akira Hatanakaf139ae32019-12-03 15:17:01 -08004848 !Update.getScalarVal()->getType()->isIntegerTy() || !X.isSimple() ||
4849 (!isa<llvm::ConstantInt>(Update.getScalarVal()) &&
4850 (Update.getScalarVal()->getType() !=
4851 X.getAddress(CGF).getElementType())) ||
4852 !X.getAddress(CGF).getElementType()->isIntegerTy() ||
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004853 !Context.getTargetInfo().hasBuiltinAtomic(
4854 Context.getTypeSize(X.getType()), Context.toBits(X.getAlignment())))
Alexey Bataev5e018f92015-04-23 06:35:10 +00004855 return std::make_pair(false, RValue::get(nullptr));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004856
4857 llvm::AtomicRMWInst::BinOp RMWOp;
4858 switch (BO) {
4859 case BO_Add:
4860 RMWOp = llvm::AtomicRMWInst::Add;
4861 break;
4862 case BO_Sub:
4863 if (!IsXLHSInRHSPart)
Alexey Bataev5e018f92015-04-23 06:35:10 +00004864 return std::make_pair(false, RValue::get(nullptr));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004865 RMWOp = llvm::AtomicRMWInst::Sub;
4866 break;
4867 case BO_And:
4868 RMWOp = llvm::AtomicRMWInst::And;
4869 break;
4870 case BO_Or:
4871 RMWOp = llvm::AtomicRMWInst::Or;
4872 break;
4873 case BO_Xor:
4874 RMWOp = llvm::AtomicRMWInst::Xor;
4875 break;
4876 case BO_LT:
4877 RMWOp = X.getType()->hasSignedIntegerRepresentation()
4878 ? (IsXLHSInRHSPart ? llvm::AtomicRMWInst::Min
4879 : llvm::AtomicRMWInst::Max)
4880 : (IsXLHSInRHSPart ? llvm::AtomicRMWInst::UMin
4881 : llvm::AtomicRMWInst::UMax);
4882 break;
4883 case BO_GT:
4884 RMWOp = X.getType()->hasSignedIntegerRepresentation()
4885 ? (IsXLHSInRHSPart ? llvm::AtomicRMWInst::Max
4886 : llvm::AtomicRMWInst::Min)
4887 : (IsXLHSInRHSPart ? llvm::AtomicRMWInst::UMax
4888 : llvm::AtomicRMWInst::UMin);
4889 break;
Alexey Bataev5e018f92015-04-23 06:35:10 +00004890 case BO_Assign:
4891 RMWOp = llvm::AtomicRMWInst::Xchg;
4892 break;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004893 case BO_Mul:
4894 case BO_Div:
4895 case BO_Rem:
4896 case BO_Shl:
4897 case BO_Shr:
4898 case BO_LAnd:
4899 case BO_LOr:
Alexey Bataev5e018f92015-04-23 06:35:10 +00004900 return std::make_pair(false, RValue::get(nullptr));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004901 case BO_PtrMemD:
4902 case BO_PtrMemI:
4903 case BO_LE:
4904 case BO_GE:
4905 case BO_EQ:
4906 case BO_NE:
Richard Smithc70f1d62017-12-14 15:16:18 +00004907 case BO_Cmp:
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004908 case BO_AddAssign:
4909 case BO_SubAssign:
4910 case BO_AndAssign:
4911 case BO_OrAssign:
4912 case BO_XorAssign:
4913 case BO_MulAssign:
4914 case BO_DivAssign:
4915 case BO_RemAssign:
4916 case BO_ShlAssign:
4917 case BO_ShrAssign:
4918 case BO_Comma:
4919 llvm_unreachable("Unsupported atomic update operation");
4920 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00004921 llvm::Value *UpdateVal = Update.getScalarVal();
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004922 if (auto *IC = dyn_cast<llvm::ConstantInt>(UpdateVal)) {
4923 UpdateVal = CGF.Builder.CreateIntCast(
Akira Hatanakaf139ae32019-12-03 15:17:01 -08004924 IC, X.getAddress(CGF).getElementType(),
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004925 X.getType()->hasSignedIntegerRepresentation());
4926 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00004927 llvm::Value *Res =
Akira Hatanakaf139ae32019-12-03 15:17:01 -08004928 CGF.Builder.CreateAtomicRMW(RMWOp, X.getPointer(CGF), UpdateVal, AO);
Alexey Bataev5e018f92015-04-23 06:35:10 +00004929 return std::make_pair(true, RValue::get(Res));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004930}
4931
Alexey Bataev5e018f92015-04-23 06:35:10 +00004932std::pair<bool, RValue> CodeGenFunction::EmitOMPAtomicSimpleUpdateExpr(
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004933 LValue X, RValue E, BinaryOperatorKind BO, bool IsXLHSInRHSPart,
4934 llvm::AtomicOrdering AO, SourceLocation Loc,
Alexey Bataevddf3db92018-04-13 17:31:06 +00004935 const llvm::function_ref<RValue(RValue)> CommonGen) {
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004936 // Update expressions are allowed to have the following forms:
4937 // x binop= expr; -> xrval + expr;
4938 // x++, ++x -> xrval + 1;
4939 // x--, --x -> xrval - 1;
4940 // x = x binop expr; -> xrval binop expr
4941 // x = expr Op x; - > expr binop xrval;
Alexey Bataev5e018f92015-04-23 06:35:10 +00004942 auto Res = emitOMPAtomicRMW(*this, X, E, BO, AO, IsXLHSInRHSPart);
4943 if (!Res.first) {
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004944 if (X.isGlobalReg()) {
4945 // Emit an update expression: 'xrval' binop 'expr' or 'expr' binop
4946 // 'xrval'.
4947 EmitStoreThroughLValue(CommonGen(EmitLoadOfLValue(X, Loc)), X);
4948 } else {
4949 // Perform compare-and-swap procedure.
4950 EmitAtomicUpdate(X, AO, CommonGen, X.getType().isVolatileQualified());
Alexey Bataevb4505a72015-03-30 05:20:59 +00004951 }
4952 }
Alexey Bataev5e018f92015-04-23 06:35:10 +00004953 return Res;
Alexey Bataevb4505a72015-03-30 05:20:59 +00004954}
4955
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004956static void emitOMPAtomicUpdateExpr(CodeGenFunction &CGF,
4957 llvm::AtomicOrdering AO, const Expr *X,
4958 const Expr *E, const Expr *UE,
4959 bool IsXLHSInRHSPart, SourceLocation Loc) {
Alexey Bataevb4505a72015-03-30 05:20:59 +00004960 assert(isa<BinaryOperator>(UE->IgnoreImpCasts()) &&
4961 "Update expr in 'atomic update' must be a binary operator.");
Alexey Bataevddf3db92018-04-13 17:31:06 +00004962 const auto *BOUE = cast<BinaryOperator>(UE->IgnoreImpCasts());
Alexey Bataevb4505a72015-03-30 05:20:59 +00004963 // Update expressions are allowed to have the following forms:
4964 // x binop= expr; -> xrval + expr;
4965 // x++, ++x -> xrval + 1;
4966 // x--, --x -> xrval - 1;
4967 // x = x binop expr; -> xrval binop expr
4968 // x = expr Op x; - > expr binop xrval;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00004969 assert(X->isLValue() && "X of 'omp atomic update' is not lvalue");
Alexey Bataevb4505a72015-03-30 05:20:59 +00004970 LValue XLValue = CGF.EmitLValue(X);
4971 RValue ExprRValue = CGF.EmitAnyExpr(E);
Alexey Bataevddf3db92018-04-13 17:31:06 +00004972 const auto *LHS = cast<OpaqueValueExpr>(BOUE->getLHS()->IgnoreImpCasts());
4973 const auto *RHS = cast<OpaqueValueExpr>(BOUE->getRHS()->IgnoreImpCasts());
4974 const OpaqueValueExpr *XRValExpr = IsXLHSInRHSPart ? LHS : RHS;
4975 const OpaqueValueExpr *ERValExpr = IsXLHSInRHSPart ? RHS : LHS;
4976 auto &&Gen = [&CGF, UE, ExprRValue, XRValExpr, ERValExpr](RValue XRValue) {
4977 CodeGenFunction::OpaqueValueMapping MapExpr(CGF, ERValExpr, ExprRValue);
4978 CodeGenFunction::OpaqueValueMapping MapX(CGF, XRValExpr, XRValue);
4979 return CGF.EmitAnyExpr(UE);
4980 };
Alexey Bataev5e018f92015-04-23 06:35:10 +00004981 (void)CGF.EmitOMPAtomicSimpleUpdateExpr(
4982 XLValue, ExprRValue, BOUE->getOpcode(), IsXLHSInRHSPart, AO, Loc, Gen);
Alexey Bataevf117f2c2020-01-28 11:19:45 -05004983 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, X);
Alexey Bataeve8e05de2020-02-07 12:22:23 -05004984 // OpenMP, 2.17.7, atomic Construct
4985 // If the write, update, or capture clause is specified and the release,
4986 // acq_rel, or seq_cst clause is specified then the strong flush on entry to
4987 // the atomic operation is also a release flush.
4988 switch (AO) {
4989 case llvm::AtomicOrdering::Release:
4990 case llvm::AtomicOrdering::AcquireRelease:
4991 case llvm::AtomicOrdering::SequentiallyConsistent:
4992 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc,
4993 llvm::AtomicOrdering::Release);
4994 break;
4995 case llvm::AtomicOrdering::Acquire:
4996 case llvm::AtomicOrdering::Monotonic:
4997 break;
4998 case llvm::AtomicOrdering::NotAtomic:
4999 case llvm::AtomicOrdering::Unordered:
5000 llvm_unreachable("Unexpected ordering.");
5001 }
Alexey Bataev5e018f92015-04-23 06:35:10 +00005002}
5003
5004static RValue convertToType(CodeGenFunction &CGF, RValue Value,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00005005 QualType SourceType, QualType ResType,
5006 SourceLocation Loc) {
Alexey Bataev5e018f92015-04-23 06:35:10 +00005007 switch (CGF.getEvaluationKind(ResType)) {
5008 case TEK_Scalar:
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00005009 return RValue::get(
5010 convertToScalarValue(CGF, Value, SourceType, ResType, Loc));
Alexey Bataev5e018f92015-04-23 06:35:10 +00005011 case TEK_Complex: {
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00005012 auto Res = convertToComplexValue(CGF, Value, SourceType, ResType, Loc);
Alexey Bataev5e018f92015-04-23 06:35:10 +00005013 return RValue::getComplex(Res.first, Res.second);
5014 }
5015 case TEK_Aggregate:
5016 break;
5017 }
5018 llvm_unreachable("Must be a scalar or complex.");
5019}
5020
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005021static void emitOMPAtomicCaptureExpr(CodeGenFunction &CGF,
5022 llvm::AtomicOrdering AO,
Alexey Bataev5e018f92015-04-23 06:35:10 +00005023 bool IsPostfixUpdate, const Expr *V,
5024 const Expr *X, const Expr *E,
5025 const Expr *UE, bool IsXLHSInRHSPart,
5026 SourceLocation Loc) {
5027 assert(X->isLValue() && "X of 'omp atomic capture' is not lvalue");
5028 assert(V->isLValue() && "V of 'omp atomic capture' is not lvalue");
5029 RValue NewVVal;
5030 LValue VLValue = CGF.EmitLValue(V);
5031 LValue XLValue = CGF.EmitLValue(X);
5032 RValue ExprRValue = CGF.EmitAnyExpr(E);
Alexey Bataev5e018f92015-04-23 06:35:10 +00005033 QualType NewVValType;
5034 if (UE) {
5035 // 'x' is updated with some additional value.
5036 assert(isa<BinaryOperator>(UE->IgnoreImpCasts()) &&
5037 "Update expr in 'atomic capture' must be a binary operator.");
Alexey Bataevddf3db92018-04-13 17:31:06 +00005038 const auto *BOUE = cast<BinaryOperator>(UE->IgnoreImpCasts());
Alexey Bataev5e018f92015-04-23 06:35:10 +00005039 // Update expressions are allowed to have the following forms:
5040 // x binop= expr; -> xrval + expr;
5041 // x++, ++x -> xrval + 1;
5042 // x--, --x -> xrval - 1;
5043 // x = x binop expr; -> xrval binop expr
5044 // x = expr Op x; - > expr binop xrval;
Alexey Bataevddf3db92018-04-13 17:31:06 +00005045 const auto *LHS = cast<OpaqueValueExpr>(BOUE->getLHS()->IgnoreImpCasts());
5046 const auto *RHS = cast<OpaqueValueExpr>(BOUE->getRHS()->IgnoreImpCasts());
5047 const OpaqueValueExpr *XRValExpr = IsXLHSInRHSPart ? LHS : RHS;
Alexey Bataev5e018f92015-04-23 06:35:10 +00005048 NewVValType = XRValExpr->getType();
Alexey Bataevddf3db92018-04-13 17:31:06 +00005049 const OpaqueValueExpr *ERValExpr = IsXLHSInRHSPart ? RHS : LHS;
Alexey Bataev5e018f92015-04-23 06:35:10 +00005050 auto &&Gen = [&CGF, &NewVVal, UE, ExprRValue, XRValExpr, ERValExpr,
Alexey Bataevddf3db92018-04-13 17:31:06 +00005051 IsPostfixUpdate](RValue XRValue) {
Alexey Bataev5e018f92015-04-23 06:35:10 +00005052 CodeGenFunction::OpaqueValueMapping MapExpr(CGF, ERValExpr, ExprRValue);
5053 CodeGenFunction::OpaqueValueMapping MapX(CGF, XRValExpr, XRValue);
5054 RValue Res = CGF.EmitAnyExpr(UE);
5055 NewVVal = IsPostfixUpdate ? XRValue : Res;
5056 return Res;
5057 };
5058 auto Res = CGF.EmitOMPAtomicSimpleUpdateExpr(
5059 XLValue, ExprRValue, BOUE->getOpcode(), IsXLHSInRHSPart, AO, Loc, Gen);
Alexey Bataevf117f2c2020-01-28 11:19:45 -05005060 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, X);
Alexey Bataev5e018f92015-04-23 06:35:10 +00005061 if (Res.first) {
5062 // 'atomicrmw' instruction was generated.
5063 if (IsPostfixUpdate) {
5064 // Use old value from 'atomicrmw'.
5065 NewVVal = Res.second;
5066 } else {
5067 // 'atomicrmw' does not provide new value, so evaluate it using old
5068 // value of 'x'.
5069 CodeGenFunction::OpaqueValueMapping MapExpr(CGF, ERValExpr, ExprRValue);
5070 CodeGenFunction::OpaqueValueMapping MapX(CGF, XRValExpr, Res.second);
5071 NewVVal = CGF.EmitAnyExpr(UE);
5072 }
5073 }
5074 } else {
5075 // 'x' is simply rewritten with some 'expr'.
5076 NewVValType = X->getType().getNonReferenceType();
5077 ExprRValue = convertToType(CGF, ExprRValue, E->getType(),
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00005078 X->getType().getNonReferenceType(), Loc);
Alexey Bataevddf3db92018-04-13 17:31:06 +00005079 auto &&Gen = [&NewVVal, ExprRValue](RValue XRValue) {
Alexey Bataev5e018f92015-04-23 06:35:10 +00005080 NewVVal = XRValue;
5081 return ExprRValue;
5082 };
5083 // Try to perform atomicrmw xchg, otherwise simple exchange.
5084 auto Res = CGF.EmitOMPAtomicSimpleUpdateExpr(
5085 XLValue, ExprRValue, /*BO=*/BO_Assign, /*IsXLHSInRHSPart=*/false, AO,
5086 Loc, Gen);
Alexey Bataevf117f2c2020-01-28 11:19:45 -05005087 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, X);
Alexey Bataev5e018f92015-04-23 06:35:10 +00005088 if (Res.first) {
5089 // 'atomicrmw' instruction was generated.
5090 NewVVal = IsPostfixUpdate ? Res.second : ExprRValue;
5091 }
5092 }
5093 // Emit post-update store to 'v' of old/new 'x' value.
Alexey Bataev8524d152016-01-21 12:35:58 +00005094 CGF.emitOMPSimpleStore(VLValue, NewVVal, NewVValType, Loc);
Alexey Bataevf117f2c2020-01-28 11:19:45 -05005095 CGF.CGM.getOpenMPRuntime().checkAndEmitLastprivateConditional(CGF, V);
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005096 // OpenMP, 2.17.7, atomic Construct
5097 // If the write, update, or capture clause is specified and the release,
5098 // acq_rel, or seq_cst clause is specified then the strong flush on entry to
5099 // the atomic operation is also a release flush.
5100 // If the read or capture clause is specified and the acquire, acq_rel, or
5101 // seq_cst clause is specified then the strong flush on exit from the atomic
5102 // operation is also an acquire flush.
5103 switch (AO) {
5104 case llvm::AtomicOrdering::Release:
5105 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc,
5106 llvm::AtomicOrdering::Release);
5107 break;
5108 case llvm::AtomicOrdering::Acquire:
5109 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc,
5110 llvm::AtomicOrdering::Acquire);
5111 break;
5112 case llvm::AtomicOrdering::AcquireRelease:
5113 case llvm::AtomicOrdering::SequentiallyConsistent:
5114 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc,
5115 llvm::AtomicOrdering::AcquireRelease);
5116 break;
5117 case llvm::AtomicOrdering::Monotonic:
5118 break;
5119 case llvm::AtomicOrdering::NotAtomic:
5120 case llvm::AtomicOrdering::Unordered:
5121 llvm_unreachable("Unexpected ordering.");
5122 }
Alexey Bataevb4505a72015-03-30 05:20:59 +00005123}
5124
Alexey Bataevddf3db92018-04-13 17:31:06 +00005125static void emitOMPAtomicExpr(CodeGenFunction &CGF, OpenMPClauseKind Kind,
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005126 llvm::AtomicOrdering AO, bool IsPostfixUpdate,
Alexey Bataev5e018f92015-04-23 06:35:10 +00005127 const Expr *X, const Expr *V, const Expr *E,
5128 const Expr *UE, bool IsXLHSInRHSPart,
5129 SourceLocation Loc) {
Alexey Bataevb57056f2015-01-22 06:17:56 +00005130 switch (Kind) {
5131 case OMPC_read:
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005132 emitOMPAtomicReadExpr(CGF, AO, X, V, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00005133 break;
5134 case OMPC_write:
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005135 emitOMPAtomicWriteExpr(CGF, AO, X, E, Loc);
Alexey Bataevb8329262015-02-27 06:33:30 +00005136 break;
Alexey Bataevb4505a72015-03-30 05:20:59 +00005137 case OMPC_unknown:
Alexey Bataevb57056f2015-01-22 06:17:56 +00005138 case OMPC_update:
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005139 emitOMPAtomicUpdateExpr(CGF, AO, X, E, UE, IsXLHSInRHSPart, Loc);
Alexey Bataevb4505a72015-03-30 05:20:59 +00005140 break;
Alexey Bataevb57056f2015-01-22 06:17:56 +00005141 case OMPC_capture:
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005142 emitOMPAtomicCaptureExpr(CGF, AO, IsPostfixUpdate, V, X, E, UE,
Alexey Bataev5e018f92015-04-23 06:35:10 +00005143 IsXLHSInRHSPart, Loc);
5144 break;
Alexey Bataevb57056f2015-01-22 06:17:56 +00005145 case OMPC_if:
5146 case OMPC_final:
5147 case OMPC_num_threads:
5148 case OMPC_private:
5149 case OMPC_firstprivate:
5150 case OMPC_lastprivate:
5151 case OMPC_reduction:
Alexey Bataev169d96a2017-07-18 20:17:46 +00005152 case OMPC_task_reduction:
Alexey Bataevfa312f32017-07-21 18:48:21 +00005153 case OMPC_in_reduction:
Alexey Bataevb57056f2015-01-22 06:17:56 +00005154 case OMPC_safelen:
Alexey Bataev66b15b52015-08-21 11:14:16 +00005155 case OMPC_simdlen:
Alexey Bataev9cc10fc2019-03-12 18:52:33 +00005156 case OMPC_allocator:
Alexey Bataeve04483e2019-03-27 14:14:31 +00005157 case OMPC_allocate:
Alexey Bataevb57056f2015-01-22 06:17:56 +00005158 case OMPC_collapse:
5159 case OMPC_default:
5160 case OMPC_seq_cst:
Alexey Bataevea9166b2020-02-06 16:30:23 -05005161 case OMPC_acq_rel:
Alexey Bataev04a830f2020-02-10 14:30:39 -05005162 case OMPC_acquire:
Alexey Bataev95598342020-02-10 15:49:05 -05005163 case OMPC_release:
Alexey Bataev9a8defc2020-02-11 11:10:43 -05005164 case OMPC_relaxed:
Alexey Bataevb57056f2015-01-22 06:17:56 +00005165 case OMPC_shared:
5166 case OMPC_linear:
5167 case OMPC_aligned:
5168 case OMPC_copyin:
5169 case OMPC_copyprivate:
5170 case OMPC_flush:
Alexey Bataevc112e942020-02-28 09:52:15 -05005171 case OMPC_depobj:
Alexey Bataevb57056f2015-01-22 06:17:56 +00005172 case OMPC_proc_bind:
5173 case OMPC_schedule:
5174 case OMPC_ordered:
5175 case OMPC_nowait:
5176 case OMPC_untied:
5177 case OMPC_threadprivate:
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +00005178 case OMPC_depend:
Alexey Bataevb57056f2015-01-22 06:17:56 +00005179 case OMPC_mergeable:
Michael Wonge710d542015-08-07 16:16:36 +00005180 case OMPC_device:
Alexey Bataev346265e2015-09-25 10:37:12 +00005181 case OMPC_threads:
Alexey Bataevd14d1e62015-09-28 06:39:35 +00005182 case OMPC_simd:
Kelvin Li0bff7af2015-11-23 05:32:03 +00005183 case OMPC_map:
Kelvin Li099bb8c2015-11-24 20:50:12 +00005184 case OMPC_num_teams:
Kelvin Lia15fb1a2015-11-27 18:47:36 +00005185 case OMPC_thread_limit:
Alexey Bataeva0569352015-12-01 10:17:31 +00005186 case OMPC_priority:
Alexey Bataev1fd4aed2015-12-07 12:52:51 +00005187 case OMPC_grainsize:
Alexey Bataevb825de12015-12-07 10:51:44 +00005188 case OMPC_nogroup:
Alexey Bataev382967a2015-12-08 12:06:20 +00005189 case OMPC_num_tasks:
Alexey Bataev28c75412015-12-15 08:19:24 +00005190 case OMPC_hint:
Carlo Bertollib4adf552016-01-15 18:50:31 +00005191 case OMPC_dist_schedule:
Arpith Chacko Jacob3cf89042016-01-26 16:37:23 +00005192 case OMPC_defaultmap:
Alexey Bataeve48a5fc2016-04-12 05:28:34 +00005193 case OMPC_uniform:
Samuel Antao661c0902016-05-26 17:39:58 +00005194 case OMPC_to:
Samuel Antaoec172c62016-05-26 17:49:04 +00005195 case OMPC_from:
Carlo Bertolli2404b172016-07-13 15:37:16 +00005196 case OMPC_use_device_ptr:
Alexey Bataeva888fc62020-05-21 08:30:23 -04005197 case OMPC_use_device_addr:
Carlo Bertolli70594e92016-07-13 17:16:49 +00005198 case OMPC_is_device_ptr:
Kelvin Li1408f912018-09-26 04:28:39 +00005199 case OMPC_unified_address:
Alexey Bataev94c50642018-10-01 14:26:31 +00005200 case OMPC_unified_shared_memory:
Patrick Lyster6bdf63b2018-10-03 20:07:58 +00005201 case OMPC_reverse_offload:
Patrick Lyster3fe9e392018-10-11 14:41:10 +00005202 case OMPC_dynamic_allocators:
Patrick Lyster7a2a27c2018-11-02 12:18:11 +00005203 case OMPC_atomic_default_mem_order:
Alexey Bataev729e2422019-08-23 16:11:14 +00005204 case OMPC_device_type:
Alexey Bataevdba792c2019-09-23 18:13:31 +00005205 case OMPC_match:
Alexey Bataevb6e70842019-12-16 15:54:17 -05005206 case OMPC_nontemporal:
Alexey Bataevcb8e6912020-01-31 16:09:26 -05005207 case OMPC_order:
Alexey Bataev375437a2020-03-02 14:21:20 -05005208 case OMPC_destroy:
Alexey Bataev0f0564b2020-03-17 09:17:42 -04005209 case OMPC_detach:
Alexey Bataev06dea732020-03-20 09:41:22 -04005210 case OMPC_inclusive:
Alexey Bataev63828a32020-03-23 10:41:08 -04005211 case OMPC_exclusive:
Alexey Bataevb5be1c52020-04-21 13:21:00 -04005212 case OMPC_uses_allocators:
Alexey Bataev2e499ee2020-05-18 13:37:53 -04005213 case OMPC_affinity:
Alexey Bataevb57056f2015-01-22 06:17:56 +00005214 llvm_unreachable("Clause is not allowed in 'omp atomic'.");
5215 }
5216}
5217
5218void CodeGenFunction::EmitOMPAtomicDirective(const OMPAtomicDirective &S) {
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005219 llvm::AtomicOrdering AO = llvm::AtomicOrdering::Monotonic;
Alexey Bataev2d4f80f2020-02-11 15:15:21 -05005220 bool MemOrderingSpecified = false;
5221 if (S.getSingleClause<OMPSeqCstClause>()) {
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005222 AO = llvm::AtomicOrdering::SequentiallyConsistent;
Alexey Bataev2d4f80f2020-02-11 15:15:21 -05005223 MemOrderingSpecified = true;
5224 } else if (S.getSingleClause<OMPAcqRelClause>()) {
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005225 AO = llvm::AtomicOrdering::AcquireRelease;
Alexey Bataev2d4f80f2020-02-11 15:15:21 -05005226 MemOrderingSpecified = true;
5227 } else if (S.getSingleClause<OMPAcquireClause>()) {
Alexey Bataev04a830f2020-02-10 14:30:39 -05005228 AO = llvm::AtomicOrdering::Acquire;
Alexey Bataev2d4f80f2020-02-11 15:15:21 -05005229 MemOrderingSpecified = true;
5230 } else if (S.getSingleClause<OMPReleaseClause>()) {
Alexey Bataev95598342020-02-10 15:49:05 -05005231 AO = llvm::AtomicOrdering::Release;
Alexey Bataev2d4f80f2020-02-11 15:15:21 -05005232 MemOrderingSpecified = true;
5233 } else if (S.getSingleClause<OMPRelaxedClause>()) {
Alexey Bataev9a8defc2020-02-11 11:10:43 -05005234 AO = llvm::AtomicOrdering::Monotonic;
Alexey Bataev2d4f80f2020-02-11 15:15:21 -05005235 MemOrderingSpecified = true;
5236 }
Alexey Bataevb57056f2015-01-22 06:17:56 +00005237 OpenMPClauseKind Kind = OMPC_unknown;
Alexey Bataevddf3db92018-04-13 17:31:06 +00005238 for (const OMPClause *C : S.clauses()) {
Alexey Bataev9a8defc2020-02-11 11:10:43 -05005239 // Find first clause (skip seq_cst|acq_rel|aqcuire|release|relaxed clause,
5240 // if it is first).
Alexey Bataevea9166b2020-02-06 16:30:23 -05005241 if (C->getClauseKind() != OMPC_seq_cst &&
Alexey Bataev04a830f2020-02-10 14:30:39 -05005242 C->getClauseKind() != OMPC_acq_rel &&
Alexey Bataev95598342020-02-10 15:49:05 -05005243 C->getClauseKind() != OMPC_acquire &&
Alexey Bataev9a8defc2020-02-11 11:10:43 -05005244 C->getClauseKind() != OMPC_release &&
5245 C->getClauseKind() != OMPC_relaxed) {
Alexey Bataevb57056f2015-01-22 06:17:56 +00005246 Kind = C->getClauseKind();
5247 break;
5248 }
5249 }
Alexey Bataev2d4f80f2020-02-11 15:15:21 -05005250 if (!MemOrderingSpecified) {
5251 llvm::AtomicOrdering DefaultOrder =
5252 CGM.getOpenMPRuntime().getDefaultMemoryOrdering();
5253 if (DefaultOrder == llvm::AtomicOrdering::Monotonic ||
5254 DefaultOrder == llvm::AtomicOrdering::SequentiallyConsistent ||
5255 (DefaultOrder == llvm::AtomicOrdering::AcquireRelease &&
5256 Kind == OMPC_capture)) {
5257 AO = DefaultOrder;
5258 } else if (DefaultOrder == llvm::AtomicOrdering::AcquireRelease) {
5259 if (Kind == OMPC_unknown || Kind == OMPC_update || Kind == OMPC_write) {
5260 AO = llvm::AtomicOrdering::Release;
5261 } else if (Kind == OMPC_read) {
5262 assert(Kind == OMPC_read && "Unexpected atomic kind.");
5263 AO = llvm::AtomicOrdering::Acquire;
5264 }
5265 }
5266 }
Alexey Bataev10fec572015-03-11 04:48:56 +00005267
Alexey Bataevddf3db92018-04-13 17:31:06 +00005268 const Stmt *CS = S.getInnermostCapturedStmt()->IgnoreContainers();
Alexey Bataev10fec572015-03-11 04:48:56 +00005269
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005270 auto &&CodeGen = [&S, Kind, AO, CS](CodeGenFunction &CGF,
Alexey Bataev14fa1c62016-03-29 05:34:15 +00005271 PrePostActionTy &) {
Alexey Bataev33c56402015-12-14 09:26:19 +00005272 CGF.EmitStopPoint(CS);
Alexey Bataeve8e05de2020-02-07 12:22:23 -05005273 emitOMPAtomicExpr(CGF, Kind, AO, S.isPostfixUpdate(), S.getX(), S.getV(),
5274 S.getExpr(), S.getUpdateExpr(), S.isXLHSInRHSPart(),
5275 S.getBeginLoc());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00005276 };
Alexey Bataev475a7442018-01-12 19:39:11 +00005277 OMPLexicalScope Scope(*this, S, OMPD_unknown);
Alexey Bataev81c7ea02015-07-03 09:56:58 +00005278 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_atomic, CodeGen);
Alexey Bataev0162e452014-07-22 10:10:35 +00005279}
5280
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005281static void emitCommonOMPTargetDirective(CodeGenFunction &CGF,
5282 const OMPExecutableDirective &S,
5283 const RegionCodeGenTy &CodeGen) {
5284 assert(isOpenMPTargetExecutionDirective(S.getDirectiveKind()));
5285 CodeGenModule &CGM = CGF.CGM;
Samuel Antaobed3c462015-10-02 16:14:20 +00005286
Alexey Bataev4f4bf7c2018-03-15 15:47:20 +00005287 // On device emit this construct as inlined code.
5288 if (CGM.getLangOpts().OpenMPIsDevice) {
5289 OMPLexicalScope Scope(CGF, S, OMPD_target);
5290 CGM.getOpenMPRuntime().emitInlinedDirective(
5291 CGF, OMPD_target, [&S](CodeGenFunction &CGF, PrePostActionTy &) {
Alexey Bataev4ac68a22018-05-16 15:08:32 +00005292 CGF.EmitStmt(S.getInnermostCapturedStmt()->getCapturedStmt());
Alexey Bataev4f4bf7c2018-03-15 15:47:20 +00005293 });
5294 return;
5295 }
5296
Alexey Bataev46978742020-01-30 10:46:11 -05005297 auto LPCRegion =
5298 CGOpenMPRuntime::LastprivateConditionalRAII::disable(CGF, S);
Samuel Antaoee8fb302016-01-06 13:42:12 +00005299 llvm::Function *Fn = nullptr;
5300 llvm::Constant *FnID = nullptr;
Samuel Antaobed3c462015-10-02 16:14:20 +00005301
Samuel Antaobed3c462015-10-02 16:14:20 +00005302 const Expr *IfCond = nullptr;
Arpith Chacko Jacobfe4890a2017-01-18 20:40:48 +00005303 // Check for the at most one if clause associated with the target region.
5304 for (const auto *C : S.getClausesOfKind<OMPIfClause>()) {
5305 if (C->getNameModifier() == OMPD_unknown ||
5306 C->getNameModifier() == OMPD_target) {
5307 IfCond = C->getCondition();
5308 break;
5309 }
Samuel Antaobed3c462015-10-02 16:14:20 +00005310 }
5311
5312 // Check if we have any device clause associated with the directive.
Alexey Bataevf3c857f2020-03-18 17:52:41 -04005313 llvm::PointerIntPair<const Expr *, 2, OpenMPDeviceClauseModifier> Device(
5314 nullptr, OMPC_DEVICE_unknown);
5315 if (auto *C = S.getSingleClause<OMPDeviceClause>())
5316 Device.setPointerAndInt(C->getDevice(), C->getModifier());
Samuel Antaobed3c462015-10-02 16:14:20 +00005317
Samuel Antaoee8fb302016-01-06 13:42:12 +00005318 // Check if we have an if clause whose conditional always evaluates to false
5319 // or if we do not have any targets specified. If so the target region is not
5320 // an offload entry point.
5321 bool IsOffloadEntry = true;
5322 if (IfCond) {
5323 bool Val;
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005324 if (CGF.ConstantFoldsToSimpleInteger(IfCond, Val) && !Val)
Samuel Antaoee8fb302016-01-06 13:42:12 +00005325 IsOffloadEntry = false;
5326 }
5327 if (CGM.getLangOpts().OMPTargetTriples.empty())
5328 IsOffloadEntry = false;
5329
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005330 assert(CGF.CurFuncDecl && "No parent declaration for target region!");
Samuel Antaoee8fb302016-01-06 13:42:12 +00005331 StringRef ParentName;
5332 // In case we have Ctors/Dtors we use the complete type variant to produce
5333 // the mangling of the device outlined kernel.
Alexey Bataevddf3db92018-04-13 17:31:06 +00005334 if (const auto *D = dyn_cast<CXXConstructorDecl>(CGF.CurFuncDecl))
Samuel Antaoee8fb302016-01-06 13:42:12 +00005335 ParentName = CGM.getMangledName(GlobalDecl(D, Ctor_Complete));
Alexey Bataevddf3db92018-04-13 17:31:06 +00005336 else if (const auto *D = dyn_cast<CXXDestructorDecl>(CGF.CurFuncDecl))
Samuel Antaoee8fb302016-01-06 13:42:12 +00005337 ParentName = CGM.getMangledName(GlobalDecl(D, Dtor_Complete));
5338 else
5339 ParentName =
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005340 CGM.getMangledName(GlobalDecl(cast<FunctionDecl>(CGF.CurFuncDecl)));
Samuel Antaoee8fb302016-01-06 13:42:12 +00005341
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005342 // Emit target region as a standalone region.
5343 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(S, ParentName, Fn, FnID,
5344 IsOffloadEntry, CodeGen);
Alexey Bataev8451efa2018-01-15 19:06:12 +00005345 OMPLexicalScope Scope(CGF, S, OMPD_task);
Alexey Bataevec7946e2019-09-23 14:06:51 +00005346 auto &&SizeEmitter =
5347 [IsOffloadEntry](CodeGenFunction &CGF,
5348 const OMPLoopDirective &D) -> llvm::Value * {
5349 if (IsOffloadEntry) {
5350 OMPLoopScope(CGF, D);
5351 // Emit calculation of the iterations count.
5352 llvm::Value *NumIterations = CGF.EmitScalarExpr(D.getNumIterations());
5353 NumIterations = CGF.Builder.CreateIntCast(NumIterations, CGF.Int64Ty,
5354 /*isSigned=*/false);
5355 return NumIterations;
5356 }
5357 return nullptr;
Alexey Bataev7bb33532019-01-07 21:30:43 +00005358 };
Alexey Bataevec7946e2019-09-23 14:06:51 +00005359 CGM.getOpenMPRuntime().emitTargetCall(CGF, S, Fn, FnID, IfCond, Device,
5360 SizeEmitter);
Alexey Bataev0bd520b2014-09-19 08:19:49 +00005361}
5362
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005363static void emitTargetRegion(CodeGenFunction &CGF, const OMPTargetDirective &S,
5364 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005365 Action.Enter(CGF);
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005366 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
5367 (void)CGF.EmitOMPFirstprivateClause(S, PrivateScope);
5368 CGF.EmitOMPPrivateClause(S, PrivateScope);
5369 (void)PrivateScope.Privatize();
Alexey Bataev60705422018-10-30 15:50:12 +00005370 if (isOpenMPTargetExecutionDirective(S.getDirectiveKind()))
5371 CGF.CGM.getOpenMPRuntime().adjustTargetSpecificDataForLambdas(CGF, S);
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005372
Alexey Bataev475a7442018-01-12 19:39:11 +00005373 CGF.EmitStmt(S.getCapturedStmt(OMPD_target)->getCapturedStmt());
Arpith Chacko Jacob43a8b7b2017-01-16 15:26:02 +00005374}
5375
5376void CodeGenFunction::EmitOMPTargetDeviceFunction(CodeGenModule &CGM,
5377 StringRef ParentName,
5378 const OMPTargetDirective &S) {
5379 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5380 emitTargetRegion(CGF, S, Action);
5381 };
5382 llvm::Function *Fn;
5383 llvm::Constant *Addr;
5384 // Emit target region as a standalone region.
5385 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
5386 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
5387 assert(Fn && Addr && "Target device function emission failed.");
5388}
5389
5390void CodeGenFunction::EmitOMPTargetDirective(const OMPTargetDirective &S) {
5391 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5392 emitTargetRegion(CGF, S, Action);
5393 };
5394 emitCommonOMPTargetDirective(*this, S, CodeGen);
5395}
5396
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005397static void emitCommonOMPTeamsDirective(CodeGenFunction &CGF,
5398 const OMPExecutableDirective &S,
5399 OpenMPDirectiveKind InnermostKind,
5400 const RegionCodeGenTy &CodeGen) {
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00005401 const CapturedStmt *CS = S.getCapturedStmt(OMPD_teams);
James Y Knight9871db02019-02-05 16:42:33 +00005402 llvm::Function *OutlinedFn =
Alexey Bataevddf3db92018-04-13 17:31:06 +00005403 CGF.CGM.getOpenMPRuntime().emitTeamsOutlinedFunction(
5404 S, *CS->getCapturedDecl()->param_begin(), InnermostKind, CodeGen);
Samuel Antaob68e2db2016-03-03 16:20:23 +00005405
Alexey Bataevddf3db92018-04-13 17:31:06 +00005406 const auto *NT = S.getSingleClause<OMPNumTeamsClause>();
5407 const auto *TL = S.getSingleClause<OMPThreadLimitClause>();
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005408 if (NT || TL) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00005409 const Expr *NumTeams = NT ? NT->getNumTeams() : nullptr;
5410 const Expr *ThreadLimit = TL ? TL->getThreadLimit() : nullptr;
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005411
Carlo Bertollic6872252016-04-04 15:55:02 +00005412 CGF.CGM.getOpenMPRuntime().emitNumTeamsClause(CGF, NumTeams, ThreadLimit,
Stephen Kellyf2ceec42018-08-09 21:08:08 +00005413 S.getBeginLoc());
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005414 }
5415
Arpith Chacko Jacob99a1e0e2017-01-25 02:18:43 +00005416 OMPTeamsScope Scope(CGF, S);
Alexey Bataev14fa1c62016-03-29 05:34:15 +00005417 llvm::SmallVector<llvm::Value *, 16> CapturedVars;
5418 CGF.GenerateOpenMPCapturedVars(*CS, CapturedVars);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00005419 CGF.CGM.getOpenMPRuntime().emitTeamsCall(CGF, S, S.getBeginLoc(), OutlinedFn,
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005420 CapturedVars);
5421}
5422
5423void CodeGenFunction::EmitOMPTeamsDirective(const OMPTeamsDirective &S) {
Kelvin Li51336dd2016-12-15 17:55:32 +00005424 // Emit teams region as a standalone region.
Alexey Bataevc99042b2018-03-15 18:10:54 +00005425 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005426 Action.Enter(CGF);
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005427 OMPPrivateScope PrivateScope(CGF);
Carlo Bertolli6ad7b5a2016-03-03 22:09:40 +00005428 (void)CGF.EmitOMPFirstprivateClause(S, PrivateScope);
5429 CGF.EmitOMPPrivateClause(S, PrivateScope);
Arpith Chacko Jacobfc711b12017-02-16 16:48:49 +00005430 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005431 (void)PrivateScope.Privatize();
Alexey Bataev475a7442018-01-12 19:39:11 +00005432 CGF.EmitStmt(S.getCapturedStmt(OMPD_teams)->getCapturedStmt());
Arpith Chacko Jacobfc711b12017-02-16 16:48:49 +00005433 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
Carlo Bertolli430d8ec2016-03-03 20:34:23 +00005434 };
Alexey Bataev2139ed62017-11-16 18:20:21 +00005435 emitCommonOMPTeamsDirective(*this, S, OMPD_distribute, CodeGen);
Alexey Bataevddf3db92018-04-13 17:31:06 +00005436 emitPostUpdateForReductionClause(*this, S,
5437 [](CodeGenFunction &) { return nullptr; });
Alexey Bataev13314bf2014-10-09 04:18:56 +00005438}
Alexey Bataev6d4ed052015-07-01 06:57:41 +00005439
Arpith Chacko Jacob99a1e0e2017-01-25 02:18:43 +00005440static void emitTargetTeamsRegion(CodeGenFunction &CGF, PrePostActionTy &Action,
5441 const OMPTargetTeamsDirective &S) {
5442 auto *CS = S.getCapturedStmt(OMPD_teams);
5443 Action.Enter(CGF);
Alexey Bataevf9fc42e2017-11-22 14:25:55 +00005444 // Emit teams region as a standalone region.
5445 auto &&CodeGen = [&S, CS](CodeGenFunction &CGF, PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005446 Action.Enter(CGF);
Alexey Bataevf9fc42e2017-11-22 14:25:55 +00005447 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
5448 (void)CGF.EmitOMPFirstprivateClause(S, PrivateScope);
5449 CGF.EmitOMPPrivateClause(S, PrivateScope);
5450 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5451 (void)PrivateScope.Privatize();
Alexey Bataev60705422018-10-30 15:50:12 +00005452 if (isOpenMPTargetExecutionDirective(S.getDirectiveKind()))
5453 CGF.CGM.getOpenMPRuntime().adjustTargetSpecificDataForLambdas(CGF, S);
Arpith Chacko Jacob99a1e0e2017-01-25 02:18:43 +00005454 CGF.EmitStmt(CS->getCapturedStmt());
Alexey Bataevf9fc42e2017-11-22 14:25:55 +00005455 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
Arpith Chacko Jacob99a1e0e2017-01-25 02:18:43 +00005456 };
5457 emitCommonOMPTeamsDirective(CGF, S, OMPD_teams, CodeGen);
Alexey Bataevddf3db92018-04-13 17:31:06 +00005458 emitPostUpdateForReductionClause(CGF, S,
5459 [](CodeGenFunction &) { return nullptr; });
Arpith Chacko Jacob99a1e0e2017-01-25 02:18:43 +00005460}
5461
5462void CodeGenFunction::EmitOMPTargetTeamsDeviceFunction(
5463 CodeGenModule &CGM, StringRef ParentName,
5464 const OMPTargetTeamsDirective &S) {
5465 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5466 emitTargetTeamsRegion(CGF, Action, S);
5467 };
5468 llvm::Function *Fn;
5469 llvm::Constant *Addr;
5470 // Emit target region as a standalone region.
5471 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
5472 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
5473 assert(Fn && Addr && "Target device function emission failed.");
5474}
5475
5476void CodeGenFunction::EmitOMPTargetTeamsDirective(
5477 const OMPTargetTeamsDirective &S) {
5478 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5479 emitTargetTeamsRegion(CGF, Action, S);
5480 };
5481 emitCommonOMPTargetDirective(*this, S, CodeGen);
5482}
5483
Alexey Bataevdfa430f2017-12-08 15:03:50 +00005484static void
5485emitTargetTeamsDistributeRegion(CodeGenFunction &CGF, PrePostActionTy &Action,
5486 const OMPTargetTeamsDistributeDirective &S) {
5487 Action.Enter(CGF);
5488 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5489 CGF.EmitOMPDistributeLoop(S, emitOMPLoopBodyWithStopPoint, S.getInc());
5490 };
5491
5492 // Emit teams region as a standalone region.
5493 auto &&CodeGen = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005494 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005495 Action.Enter(CGF);
Alexey Bataevdfa430f2017-12-08 15:03:50 +00005496 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
5497 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5498 (void)PrivateScope.Privatize();
5499 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, OMPD_distribute,
5500 CodeGenDistribute);
5501 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5502 };
5503 emitCommonOMPTeamsDirective(CGF, S, OMPD_distribute, CodeGen);
5504 emitPostUpdateForReductionClause(CGF, S,
5505 [](CodeGenFunction &) { return nullptr; });
5506}
5507
5508void CodeGenFunction::EmitOMPTargetTeamsDistributeDeviceFunction(
5509 CodeGenModule &CGM, StringRef ParentName,
5510 const OMPTargetTeamsDistributeDirective &S) {
5511 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5512 emitTargetTeamsDistributeRegion(CGF, Action, S);
5513 };
5514 llvm::Function *Fn;
5515 llvm::Constant *Addr;
5516 // Emit target region as a standalone region.
5517 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
5518 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
5519 assert(Fn && Addr && "Target device function emission failed.");
5520}
5521
5522void CodeGenFunction::EmitOMPTargetTeamsDistributeDirective(
5523 const OMPTargetTeamsDistributeDirective &S) {
5524 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5525 emitTargetTeamsDistributeRegion(CGF, Action, S);
5526 };
5527 emitCommonOMPTargetDirective(*this, S, CodeGen);
5528}
5529
Alexey Bataevfbe17fb2017-12-13 19:45:06 +00005530static void emitTargetTeamsDistributeSimdRegion(
5531 CodeGenFunction &CGF, PrePostActionTy &Action,
5532 const OMPTargetTeamsDistributeSimdDirective &S) {
5533 Action.Enter(CGF);
5534 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5535 CGF.EmitOMPDistributeLoop(S, emitOMPLoopBodyWithStopPoint, S.getInc());
5536 };
5537
5538 // Emit teams region as a standalone region.
5539 auto &&CodeGen = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005540 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005541 Action.Enter(CGF);
Alexey Bataevfbe17fb2017-12-13 19:45:06 +00005542 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
5543 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5544 (void)PrivateScope.Privatize();
5545 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, OMPD_distribute,
5546 CodeGenDistribute);
5547 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5548 };
5549 emitCommonOMPTeamsDirective(CGF, S, OMPD_distribute_simd, CodeGen);
5550 emitPostUpdateForReductionClause(CGF, S,
5551 [](CodeGenFunction &) { return nullptr; });
5552}
5553
5554void CodeGenFunction::EmitOMPTargetTeamsDistributeSimdDeviceFunction(
5555 CodeGenModule &CGM, StringRef ParentName,
5556 const OMPTargetTeamsDistributeSimdDirective &S) {
5557 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5558 emitTargetTeamsDistributeSimdRegion(CGF, Action, S);
5559 };
5560 llvm::Function *Fn;
5561 llvm::Constant *Addr;
5562 // Emit target region as a standalone region.
5563 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
5564 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
5565 assert(Fn && Addr && "Target device function emission failed.");
5566}
5567
5568void CodeGenFunction::EmitOMPTargetTeamsDistributeSimdDirective(
5569 const OMPTargetTeamsDistributeSimdDirective &S) {
5570 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5571 emitTargetTeamsDistributeSimdRegion(CGF, Action, S);
5572 };
5573 emitCommonOMPTargetDirective(*this, S, CodeGen);
5574}
5575
Carlo Bertolliba1487b2017-10-04 14:12:09 +00005576void CodeGenFunction::EmitOMPTeamsDistributeDirective(
5577 const OMPTeamsDistributeDirective &S) {
5578
5579 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5580 CGF.EmitOMPDistributeLoop(S, emitOMPLoopBodyWithStopPoint, S.getInc());
5581 };
5582
5583 // Emit teams region as a standalone region.
5584 auto &&CodeGen = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005585 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005586 Action.Enter(CGF);
Carlo Bertolliba1487b2017-10-04 14:12:09 +00005587 OMPPrivateScope PrivateScope(CGF);
5588 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5589 (void)PrivateScope.Privatize();
5590 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, OMPD_distribute,
5591 CodeGenDistribute);
5592 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5593 };
Alexey Bataev95c6dd42017-11-29 15:14:16 +00005594 emitCommonOMPTeamsDirective(*this, S, OMPD_distribute, CodeGen);
Carlo Bertolliba1487b2017-10-04 14:12:09 +00005595 emitPostUpdateForReductionClause(*this, S,
5596 [](CodeGenFunction &) { return nullptr; });
5597}
5598
Alexey Bataev999277a2017-12-06 14:31:09 +00005599void CodeGenFunction::EmitOMPTeamsDistributeSimdDirective(
5600 const OMPTeamsDistributeSimdDirective &S) {
5601 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5602 CGF.EmitOMPDistributeLoop(S, emitOMPLoopBodyWithStopPoint, S.getInc());
5603 };
5604
5605 // Emit teams region as a standalone region.
5606 auto &&CodeGen = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005607 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005608 Action.Enter(CGF);
Alexey Bataev999277a2017-12-06 14:31:09 +00005609 OMPPrivateScope PrivateScope(CGF);
5610 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5611 (void)PrivateScope.Privatize();
5612 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, OMPD_simd,
5613 CodeGenDistribute);
5614 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5615 };
5616 emitCommonOMPTeamsDirective(*this, S, OMPD_distribute_simd, CodeGen);
5617 emitPostUpdateForReductionClause(*this, S,
5618 [](CodeGenFunction &) { return nullptr; });
5619}
5620
Carlo Bertolli62fae152017-11-20 20:46:39 +00005621void CodeGenFunction::EmitOMPTeamsDistributeParallelForDirective(
5622 const OMPTeamsDistributeParallelForDirective &S) {
5623 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5624 CGF.EmitOMPDistributeLoop(S, emitInnerParallelForWhenCombined,
5625 S.getDistInc());
5626 };
5627
5628 // Emit teams region as a standalone region.
5629 auto &&CodeGen = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005630 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005631 Action.Enter(CGF);
Carlo Bertolli62fae152017-11-20 20:46:39 +00005632 OMPPrivateScope PrivateScope(CGF);
5633 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5634 (void)PrivateScope.Privatize();
Alexey Bataev10a54312017-11-27 16:54:08 +00005635 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, OMPD_distribute,
5636 CodeGenDistribute);
Carlo Bertolli62fae152017-11-20 20:46:39 +00005637 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5638 };
5639 emitCommonOMPTeamsDirective(*this, S, OMPD_distribute_parallel_for, CodeGen);
5640 emitPostUpdateForReductionClause(*this, S,
5641 [](CodeGenFunction &) { return nullptr; });
5642}
5643
Carlo Bertolli56a2aa42017-12-04 20:57:19 +00005644void CodeGenFunction::EmitOMPTeamsDistributeParallelForSimdDirective(
5645 const OMPTeamsDistributeParallelForSimdDirective &S) {
5646 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5647 CGF.EmitOMPDistributeLoop(S, emitInnerParallelForWhenCombined,
5648 S.getDistInc());
5649 };
5650
5651 // Emit teams region as a standalone region.
5652 auto &&CodeGen = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005653 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005654 Action.Enter(CGF);
Carlo Bertolli56a2aa42017-12-04 20:57:19 +00005655 OMPPrivateScope PrivateScope(CGF);
5656 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5657 (void)PrivateScope.Privatize();
5658 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(
5659 CGF, OMPD_distribute, CodeGenDistribute, /*HasCancel=*/false);
5660 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5661 };
Alexey Bataev0b978942019-12-11 15:26:38 -05005662 emitCommonOMPTeamsDirective(*this, S, OMPD_distribute_parallel_for_simd,
5663 CodeGen);
Carlo Bertolli56a2aa42017-12-04 20:57:19 +00005664 emitPostUpdateForReductionClause(*this, S,
5665 [](CodeGenFunction &) { return nullptr; });
5666}
5667
Carlo Bertolli52978c32018-01-03 21:12:44 +00005668static void emitTargetTeamsDistributeParallelForRegion(
5669 CodeGenFunction &CGF, const OMPTargetTeamsDistributeParallelForDirective &S,
5670 PrePostActionTy &Action) {
Carlo Bertolli79712092018-02-28 20:48:35 +00005671 Action.Enter(CGF);
Carlo Bertolli52978c32018-01-03 21:12:44 +00005672 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5673 CGF.EmitOMPDistributeLoop(S, emitInnerParallelForWhenCombined,
5674 S.getDistInc());
5675 };
5676
5677 // Emit teams region as a standalone region.
5678 auto &&CodeGenTeams = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005679 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005680 Action.Enter(CGF);
Carlo Bertolli52978c32018-01-03 21:12:44 +00005681 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
5682 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5683 (void)PrivateScope.Privatize();
5684 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(
5685 CGF, OMPD_distribute, CodeGenDistribute, /*HasCancel=*/false);
5686 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5687 };
5688
5689 emitCommonOMPTeamsDirective(CGF, S, OMPD_distribute_parallel_for,
5690 CodeGenTeams);
5691 emitPostUpdateForReductionClause(CGF, S,
5692 [](CodeGenFunction &) { return nullptr; });
5693}
5694
5695void CodeGenFunction::EmitOMPTargetTeamsDistributeParallelForDeviceFunction(
5696 CodeGenModule &CGM, StringRef ParentName,
5697 const OMPTargetTeamsDistributeParallelForDirective &S) {
5698 // Emit SPMD target teams distribute parallel for region as a standalone
5699 // region.
5700 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5701 emitTargetTeamsDistributeParallelForRegion(CGF, S, Action);
5702 };
5703 llvm::Function *Fn;
5704 llvm::Constant *Addr;
5705 // Emit target region as a standalone region.
5706 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
5707 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
5708 assert(Fn && Addr && "Target device function emission failed.");
5709}
5710
5711void CodeGenFunction::EmitOMPTargetTeamsDistributeParallelForDirective(
5712 const OMPTargetTeamsDistributeParallelForDirective &S) {
5713 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5714 emitTargetTeamsDistributeParallelForRegion(CGF, S, Action);
5715 };
5716 emitCommonOMPTargetDirective(*this, S, CodeGen);
5717}
5718
Alexey Bataev647dd842018-01-15 20:59:40 +00005719static void emitTargetTeamsDistributeParallelForSimdRegion(
5720 CodeGenFunction &CGF,
5721 const OMPTargetTeamsDistributeParallelForSimdDirective &S,
5722 PrePostActionTy &Action) {
Carlo Bertolli79712092018-02-28 20:48:35 +00005723 Action.Enter(CGF);
Alexey Bataev647dd842018-01-15 20:59:40 +00005724 auto &&CodeGenDistribute = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
5725 CGF.EmitOMPDistributeLoop(S, emitInnerParallelForWhenCombined,
5726 S.getDistInc());
5727 };
5728
5729 // Emit teams region as a standalone region.
5730 auto &&CodeGenTeams = [&S, &CodeGenDistribute](CodeGenFunction &CGF,
Alexey Bataevc99042b2018-03-15 18:10:54 +00005731 PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00005732 Action.Enter(CGF);
Alexey Bataev647dd842018-01-15 20:59:40 +00005733 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
5734 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
5735 (void)PrivateScope.Privatize();
5736 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(
5737 CGF, OMPD_distribute, CodeGenDistribute, /*HasCancel=*/false);
5738 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_teams);
5739 };
5740
5741 emitCommonOMPTeamsDirective(CGF, S, OMPD_distribute_parallel_for_simd,
5742 CodeGenTeams);
5743 emitPostUpdateForReductionClause(CGF, S,
5744 [](CodeGenFunction &) { return nullptr; });
5745}
5746
5747void CodeGenFunction::EmitOMPTargetTeamsDistributeParallelForSimdDeviceFunction(
5748 CodeGenModule &CGM, StringRef ParentName,
5749 const OMPTargetTeamsDistributeParallelForSimdDirective &S) {
5750 // Emit SPMD target teams distribute parallel for simd region as a standalone
5751 // region.
5752 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5753 emitTargetTeamsDistributeParallelForSimdRegion(CGF, S, Action);
5754 };
5755 llvm::Function *Fn;
5756 llvm::Constant *Addr;
5757 // Emit target region as a standalone region.
5758 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
5759 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
5760 assert(Fn && Addr && "Target device function emission failed.");
5761}
5762
5763void CodeGenFunction::EmitOMPTargetTeamsDistributeParallelForSimdDirective(
5764 const OMPTargetTeamsDistributeParallelForSimdDirective &S) {
5765 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
5766 emitTargetTeamsDistributeParallelForSimdRegion(CGF, S, Action);
5767 };
5768 emitCommonOMPTargetDirective(*this, S, CodeGen);
5769}
5770
Alexey Bataev6d4ed052015-07-01 06:57:41 +00005771void CodeGenFunction::EmitOMPCancellationPointDirective(
5772 const OMPCancellationPointDirective &S) {
Stephen Kellyf2ceec42018-08-09 21:08:08 +00005773 CGM.getOpenMPRuntime().emitCancellationPointCall(*this, S.getBeginLoc(),
Alexey Bataev0f34da12015-07-02 04:17:07 +00005774 S.getCancelRegion());
Alexey Bataev6d4ed052015-07-01 06:57:41 +00005775}
5776
Alexey Bataev80909872015-07-02 11:25:17 +00005777void CodeGenFunction::EmitOMPCancelDirective(const OMPCancelDirective &S) {
Alexey Bataev87933c72015-09-18 08:07:34 +00005778 const Expr *IfCond = nullptr;
5779 for (const auto *C : S.getClausesOfKind<OMPIfClause>()) {
5780 if (C->getNameModifier() == OMPD_unknown ||
5781 C->getNameModifier() == OMPD_cancel) {
5782 IfCond = C->getCondition();
5783 break;
5784 }
5785 }
Johannes Doerfert10fedd92019-12-26 11:23:38 -06005786 if (llvm::OpenMPIRBuilder *OMPBuilder = CGM.getOpenMPIRBuilder()) {
5787 // TODO: This check is necessary as we only generate `omp parallel` through
5788 // the OpenMPIRBuilder for now.
5789 if (S.getCancelRegion() == OMPD_parallel) {
5790 llvm::Value *IfCondition = nullptr;
5791 if (IfCond)
5792 IfCondition = EmitScalarExpr(IfCond,
5793 /*IgnoreResultAssign=*/true);
5794 return Builder.restoreIP(
5795 OMPBuilder->CreateCancel(Builder, IfCondition, S.getCancelRegion()));
5796 }
5797 }
5798
Stephen Kellyf2ceec42018-08-09 21:08:08 +00005799 CGM.getOpenMPRuntime().emitCancelCall(*this, S.getBeginLoc(), IfCond,
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00005800 S.getCancelRegion());
Alexey Bataev80909872015-07-02 11:25:17 +00005801}
5802
Alexey Bataev81c7ea02015-07-03 09:56:58 +00005803CodeGenFunction::JumpDest
5804CodeGenFunction::getOMPCancelDestination(OpenMPDirectiveKind Kind) {
Alexey Bataev957d8562016-11-17 15:12:05 +00005805 if (Kind == OMPD_parallel || Kind == OMPD_task ||
Alexey Bataeve0ca4792020-02-12 16:12:53 -05005806 Kind == OMPD_target_parallel || Kind == OMPD_taskloop ||
5807 Kind == OMPD_master_taskloop || Kind == OMPD_parallel_master_taskloop)
Alexey Bataev81c7ea02015-07-03 09:56:58 +00005808 return ReturnBlock;
Alexey Bataev25e5b442015-09-15 12:52:43 +00005809 assert(Kind == OMPD_for || Kind == OMPD_section || Kind == OMPD_sections ||
Alexey Bataev957d8562016-11-17 15:12:05 +00005810 Kind == OMPD_parallel_sections || Kind == OMPD_parallel_for ||
5811 Kind == OMPD_distribute_parallel_for ||
Alexey Bataevdcb4b8fb2017-11-22 20:19:50 +00005812 Kind == OMPD_target_parallel_for ||
Alexey Bataev16e79882017-11-22 21:12:03 +00005813 Kind == OMPD_teams_distribute_parallel_for ||
5814 Kind == OMPD_target_teams_distribute_parallel_for);
Alexey Bataev957d8562016-11-17 15:12:05 +00005815 return OMPCancelStack.getExitBlock();
Alexey Bataev81c7ea02015-07-03 09:56:58 +00005816}
Michael Wong65f367f2015-07-21 13:44:28 +00005817
Samuel Antaocc10b852016-07-28 14:23:26 +00005818void CodeGenFunction::EmitOMPUseDevicePtrClause(
Alexey Bataev90b54fa2020-05-25 16:06:31 -04005819 const OMPUseDevicePtrClause &C, OMPPrivateScope &PrivateScope,
Samuel Antaocc10b852016-07-28 14:23:26 +00005820 const llvm::DenseMap<const ValueDecl *, Address> &CaptureDeviceAddrMap) {
Samuel Antaocc10b852016-07-28 14:23:26 +00005821 auto OrigVarIt = C.varlist_begin();
5822 auto InitIt = C.inits().begin();
Alexey Bataevddf3db92018-04-13 17:31:06 +00005823 for (const Expr *PvtVarIt : C.private_copies()) {
5824 const auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*OrigVarIt)->getDecl());
5825 const auto *InitVD = cast<VarDecl>(cast<DeclRefExpr>(*InitIt)->getDecl());
5826 const auto *PvtVD = cast<VarDecl>(cast<DeclRefExpr>(PvtVarIt)->getDecl());
Samuel Antaocc10b852016-07-28 14:23:26 +00005827
5828 // In order to identify the right initializer we need to match the
5829 // declaration used by the mapping logic. In some cases we may get
5830 // OMPCapturedExprDecl that refers to the original declaration.
5831 const ValueDecl *MatchingVD = OrigVD;
Alexey Bataevddf3db92018-04-13 17:31:06 +00005832 if (const auto *OED = dyn_cast<OMPCapturedExprDecl>(MatchingVD)) {
Samuel Antaocc10b852016-07-28 14:23:26 +00005833 // OMPCapturedExprDecl are used to privative fields of the current
5834 // structure.
Alexey Bataevddf3db92018-04-13 17:31:06 +00005835 const auto *ME = cast<MemberExpr>(OED->getInit());
Samuel Antaocc10b852016-07-28 14:23:26 +00005836 assert(isa<CXXThisExpr>(ME->getBase()) &&
5837 "Base should be the current struct!");
5838 MatchingVD = ME->getMemberDecl();
5839 }
5840
5841 // If we don't have information about the current list item, move on to
5842 // the next one.
5843 auto InitAddrIt = CaptureDeviceAddrMap.find(MatchingVD);
5844 if (InitAddrIt == CaptureDeviceAddrMap.end())
5845 continue;
5846
Alexey Bataevddf3db92018-04-13 17:31:06 +00005847 bool IsRegistered = PrivateScope.addPrivate(OrigVD, [this, OrigVD,
5848 InitAddrIt, InitVD,
5849 PvtVD]() {
Samuel Antaocc10b852016-07-28 14:23:26 +00005850 // Initialize the temporary initialization variable with the address we
5851 // get from the runtime library. We have to cast the source address
5852 // because it is always a void *. References are materialized in the
5853 // privatization scope, so the initialization here disregards the fact
5854 // the original variable is a reference.
5855 QualType AddrQTy =
5856 getContext().getPointerType(OrigVD->getType().getNonReferenceType());
5857 llvm::Type *AddrTy = ConvertTypeForMem(AddrQTy);
5858 Address InitAddr = Builder.CreateBitCast(InitAddrIt->second, AddrTy);
5859 setAddrOfLocalVar(InitVD, InitAddr);
5860
5861 // Emit private declaration, it will be initialized by the value we
5862 // declaration we just added to the local declarations map.
5863 EmitDecl(*PvtVD);
5864
5865 // The initialization variables reached its purpose in the emission
Alexander Kornienko2a8c18d2018-04-06 15:14:32 +00005866 // of the previous declaration, so we don't need it anymore.
Samuel Antaocc10b852016-07-28 14:23:26 +00005867 LocalDeclMap.erase(InitVD);
5868
5869 // Return the address of the private variable.
5870 return GetAddrOfLocalVar(PvtVD);
5871 });
5872 assert(IsRegistered && "firstprivate var already registered as private");
5873 // Silence the warning about unused variable.
5874 (void)IsRegistered;
5875
5876 ++OrigVarIt;
5877 ++InitIt;
5878 }
5879}
5880
Alexey Bataev90b54fa2020-05-25 16:06:31 -04005881static const VarDecl *getBaseDecl(const Expr *Ref) {
5882 const Expr *Base = Ref->IgnoreParenImpCasts();
5883 while (const auto *OASE = dyn_cast<OMPArraySectionExpr>(Base))
5884 Base = OASE->getBase()->IgnoreParenImpCasts();
5885 while (const auto *ASE = dyn_cast<ArraySubscriptExpr>(Base))
5886 Base = ASE->getBase()->IgnoreParenImpCasts();
5887 return cast<VarDecl>(cast<DeclRefExpr>(Base)->getDecl());
5888}
5889
5890void CodeGenFunction::EmitOMPUseDeviceAddrClause(
5891 const OMPUseDeviceAddrClause &C, OMPPrivateScope &PrivateScope,
5892 const llvm::DenseMap<const ValueDecl *, Address> &CaptureDeviceAddrMap) {
5893 llvm::SmallDenseSet<CanonicalDeclPtr<const Decl>, 4> Processed;
5894 for (const Expr *Ref : C.varlists()) {
5895 const VarDecl *OrigVD = getBaseDecl(Ref);
5896 if (!Processed.insert(OrigVD).second)
5897 continue;
5898 // In order to identify the right initializer we need to match the
5899 // declaration used by the mapping logic. In some cases we may get
5900 // OMPCapturedExprDecl that refers to the original declaration.
5901 const ValueDecl *MatchingVD = OrigVD;
5902 if (const auto *OED = dyn_cast<OMPCapturedExprDecl>(MatchingVD)) {
5903 // OMPCapturedExprDecl are used to privative fields of the current
5904 // structure.
5905 const auto *ME = cast<MemberExpr>(OED->getInit());
5906 assert(isa<CXXThisExpr>(ME->getBase()) &&
5907 "Base should be the current struct!");
5908 MatchingVD = ME->getMemberDecl();
5909 }
5910
5911 // If we don't have information about the current list item, move on to
5912 // the next one.
5913 auto InitAddrIt = CaptureDeviceAddrMap.find(MatchingVD);
5914 if (InitAddrIt == CaptureDeviceAddrMap.end())
5915 continue;
5916
5917 Address PrivAddr = InitAddrIt->getSecond();
5918 // For declrefs and variable length array need to load the pointer for
5919 // correct mapping, since the pointer to the data was passed to the runtime.
5920 if (isa<DeclRefExpr>(Ref->IgnoreParenImpCasts()) ||
5921 MatchingVD->getType()->isArrayType())
5922 PrivAddr =
5923 EmitLoadOfPointer(PrivAddr, getContext()
5924 .getPointerType(OrigVD->getType())
5925 ->castAs<PointerType>());
5926 llvm::Type *RealTy =
5927 ConvertTypeForMem(OrigVD->getType().getNonReferenceType())
5928 ->getPointerTo();
5929 PrivAddr = Builder.CreatePointerBitCastOrAddrSpaceCast(PrivAddr, RealTy);
5930
5931 (void)PrivateScope.addPrivate(OrigVD, [PrivAddr]() { return PrivAddr; });
5932 }
5933}
5934
Michael Wong65f367f2015-07-21 13:44:28 +00005935// Generate the instructions for '#pragma omp target data' directive.
5936void CodeGenFunction::EmitOMPTargetDataDirective(
5937 const OMPTargetDataDirective &S) {
Samuel Antaocc10b852016-07-28 14:23:26 +00005938 CGOpenMPRuntime::TargetDataInfo Info(/*RequiresDevicePointerInfo=*/true);
5939
5940 // Create a pre/post action to signal the privatization of the device pointer.
5941 // This action can be replaced by the OpenMP runtime code generation to
5942 // deactivate privatization.
5943 bool PrivatizeDevicePointers = false;
5944 class DevicePointerPrivActionTy : public PrePostActionTy {
5945 bool &PrivatizeDevicePointers;
5946
5947 public:
5948 explicit DevicePointerPrivActionTy(bool &PrivatizeDevicePointers)
5949 : PrePostActionTy(), PrivatizeDevicePointers(PrivatizeDevicePointers) {}
5950 void Enter(CodeGenFunction &CGF) override {
5951 PrivatizeDevicePointers = true;
5952 }
Samuel Antaodf158d52016-04-27 22:58:19 +00005953 };
Samuel Antaocc10b852016-07-28 14:23:26 +00005954 DevicePointerPrivActionTy PrivAction(PrivatizeDevicePointers);
5955
5956 auto &&CodeGen = [&S, &Info, &PrivatizeDevicePointers](
Alexey Bataev475a7442018-01-12 19:39:11 +00005957 CodeGenFunction &CGF, PrePostActionTy &Action) {
Samuel Antaocc10b852016-07-28 14:23:26 +00005958 auto &&InnermostCodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &) {
Alexey Bataev475a7442018-01-12 19:39:11 +00005959 CGF.EmitStmt(S.getInnermostCapturedStmt()->getCapturedStmt());
Samuel Antaocc10b852016-07-28 14:23:26 +00005960 };
5961
Alexander Kornienko2a8c18d2018-04-06 15:14:32 +00005962 // Codegen that selects whether to generate the privatization code or not.
Samuel Antaocc10b852016-07-28 14:23:26 +00005963 auto &&PrivCodeGen = [&S, &Info, &PrivatizeDevicePointers,
5964 &InnermostCodeGen](CodeGenFunction &CGF,
5965 PrePostActionTy &Action) {
5966 RegionCodeGenTy RCG(InnermostCodeGen);
5967 PrivatizeDevicePointers = false;
5968
5969 // Call the pre-action to change the status of PrivatizeDevicePointers if
5970 // needed.
5971 Action.Enter(CGF);
5972
5973 if (PrivatizeDevicePointers) {
5974 OMPPrivateScope PrivateScope(CGF);
5975 // Emit all instances of the use_device_ptr clause.
5976 for (const auto *C : S.getClausesOfKind<OMPUseDevicePtrClause>())
5977 CGF.EmitOMPUseDevicePtrClause(*C, PrivateScope,
5978 Info.CaptureDeviceAddrMap);
Alexey Bataev90b54fa2020-05-25 16:06:31 -04005979 for (const auto *C : S.getClausesOfKind<OMPUseDeviceAddrClause>())
5980 CGF.EmitOMPUseDeviceAddrClause(*C, PrivateScope,
5981 Info.CaptureDeviceAddrMap);
Samuel Antaocc10b852016-07-28 14:23:26 +00005982 (void)PrivateScope.Privatize();
5983 RCG(CGF);
Alexey Bataevddf3db92018-04-13 17:31:06 +00005984 } else {
Samuel Antaocc10b852016-07-28 14:23:26 +00005985 RCG(CGF);
Alexey Bataevddf3db92018-04-13 17:31:06 +00005986 }
Samuel Antaocc10b852016-07-28 14:23:26 +00005987 };
5988
5989 // Forward the provided action to the privatization codegen.
5990 RegionCodeGenTy PrivRCG(PrivCodeGen);
5991 PrivRCG.setAction(Action);
5992
5993 // Notwithstanding the body of the region is emitted as inlined directive,
5994 // we don't use an inline scope as changes in the references inside the
5995 // region are expected to be visible outside, so we do not privative them.
5996 OMPLexicalScope Scope(CGF, S);
5997 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, OMPD_target_data,
5998 PrivRCG);
5999 };
6000
6001 RegionCodeGenTy RCG(CodeGen);
Samuel Antaodf158d52016-04-27 22:58:19 +00006002
6003 // If we don't have target devices, don't bother emitting the data mapping
6004 // code.
6005 if (CGM.getLangOpts().OMPTargetTriples.empty()) {
Samuel Antaocc10b852016-07-28 14:23:26 +00006006 RCG(*this);
Samuel Antaodf158d52016-04-27 22:58:19 +00006007 return;
6008 }
6009
6010 // Check if we have any if clause associated with the directive.
6011 const Expr *IfCond = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006012 if (const auto *C = S.getSingleClause<OMPIfClause>())
Samuel Antaodf158d52016-04-27 22:58:19 +00006013 IfCond = C->getCondition();
6014
6015 // Check if we have any device clause associated with the directive.
6016 const Expr *Device = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006017 if (const auto *C = S.getSingleClause<OMPDeviceClause>())
Samuel Antaodf158d52016-04-27 22:58:19 +00006018 Device = C->getDevice();
6019
Samuel Antaocc10b852016-07-28 14:23:26 +00006020 // Set the action to signal privatization of device pointers.
6021 RCG.setAction(PrivAction);
6022
6023 // Emit region code.
6024 CGM.getOpenMPRuntime().emitTargetDataCalls(*this, S, IfCond, Device, RCG,
6025 Info);
Michael Wong65f367f2015-07-21 13:44:28 +00006026}
Alexey Bataev49f6e782015-12-01 04:18:41 +00006027
Samuel Antaodf67fc42016-01-19 19:15:56 +00006028void CodeGenFunction::EmitOMPTargetEnterDataDirective(
6029 const OMPTargetEnterDataDirective &S) {
Samuel Antaobd0ae2e2016-04-27 23:07:29 +00006030 // If we don't have target devices, don't bother emitting the data mapping
6031 // code.
6032 if (CGM.getLangOpts().OMPTargetTriples.empty())
6033 return;
6034
6035 // Check if we have any if clause associated with the directive.
6036 const Expr *IfCond = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006037 if (const auto *C = S.getSingleClause<OMPIfClause>())
Samuel Antaobd0ae2e2016-04-27 23:07:29 +00006038 IfCond = C->getCondition();
6039
6040 // Check if we have any device clause associated with the directive.
6041 const Expr *Device = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006042 if (const auto *C = S.getSingleClause<OMPDeviceClause>())
Samuel Antaobd0ae2e2016-04-27 23:07:29 +00006043 Device = C->getDevice();
6044
Alexey Bataev475a7442018-01-12 19:39:11 +00006045 OMPLexicalScope Scope(*this, S, OMPD_task);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00006046 CGM.getOpenMPRuntime().emitTargetDataStandAloneCall(*this, S, IfCond, Device);
Samuel Antaodf67fc42016-01-19 19:15:56 +00006047}
6048
Samuel Antao72590762016-01-19 20:04:50 +00006049void CodeGenFunction::EmitOMPTargetExitDataDirective(
6050 const OMPTargetExitDataDirective &S) {
Samuel Antao8dd66282016-04-27 23:14:30 +00006051 // If we don't have target devices, don't bother emitting the data mapping
6052 // code.
6053 if (CGM.getLangOpts().OMPTargetTriples.empty())
6054 return;
6055
6056 // Check if we have any if clause associated with the directive.
6057 const Expr *IfCond = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006058 if (const auto *C = S.getSingleClause<OMPIfClause>())
Samuel Antao8dd66282016-04-27 23:14:30 +00006059 IfCond = C->getCondition();
6060
6061 // Check if we have any device clause associated with the directive.
6062 const Expr *Device = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006063 if (const auto *C = S.getSingleClause<OMPDeviceClause>())
Samuel Antao8dd66282016-04-27 23:14:30 +00006064 Device = C->getDevice();
6065
Alexey Bataev475a7442018-01-12 19:39:11 +00006066 OMPLexicalScope Scope(*this, S, OMPD_task);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00006067 CGM.getOpenMPRuntime().emitTargetDataStandAloneCall(*this, S, IfCond, Device);
Samuel Antao72590762016-01-19 20:04:50 +00006068}
6069
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00006070static void emitTargetParallelRegion(CodeGenFunction &CGF,
6071 const OMPTargetParallelDirective &S,
6072 PrePostActionTy &Action) {
6073 // Get the captured statement associated with the 'parallel' region.
Alexey Bataevddf3db92018-04-13 17:31:06 +00006074 const CapturedStmt *CS = S.getCapturedStmt(OMPD_parallel);
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00006075 Action.Enter(CGF);
Alexey Bataevc99042b2018-03-15 18:10:54 +00006076 auto &&CodeGen = [&S, CS](CodeGenFunction &CGF, PrePostActionTy &Action) {
Alexey Bataev63cc8e92018-03-20 14:45:59 +00006077 Action.Enter(CGF);
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00006078 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
6079 (void)CGF.EmitOMPFirstprivateClause(S, PrivateScope);
6080 CGF.EmitOMPPrivateClause(S, PrivateScope);
6081 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
6082 (void)PrivateScope.Privatize();
Alexey Bataev60705422018-10-30 15:50:12 +00006083 if (isOpenMPTargetExecutionDirective(S.getDirectiveKind()))
6084 CGF.CGM.getOpenMPRuntime().adjustTargetSpecificDataForLambdas(CGF, S);
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00006085 // TODO: Add support for clauses.
6086 CGF.EmitStmt(CS->getCapturedStmt());
Arpith Chacko Jacob101e8fb2017-02-16 16:20:16 +00006087 CGF.EmitOMPReductionClauseFinal(S, /*ReductionKind=*/OMPD_parallel);
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00006088 };
Carlo Bertollib0ff0a62017-04-25 17:52:12 +00006089 emitCommonOMPParallelDirective(CGF, S, OMPD_parallel, CodeGen,
6090 emitEmptyBoundParameters);
Alexey Bataevddf3db92018-04-13 17:31:06 +00006091 emitPostUpdateForReductionClause(CGF, S,
6092 [](CodeGenFunction &) { return nullptr; });
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00006093}
6094
6095void CodeGenFunction::EmitOMPTargetParallelDeviceFunction(
6096 CodeGenModule &CGM, StringRef ParentName,
6097 const OMPTargetParallelDirective &S) {
6098 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6099 emitTargetParallelRegion(CGF, S, Action);
6100 };
6101 llvm::Function *Fn;
6102 llvm::Constant *Addr;
6103 // Emit target region as a standalone region.
6104 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
6105 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
6106 assert(Fn && Addr && "Target device function emission failed.");
6107}
6108
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00006109void CodeGenFunction::EmitOMPTargetParallelDirective(
6110 const OMPTargetParallelDirective &S) {
Arpith Chacko Jacob19b911c2017-01-18 18:18:53 +00006111 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6112 emitTargetParallelRegion(CGF, S, Action);
6113 };
6114 emitCommonOMPTargetDirective(*this, S, CodeGen);
Arpith Chacko Jacobe955b3d2016-01-26 18:48:41 +00006115}
6116
Alexey Bataevfb0ebec2017-11-08 20:16:14 +00006117static void emitTargetParallelForRegion(CodeGenFunction &CGF,
6118 const OMPTargetParallelForDirective &S,
6119 PrePostActionTy &Action) {
6120 Action.Enter(CGF);
6121 // Emit directive as a combined directive that consists of two implicit
6122 // directives: 'parallel' with 'for' directive.
Alexey Bataevc99042b2018-03-15 18:10:54 +00006123 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6124 Action.Enter(CGF);
Alexey Bataev2139ed62017-11-16 18:20:21 +00006125 CodeGenFunction::OMPCancelStackRAII CancelRegion(
6126 CGF, OMPD_target_parallel_for, S.hasCancel());
Alexey Bataevfb0ebec2017-11-08 20:16:14 +00006127 CGF.EmitOMPWorksharingLoop(S, S.getEnsureUpperBound(), emitForLoopBounds,
6128 emitDispatchForLoopBounds);
6129 };
6130 emitCommonOMPParallelDirective(CGF, S, OMPD_for, CodeGen,
6131 emitEmptyBoundParameters);
6132}
6133
6134void CodeGenFunction::EmitOMPTargetParallelForDeviceFunction(
6135 CodeGenModule &CGM, StringRef ParentName,
6136 const OMPTargetParallelForDirective &S) {
6137 // Emit SPMD target parallel for region as a standalone region.
6138 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6139 emitTargetParallelForRegion(CGF, S, Action);
6140 };
6141 llvm::Function *Fn;
6142 llvm::Constant *Addr;
6143 // Emit target region as a standalone region.
6144 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
6145 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
6146 assert(Fn && Addr && "Target device function emission failed.");
6147}
6148
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00006149void CodeGenFunction::EmitOMPTargetParallelForDirective(
6150 const OMPTargetParallelForDirective &S) {
Alexey Bataevfb0ebec2017-11-08 20:16:14 +00006151 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6152 emitTargetParallelForRegion(CGF, S, Action);
6153 };
6154 emitCommonOMPTargetDirective(*this, S, CodeGen);
Arpith Chacko Jacob05bebb52016-02-03 15:46:42 +00006155}
6156
Alexey Bataev5d7edca2017-11-09 17:32:15 +00006157static void
6158emitTargetParallelForSimdRegion(CodeGenFunction &CGF,
6159 const OMPTargetParallelForSimdDirective &S,
6160 PrePostActionTy &Action) {
6161 Action.Enter(CGF);
6162 // Emit directive as a combined directive that consists of two implicit
6163 // directives: 'parallel' with 'for' directive.
Alexey Bataevc99042b2018-03-15 18:10:54 +00006164 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6165 Action.Enter(CGF);
Alexey Bataev5d7edca2017-11-09 17:32:15 +00006166 CGF.EmitOMPWorksharingLoop(S, S.getEnsureUpperBound(), emitForLoopBounds,
6167 emitDispatchForLoopBounds);
6168 };
6169 emitCommonOMPParallelDirective(CGF, S, OMPD_simd, CodeGen,
6170 emitEmptyBoundParameters);
6171}
6172
6173void CodeGenFunction::EmitOMPTargetParallelForSimdDeviceFunction(
6174 CodeGenModule &CGM, StringRef ParentName,
6175 const OMPTargetParallelForSimdDirective &S) {
6176 // Emit SPMD target parallel for region as a standalone region.
6177 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6178 emitTargetParallelForSimdRegion(CGF, S, Action);
6179 };
6180 llvm::Function *Fn;
6181 llvm::Constant *Addr;
6182 // Emit target region as a standalone region.
6183 CGM.getOpenMPRuntime().emitTargetOutlinedFunction(
6184 S, ParentName, Fn, Addr, /*IsOffloadEntry=*/true, CodeGen);
6185 assert(Fn && Addr && "Target device function emission failed.");
6186}
6187
6188void CodeGenFunction::EmitOMPTargetParallelForSimdDirective(
6189 const OMPTargetParallelForSimdDirective &S) {
6190 auto &&CodeGen = [&S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6191 emitTargetParallelForSimdRegion(CGF, S, Action);
6192 };
6193 emitCommonOMPTargetDirective(*this, S, CodeGen);
6194}
6195
Alexey Bataev7292c292016-04-25 12:22:29 +00006196/// Emit a helper variable and return corresponding lvalue.
6197static void mapParam(CodeGenFunction &CGF, const DeclRefExpr *Helper,
6198 const ImplicitParamDecl *PVD,
6199 CodeGenFunction::OMPPrivateScope &Privates) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00006200 const auto *VDecl = cast<VarDecl>(Helper->getDecl());
6201 Privates.addPrivate(VDecl,
6202 [&CGF, PVD]() { return CGF.GetAddrOfLocalVar(PVD); });
Alexey Bataev7292c292016-04-25 12:22:29 +00006203}
6204
6205void CodeGenFunction::EmitOMPTaskLoopBasedDirective(const OMPLoopDirective &S) {
6206 assert(isOpenMPTaskLoopDirective(S.getDirectiveKind()));
6207 // Emit outlined function for task construct.
Alexey Bataev475a7442018-01-12 19:39:11 +00006208 const CapturedStmt *CS = S.getCapturedStmt(OMPD_taskloop);
Alexey Bataev172f1462020-03-12 12:52:02 -04006209 Address CapturedStruct = Address::invalid();
6210 {
6211 OMPLexicalScope Scope(*this, S, OMPD_taskloop, /*EmitPreInitStmt=*/false);
6212 CapturedStruct = GenerateCapturedStmtArgument(*CS);
6213 }
Alexey Bataevddf3db92018-04-13 17:31:06 +00006214 QualType SharedsTy = getContext().getRecordType(CS->getCapturedRecordDecl());
Alexey Bataev7292c292016-04-25 12:22:29 +00006215 const Expr *IfCond = nullptr;
6216 for (const auto *C : S.getClausesOfKind<OMPIfClause>()) {
6217 if (C->getNameModifier() == OMPD_unknown ||
6218 C->getNameModifier() == OMPD_taskloop) {
6219 IfCond = C->getCondition();
6220 break;
6221 }
6222 }
Alexey Bataev24b5bae2016-04-28 09:23:51 +00006223
6224 OMPTaskDataTy Data;
6225 // Check if taskloop must be emitted without taskgroup.
6226 Data.Nogroup = S.getSingleClause<OMPNogroupClause>();
Alexey Bataev7292c292016-04-25 12:22:29 +00006227 // TODO: Check if we should emit tied or untied task.
Alexey Bataev24b5bae2016-04-28 09:23:51 +00006228 Data.Tied = true;
6229 // Set scheduling for taskloop
Alexey Bataev2b19a6f2016-04-28 09:15:06 +00006230 if (const auto* Clause = S.getSingleClause<OMPGrainsizeClause>()) {
6231 // grainsize clause
Alexey Bataev24b5bae2016-04-28 09:23:51 +00006232 Data.Schedule.setInt(/*IntVal=*/false);
6233 Data.Schedule.setPointer(EmitScalarExpr(Clause->getGrainsize()));
Alexey Bataev2b19a6f2016-04-28 09:15:06 +00006234 } else if (const auto* Clause = S.getSingleClause<OMPNumTasksClause>()) {
6235 // num_tasks clause
Alexey Bataev24b5bae2016-04-28 09:23:51 +00006236 Data.Schedule.setInt(/*IntVal=*/true);
6237 Data.Schedule.setPointer(EmitScalarExpr(Clause->getNumTasks()));
Alexey Bataev2b19a6f2016-04-28 09:15:06 +00006238 }
Alexey Bataev7292c292016-04-25 12:22:29 +00006239
6240 auto &&BodyGen = [CS, &S](CodeGenFunction &CGF, PrePostActionTy &) {
6241 // if (PreCond) {
6242 // for (IV in 0..LastIteration) BODY;
6243 // <Final counter/linear vars updates>;
6244 // }
6245 //
6246
6247 // Emit: if (PreCond) - begin.
6248 // If the condition constant folds and can be elided, avoid emitting the
6249 // whole loop.
6250 bool CondConstant;
6251 llvm::BasicBlock *ContBlock = nullptr;
6252 OMPLoopScope PreInitScope(CGF, S);
6253 if (CGF.ConstantFoldsToSimpleInteger(S.getPreCond(), CondConstant)) {
6254 if (!CondConstant)
6255 return;
6256 } else {
Alexey Bataevddf3db92018-04-13 17:31:06 +00006257 llvm::BasicBlock *ThenBlock = CGF.createBasicBlock("taskloop.if.then");
Alexey Bataev7292c292016-04-25 12:22:29 +00006258 ContBlock = CGF.createBasicBlock("taskloop.if.end");
6259 emitPreCond(CGF, S, S.getPreCond(), ThenBlock, ContBlock,
6260 CGF.getProfileCount(&S));
6261 CGF.EmitBlock(ThenBlock);
6262 CGF.incrementProfileCounter(&S);
6263 }
6264
Alexey Bataev61205822019-12-04 09:50:21 -05006265 (void)CGF.EmitOMPLinearClauseInit(S);
Alexey Bataev1e73ef32016-04-28 12:14:51 +00006266
Alexey Bataev7292c292016-04-25 12:22:29 +00006267 OMPPrivateScope LoopScope(CGF);
6268 // Emit helper vars inits.
6269 enum { LowerBound = 5, UpperBound, Stride, LastIter };
6270 auto *I = CS->getCapturedDecl()->param_begin();
6271 auto *LBP = std::next(I, LowerBound);
6272 auto *UBP = std::next(I, UpperBound);
6273 auto *STP = std::next(I, Stride);
6274 auto *LIP = std::next(I, LastIter);
6275 mapParam(CGF, cast<DeclRefExpr>(S.getLowerBoundVariable()), *LBP,
6276 LoopScope);
6277 mapParam(CGF, cast<DeclRefExpr>(S.getUpperBoundVariable()), *UBP,
6278 LoopScope);
6279 mapParam(CGF, cast<DeclRefExpr>(S.getStrideVariable()), *STP, LoopScope);
6280 mapParam(CGF, cast<DeclRefExpr>(S.getIsLastIterVariable()), *LIP,
6281 LoopScope);
6282 CGF.EmitOMPPrivateLoopCounters(S, LoopScope);
Alexey Bataev14a388f2019-10-25 10:27:13 -04006283 CGF.EmitOMPLinearClause(S, LoopScope);
Alexey Bataevf93095a2016-05-05 08:46:22 +00006284 bool HasLastprivateClause = CGF.EmitOMPLastprivateClauseInit(S, LoopScope);
Alexey Bataev7292c292016-04-25 12:22:29 +00006285 (void)LoopScope.Privatize();
6286 // Emit the loop iteration variable.
6287 const Expr *IVExpr = S.getIterationVariable();
Alexey Bataevddf3db92018-04-13 17:31:06 +00006288 const auto *IVDecl = cast<VarDecl>(cast<DeclRefExpr>(IVExpr)->getDecl());
Alexey Bataev7292c292016-04-25 12:22:29 +00006289 CGF.EmitVarDecl(*IVDecl);
6290 CGF.EmitIgnoredExpr(S.getInit());
6291
6292 // Emit the iterations count variable.
6293 // If it is not a variable, Sema decided to calculate iterations count on
6294 // each iteration (e.g., it is foldable into a constant).
Alexey Bataevddf3db92018-04-13 17:31:06 +00006295 if (const auto *LIExpr = dyn_cast<DeclRefExpr>(S.getLastIteration())) {
Alexey Bataev7292c292016-04-25 12:22:29 +00006296 CGF.EmitVarDecl(*cast<VarDecl>(LIExpr->getDecl()));
6297 // Emit calculation of the iterations count.
6298 CGF.EmitIgnoredExpr(S.getCalcLastIteration());
6299 }
6300
Alexey Bataev61205822019-12-04 09:50:21 -05006301 {
6302 OMPLexicalScope Scope(CGF, S, OMPD_taskloop, /*EmitPreInitStmt=*/false);
6303 emitCommonSimdLoop(
6304 CGF, S,
6305 [&S](CodeGenFunction &CGF, PrePostActionTy &) {
6306 if (isOpenMPSimdDirective(S.getDirectiveKind()))
6307 CGF.EmitOMPSimdInit(S);
6308 },
6309 [&S, &LoopScope](CodeGenFunction &CGF, PrePostActionTy &) {
6310 CGF.EmitOMPInnerLoop(
6311 S, LoopScope.requiresCleanups(), S.getCond(), S.getInc(),
6312 [&S](CodeGenFunction &CGF) {
Alexey Bataevcb9191c2020-06-09 08:04:48 -04006313 emitOMPLoopBodyWithStopPoint(CGF, S,
6314 CodeGenFunction::JumpDest());
Alexey Bataev61205822019-12-04 09:50:21 -05006315 },
6316 [](CodeGenFunction &) {});
6317 });
6318 }
Alexey Bataev7292c292016-04-25 12:22:29 +00006319 // Emit: if (PreCond) - end.
6320 if (ContBlock) {
6321 CGF.EmitBranch(ContBlock);
6322 CGF.EmitBlock(ContBlock, true);
6323 }
Alexey Bataevf93095a2016-05-05 08:46:22 +00006324 // Emit final copy of the lastprivate variables if IsLastIter != 0.
6325 if (HasLastprivateClause) {
6326 CGF.EmitOMPLastprivateClauseFinal(
6327 S, isOpenMPSimdDirective(S.getDirectiveKind()),
6328 CGF.Builder.CreateIsNotNull(CGF.EmitLoadOfScalar(
6329 CGF.GetAddrOfLocalVar(*LIP), /*Volatile=*/false,
Stephen Kellyf2ceec42018-08-09 21:08:08 +00006330 (*LIP)->getType(), S.getBeginLoc())));
Alexey Bataevf93095a2016-05-05 08:46:22 +00006331 }
Alexey Bataev14a388f2019-10-25 10:27:13 -04006332 CGF.EmitOMPLinearClauseFinal(S, [LIP, &S](CodeGenFunction &CGF) {
6333 return CGF.Builder.CreateIsNotNull(
6334 CGF.EmitLoadOfScalar(CGF.GetAddrOfLocalVar(*LIP), /*Volatile=*/false,
6335 (*LIP)->getType(), S.getBeginLoc()));
6336 });
Alexey Bataev7292c292016-04-25 12:22:29 +00006337 };
Alexey Bataev24b5bae2016-04-28 09:23:51 +00006338 auto &&TaskGen = [&S, SharedsTy, CapturedStruct,
James Y Knight9871db02019-02-05 16:42:33 +00006339 IfCond](CodeGenFunction &CGF, llvm::Function *OutlinedFn,
Alexey Bataev24b5bae2016-04-28 09:23:51 +00006340 const OMPTaskDataTy &Data) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00006341 auto &&CodeGen = [&S, OutlinedFn, SharedsTy, CapturedStruct, IfCond,
6342 &Data](CodeGenFunction &CGF, PrePostActionTy &) {
Alexey Bataev7292c292016-04-25 12:22:29 +00006343 OMPLoopScope PreInitScope(CGF, S);
Stephen Kellyf2ceec42018-08-09 21:08:08 +00006344 CGF.CGM.getOpenMPRuntime().emitTaskLoopCall(CGF, S.getBeginLoc(), S,
Alexey Bataev24b5bae2016-04-28 09:23:51 +00006345 OutlinedFn, SharedsTy,
6346 CapturedStruct, IfCond, Data);
Alexey Bataev7292c292016-04-25 12:22:29 +00006347 };
6348 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, OMPD_taskloop,
6349 CodeGen);
6350 };
Alexey Bataev475a7442018-01-12 19:39:11 +00006351 if (Data.Nogroup) {
6352 EmitOMPTaskBasedDirective(S, OMPD_taskloop, BodyGen, TaskGen, Data);
6353 } else {
Alexey Bataev33446032017-07-12 18:09:32 +00006354 CGM.getOpenMPRuntime().emitTaskgroupRegion(
6355 *this,
6356 [&S, &BodyGen, &TaskGen, &Data](CodeGenFunction &CGF,
6357 PrePostActionTy &Action) {
6358 Action.Enter(CGF);
Alexey Bataev475a7442018-01-12 19:39:11 +00006359 CGF.EmitOMPTaskBasedDirective(S, OMPD_taskloop, BodyGen, TaskGen,
6360 Data);
Alexey Bataev33446032017-07-12 18:09:32 +00006361 },
Stephen Kellyf2ceec42018-08-09 21:08:08 +00006362 S.getBeginLoc());
Alexey Bataev33446032017-07-12 18:09:32 +00006363 }
Alexey Bataev7292c292016-04-25 12:22:29 +00006364}
6365
Alexey Bataev49f6e782015-12-01 04:18:41 +00006366void CodeGenFunction::EmitOMPTaskLoopDirective(const OMPTaskLoopDirective &S) {
Alexey Bataev46978742020-01-30 10:46:11 -05006367 auto LPCRegion =
6368 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev7292c292016-04-25 12:22:29 +00006369 EmitOMPTaskLoopBasedDirective(S);
Alexey Bataev49f6e782015-12-01 04:18:41 +00006370}
6371
Alexey Bataev0a6ed842015-12-03 09:40:15 +00006372void CodeGenFunction::EmitOMPTaskLoopSimdDirective(
6373 const OMPTaskLoopSimdDirective &S) {
Alexey Bataev46978742020-01-30 10:46:11 -05006374 auto LPCRegion =
6375 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev61205822019-12-04 09:50:21 -05006376 OMPLexicalScope Scope(*this, S);
Alexey Bataev1e73ef32016-04-28 12:14:51 +00006377 EmitOMPTaskLoopBasedDirective(S);
Alexey Bataev0a6ed842015-12-03 09:40:15 +00006378}
Samuel Antao686c70c2016-05-26 17:30:50 +00006379
Alexey Bataev60e51c42019-10-10 20:13:02 +00006380void CodeGenFunction::EmitOMPMasterTaskLoopDirective(
6381 const OMPMasterTaskLoopDirective &S) {
6382 auto &&CodeGen = [this, &S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6383 Action.Enter(CGF);
6384 EmitOMPTaskLoopBasedDirective(S);
6385 };
Alexey Bataev46978742020-01-30 10:46:11 -05006386 auto LPCRegion =
6387 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev60e51c42019-10-10 20:13:02 +00006388 OMPLexicalScope Scope(*this, S, llvm::None, /*EmitPreInitStmt=*/false);
6389 CGM.getOpenMPRuntime().emitMasterRegion(*this, CodeGen, S.getBeginLoc());
6390}
6391
Alexey Bataevb8552ab2019-10-18 16:47:35 +00006392void CodeGenFunction::EmitOMPMasterTaskLoopSimdDirective(
6393 const OMPMasterTaskLoopSimdDirective &S) {
6394 auto &&CodeGen = [this, &S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6395 Action.Enter(CGF);
6396 EmitOMPTaskLoopBasedDirective(S);
6397 };
Alexey Bataev46978742020-01-30 10:46:11 -05006398 auto LPCRegion =
6399 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev61205822019-12-04 09:50:21 -05006400 OMPLexicalScope Scope(*this, S);
Alexey Bataevb8552ab2019-10-18 16:47:35 +00006401 CGM.getOpenMPRuntime().emitMasterRegion(*this, CodeGen, S.getBeginLoc());
6402}
6403
Alexey Bataev5bbcead2019-10-14 17:17:41 +00006404void CodeGenFunction::EmitOMPParallelMasterTaskLoopDirective(
6405 const OMPParallelMasterTaskLoopDirective &S) {
6406 auto &&CodeGen = [this, &S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6407 auto &&TaskLoopCodeGen = [&S](CodeGenFunction &CGF,
6408 PrePostActionTy &Action) {
6409 Action.Enter(CGF);
6410 CGF.EmitOMPTaskLoopBasedDirective(S);
6411 };
Alexey Bataev18789bf2020-02-13 09:21:15 -05006412 OMPLexicalScope Scope(CGF, S, OMPD_parallel, /*EmitPreInitStmt=*/false);
Alexey Bataev5bbcead2019-10-14 17:17:41 +00006413 CGM.getOpenMPRuntime().emitMasterRegion(CGF, TaskLoopCodeGen,
6414 S.getBeginLoc());
6415 };
Alexey Bataev46978742020-01-30 10:46:11 -05006416 auto LPCRegion =
6417 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev5bbcead2019-10-14 17:17:41 +00006418 emitCommonOMPParallelDirective(*this, S, OMPD_master_taskloop, CodeGen,
6419 emitEmptyBoundParameters);
6420}
6421
Alexey Bataev14a388f2019-10-25 10:27:13 -04006422void CodeGenFunction::EmitOMPParallelMasterTaskLoopSimdDirective(
6423 const OMPParallelMasterTaskLoopSimdDirective &S) {
6424 auto &&CodeGen = [this, &S](CodeGenFunction &CGF, PrePostActionTy &Action) {
6425 auto &&TaskLoopCodeGen = [&S](CodeGenFunction &CGF,
6426 PrePostActionTy &Action) {
6427 Action.Enter(CGF);
6428 CGF.EmitOMPTaskLoopBasedDirective(S);
6429 };
6430 OMPLexicalScope Scope(CGF, S, OMPD_parallel, /*EmitPreInitStmt=*/false);
6431 CGM.getOpenMPRuntime().emitMasterRegion(CGF, TaskLoopCodeGen,
6432 S.getBeginLoc());
6433 };
Alexey Bataev46978742020-01-30 10:46:11 -05006434 auto LPCRegion =
6435 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, S);
Alexey Bataev14a388f2019-10-25 10:27:13 -04006436 emitCommonOMPParallelDirective(*this, S, OMPD_master_taskloop_simd, CodeGen,
6437 emitEmptyBoundParameters);
6438}
6439
Samuel Antao686c70c2016-05-26 17:30:50 +00006440// Generate the instructions for '#pragma omp target update' directive.
6441void CodeGenFunction::EmitOMPTargetUpdateDirective(
6442 const OMPTargetUpdateDirective &S) {
Samuel Antao8d2d7302016-05-26 18:30:22 +00006443 // If we don't have target devices, don't bother emitting the data mapping
6444 // code.
6445 if (CGM.getLangOpts().OMPTargetTriples.empty())
6446 return;
6447
6448 // Check if we have any if clause associated with the directive.
6449 const Expr *IfCond = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006450 if (const auto *C = S.getSingleClause<OMPIfClause>())
Samuel Antao8d2d7302016-05-26 18:30:22 +00006451 IfCond = C->getCondition();
6452
6453 // Check if we have any device clause associated with the directive.
6454 const Expr *Device = nullptr;
Alexey Bataevddf3db92018-04-13 17:31:06 +00006455 if (const auto *C = S.getSingleClause<OMPDeviceClause>())
Samuel Antao8d2d7302016-05-26 18:30:22 +00006456 Device = C->getDevice();
6457
Alexey Bataev475a7442018-01-12 19:39:11 +00006458 OMPLexicalScope Scope(*this, S, OMPD_task);
Alexey Bataevd2202ca2017-12-27 17:58:32 +00006459 CGM.getOpenMPRuntime().emitTargetDataStandAloneCall(*this, S, IfCond, Device);
Samuel Antao686c70c2016-05-26 17:30:50 +00006460}
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006461
6462void CodeGenFunction::EmitSimpleOMPExecutableDirective(
6463 const OMPExecutableDirective &D) {
Alexey Bataevbd1c03d2020-05-04 16:19:31 -04006464 if (const auto *SD = dyn_cast<OMPScanDirective>(&D)) {
6465 EmitOMPScanDirective(*SD);
6466 return;
6467 }
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006468 if (!D.hasAssociatedStmt() || !D.getAssociatedStmt())
6469 return;
Akira Hatanaka9f37c0e2019-12-03 13:07:22 -08006470 auto &&CodeGen = [&D](CodeGenFunction &CGF, PrePostActionTy &Action) {
Alexey Bataevb3998a02020-03-09 08:55:57 -04006471 OMPPrivateScope GlobalsScope(CGF);
6472 if (isOpenMPTaskingDirective(D.getDirectiveKind())) {
6473 // Capture global firstprivates to avoid crash.
6474 for (const auto *C : D.getClausesOfKind<OMPFirstprivateClause>()) {
6475 for (const Expr *Ref : C->varlists()) {
6476 const auto *DRE = cast<DeclRefExpr>(Ref->IgnoreParenImpCasts());
6477 if (!DRE)
6478 continue;
6479 const auto *VD = dyn_cast<VarDecl>(DRE->getDecl());
6480 if (!VD || VD->hasLocalStorage())
6481 continue;
6482 if (!CGF.LocalDeclMap.count(VD)) {
6483 LValue GlobLVal = CGF.EmitLValue(Ref);
6484 GlobalsScope.addPrivate(
6485 VD, [&GlobLVal, &CGF]() { return GlobLVal.getAddress(CGF); });
6486 }
6487 }
6488 }
6489 }
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006490 if (isOpenMPSimdDirective(D.getDirectiveKind())) {
Alexey Bataevb3998a02020-03-09 08:55:57 -04006491 (void)GlobalsScope.Privatize();
Alexey Bataev43101d12020-06-11 11:28:59 -04006492 ParentLoopDirectiveForScanRegion ScanRegion(CGF, D);
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006493 emitOMPSimdRegion(CGF, cast<OMPLoopDirective>(D), Action);
6494 } else {
6495 if (const auto *LD = dyn_cast<OMPLoopDirective>(&D)) {
Alexey Bataevddf3db92018-04-13 17:31:06 +00006496 for (const Expr *E : LD->counters()) {
Simon Pilgrim93431f92020-01-11 16:00:17 +00006497 const auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
Alexey Bataev6ab5bb12018-10-29 15:01:58 +00006498 if (!VD->hasLocalStorage() && !CGF.LocalDeclMap.count(VD)) {
6499 LValue GlobLVal = CGF.EmitLValue(E);
Alexey Bataevb3998a02020-03-09 08:55:57 -04006500 GlobalsScope.addPrivate(
Akira Hatanakaf139ae32019-12-03 15:17:01 -08006501 VD, [&GlobLVal, &CGF]() { return GlobLVal.getAddress(CGF); });
Alexey Bataev6ab5bb12018-10-29 15:01:58 +00006502 }
Bjorn Pettersson6c2d83b2018-10-30 08:49:26 +00006503 if (isa<OMPCapturedExprDecl>(VD)) {
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006504 // Emit only those that were not explicitly referenced in clauses.
6505 if (!CGF.LocalDeclMap.count(VD))
6506 CGF.EmitVarDecl(*VD);
6507 }
6508 }
Alexey Bataevf138fda2018-08-13 19:04:24 +00006509 for (const auto *C : D.getClausesOfKind<OMPOrderedClause>()) {
6510 if (!C->getNumForLoops())
6511 continue;
6512 for (unsigned I = LD->getCollapsedNumber(),
6513 E = C->getLoopNumIterations().size();
6514 I < E; ++I) {
6515 if (const auto *VD = dyn_cast<OMPCapturedExprDecl>(
Mike Rice0ed46662018-09-20 17:19:41 +00006516 cast<DeclRefExpr>(C->getLoopCounter(I))->getDecl())) {
Alexey Bataevf138fda2018-08-13 19:04:24 +00006517 // Emit only those that were not explicitly referenced in clauses.
6518 if (!CGF.LocalDeclMap.count(VD))
6519 CGF.EmitVarDecl(*VD);
6520 }
6521 }
6522 }
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006523 }
Alexey Bataevb3998a02020-03-09 08:55:57 -04006524 (void)GlobalsScope.Privatize();
Alexey Bataev475a7442018-01-12 19:39:11 +00006525 CGF.EmitStmt(D.getInnermostCapturedStmt()->getCapturedStmt());
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006526 }
6527 };
Alexey Bataev46978742020-01-30 10:46:11 -05006528 {
6529 auto LPCRegion =
6530 CGOpenMPRuntime::LastprivateConditionalRAII::disable(*this, D);
6531 OMPSimdLexicalScope Scope(*this, D);
6532 CGM.getOpenMPRuntime().emitInlinedDirective(
6533 *this,
6534 isOpenMPSimdDirective(D.getDirectiveKind()) ? OMPD_simd
6535 : D.getDirectiveKind(),
6536 CodeGen);
6537 }
6538 // Check for outer lastprivate conditional update.
6539 checkForLastprivateConditionalUpdate(*this, D);
Alexey Bataeva8a9153a2017-12-29 18:07:07 +00006540}