blob: 65ba486d852a16593da5112635226b9d2c492c4a [file] [log] [blame]
Alexey Bataev9959db52014-05-06 10:08:46 +00001//===--- CGStmtOpenMP.cpp - Emit LLVM Code from Statements ----------------===//
2//
3// The LLVM Compiler Infrastructure
4//
5// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
10// This contains code to emit OpenMP nodes as LLVM code.
11//
12//===----------------------------------------------------------------------===//
13
14#include "CGOpenMPRuntime.h"
15#include "CodeGenFunction.h"
16#include "CodeGenModule.h"
Chandler Carruth0d9593d2015-01-14 11:29:14 +000017#include "TargetInfo.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000018#include "clang/AST/Stmt.h"
19#include "clang/AST/StmtOpenMP.h"
20using namespace clang;
21using namespace CodeGen;
22
23//===----------------------------------------------------------------------===//
24// OpenMP Directive Emission
25//===----------------------------------------------------------------------===//
Alexey Bataev420d45b2015-04-14 05:11:24 +000026void CodeGenFunction::EmitOMPAggregateAssign(
27 llvm::Value *DestAddr, llvm::Value *SrcAddr, QualType OriginalType,
28 const llvm::function_ref<void(llvm::Value *, llvm::Value *)> &CopyGen) {
29 // Perform element-by-element initialization.
30 QualType ElementTy;
31 auto SrcBegin = SrcAddr;
32 auto DestBegin = DestAddr;
33 auto ArrayTy = OriginalType->getAsArrayTypeUnsafe();
34 auto NumElements = emitArrayLength(ArrayTy, ElementTy, DestBegin);
35 // Cast from pointer to array type to pointer to single element.
36 SrcBegin = Builder.CreatePointerBitCastOrAddrSpaceCast(SrcBegin,
37 DestBegin->getType());
38 auto DestEnd = Builder.CreateGEP(DestBegin, NumElements);
39 // The basic structure here is a while-do loop.
40 auto BodyBB = createBasicBlock("omp.arraycpy.body");
41 auto DoneBB = createBasicBlock("omp.arraycpy.done");
42 auto IsEmpty =
43 Builder.CreateICmpEQ(DestBegin, DestEnd, "omp.arraycpy.isempty");
44 Builder.CreateCondBr(IsEmpty, DoneBB, BodyBB);
Alexey Bataev4a5bb772014-10-08 14:01:46 +000045
Alexey Bataev420d45b2015-04-14 05:11:24 +000046 // Enter the loop body, making that address the current address.
47 auto EntryBB = Builder.GetInsertBlock();
48 EmitBlock(BodyBB);
49 auto SrcElementCurrent =
50 Builder.CreatePHI(SrcBegin->getType(), 2, "omp.arraycpy.srcElementPast");
51 SrcElementCurrent->addIncoming(SrcBegin, EntryBB);
52 auto DestElementCurrent = Builder.CreatePHI(DestBegin->getType(), 2,
53 "omp.arraycpy.destElementPast");
54 DestElementCurrent->addIncoming(DestBegin, EntryBB);
Alexey Bataev4a5bb772014-10-08 14:01:46 +000055
Alexey Bataev420d45b2015-04-14 05:11:24 +000056 // Emit copy.
57 CopyGen(DestElementCurrent, SrcElementCurrent);
58
59 // Shift the address forward by one element.
60 auto DestElementNext = Builder.CreateConstGEP1_32(
61 DestElementCurrent, /*Idx0=*/1, "omp.arraycpy.dest.element");
62 auto SrcElementNext = Builder.CreateConstGEP1_32(
63 SrcElementCurrent, /*Idx0=*/1, "omp.arraycpy.src.element");
64 // Check whether we've reached the end.
65 auto Done =
66 Builder.CreateICmpEQ(DestElementNext, DestEnd, "omp.arraycpy.done");
67 Builder.CreateCondBr(Done, DoneBB, BodyBB);
68 DestElementCurrent->addIncoming(DestElementNext, Builder.GetInsertBlock());
69 SrcElementCurrent->addIncoming(SrcElementNext, Builder.GetInsertBlock());
70
71 // Done.
72 EmitBlock(DoneBB, /*IsFinished=*/true);
73}
74
75void CodeGenFunction::EmitOMPCopy(CodeGenFunction &CGF,
76 QualType OriginalType, llvm::Value *DestAddr,
77 llvm::Value *SrcAddr, const VarDecl *DestVD,
78 const VarDecl *SrcVD, const Expr *Copy) {
79 if (OriginalType->isArrayType()) {
80 auto *BO = dyn_cast<BinaryOperator>(Copy);
81 if (BO && BO->getOpcode() == BO_Assign) {
82 // Perform simple memcpy for simple copying.
83 CGF.EmitAggregateAssign(DestAddr, SrcAddr, OriginalType);
84 } else {
85 // For arrays with complex element types perform element by element
86 // copying.
87 CGF.EmitOMPAggregateAssign(
88 DestAddr, SrcAddr, OriginalType,
89 [&CGF, Copy, SrcVD, DestVD](llvm::Value *DestElement,
90 llvm::Value *SrcElement) {
91 // Working with the single array element, so have to remap
92 // destination and source variables to corresponding array
93 // elements.
94 CodeGenFunction::OMPPrivateScope Remap(CGF);
95 Remap.addPrivate(DestVD, [DestElement]() -> llvm::Value *{
96 return DestElement;
97 });
98 Remap.addPrivate(
99 SrcVD, [SrcElement]() -> llvm::Value *{ return SrcElement; });
100 (void)Remap.Privatize();
101 CGF.EmitIgnoredExpr(Copy);
102 });
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000103 }
Alexey Bataev420d45b2015-04-14 05:11:24 +0000104 } else {
105 // Remap pseudo source variable to private copy.
106 CodeGenFunction::OMPPrivateScope Remap(CGF);
107 Remap.addPrivate(SrcVD, [SrcAddr]() -> llvm::Value *{ return SrcAddr; });
108 Remap.addPrivate(DestVD, [DestAddr]() -> llvm::Value *{ return DestAddr; });
109 (void)Remap.Privatize();
110 // Emit copying of the whole variable.
111 CGF.EmitIgnoredExpr(Copy);
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000112 }
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000113}
114
Alexey Bataev69c62a92015-04-15 04:52:20 +0000115bool CodeGenFunction::EmitOMPFirstprivateClause(const OMPExecutableDirective &D,
116 OMPPrivateScope &PrivateScope) {
Alexey Bataev69c62a92015-04-15 04:52:20 +0000117 llvm::DenseSet<const VarDecl *> EmittedAsFirstprivate;
Alexey Bataevc925aa32015-04-27 08:00:32 +0000118 for (auto &&I = D.getClausesOfKind(OMPC_firstprivate); I; ++I) {
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000119 auto *C = cast<OMPFirstprivateClause>(*I);
120 auto IRef = C->varlist_begin();
121 auto InitsRef = C->inits().begin();
122 for (auto IInit : C->private_copies()) {
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000123 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev69c62a92015-04-15 04:52:20 +0000124 if (EmittedAsFirstprivate.count(OrigVD) == 0) {
125 EmittedAsFirstprivate.insert(OrigVD);
126 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(IInit)->getDecl());
127 auto *VDInit = cast<VarDecl>(cast<DeclRefExpr>(*InitsRef)->getDecl());
128 bool IsRegistered;
129 DeclRefExpr DRE(
130 const_cast<VarDecl *>(OrigVD),
131 /*RefersToEnclosingVariableOrCapture=*/CapturedStmtInfo->lookup(
132 OrigVD) != nullptr,
133 (*IRef)->getType(), VK_LValue, (*IRef)->getExprLoc());
134 auto *OriginalAddr = EmitLValue(&DRE).getAddress();
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000135 QualType Type = OrigVD->getType();
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000136 if (Type->isArrayType()) {
Alexey Bataev69c62a92015-04-15 04:52:20 +0000137 // Emit VarDecl with copy init for arrays.
138 // Get the address of the original variable captured in current
139 // captured region.
140 IsRegistered = PrivateScope.addPrivate(OrigVD, [&]() -> llvm::Value *{
141 auto Emission = EmitAutoVarAlloca(*VD);
142 auto *Init = VD->getInit();
143 if (!isa<CXXConstructExpr>(Init) || isTrivialInitializer(Init)) {
144 // Perform simple memcpy.
145 EmitAggregateAssign(Emission.getAllocatedAddress(), OriginalAddr,
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000146 Type);
Alexey Bataev69c62a92015-04-15 04:52:20 +0000147 } else {
148 EmitOMPAggregateAssign(
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000149 Emission.getAllocatedAddress(), OriginalAddr, Type,
Alexey Bataev69c62a92015-04-15 04:52:20 +0000150 [this, VDInit, Init](llvm::Value *DestElement,
151 llvm::Value *SrcElement) {
152 // Clean up any temporaries needed by the initialization.
153 RunCleanupsScope InitScope(*this);
154 // Emit initialization for single element.
155 LocalDeclMap[VDInit] = SrcElement;
156 EmitAnyExprToMem(Init, DestElement,
157 Init->getType().getQualifiers(),
158 /*IsInitializer*/ false);
159 LocalDeclMap.erase(VDInit);
160 });
161 }
162 EmitAutoVarCleanups(Emission);
163 return Emission.getAllocatedAddress();
164 });
165 } else {
166 IsRegistered = PrivateScope.addPrivate(OrigVD, [&]() -> llvm::Value *{
167 // Emit private VarDecl with copy init.
168 // Remap temp VDInit variable to the address of the original
169 // variable
170 // (for proper handling of captured global variables).
171 LocalDeclMap[VDInit] = OriginalAddr;
172 EmitDecl(*VD);
173 LocalDeclMap.erase(VDInit);
174 return GetAddrOfLocalVar(VD);
175 });
176 }
177 assert(IsRegistered &&
178 "firstprivate var already registered as private");
179 // Silence the warning about unused variable.
180 (void)IsRegistered;
181 }
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000182 ++IRef, ++InitsRef;
183 }
184 }
Alexey Bataev69c62a92015-04-15 04:52:20 +0000185 return !EmittedAsFirstprivate.empty();
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000186}
187
Alexey Bataev03b340a2014-10-21 03:16:40 +0000188void CodeGenFunction::EmitOMPPrivateClause(
189 const OMPExecutableDirective &D,
190 CodeGenFunction::OMPPrivateScope &PrivateScope) {
Alexey Bataev50a64582015-04-22 12:24:45 +0000191 llvm::DenseSet<const VarDecl *> EmittedAsPrivate;
Alexey Bataevc925aa32015-04-27 08:00:32 +0000192 for (auto &&I = D.getClausesOfKind(OMPC_private); I; ++I) {
Alexey Bataev03b340a2014-10-21 03:16:40 +0000193 auto *C = cast<OMPPrivateClause>(*I);
194 auto IRef = C->varlist_begin();
195 for (auto IInit : C->private_copies()) {
196 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev50a64582015-04-22 12:24:45 +0000197 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
198 auto VD = cast<VarDecl>(cast<DeclRefExpr>(IInit)->getDecl());
199 bool IsRegistered =
200 PrivateScope.addPrivate(OrigVD, [&]() -> llvm::Value *{
201 // Emit private VarDecl with copy init.
202 EmitDecl(*VD);
203 return GetAddrOfLocalVar(VD);
204 });
205 assert(IsRegistered && "private var already registered as private");
206 // Silence the warning about unused variable.
207 (void)IsRegistered;
208 }
Alexey Bataev03b340a2014-10-21 03:16:40 +0000209 ++IRef;
210 }
211 }
212}
213
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000214bool CodeGenFunction::EmitOMPCopyinClause(const OMPExecutableDirective &D) {
215 // threadprivate_var1 = master_threadprivate_var1;
216 // operator=(threadprivate_var2, master_threadprivate_var2);
217 // ...
218 // __kmpc_barrier(&loc, global_tid);
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000219 llvm::DenseSet<const VarDecl *> CopiedVars;
220 llvm::BasicBlock *CopyBegin = nullptr, *CopyEnd = nullptr;
Alexey Bataevc925aa32015-04-27 08:00:32 +0000221 for (auto &&I = D.getClausesOfKind(OMPC_copyin); I; ++I) {
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000222 auto *C = cast<OMPCopyinClause>(*I);
223 auto IRef = C->varlist_begin();
224 auto ISrcRef = C->source_exprs().begin();
225 auto IDestRef = C->destination_exprs().begin();
226 for (auto *AssignOp : C->assignment_ops()) {
227 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000228 QualType Type = VD->getType();
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000229 if (CopiedVars.insert(VD->getCanonicalDecl()).second) {
Samuel Antao9c75cfe2015-07-27 16:38:06 +0000230
231 // Get the address of the master variable. If we are emitting code with
232 // TLS support, the address is passed from the master as field in the
233 // captured declaration.
234 llvm::Value *MasterAddr;
235 if (getLangOpts().OpenMPUseTLS &&
236 getContext().getTargetInfo().isTLSSupported()) {
237 assert(CapturedStmtInfo->lookup(VD) &&
238 "Copyin threadprivates should have been captured!");
239 DeclRefExpr DRE(const_cast<VarDecl *>(VD), true, (*IRef)->getType(),
240 VK_LValue, (*IRef)->getExprLoc());
241 MasterAddr = EmitLValue(&DRE).getAddress();
242 } else {
243 MasterAddr = VD->isStaticLocal() ? CGM.getStaticLocalDeclAddress(VD)
244 : CGM.GetAddrOfGlobal(VD);
245 }
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000246 // Get the address of the threadprivate variable.
247 auto *PrivateAddr = EmitLValue(*IRef).getAddress();
248 if (CopiedVars.size() == 1) {
249 // At first check if current thread is a master thread. If it is, no
250 // need to copy data.
251 CopyBegin = createBasicBlock("copyin.not.master");
252 CopyEnd = createBasicBlock("copyin.not.master.end");
253 Builder.CreateCondBr(
254 Builder.CreateICmpNE(
255 Builder.CreatePtrToInt(MasterAddr, CGM.IntPtrTy),
256 Builder.CreatePtrToInt(PrivateAddr, CGM.IntPtrTy)),
257 CopyBegin, CopyEnd);
258 EmitBlock(CopyBegin);
259 }
260 auto *SrcVD = cast<VarDecl>(cast<DeclRefExpr>(*ISrcRef)->getDecl());
261 auto *DestVD = cast<VarDecl>(cast<DeclRefExpr>(*IDestRef)->getDecl());
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000262 EmitOMPCopy(*this, Type, PrivateAddr, MasterAddr, DestVD, SrcVD,
263 AssignOp);
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000264 }
265 ++IRef;
266 ++ISrcRef;
267 ++IDestRef;
268 }
269 }
270 if (CopyEnd) {
271 // Exit out of copying procedure for non-master thread.
272 EmitBlock(CopyEnd, /*IsFinished=*/true);
273 return true;
274 }
275 return false;
276}
277
Alexey Bataev38e89532015-04-16 04:54:05 +0000278bool CodeGenFunction::EmitOMPLastprivateClauseInit(
279 const OMPExecutableDirective &D, OMPPrivateScope &PrivateScope) {
Alexey Bataev38e89532015-04-16 04:54:05 +0000280 bool HasAtLeastOneLastprivate = false;
281 llvm::DenseSet<const VarDecl *> AlreadyEmittedVars;
Alexey Bataevc925aa32015-04-27 08:00:32 +0000282 for (auto &&I = D.getClausesOfKind(OMPC_lastprivate); I; ++I) {
Alexey Bataevd130fd12015-05-13 10:23:02 +0000283 HasAtLeastOneLastprivate = true;
Alexey Bataev38e89532015-04-16 04:54:05 +0000284 auto *C = cast<OMPLastprivateClause>(*I);
285 auto IRef = C->varlist_begin();
286 auto IDestRef = C->destination_exprs().begin();
287 for (auto *IInit : C->private_copies()) {
288 // Keep the address of the original variable for future update at the end
289 // of the loop.
290 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
291 if (AlreadyEmittedVars.insert(OrigVD->getCanonicalDecl()).second) {
292 auto *DestVD = cast<VarDecl>(cast<DeclRefExpr>(*IDestRef)->getDecl());
293 PrivateScope.addPrivate(DestVD, [this, OrigVD, IRef]() -> llvm::Value *{
294 DeclRefExpr DRE(
295 const_cast<VarDecl *>(OrigVD),
296 /*RefersToEnclosingVariableOrCapture=*/CapturedStmtInfo->lookup(
297 OrigVD) != nullptr,
298 (*IRef)->getType(), VK_LValue, (*IRef)->getExprLoc());
299 return EmitLValue(&DRE).getAddress();
300 });
301 // Check if the variable is also a firstprivate: in this case IInit is
302 // not generated. Initialization of this variable will happen in codegen
303 // for 'firstprivate' clause.
Alexey Bataevd130fd12015-05-13 10:23:02 +0000304 if (IInit) {
305 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(IInit)->getDecl());
306 bool IsRegistered =
307 PrivateScope.addPrivate(OrigVD, [&]() -> llvm::Value *{
308 // Emit private VarDecl with copy init.
309 EmitDecl(*VD);
310 return GetAddrOfLocalVar(VD);
311 });
312 assert(IsRegistered &&
313 "lastprivate var already registered as private");
314 (void)IsRegistered;
315 }
Alexey Bataev38e89532015-04-16 04:54:05 +0000316 }
317 ++IRef, ++IDestRef;
318 }
319 }
320 return HasAtLeastOneLastprivate;
321}
322
323void CodeGenFunction::EmitOMPLastprivateClauseFinal(
324 const OMPExecutableDirective &D, llvm::Value *IsLastIterCond) {
325 // Emit following code:
326 // if (<IsLastIterCond>) {
327 // orig_var1 = private_orig_var1;
328 // ...
329 // orig_varn = private_orig_varn;
330 // }
Alexey Bataevfc087ec2015-06-16 13:14:42 +0000331 llvm::BasicBlock *ThenBB = nullptr;
332 llvm::BasicBlock *DoneBB = nullptr;
333 if (IsLastIterCond) {
334 ThenBB = createBasicBlock(".omp.lastprivate.then");
335 DoneBB = createBasicBlock(".omp.lastprivate.done");
336 Builder.CreateCondBr(IsLastIterCond, ThenBB, DoneBB);
337 EmitBlock(ThenBB);
338 }
Alexey Bataev7a228ff2015-05-21 07:59:51 +0000339 llvm::DenseMap<const Decl *, const Expr *> LoopCountersAndUpdates;
340 const Expr *LastIterVal = nullptr;
341 const Expr *IVExpr = nullptr;
342 const Expr *IncExpr = nullptr;
343 if (auto *LoopDirective = dyn_cast<OMPLoopDirective>(&D)) {
Alexey Bataevfc087ec2015-06-16 13:14:42 +0000344 if (isOpenMPWorksharingDirective(D.getDirectiveKind())) {
345 LastIterVal = cast<VarDecl>(cast<DeclRefExpr>(
346 LoopDirective->getUpperBoundVariable())
347 ->getDecl())
348 ->getAnyInitializer();
349 IVExpr = LoopDirective->getIterationVariable();
350 IncExpr = LoopDirective->getInc();
351 auto IUpdate = LoopDirective->updates().begin();
352 for (auto *E : LoopDirective->counters()) {
353 auto *D = cast<DeclRefExpr>(E)->getDecl()->getCanonicalDecl();
354 LoopCountersAndUpdates[D] = *IUpdate;
355 ++IUpdate;
356 }
Alexey Bataev7a228ff2015-05-21 07:59:51 +0000357 }
358 }
Alexey Bataev38e89532015-04-16 04:54:05 +0000359 {
Alexey Bataev38e89532015-04-16 04:54:05 +0000360 llvm::DenseSet<const VarDecl *> AlreadyEmittedVars;
Alexey Bataev7a228ff2015-05-21 07:59:51 +0000361 bool FirstLCV = true;
Alexey Bataevc925aa32015-04-27 08:00:32 +0000362 for (auto &&I = D.getClausesOfKind(OMPC_lastprivate); I; ++I) {
Alexey Bataev38e89532015-04-16 04:54:05 +0000363 auto *C = cast<OMPLastprivateClause>(*I);
364 auto IRef = C->varlist_begin();
365 auto ISrcRef = C->source_exprs().begin();
366 auto IDestRef = C->destination_exprs().begin();
367 for (auto *AssignOp : C->assignment_ops()) {
368 auto *PrivateVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000369 QualType Type = PrivateVD->getType();
Alexey Bataev7a228ff2015-05-21 07:59:51 +0000370 auto *CanonicalVD = PrivateVD->getCanonicalDecl();
371 if (AlreadyEmittedVars.insert(CanonicalVD).second) {
372 // If lastprivate variable is a loop control variable for loop-based
373 // directive, update its value before copyin back to original
374 // variable.
375 if (auto *UpExpr = LoopCountersAndUpdates.lookup(CanonicalVD)) {
Alexey Bataevfc087ec2015-06-16 13:14:42 +0000376 if (FirstLCV && LastIterVal) {
Alexey Bataev7a228ff2015-05-21 07:59:51 +0000377 EmitAnyExprToMem(LastIterVal, EmitLValue(IVExpr).getAddress(),
378 IVExpr->getType().getQualifiers(),
379 /*IsInitializer=*/false);
380 EmitIgnoredExpr(IncExpr);
381 FirstLCV = false;
382 }
383 EmitIgnoredExpr(UpExpr);
384 }
Alexey Bataev38e89532015-04-16 04:54:05 +0000385 auto *SrcVD = cast<VarDecl>(cast<DeclRefExpr>(*ISrcRef)->getDecl());
386 auto *DestVD = cast<VarDecl>(cast<DeclRefExpr>(*IDestRef)->getDecl());
387 // Get the address of the original variable.
388 auto *OriginalAddr = GetAddrOfLocalVar(DestVD);
389 // Get the address of the private variable.
390 auto *PrivateAddr = GetAddrOfLocalVar(PrivateVD);
Alexey Bataev1d9c15c2015-05-19 12:31:28 +0000391 EmitOMPCopy(*this, Type, OriginalAddr, PrivateAddr, DestVD, SrcVD,
392 AssignOp);
Alexey Bataev38e89532015-04-16 04:54:05 +0000393 }
394 ++IRef;
395 ++ISrcRef;
396 ++IDestRef;
397 }
398 }
399 }
Alexey Bataevfc087ec2015-06-16 13:14:42 +0000400 if (IsLastIterCond) {
401 EmitBlock(DoneBB, /*IsFinished=*/true);
402 }
Alexey Bataev38e89532015-04-16 04:54:05 +0000403}
404
Alexey Bataev794ba0d2015-04-10 10:43:45 +0000405void CodeGenFunction::EmitOMPReductionClauseInit(
406 const OMPExecutableDirective &D,
407 CodeGenFunction::OMPPrivateScope &PrivateScope) {
Alexey Bataevc925aa32015-04-27 08:00:32 +0000408 for (auto &&I = D.getClausesOfKind(OMPC_reduction); I; ++I) {
Alexey Bataev794ba0d2015-04-10 10:43:45 +0000409 auto *C = cast<OMPReductionClause>(*I);
410 auto ILHS = C->lhs_exprs().begin();
411 auto IRHS = C->rhs_exprs().begin();
412 for (auto IRef : C->varlists()) {
413 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(IRef)->getDecl());
414 auto *LHSVD = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
415 auto *PrivateVD = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
416 // Store the address of the original variable associated with the LHS
417 // implicit variable.
418 PrivateScope.addPrivate(LHSVD, [this, OrigVD, IRef]() -> llvm::Value *{
419 DeclRefExpr DRE(const_cast<VarDecl *>(OrigVD),
420 CapturedStmtInfo->lookup(OrigVD) != nullptr,
421 IRef->getType(), VK_LValue, IRef->getExprLoc());
422 return EmitLValue(&DRE).getAddress();
423 });
424 // Emit reduction copy.
425 bool IsRegistered =
426 PrivateScope.addPrivate(OrigVD, [this, PrivateVD]() -> llvm::Value *{
427 // Emit private VarDecl with reduction init.
428 EmitDecl(*PrivateVD);
429 return GetAddrOfLocalVar(PrivateVD);
430 });
431 assert(IsRegistered && "private var already registered as private");
432 // Silence the warning about unused variable.
433 (void)IsRegistered;
434 ++ILHS, ++IRHS;
435 }
436 }
437}
438
439void CodeGenFunction::EmitOMPReductionClauseFinal(
440 const OMPExecutableDirective &D) {
441 llvm::SmallVector<const Expr *, 8> LHSExprs;
442 llvm::SmallVector<const Expr *, 8> RHSExprs;
443 llvm::SmallVector<const Expr *, 8> ReductionOps;
Alexey Bataev794ba0d2015-04-10 10:43:45 +0000444 bool HasAtLeastOneReduction = false;
Alexey Bataevc925aa32015-04-27 08:00:32 +0000445 for (auto &&I = D.getClausesOfKind(OMPC_reduction); I; ++I) {
Alexey Bataev794ba0d2015-04-10 10:43:45 +0000446 HasAtLeastOneReduction = true;
447 auto *C = cast<OMPReductionClause>(*I);
448 LHSExprs.append(C->lhs_exprs().begin(), C->lhs_exprs().end());
449 RHSExprs.append(C->rhs_exprs().begin(), C->rhs_exprs().end());
450 ReductionOps.append(C->reduction_ops().begin(), C->reduction_ops().end());
451 }
452 if (HasAtLeastOneReduction) {
453 // Emit nowait reduction if nowait clause is present or directive is a
454 // parallel directive (it always has implicit barrier).
455 CGM.getOpenMPRuntime().emitReduction(
456 *this, D.getLocEnd(), LHSExprs, RHSExprs, ReductionOps,
457 D.getSingleClause(OMPC_nowait) ||
Alexey Bataev89e7e8e2015-06-17 06:21:39 +0000458 isOpenMPParallelDirective(D.getDirectiveKind()) ||
459 D.getDirectiveKind() == OMPD_simd,
460 D.getDirectiveKind() == OMPD_simd);
Alexey Bataev794ba0d2015-04-10 10:43:45 +0000461 }
462}
463
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000464static void emitCommonOMPParallelDirective(CodeGenFunction &CGF,
465 const OMPExecutableDirective &S,
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000466 OpenMPDirectiveKind InnermostKind,
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000467 const RegionCodeGenTy &CodeGen) {
Alexey Bataev18095712014-10-10 12:19:54 +0000468 auto CS = cast<CapturedStmt>(S.getAssociatedStmt());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000469 auto CapturedStruct = CGF.GenerateCapturedStmtArgument(*CS);
470 auto OutlinedFn = CGF.CGM.getOpenMPRuntime().emitParallelOutlinedFunction(
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000471 S, *CS->getCapturedDecl()->param_begin(), InnermostKind, CodeGen);
Alexey Bataev1d677132015-04-22 13:57:31 +0000472 if (auto C = S.getSingleClause(OMPC_num_threads)) {
473 CodeGenFunction::RunCleanupsScope NumThreadsScope(CGF);
474 auto NumThreadsClause = cast<OMPNumThreadsClause>(C);
475 auto NumThreads = CGF.EmitScalarExpr(NumThreadsClause->getNumThreads(),
476 /*IgnoreResultAssign*/ true);
477 CGF.CGM.getOpenMPRuntime().emitNumThreadsClause(
478 CGF, NumThreads, NumThreadsClause->getLocStart());
479 }
Alexey Bataev7f210c62015-06-18 13:40:03 +0000480 if (auto *C = S.getSingleClause(OMPC_proc_bind)) {
481 CodeGenFunction::RunCleanupsScope NumThreadsScope(CGF);
482 auto *ProcBindClause = cast<OMPProcBindClause>(C);
483 CGF.CGM.getOpenMPRuntime().emitProcBindClause(
484 CGF, ProcBindClause->getProcBindKind(), ProcBindClause->getLocStart());
485 }
Alexey Bataev1d677132015-04-22 13:57:31 +0000486 const Expr *IfCond = nullptr;
487 if (auto C = S.getSingleClause(OMPC_if)) {
488 IfCond = cast<OMPIfClause>(C)->getCondition();
489 }
490 CGF.CGM.getOpenMPRuntime().emitParallelCall(CGF, S.getLocStart(), OutlinedFn,
491 CapturedStruct, IfCond);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000492}
493
494void CodeGenFunction::EmitOMPParallelDirective(const OMPParallelDirective &S) {
495 LexicalScope Scope(*this, S.getSourceRange());
496 // Emit parallel region as a standalone region.
497 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
498 OMPPrivateScope PrivateScope(CGF);
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000499 bool Copyins = CGF.EmitOMPCopyinClause(S);
500 bool Firstprivates = CGF.EmitOMPFirstprivateClause(S, PrivateScope);
501 if (Copyins || Firstprivates) {
Alexey Bataev69c62a92015-04-15 04:52:20 +0000502 // Emit implicit barrier to synchronize threads and avoid data races on
Alexey Bataevf56f98c2015-04-16 05:39:01 +0000503 // initialization of firstprivate variables or propagation master's thread
504 // values of threadprivate variables to local instances of that variables
505 // of all other implicit threads.
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000506 CGF.CGM.getOpenMPRuntime().emitBarrierCall(CGF, S.getLocStart(),
507 OMPD_unknown);
Alexey Bataev69c62a92015-04-15 04:52:20 +0000508 }
509 CGF.EmitOMPPrivateClause(S, PrivateScope);
510 CGF.EmitOMPReductionClauseInit(S, PrivateScope);
511 (void)PrivateScope.Privatize();
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000512 CGF.EmitStmt(cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt());
Alexey Bataev794ba0d2015-04-10 10:43:45 +0000513 CGF.EmitOMPReductionClauseFinal(S);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000514 // Emit implicit barrier at the end of the 'parallel' directive.
515 CGF.CGM.getOpenMPRuntime().emitBarrierCall(CGF, S.getLocStart(),
516 OMPD_unknown);
517 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000518 emitCommonOMPParallelDirective(*this, S, OMPD_parallel, CodeGen);
Alexey Bataev9959db52014-05-06 10:08:46 +0000519}
Alexander Musman515ad8c2014-05-22 08:54:05 +0000520
Alexey Bataev0f34da12015-07-02 04:17:07 +0000521void CodeGenFunction::EmitOMPLoopBody(const OMPLoopDirective &D,
522 JumpDest LoopExit) {
Alexander Musmana5f070a2014-10-01 06:03:56 +0000523 RunCleanupsScope BodyScope(*this);
524 // Update counters values on current iteration.
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000525 for (auto I : D.updates()) {
Alexander Musmana5f070a2014-10-01 06:03:56 +0000526 EmitIgnoredExpr(I);
527 }
Alexander Musman3276a272015-03-21 10:12:56 +0000528 // Update the linear variables.
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000529 for (auto &&I = D.getClausesOfKind(OMPC_linear); I; ++I) {
Alexey Bataevc925aa32015-04-27 08:00:32 +0000530 auto *C = cast<OMPLinearClause>(*I);
Alexander Musman3276a272015-03-21 10:12:56 +0000531 for (auto U : C->updates()) {
532 EmitIgnoredExpr(U);
533 }
534 }
535
Alexander Musmana5f070a2014-10-01 06:03:56 +0000536 // On a continue in the body, jump to the end.
Alexander Musmand196ef22014-10-07 08:57:09 +0000537 auto Continue = getJumpDestInCurrentScope("omp.body.continue");
Alexey Bataev0f34da12015-07-02 04:17:07 +0000538 BreakContinueStack.push_back(BreakContinue(LoopExit, Continue));
Alexander Musmana5f070a2014-10-01 06:03:56 +0000539 // Emit loop body.
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000540 EmitStmt(D.getBody());
Alexander Musmana5f070a2014-10-01 06:03:56 +0000541 // The end (updates/cleanups).
542 EmitBlock(Continue.getBlock());
543 BreakContinueStack.pop_back();
Alexander Musmana5f070a2014-10-01 06:03:56 +0000544 // TODO: Update lastprivates if the SeparateIter flag is true.
545 // This will be implemented in a follow-up OMPLastprivateClause patch, but
546 // result should be still correct without it, as we do not make these
547 // variables private yet.
Alexander Musmana5f070a2014-10-01 06:03:56 +0000548}
549
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000550void CodeGenFunction::EmitOMPInnerLoop(
551 const Stmt &S, bool RequiresCleanup, const Expr *LoopCond,
552 const Expr *IncExpr,
Alexey Bataev98eb6e32015-04-22 11:15:40 +0000553 const llvm::function_ref<void(CodeGenFunction &)> &BodyGen,
554 const llvm::function_ref<void(CodeGenFunction &)> &PostIncGen) {
Alexander Musmand196ef22014-10-07 08:57:09 +0000555 auto LoopExit = getJumpDestInCurrentScope("omp.inner.for.end");
Alexander Musmana5f070a2014-10-01 06:03:56 +0000556
557 // Start the loop with a block that tests the condition.
Alexander Musmand196ef22014-10-07 08:57:09 +0000558 auto CondBlock = createBasicBlock("omp.inner.for.cond");
Alexander Musmana5f070a2014-10-01 06:03:56 +0000559 EmitBlock(CondBlock);
560 LoopStack.push(CondBlock);
561
562 // If there are any cleanups between here and the loop-exit scope,
563 // create a block to stage a loop exit along.
564 auto ExitBlock = LoopExit.getBlock();
Alexey Bataev2df54a02015-03-12 08:53:29 +0000565 if (RequiresCleanup)
Alexander Musmand196ef22014-10-07 08:57:09 +0000566 ExitBlock = createBasicBlock("omp.inner.for.cond.cleanup");
Alexander Musmana5f070a2014-10-01 06:03:56 +0000567
Alexander Musmand196ef22014-10-07 08:57:09 +0000568 auto LoopBody = createBasicBlock("omp.inner.for.body");
Alexander Musmana5f070a2014-10-01 06:03:56 +0000569
Alexey Bataev2df54a02015-03-12 08:53:29 +0000570 // Emit condition.
Justin Bogner66242d62015-04-23 23:06:47 +0000571 EmitBranchOnBoolExpr(LoopCond, LoopBody, ExitBlock, getProfileCount(&S));
Alexander Musmana5f070a2014-10-01 06:03:56 +0000572 if (ExitBlock != LoopExit.getBlock()) {
573 EmitBlock(ExitBlock);
574 EmitBranchThroughCleanup(LoopExit);
575 }
576
577 EmitBlock(LoopBody);
Justin Bogner66242d62015-04-23 23:06:47 +0000578 incrementProfileCounter(&S);
Alexander Musmana5f070a2014-10-01 06:03:56 +0000579
580 // Create a block for the increment.
Alexander Musmand196ef22014-10-07 08:57:09 +0000581 auto Continue = getJumpDestInCurrentScope("omp.inner.for.inc");
Alexander Musmana5f070a2014-10-01 06:03:56 +0000582 BreakContinueStack.push_back(BreakContinue(LoopExit, Continue));
583
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000584 BodyGen(*this);
Alexander Musmana5f070a2014-10-01 06:03:56 +0000585
586 // Emit "IV = IV + 1" and a back-edge to the condition block.
587 EmitBlock(Continue.getBlock());
Alexey Bataev2df54a02015-03-12 08:53:29 +0000588 EmitIgnoredExpr(IncExpr);
Alexey Bataev98eb6e32015-04-22 11:15:40 +0000589 PostIncGen(*this);
Alexander Musmana5f070a2014-10-01 06:03:56 +0000590 BreakContinueStack.pop_back();
591 EmitBranch(CondBlock);
592 LoopStack.pop();
593 // Emit the fall-through block.
594 EmitBlock(LoopExit.getBlock());
595}
596
Alexey Bataev3b5b5c42015-06-18 10:10:12 +0000597void CodeGenFunction::EmitOMPLinearClauseInit(const OMPLoopDirective &D) {
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000598 // Emit inits for the linear variables.
599 for (auto &&I = D.getClausesOfKind(OMPC_linear); I; ++I) {
600 auto *C = cast<OMPLinearClause>(*I);
601 for (auto Init : C->inits()) {
602 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(Init)->getDecl());
Alexey Bataev3b5b5c42015-06-18 10:10:12 +0000603 auto *OrigVD = cast<VarDecl>(
604 cast<DeclRefExpr>(VD->getInit()->IgnoreImpCasts())->getDecl());
605 DeclRefExpr DRE(const_cast<VarDecl *>(OrigVD),
606 CapturedStmtInfo->lookup(OrigVD) != nullptr,
607 VD->getInit()->getType(), VK_LValue,
608 VD->getInit()->getExprLoc());
609 AutoVarEmission Emission = EmitAutoVarAlloca(*VD);
610 EmitExprAsInit(&DRE, VD,
611 MakeAddrLValue(Emission.getAllocatedAddress(),
612 VD->getType(), Emission.Alignment),
613 /*capturedByInit=*/false);
614 EmitAutoVarCleanups(Emission);
Alexander Musmana5f070a2014-10-01 06:03:56 +0000615 }
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000616 // Emit the linear steps for the linear clauses.
617 // If a step is not constant, it is pre-calculated before the loop.
618 if (auto CS = cast_or_null<BinaryOperator>(C->getCalcStep()))
619 if (auto SaveRef = cast<DeclRefExpr>(CS->getLHS())) {
Alexey Bataev3b5b5c42015-06-18 10:10:12 +0000620 EmitVarDecl(*cast<VarDecl>(SaveRef->getDecl()));
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000621 // Emit calculation of the linear step.
Alexey Bataev3b5b5c42015-06-18 10:10:12 +0000622 EmitIgnoredExpr(CS);
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000623 }
Alexander Musmana5f070a2014-10-01 06:03:56 +0000624 }
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000625}
626
627static void emitLinearClauseFinal(CodeGenFunction &CGF,
628 const OMPLoopDirective &D) {
Alexander Musman3276a272015-03-21 10:12:56 +0000629 // Emit the final values of the linear variables.
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000630 for (auto &&I = D.getClausesOfKind(OMPC_linear); I; ++I) {
Alexey Bataevc925aa32015-04-27 08:00:32 +0000631 auto *C = cast<OMPLinearClause>(*I);
Alexey Bataev39f915b82015-05-08 10:41:21 +0000632 auto IC = C->varlist_begin();
Alexander Musman3276a272015-03-21 10:12:56 +0000633 for (auto F : C->finals()) {
Alexey Bataev39f915b82015-05-08 10:41:21 +0000634 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IC)->getDecl());
635 DeclRefExpr DRE(const_cast<VarDecl *>(OrigVD),
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000636 CGF.CapturedStmtInfo->lookup(OrigVD) != nullptr,
Alexey Bataev39f915b82015-05-08 10:41:21 +0000637 (*IC)->getType(), VK_LValue, (*IC)->getExprLoc());
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000638 auto *OrigAddr = CGF.EmitLValue(&DRE).getAddress();
639 CodeGenFunction::OMPPrivateScope VarScope(CGF);
Alexey Bataev39f915b82015-05-08 10:41:21 +0000640 VarScope.addPrivate(OrigVD,
641 [OrigAddr]() -> llvm::Value *{ return OrigAddr; });
642 (void)VarScope.Privatize();
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000643 CGF.EmitIgnoredExpr(F);
Alexey Bataev39f915b82015-05-08 10:41:21 +0000644 ++IC;
Alexander Musman3276a272015-03-21 10:12:56 +0000645 }
646 }
Alexander Musmana5f070a2014-10-01 06:03:56 +0000647}
648
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000649static void emitAlignedClause(CodeGenFunction &CGF,
650 const OMPExecutableDirective &D) {
651 for (auto &&I = D.getClausesOfKind(OMPC_aligned); I; ++I) {
652 auto *Clause = cast<OMPAlignedClause>(*I);
653 unsigned ClauseAlignment = 0;
654 if (auto AlignmentExpr = Clause->getAlignment()) {
655 auto AlignmentCI =
656 cast<llvm::ConstantInt>(CGF.EmitScalarExpr(AlignmentExpr));
657 ClauseAlignment = static_cast<unsigned>(AlignmentCI->getZExtValue());
Alexander Musman09184fe2014-09-30 05:29:28 +0000658 }
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000659 for (auto E : Clause->varlists()) {
660 unsigned Alignment = ClauseAlignment;
661 if (Alignment == 0) {
662 // OpenMP [2.8.1, Description]
663 // If no optional parameter is specified, implementation-defined default
664 // alignments for SIMD instructions on the target platforms are assumed.
665 Alignment =
Alexey Bataev00396512015-07-02 03:40:19 +0000666 CGF.getContext()
667 .toCharUnitsFromBits(CGF.getContext().getOpenMPDefaultSimdAlign(
668 E->getType()->getPointeeType()))
669 .getQuantity();
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000670 }
671 assert((Alignment == 0 || llvm::isPowerOf2_32(Alignment)) &&
672 "alignment is not power of 2");
673 if (Alignment != 0) {
674 llvm::Value *PtrValue = CGF.EmitScalarExpr(E);
675 CGF.EmitAlignmentAssumption(PtrValue, Alignment);
676 }
Alexander Musman09184fe2014-09-30 05:29:28 +0000677 }
678 }
679}
680
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000681static void emitPrivateLoopCounters(CodeGenFunction &CGF,
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000682 CodeGenFunction::OMPPrivateScope &LoopScope,
Alexey Bataeva8899172015-08-06 12:30:57 +0000683 ArrayRef<Expr *> Counters,
684 ArrayRef<Expr *> PrivateCounters) {
685 auto I = PrivateCounters.begin();
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000686 for (auto *E : Counters) {
Alexey Bataeva8899172015-08-06 12:30:57 +0000687 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
688 auto *PrivateVD = cast<VarDecl>(cast<DeclRefExpr>(*I)->getDecl());
689 llvm::Value *Addr;
690 (void)LoopScope.addPrivate(PrivateVD, [&]() -> llvm::Value * {
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000691 // Emit var without initialization.
Alexey Bataeva8899172015-08-06 12:30:57 +0000692 auto VarEmission = CGF.EmitAutoVarAlloca(*PrivateVD);
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000693 CGF.EmitAutoVarCleanups(VarEmission);
Alexey Bataeva8899172015-08-06 12:30:57 +0000694 Addr = VarEmission.getAllocatedAddress();
695 return Addr;
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000696 });
Alexey Bataeva8899172015-08-06 12:30:57 +0000697 (void)LoopScope.addPrivate(VD, [&]() -> llvm::Value * { return Addr; });
698 ++I;
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000699 }
Alexey Bataev435ad7b2014-10-10 09:48:26 +0000700}
701
Alexey Bataev62dbb972015-04-22 11:59:37 +0000702static void emitPreCond(CodeGenFunction &CGF, const OMPLoopDirective &S,
703 const Expr *Cond, llvm::BasicBlock *TrueBlock,
704 llvm::BasicBlock *FalseBlock, uint64_t TrueCount) {
Alexey Bataev6e8248f2015-06-11 10:53:56 +0000705 {
706 CodeGenFunction::OMPPrivateScope PreCondScope(CGF);
Alexey Bataeva8899172015-08-06 12:30:57 +0000707 emitPrivateLoopCounters(CGF, PreCondScope, S.counters(),
708 S.private_counters());
Alexey Bataev6e8248f2015-06-11 10:53:56 +0000709 const VarDecl *IVDecl =
710 cast<VarDecl>(cast<DeclRefExpr>(S.getIterationVariable())->getDecl());
711 bool IsRegistered = PreCondScope.addPrivate(IVDecl, [&]() -> llvm::Value *{
712 // Emit var without initialization.
713 auto VarEmission = CGF.EmitAutoVarAlloca(*IVDecl);
714 CGF.EmitAutoVarCleanups(VarEmission);
715 return VarEmission.getAllocatedAddress();
716 });
717 assert(IsRegistered && "counter already registered as private");
718 // Silence the warning about unused variable.
719 (void)IsRegistered;
720 (void)PreCondScope.Privatize();
721 // Initialize internal counter to 0 to calculate initial values of real
722 // counters.
723 LValue IV = CGF.EmitLValue(S.getIterationVariable());
724 CGF.EmitStoreOfScalar(
725 llvm::ConstantInt::getNullValue(
726 IV.getAddress()->getType()->getPointerElementType()),
727 CGF.EmitLValue(S.getIterationVariable()), /*isInit=*/true);
728 // Get initial values of real counters.
729 for (auto I : S.updates()) {
730 CGF.EmitIgnoredExpr(I);
731 }
Alexey Bataev62dbb972015-04-22 11:59:37 +0000732 }
733 // Check that loop is executed at least one time.
734 CGF.EmitBranchOnBoolExpr(Cond, TrueBlock, FalseBlock, TrueCount);
735}
736
Alexander Musman3276a272015-03-21 10:12:56 +0000737static void
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000738emitPrivateLinearVars(CodeGenFunction &CGF, const OMPExecutableDirective &D,
Alexander Musman3276a272015-03-21 10:12:56 +0000739 CodeGenFunction::OMPPrivateScope &PrivateScope) {
Alexey Bataevc925aa32015-04-27 08:00:32 +0000740 for (auto &&I = D.getClausesOfKind(OMPC_linear); I; ++I) {
741 auto *C = cast<OMPLinearClause>(*I);
742 for (auto *E : C->varlists()) {
Alexander Musman3276a272015-03-21 10:12:56 +0000743 auto VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
744 bool IsRegistered = PrivateScope.addPrivate(VD, [&]()->llvm::Value * {
745 // Emit var without initialization.
746 auto VarEmission = CGF.EmitAutoVarAlloca(*VD);
747 CGF.EmitAutoVarCleanups(VarEmission);
748 return VarEmission.getAllocatedAddress();
749 });
750 assert(IsRegistered && "linear var already registered as private");
751 // Silence the warning about unused variable.
752 (void)IsRegistered;
753 }
754 }
755}
756
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000757static void emitSafelenClause(CodeGenFunction &CGF,
758 const OMPExecutableDirective &D) {
759 if (auto *C =
760 cast_or_null<OMPSafelenClause>(D.getSingleClause(OMPC_safelen))) {
761 RValue Len = CGF.EmitAnyExpr(C->getSafelen(), AggValueSlot::ignored(),
762 /*ignoreResult=*/true);
763 llvm::ConstantInt *Val = cast<llvm::ConstantInt>(Len.getScalarVal());
Tyler Nowickida46d0e2015-07-14 23:03:09 +0000764 CGF.LoopStack.setVectorizeWidth(Val->getZExtValue());
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000765 // In presence of finite 'safelen', it may be unsafe to mark all
766 // the memory instructions parallel, because loop-carried
767 // dependences of 'safelen' iterations are possible.
768 CGF.LoopStack.setParallel(false);
769 }
770}
771
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000772void CodeGenFunction::EmitOMPSimdInit(const OMPLoopDirective &D) {
773 // Walk clauses and process safelen/lastprivate.
774 LoopStack.setParallel();
Tyler Nowickida46d0e2015-07-14 23:03:09 +0000775 LoopStack.setVectorizeEnable(true);
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000776 emitSafelenClause(*this, D);
777}
778
779void CodeGenFunction::EmitOMPSimdFinal(const OMPLoopDirective &D) {
780 auto IC = D.counters().begin();
781 for (auto F : D.finals()) {
782 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>((*IC))->getDecl());
Alexey Bataev3b5b5c42015-06-18 10:10:12 +0000783 if (LocalDeclMap.lookup(OrigVD) || CapturedStmtInfo->lookup(OrigVD)) {
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000784 DeclRefExpr DRE(const_cast<VarDecl *>(OrigVD),
785 CapturedStmtInfo->lookup(OrigVD) != nullptr,
786 (*IC)->getType(), VK_LValue, (*IC)->getExprLoc());
787 auto *OrigAddr = EmitLValue(&DRE).getAddress();
788 OMPPrivateScope VarScope(*this);
789 VarScope.addPrivate(OrigVD,
790 [OrigAddr]() -> llvm::Value *{ return OrigAddr; });
791 (void)VarScope.Privatize();
792 EmitIgnoredExpr(F);
793 }
794 ++IC;
795 }
796 emitLinearClauseFinal(*this, D);
797}
798
Alexander Musman515ad8c2014-05-22 08:54:05 +0000799void CodeGenFunction::EmitOMPSimdDirective(const OMPSimdDirective &S) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000800 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
Alexey Bataev62dbb972015-04-22 11:59:37 +0000801 // if (PreCond) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000802 // for (IV in 0..LastIteration) BODY;
803 // <Final counter/linear vars updates>;
Alexey Bataev62dbb972015-04-22 11:59:37 +0000804 // }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000805 //
Alexander Musmana5f070a2014-10-01 06:03:56 +0000806
Alexey Bataev62dbb972015-04-22 11:59:37 +0000807 // Emit: if (PreCond) - begin.
808 // If the condition constant folds and can be elided, avoid emitting the
809 // whole loop.
810 bool CondConstant;
811 llvm::BasicBlock *ContBlock = nullptr;
812 if (CGF.ConstantFoldsToSimpleInteger(S.getPreCond(), CondConstant)) {
813 if (!CondConstant)
814 return;
815 } else {
Alexey Bataev62dbb972015-04-22 11:59:37 +0000816 auto *ThenBlock = CGF.createBasicBlock("simd.if.then");
817 ContBlock = CGF.createBasicBlock("simd.if.end");
Justin Bogner66242d62015-04-23 23:06:47 +0000818 emitPreCond(CGF, S, S.getPreCond(), ThenBlock, ContBlock,
819 CGF.getProfileCount(&S));
Alexey Bataev62dbb972015-04-22 11:59:37 +0000820 CGF.EmitBlock(ThenBlock);
Justin Bogner66242d62015-04-23 23:06:47 +0000821 CGF.incrementProfileCounter(&S);
Alexey Bataev62dbb972015-04-22 11:59:37 +0000822 }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000823
824 // Emit the loop iteration variable.
825 const Expr *IVExpr = S.getIterationVariable();
826 const VarDecl *IVDecl = cast<VarDecl>(cast<DeclRefExpr>(IVExpr)->getDecl());
827 CGF.EmitVarDecl(*IVDecl);
828 CGF.EmitIgnoredExpr(S.getInit());
829
830 // Emit the iterations count variable.
831 // If it is not a variable, Sema decided to calculate iterations count on
Alexey Bataev7a228ff2015-05-21 07:59:51 +0000832 // each iteration (e.g., it is foldable into a constant).
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000833 if (auto LIExpr = dyn_cast<DeclRefExpr>(S.getLastIteration())) {
834 CGF.EmitVarDecl(*cast<VarDecl>(LIExpr->getDecl()));
835 // Emit calculation of the iterations count.
836 CGF.EmitIgnoredExpr(S.getCalcLastIteration());
Alexander Musmana5f070a2014-10-01 06:03:56 +0000837 }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000838
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000839 CGF.EmitOMPSimdInit(S);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000840
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000841 emitAlignedClause(CGF, S);
Alexey Bataev3b5b5c42015-06-18 10:10:12 +0000842 CGF.EmitOMPLinearClauseInit(S);
Alexey Bataevfc087ec2015-06-16 13:14:42 +0000843 bool HasLastprivateClause;
Alexey Bataev62dbb972015-04-22 11:59:37 +0000844 {
845 OMPPrivateScope LoopScope(CGF);
Alexey Bataeva8899172015-08-06 12:30:57 +0000846 emitPrivateLoopCounters(CGF, LoopScope, S.counters(),
847 S.private_counters());
Alexey Bataevcbdcbb72015-06-17 07:45:51 +0000848 emitPrivateLinearVars(CGF, S, LoopScope);
Alexey Bataev62dbb972015-04-22 11:59:37 +0000849 CGF.EmitOMPPrivateClause(S, LoopScope);
Alexey Bataev89e7e8e2015-06-17 06:21:39 +0000850 CGF.EmitOMPReductionClauseInit(S, LoopScope);
Alexey Bataevfc087ec2015-06-16 13:14:42 +0000851 HasLastprivateClause = CGF.EmitOMPLastprivateClauseInit(S, LoopScope);
Alexey Bataev62dbb972015-04-22 11:59:37 +0000852 (void)LoopScope.Privatize();
Alexey Bataev0f34da12015-07-02 04:17:07 +0000853 CGF.EmitOMPInnerLoop(S, LoopScope.requiresCleanups(), S.getCond(),
854 S.getInc(),
Alexey Bataev62dbb972015-04-22 11:59:37 +0000855 [&S](CodeGenFunction &CGF) {
Alexey Bataev0f34da12015-07-02 04:17:07 +0000856 CGF.EmitOMPLoopBody(S, JumpDest());
Alexey Bataev62dbb972015-04-22 11:59:37 +0000857 CGF.EmitStopPoint(&S);
858 },
859 [](CodeGenFunction &) {});
Alexey Bataevfc087ec2015-06-16 13:14:42 +0000860 // Emit final copy of the lastprivate variables at the end of loops.
861 if (HasLastprivateClause) {
862 CGF.EmitOMPLastprivateClauseFinal(S);
863 }
Alexey Bataev89e7e8e2015-06-17 06:21:39 +0000864 CGF.EmitOMPReductionClauseFinal(S);
Alexey Bataev62dbb972015-04-22 11:59:37 +0000865 }
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000866 CGF.EmitOMPSimdFinal(S);
Alexey Bataev62dbb972015-04-22 11:59:37 +0000867 // Emit: if (PreCond) - end.
868 if (ContBlock) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000869 CGF.EmitBranch(ContBlock);
870 CGF.EmitBlock(ContBlock, true);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000871 }
872 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000873 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_simd, CodeGen);
Alexander Musman515ad8c2014-05-22 08:54:05 +0000874}
875
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000876void CodeGenFunction::EmitOMPForOuterLoop(OpenMPScheduleClauseKind ScheduleKind,
877 const OMPLoopDirective &S,
878 OMPPrivateScope &LoopScope,
Alexey Bataevd7589ffe2015-05-20 13:12:48 +0000879 bool Ordered, llvm::Value *LB,
880 llvm::Value *UB, llvm::Value *ST,
881 llvm::Value *IL, llvm::Value *Chunk) {
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000882 auto &RT = CGM.getOpenMPRuntime();
Alexander Musman92bdaab2015-03-12 13:37:50 +0000883
884 // Dynamic scheduling of the outer loop (dynamic, guided, auto, runtime).
Alexey Bataevd7589ffe2015-05-20 13:12:48 +0000885 const bool DynamicOrOrdered = Ordered || RT.isDynamic(ScheduleKind);
Alexander Musman92bdaab2015-03-12 13:37:50 +0000886
Alexey Bataevd7589ffe2015-05-20 13:12:48 +0000887 assert((Ordered ||
888 !RT.isStaticNonchunked(ScheduleKind, /*Chunked=*/Chunk != nullptr)) &&
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000889 "static non-chunked schedule does not need outer loop");
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000890
891 // Emit outer loop.
892 //
893 // OpenMP [2.7.1, Loop Construct, Description, table 2-1]
Alexander Musman92bdaab2015-03-12 13:37:50 +0000894 // When schedule(dynamic,chunk_size) is specified, the iterations are
895 // distributed to threads in the team in chunks as the threads request them.
896 // Each thread executes a chunk of iterations, then requests another chunk,
897 // until no chunks remain to be distributed. Each chunk contains chunk_size
898 // iterations, except for the last chunk to be distributed, which may have
899 // fewer iterations. When no chunk_size is specified, it defaults to 1.
900 //
901 // When schedule(guided,chunk_size) is specified, the iterations are assigned
902 // to threads in the team in chunks as the executing threads request them.
903 // Each thread executes a chunk of iterations, then requests another chunk,
904 // until no chunks remain to be assigned. For a chunk_size of 1, the size of
905 // each chunk is proportional to the number of unassigned iterations divided
906 // by the number of threads in the team, decreasing to 1. For a chunk_size
907 // with value k (greater than 1), the size of each chunk is determined in the
908 // same way, with the restriction that the chunks do not contain fewer than k
909 // iterations (except for the last chunk to be assigned, which may have fewer
910 // than k iterations).
911 //
912 // When schedule(auto) is specified, the decision regarding scheduling is
913 // delegated to the compiler and/or runtime system. The programmer gives the
914 // implementation the freedom to choose any possible mapping of iterations to
915 // threads in the team.
916 //
917 // When schedule(runtime) is specified, the decision regarding scheduling is
918 // deferred until run time, and the schedule and chunk size are taken from the
919 // run-sched-var ICV. If the ICV is set to auto, the schedule is
920 // implementation defined
921 //
922 // while(__kmpc_dispatch_next(&LB, &UB)) {
923 // idx = LB;
Alexey Bataev98eb6e32015-04-22 11:15:40 +0000924 // while (idx <= UB) { BODY; ++idx;
925 // __kmpc_dispatch_fini_(4|8)[u](); // For ordered loops only.
926 // } // inner loop
Alexander Musman92bdaab2015-03-12 13:37:50 +0000927 // }
928 //
929 // OpenMP [2.7.1, Loop Construct, Description, table 2-1]
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000930 // When schedule(static, chunk_size) is specified, iterations are divided into
931 // chunks of size chunk_size, and the chunks are assigned to the threads in
932 // the team in a round-robin fashion in the order of the thread number.
933 //
934 // while(UB = min(UB, GlobalUB), idx = LB, idx < UB) {
935 // while (idx <= UB) { BODY; ++idx; } // inner loop
936 // LB = LB + ST;
937 // UB = UB + ST;
938 // }
939 //
Alexander Musman92bdaab2015-03-12 13:37:50 +0000940
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000941 const Expr *IVExpr = S.getIterationVariable();
942 const unsigned IVSize = getContext().getTypeSize(IVExpr->getType());
943 const bool IVSigned = IVExpr->getType()->hasSignedIntegerRepresentation();
944
Alexander Musman92bdaab2015-03-12 13:37:50 +0000945 RT.emitForInit(
Alexey Bataevd7589ffe2015-05-20 13:12:48 +0000946 *this, S.getLocStart(), ScheduleKind, IVSize, IVSigned, Ordered, IL, LB,
947 (DynamicOrOrdered ? EmitAnyExpr(S.getLastIteration()).getScalarVal()
948 : UB),
949 ST, Chunk);
Alexander Musman92bdaab2015-03-12 13:37:50 +0000950
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000951 auto LoopExit = getJumpDestInCurrentScope("omp.dispatch.end");
952
953 // Start the loop with a block that tests the condition.
954 auto CondBlock = createBasicBlock("omp.dispatch.cond");
955 EmitBlock(CondBlock);
956 LoopStack.push(CondBlock);
957
958 llvm::Value *BoolCondVal = nullptr;
Alexey Bataevd7589ffe2015-05-20 13:12:48 +0000959 if (!DynamicOrOrdered) {
Alexander Musman92bdaab2015-03-12 13:37:50 +0000960 // UB = min(UB, GlobalUB)
961 EmitIgnoredExpr(S.getEnsureUpperBound());
962 // IV = LB
963 EmitIgnoredExpr(S.getInit());
964 // IV < UB
Alexey Bataevae05c292015-06-16 11:59:36 +0000965 BoolCondVal = EvaluateExprAsBool(S.getCond());
Alexander Musman92bdaab2015-03-12 13:37:50 +0000966 } else {
967 BoolCondVal = RT.emitForNext(*this, S.getLocStart(), IVSize, IVSigned,
968 IL, LB, UB, ST);
969 }
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000970
971 // If there are any cleanups between here and the loop-exit scope,
972 // create a block to stage a loop exit along.
973 auto ExitBlock = LoopExit.getBlock();
974 if (LoopScope.requiresCleanups())
975 ExitBlock = createBasicBlock("omp.dispatch.cleanup");
976
977 auto LoopBody = createBasicBlock("omp.dispatch.body");
978 Builder.CreateCondBr(BoolCondVal, LoopBody, ExitBlock);
979 if (ExitBlock != LoopExit.getBlock()) {
980 EmitBlock(ExitBlock);
981 EmitBranchThroughCleanup(LoopExit);
982 }
983 EmitBlock(LoopBody);
984
Alexander Musman92bdaab2015-03-12 13:37:50 +0000985 // Emit "IV = LB" (in case of static schedule, we have already calculated new
986 // LB for loop condition and emitted it above).
Alexey Bataevd7589ffe2015-05-20 13:12:48 +0000987 if (DynamicOrOrdered)
Alexander Musman92bdaab2015-03-12 13:37:50 +0000988 EmitIgnoredExpr(S.getInit());
989
Alexander Musmandf7a8e22015-01-22 08:49:35 +0000990 // Create a block for the increment.
991 auto Continue = getJumpDestInCurrentScope("omp.dispatch.inc");
992 BreakContinueStack.push_back(BreakContinue(LoopExit, Continue));
993
Alexey Bataev58e5bdb2015-06-18 04:45:29 +0000994 // Generate !llvm.loop.parallel metadata for loads and stores for loops
995 // with dynamic/guided scheduling and without ordered clause.
996 if (!isOpenMPSimdDirective(S.getDirectiveKind())) {
997 LoopStack.setParallel((ScheduleKind == OMPC_SCHEDULE_dynamic ||
998 ScheduleKind == OMPC_SCHEDULE_guided) &&
999 !Ordered);
1000 } else {
1001 EmitOMPSimdInit(S);
1002 }
1003
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001004 SourceLocation Loc = S.getLocStart();
Alexey Bataev0f34da12015-07-02 04:17:07 +00001005 EmitOMPInnerLoop(S, LoopScope.requiresCleanups(), S.getCond(), S.getInc(),
1006 [&S, LoopExit](CodeGenFunction &CGF) {
1007 CGF.EmitOMPLoopBody(S, LoopExit);
1008 CGF.EmitStopPoint(&S);
1009 },
1010 [Ordered, IVSize, IVSigned, Loc](CodeGenFunction &CGF) {
1011 if (Ordered) {
1012 CGF.CGM.getOpenMPRuntime().emitForOrderedIterationEnd(
1013 CGF, Loc, IVSize, IVSigned);
1014 }
1015 });
Alexander Musmandf7a8e22015-01-22 08:49:35 +00001016
1017 EmitBlock(Continue.getBlock());
1018 BreakContinueStack.pop_back();
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001019 if (!DynamicOrOrdered) {
Alexander Musman92bdaab2015-03-12 13:37:50 +00001020 // Emit "LB = LB + Stride", "UB = UB + Stride".
1021 EmitIgnoredExpr(S.getNextLowerBound());
1022 EmitIgnoredExpr(S.getNextUpperBound());
1023 }
Alexander Musmandf7a8e22015-01-22 08:49:35 +00001024
1025 EmitBranch(CondBlock);
1026 LoopStack.pop();
1027 // Emit the fall-through block.
1028 EmitBlock(LoopExit.getBlock());
1029
1030 // Tell the runtime we are done.
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001031 if (!DynamicOrOrdered)
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001032 RT.emitForStaticFinish(*this, S.getLocEnd());
Alexander Musmandf7a8e22015-01-22 08:49:35 +00001033}
1034
Alexander Musmanc6388682014-12-15 07:07:06 +00001035/// \brief Emit a helper variable and return corresponding lvalue.
1036static LValue EmitOMPHelperVar(CodeGenFunction &CGF,
1037 const DeclRefExpr *Helper) {
1038 auto VDecl = cast<VarDecl>(Helper->getDecl());
1039 CGF.EmitVarDecl(*VDecl);
1040 return CGF.EmitLValue(Helper);
1041}
1042
Alexey Bataev040d5402015-05-12 08:35:28 +00001043static std::pair<llvm::Value * /*Chunk*/, OpenMPScheduleClauseKind>
1044emitScheduleClause(CodeGenFunction &CGF, const OMPLoopDirective &S,
1045 bool OuterRegion) {
1046 // Detect the loop schedule kind and chunk.
1047 auto ScheduleKind = OMPC_SCHEDULE_unknown;
1048 llvm::Value *Chunk = nullptr;
1049 if (auto *C =
1050 cast_or_null<OMPScheduleClause>(S.getSingleClause(OMPC_schedule))) {
1051 ScheduleKind = C->getScheduleKind();
1052 if (const auto *Ch = C->getChunkSize()) {
1053 if (auto *ImpRef = cast_or_null<DeclRefExpr>(C->getHelperChunkSize())) {
1054 if (OuterRegion) {
1055 const VarDecl *ImpVar = cast<VarDecl>(ImpRef->getDecl());
1056 CGF.EmitVarDecl(*ImpVar);
1057 CGF.EmitStoreThroughLValue(
1058 CGF.EmitAnyExpr(Ch),
1059 CGF.MakeNaturalAlignAddrLValue(CGF.GetAddrOfLocalVar(ImpVar),
1060 ImpVar->getType()));
1061 } else {
1062 Ch = ImpRef;
1063 }
1064 }
1065 if (!C->getHelperChunkSize() || !OuterRegion) {
1066 Chunk = CGF.EmitScalarExpr(Ch);
1067 Chunk = CGF.EmitScalarConversion(Chunk, Ch->getType(),
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001068 S.getIterationVariable()->getType(),
1069 S.getLocStart());
Alexey Bataev040d5402015-05-12 08:35:28 +00001070 }
1071 }
1072 }
1073 return std::make_pair(Chunk, ScheduleKind);
1074}
1075
Alexey Bataev38e89532015-04-16 04:54:05 +00001076bool CodeGenFunction::EmitOMPWorksharingLoop(const OMPLoopDirective &S) {
Alexander Musmanc6388682014-12-15 07:07:06 +00001077 // Emit the loop iteration variable.
1078 auto IVExpr = cast<DeclRefExpr>(S.getIterationVariable());
1079 auto IVDecl = cast<VarDecl>(IVExpr->getDecl());
1080 EmitVarDecl(*IVDecl);
1081
1082 // Emit the iterations count variable.
1083 // If it is not a variable, Sema decided to calculate iterations count on each
1084 // iteration (e.g., it is foldable into a constant).
1085 if (auto LIExpr = dyn_cast<DeclRefExpr>(S.getLastIteration())) {
1086 EmitVarDecl(*cast<VarDecl>(LIExpr->getDecl()));
1087 // Emit calculation of the iterations count.
1088 EmitIgnoredExpr(S.getCalcLastIteration());
1089 }
1090
1091 auto &RT = CGM.getOpenMPRuntime();
1092
Alexey Bataev38e89532015-04-16 04:54:05 +00001093 bool HasLastprivateClause;
Alexander Musmanc6388682014-12-15 07:07:06 +00001094 // Check pre-condition.
1095 {
1096 // Skip the entire loop if we don't meet the precondition.
Alexey Bataev62dbb972015-04-22 11:59:37 +00001097 // If the condition constant folds and can be elided, avoid emitting the
1098 // whole loop.
1099 bool CondConstant;
1100 llvm::BasicBlock *ContBlock = nullptr;
1101 if (ConstantFoldsToSimpleInteger(S.getPreCond(), CondConstant)) {
1102 if (!CondConstant)
1103 return false;
1104 } else {
Alexey Bataev62dbb972015-04-22 11:59:37 +00001105 auto *ThenBlock = createBasicBlock("omp.precond.then");
1106 ContBlock = createBasicBlock("omp.precond.end");
1107 emitPreCond(*this, S, S.getPreCond(), ThenBlock, ContBlock,
Justin Bogner66242d62015-04-23 23:06:47 +00001108 getProfileCount(&S));
Alexey Bataev62dbb972015-04-22 11:59:37 +00001109 EmitBlock(ThenBlock);
Justin Bogner66242d62015-04-23 23:06:47 +00001110 incrementProfileCounter(&S);
Alexey Bataev62dbb972015-04-22 11:59:37 +00001111 }
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00001112
1113 emitAlignedClause(*this, S);
Alexey Bataev3b5b5c42015-06-18 10:10:12 +00001114 EmitOMPLinearClauseInit(S);
Alexander Musmanc6388682014-12-15 07:07:06 +00001115 // Emit 'then' code.
1116 {
1117 // Emit helper vars inits.
1118 LValue LB =
1119 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getLowerBoundVariable()));
1120 LValue UB =
1121 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getUpperBoundVariable()));
1122 LValue ST =
1123 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getStrideVariable()));
1124 LValue IL =
1125 EmitOMPHelperVar(*this, cast<DeclRefExpr>(S.getIsLastIterVariable()));
1126
1127 OMPPrivateScope LoopScope(*this);
Alexey Bataev69c62a92015-04-15 04:52:20 +00001128 if (EmitOMPFirstprivateClause(S, LoopScope)) {
1129 // Emit implicit barrier to synchronize threads and avoid data races on
1130 // initialization of firstprivate variables.
1131 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getLocStart(),
1132 OMPD_unknown);
1133 }
Alexey Bataev50a64582015-04-22 12:24:45 +00001134 EmitOMPPrivateClause(S, LoopScope);
Alexey Bataev38e89532015-04-16 04:54:05 +00001135 HasLastprivateClause = EmitOMPLastprivateClauseInit(S, LoopScope);
Alexey Bataev7ebe5fd2015-04-22 13:43:03 +00001136 EmitOMPReductionClauseInit(S, LoopScope);
Alexey Bataeva8899172015-08-06 12:30:57 +00001137 emitPrivateLoopCounters(*this, LoopScope, S.counters(),
1138 S.private_counters());
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00001139 emitPrivateLinearVars(*this, S, LoopScope);
Alexander Musman7931b982015-03-16 07:14:41 +00001140 (void)LoopScope.Privatize();
Alexander Musmanc6388682014-12-15 07:07:06 +00001141
1142 // Detect the loop schedule kind and chunk.
Alexey Bataev040d5402015-05-12 08:35:28 +00001143 llvm::Value *Chunk;
1144 OpenMPScheduleClauseKind ScheduleKind;
1145 auto ScheduleInfo =
1146 emitScheduleClause(*this, S, /*OuterRegion=*/false);
1147 Chunk = ScheduleInfo.first;
1148 ScheduleKind = ScheduleInfo.second;
Alexander Musmanc6388682014-12-15 07:07:06 +00001149 const unsigned IVSize = getContext().getTypeSize(IVExpr->getType());
1150 const bool IVSigned = IVExpr->getType()->hasSignedIntegerRepresentation();
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001151 const bool Ordered = S.getSingleClause(OMPC_ordered) != nullptr;
Alexander Musmanc6388682014-12-15 07:07:06 +00001152 if (RT.isStaticNonchunked(ScheduleKind,
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001153 /* Chunked */ Chunk != nullptr) &&
1154 !Ordered) {
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00001155 if (isOpenMPSimdDirective(S.getDirectiveKind())) {
1156 EmitOMPSimdInit(S);
1157 }
Alexander Musmanc6388682014-12-15 07:07:06 +00001158 // OpenMP [2.7.1, Loop Construct, Description, table 2-1]
1159 // When no chunk_size is specified, the iteration space is divided into
1160 // chunks that are approximately equal in size, and at most one chunk is
1161 // distributed to each thread. Note that the size of the chunks is
1162 // unspecified in this case.
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001163 RT.emitForInit(*this, S.getLocStart(), ScheduleKind, IVSize, IVSigned,
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001164 Ordered, IL.getAddress(), LB.getAddress(),
1165 UB.getAddress(), ST.getAddress());
Alexey Bataev0f34da12015-07-02 04:17:07 +00001166 auto LoopExit = getJumpDestInCurrentScope(createBasicBlock("omp.loop.exit"));
Alexander Musmanc6388682014-12-15 07:07:06 +00001167 // UB = min(UB, GlobalUB);
1168 EmitIgnoredExpr(S.getEnsureUpperBound());
1169 // IV = LB;
1170 EmitIgnoredExpr(S.getInit());
1171 // while (idx <= UB) { BODY; ++idx; }
Alexey Bataevae05c292015-06-16 11:59:36 +00001172 EmitOMPInnerLoop(S, LoopScope.requiresCleanups(), S.getCond(),
1173 S.getInc(),
Alexey Bataev0f34da12015-07-02 04:17:07 +00001174 [&S, LoopExit](CodeGenFunction &CGF) {
1175 CGF.EmitOMPLoopBody(S, LoopExit);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001176 CGF.EmitStopPoint(&S);
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001177 },
1178 [](CodeGenFunction &) {});
Alexey Bataev0f34da12015-07-02 04:17:07 +00001179 EmitBlock(LoopExit.getBlock());
Alexander Musmanc6388682014-12-15 07:07:06 +00001180 // Tell the runtime we are done.
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001181 RT.emitForStaticFinish(*this, S.getLocStart());
Alexander Musmandf7a8e22015-01-22 08:49:35 +00001182 } else {
1183 // Emit the outer loop, which requests its work chunk [LB..UB] from
1184 // runtime and runs the inner loop to process it.
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001185 EmitOMPForOuterLoop(ScheduleKind, S, LoopScope, Ordered,
1186 LB.getAddress(), UB.getAddress(), ST.getAddress(),
1187 IL.getAddress(), Chunk);
Alexander Musmandf7a8e22015-01-22 08:49:35 +00001188 }
Alexey Bataev7ebe5fd2015-04-22 13:43:03 +00001189 EmitOMPReductionClauseFinal(S);
Alexey Bataev38e89532015-04-16 04:54:05 +00001190 // Emit final copy of the lastprivate variables if IsLastIter != 0.
1191 if (HasLastprivateClause)
1192 EmitOMPLastprivateClauseFinal(
1193 S, Builder.CreateIsNotNull(EmitLoadOfScalar(IL, S.getLocStart())));
Alexander Musmanc6388682014-12-15 07:07:06 +00001194 }
Alexey Bataev58e5bdb2015-06-18 04:45:29 +00001195 if (isOpenMPSimdDirective(S.getDirectiveKind())) {
1196 EmitOMPSimdFinal(S);
1197 }
Alexander Musmanc6388682014-12-15 07:07:06 +00001198 // We're now done with the loop, so jump to the continuation block.
Alexey Bataev62dbb972015-04-22 11:59:37 +00001199 if (ContBlock) {
1200 EmitBranch(ContBlock);
1201 EmitBlock(ContBlock, true);
1202 }
Alexander Musmanc6388682014-12-15 07:07:06 +00001203 }
Alexey Bataev38e89532015-04-16 04:54:05 +00001204 return HasLastprivateClause;
Alexander Musmanc6388682014-12-15 07:07:06 +00001205}
1206
1207void CodeGenFunction::EmitOMPForDirective(const OMPForDirective &S) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001208 LexicalScope Scope(*this, S.getSourceRange());
Alexey Bataev38e89532015-04-16 04:54:05 +00001209 bool HasLastprivates = false;
1210 auto &&CodeGen = [&S, &HasLastprivates](CodeGenFunction &CGF) {
1211 HasLastprivates = CGF.EmitOMPWorksharingLoop(S);
1212 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001213 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_for, CodeGen);
Alexander Musmanc6388682014-12-15 07:07:06 +00001214
1215 // Emit an implicit barrier at the end.
Alexey Bataev38e89532015-04-16 04:54:05 +00001216 if (!S.getSingleClause(OMPC_nowait) || HasLastprivates) {
Alexey Bataevf2685682015-03-30 04:30:22 +00001217 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getLocStart(), OMPD_for);
1218 }
Alexey Bataevf29276e2014-06-18 04:14:57 +00001219}
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00001220
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001221void CodeGenFunction::EmitOMPForSimdDirective(const OMPForSimdDirective &S) {
1222 LexicalScope Scope(*this, S.getSourceRange());
1223 bool HasLastprivates = false;
1224 auto &&CodeGen = [&S, &HasLastprivates](CodeGenFunction &CGF) {
1225 HasLastprivates = CGF.EmitOMPWorksharingLoop(S);
1226 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001227 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_simd, CodeGen);
Alexey Bataevcbdcbb72015-06-17 07:45:51 +00001228
1229 // Emit an implicit barrier at the end.
1230 if (!S.getSingleClause(OMPC_nowait) || HasLastprivates) {
1231 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getLocStart(), OMPD_for);
1232 }
Alexander Musmanf82886e2014-09-18 05:12:34 +00001233}
1234
Alexey Bataev2df54a02015-03-12 08:53:29 +00001235static LValue createSectionLVal(CodeGenFunction &CGF, QualType Ty,
1236 const Twine &Name,
1237 llvm::Value *Init = nullptr) {
1238 auto LVal = CGF.MakeNaturalAlignAddrLValue(CGF.CreateMemTemp(Ty, Name), Ty);
1239 if (Init)
1240 CGF.EmitScalarInit(Init, LVal);
1241 return LVal;
Alexey Bataevd3f8dd22014-06-25 11:44:49 +00001242}
1243
Alexey Bataev0f34da12015-07-02 04:17:07 +00001244OpenMPDirectiveKind
1245CodeGenFunction::EmitSections(const OMPExecutableDirective &S) {
Alexey Bataev2df54a02015-03-12 08:53:29 +00001246 auto *Stmt = cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt();
1247 auto *CS = dyn_cast<CompoundStmt>(Stmt);
1248 if (CS && CS->size() > 1) {
Alexey Bataev9efc03b2015-04-27 04:34:03 +00001249 bool HasLastprivates = false;
1250 auto &&CodeGen = [&S, CS, &HasLastprivates](CodeGenFunction &CGF) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001251 auto &C = CGF.CGM.getContext();
1252 auto KmpInt32Ty = C.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/1);
1253 // Emit helper vars inits.
1254 LValue LB = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.lb.",
1255 CGF.Builder.getInt32(0));
1256 auto *GlobalUBVal = CGF.Builder.getInt32(CS->size() - 1);
1257 LValue UB =
1258 createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.ub.", GlobalUBVal);
1259 LValue ST = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.st.",
1260 CGF.Builder.getInt32(1));
1261 LValue IL = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.il.",
1262 CGF.Builder.getInt32(0));
1263 // Loop counter.
1264 LValue IV = createSectionLVal(CGF, KmpInt32Ty, ".omp.sections.iv.");
1265 OpaqueValueExpr IVRefExpr(S.getLocStart(), KmpInt32Ty, VK_LValue);
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001266 CodeGenFunction::OpaqueValueMapping OpaqueIV(CGF, &IVRefExpr, IV);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001267 OpaqueValueExpr UBRefExpr(S.getLocStart(), KmpInt32Ty, VK_LValue);
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001268 CodeGenFunction::OpaqueValueMapping OpaqueUB(CGF, &UBRefExpr, UB);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001269 // Generate condition for loop.
1270 BinaryOperator Cond(&IVRefExpr, &UBRefExpr, BO_LE, C.BoolTy, VK_RValue,
1271 OK_Ordinary, S.getLocStart(),
1272 /*fpContractable=*/false);
1273 // Increment for loop counter.
1274 UnaryOperator Inc(&IVRefExpr, UO_PreInc, KmpInt32Ty, VK_RValue,
1275 OK_Ordinary, S.getLocStart());
1276 auto BodyGen = [CS, &S, &IV](CodeGenFunction &CGF) {
1277 // Iterate through all sections and emit a switch construct:
1278 // switch (IV) {
1279 // case 0:
1280 // <SectionStmt[0]>;
1281 // break;
1282 // ...
1283 // case <NumSection> - 1:
1284 // <SectionStmt[<NumSection> - 1]>;
1285 // break;
1286 // }
1287 // .omp.sections.exit:
1288 auto *ExitBB = CGF.createBasicBlock(".omp.sections.exit");
1289 auto *SwitchStmt = CGF.Builder.CreateSwitch(
1290 CGF.EmitLoadOfLValue(IV, S.getLocStart()).getScalarVal(), ExitBB,
1291 CS->size());
1292 unsigned CaseNumber = 0;
Benjamin Kramer642f1732015-07-02 21:03:14 +00001293 for (auto *SubStmt : CS->children()) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001294 auto CaseBB = CGF.createBasicBlock(".omp.sections.case");
1295 CGF.EmitBlock(CaseBB);
1296 SwitchStmt->addCase(CGF.Builder.getInt32(CaseNumber), CaseBB);
Benjamin Kramer642f1732015-07-02 21:03:14 +00001297 CGF.EmitStmt(SubStmt);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001298 CGF.EmitBranch(ExitBB);
Benjamin Kramer642f1732015-07-02 21:03:14 +00001299 ++CaseNumber;
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001300 }
1301 CGF.EmitBlock(ExitBB, /*IsFinished=*/true);
1302 };
Alexey Bataev2cb9b952015-04-24 03:37:03 +00001303
1304 CodeGenFunction::OMPPrivateScope LoopScope(CGF);
1305 if (CGF.EmitOMPFirstprivateClause(S, LoopScope)) {
1306 // Emit implicit barrier to synchronize threads and avoid data races on
1307 // initialization of firstprivate variables.
1308 CGF.CGM.getOpenMPRuntime().emitBarrierCall(CGF, S.getLocStart(),
1309 OMPD_unknown);
1310 }
Alexey Bataev73870832015-04-27 04:12:12 +00001311 CGF.EmitOMPPrivateClause(S, LoopScope);
Alexey Bataev9efc03b2015-04-27 04:34:03 +00001312 HasLastprivates = CGF.EmitOMPLastprivateClauseInit(S, LoopScope);
Alexey Bataeva89adf22015-04-27 05:04:13 +00001313 CGF.EmitOMPReductionClauseInit(S, LoopScope);
Alexey Bataev2cb9b952015-04-24 03:37:03 +00001314 (void)LoopScope.Privatize();
1315
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001316 // Emit static non-chunked loop.
1317 CGF.CGM.getOpenMPRuntime().emitForInit(
1318 CGF, S.getLocStart(), OMPC_SCHEDULE_static, /*IVSize=*/32,
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001319 /*IVSigned=*/true, /*Ordered=*/false, IL.getAddress(),
1320 LB.getAddress(), UB.getAddress(), ST.getAddress());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001321 // UB = min(UB, GlobalUB);
1322 auto *UBVal = CGF.EmitLoadOfScalar(UB, S.getLocStart());
1323 auto *MinUBGlobalUB = CGF.Builder.CreateSelect(
1324 CGF.Builder.CreateICmpSLT(UBVal, GlobalUBVal), UBVal, GlobalUBVal);
1325 CGF.EmitStoreOfScalar(MinUBGlobalUB, UB);
1326 // IV = LB;
1327 CGF.EmitStoreOfScalar(CGF.EmitLoadOfScalar(LB, S.getLocStart()), IV);
1328 // while (idx <= UB) { BODY; ++idx; }
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001329 CGF.EmitOMPInnerLoop(S, /*RequiresCleanup=*/false, &Cond, &Inc, BodyGen,
1330 [](CodeGenFunction &) {});
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001331 // Tell the runtime we are done.
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001332 CGF.CGM.getOpenMPRuntime().emitForStaticFinish(CGF, S.getLocStart());
Alexey Bataeva89adf22015-04-27 05:04:13 +00001333 CGF.EmitOMPReductionClauseFinal(S);
Alexey Bataev9efc03b2015-04-27 04:34:03 +00001334
1335 // Emit final copy of the lastprivate variables if IsLastIter != 0.
1336 if (HasLastprivates)
1337 CGF.EmitOMPLastprivateClauseFinal(
1338 S, CGF.Builder.CreateIsNotNull(
1339 CGF.EmitLoadOfScalar(IL, S.getLocStart())));
Alexey Bataev2df54a02015-03-12 08:53:29 +00001340 };
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001341
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001342 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_sections, CodeGen);
Alexey Bataev9efc03b2015-04-27 04:34:03 +00001343 // Emit barrier for lastprivates only if 'sections' directive has 'nowait'
1344 // clause. Otherwise the barrier will be generated by the codegen for the
1345 // directive.
1346 if (HasLastprivates && S.getSingleClause(OMPC_nowait)) {
1347 // Emit implicit barrier to synchronize threads and avoid data races on
1348 // initialization of firstprivate variables.
Alexey Bataev0f34da12015-07-02 04:17:07 +00001349 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getLocStart(),
1350 OMPD_unknown);
Alexey Bataev9efc03b2015-04-27 04:34:03 +00001351 }
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001352 return OMPD_sections;
Alexey Bataev2df54a02015-03-12 08:53:29 +00001353 }
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001354 // If only one section is found - no need to generate loop, emit as a single
1355 // region.
Alexey Bataev2cb9b952015-04-24 03:37:03 +00001356 bool HasFirstprivates;
Alexey Bataeva89adf22015-04-27 05:04:13 +00001357 // No need to generate reductions for sections with single section region, we
1358 // can use original shared variables for all operations.
Alexey Bataevc925aa32015-04-27 08:00:32 +00001359 bool HasReductions = !S.getClausesOfKind(OMPC_reduction).empty();
Alexey Bataev9efc03b2015-04-27 04:34:03 +00001360 // No need to generate lastprivates for sections with single section region,
1361 // we can use original shared variable for all calculations with barrier at
1362 // the end of the sections.
Alexey Bataevc925aa32015-04-27 08:00:32 +00001363 bool HasLastprivates = !S.getClausesOfKind(OMPC_lastprivate).empty();
Alexey Bataev2cb9b952015-04-24 03:37:03 +00001364 auto &&CodeGen = [Stmt, &S, &HasFirstprivates](CodeGenFunction &CGF) {
1365 CodeGenFunction::OMPPrivateScope SingleScope(CGF);
1366 HasFirstprivates = CGF.EmitOMPFirstprivateClause(S, SingleScope);
Alexey Bataev73870832015-04-27 04:12:12 +00001367 CGF.EmitOMPPrivateClause(S, SingleScope);
Alexey Bataev2cb9b952015-04-24 03:37:03 +00001368 (void)SingleScope.Privatize();
1369
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001370 CGF.EmitStmt(Stmt);
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001371 };
Alexey Bataev0f34da12015-07-02 04:17:07 +00001372 CGM.getOpenMPRuntime().emitSingleRegion(*this, CodeGen, S.getLocStart(),
1373 llvm::None, llvm::None, llvm::None,
1374 llvm::None);
Alexey Bataeva89adf22015-04-27 05:04:13 +00001375 // Emit barrier for firstprivates, lastprivates or reductions only if
1376 // 'sections' directive has 'nowait' clause. Otherwise the barrier will be
1377 // generated by the codegen for the directive.
1378 if ((HasFirstprivates || HasLastprivates || HasReductions) &&
1379 S.getSingleClause(OMPC_nowait)) {
Alexey Bataev2cb9b952015-04-24 03:37:03 +00001380 // Emit implicit barrier to synchronize threads and avoid data races on
1381 // initialization of firstprivate variables.
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001382 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getLocStart(), OMPD_unknown);
Alexey Bataev2cb9b952015-04-24 03:37:03 +00001383 }
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001384 return OMPD_single;
1385}
Alexey Bataev2df54a02015-03-12 08:53:29 +00001386
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001387void CodeGenFunction::EmitOMPSectionsDirective(const OMPSectionsDirective &S) {
1388 LexicalScope Scope(*this, S.getSourceRange());
Alexey Bataev0f34da12015-07-02 04:17:07 +00001389 OpenMPDirectiveKind EmittedAs = EmitSections(S);
Alexey Bataev2df54a02015-03-12 08:53:29 +00001390 // Emit an implicit barrier at the end.
Alexey Bataevf2685682015-03-30 04:30:22 +00001391 if (!S.getSingleClause(OMPC_nowait)) {
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001392 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getLocStart(), EmittedAs);
Alexey Bataevf2685682015-03-30 04:30:22 +00001393 }
Alexey Bataev2df54a02015-03-12 08:53:29 +00001394}
1395
1396void CodeGenFunction::EmitOMPSectionDirective(const OMPSectionDirective &S) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001397 LexicalScope Scope(*this, S.getSourceRange());
1398 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
1399 CGF.EmitStmt(cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt());
1400 CGF.EnsureInsertPoint();
1401 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001402 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_section, CodeGen);
Alexey Bataev1e0498a2014-06-26 08:21:58 +00001403}
1404
Alexey Bataev6956e2e2015-02-05 06:35:41 +00001405void CodeGenFunction::EmitOMPSingleDirective(const OMPSingleDirective &S) {
Alexey Bataeva63048e2015-03-23 06:18:07 +00001406 llvm::SmallVector<const Expr *, 8> CopyprivateVars;
Alexey Bataev420d45b2015-04-14 05:11:24 +00001407 llvm::SmallVector<const Expr *, 8> DestExprs;
Alexey Bataeva63048e2015-03-23 06:18:07 +00001408 llvm::SmallVector<const Expr *, 8> SrcExprs;
Alexey Bataeva63048e2015-03-23 06:18:07 +00001409 llvm::SmallVector<const Expr *, 8> AssignmentOps;
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001410 // Check if there are any 'copyprivate' clauses associated with this
1411 // 'single'
Alexey Bataeva63048e2015-03-23 06:18:07 +00001412 // construct.
Alexey Bataeva63048e2015-03-23 06:18:07 +00001413 // Build a list of copyprivate variables along with helper expressions
1414 // (<source>, <destination>, <destination>=<source> expressions)
Alexey Bataevc925aa32015-04-27 08:00:32 +00001415 for (auto &&I = S.getClausesOfKind(OMPC_copyprivate); I; ++I) {
Alexey Bataeva63048e2015-03-23 06:18:07 +00001416 auto *C = cast<OMPCopyprivateClause>(*I);
1417 CopyprivateVars.append(C->varlists().begin(), C->varlists().end());
Alexey Bataev420d45b2015-04-14 05:11:24 +00001418 DestExprs.append(C->destination_exprs().begin(),
1419 C->destination_exprs().end());
Alexey Bataeva63048e2015-03-23 06:18:07 +00001420 SrcExprs.append(C->source_exprs().begin(), C->source_exprs().end());
Alexey Bataeva63048e2015-03-23 06:18:07 +00001421 AssignmentOps.append(C->assignment_ops().begin(),
1422 C->assignment_ops().end());
1423 }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001424 LexicalScope Scope(*this, S.getSourceRange());
Alexey Bataeva63048e2015-03-23 06:18:07 +00001425 // Emit code for 'single' region along with 'copyprivate' clauses
Alexey Bataev5521d782015-04-24 04:21:15 +00001426 bool HasFirstprivates;
1427 auto &&CodeGen = [&S, &HasFirstprivates](CodeGenFunction &CGF) {
1428 CodeGenFunction::OMPPrivateScope SingleScope(CGF);
1429 HasFirstprivates = CGF.EmitOMPFirstprivateClause(S, SingleScope);
Alexey Bataev59c654a2015-04-27 03:48:52 +00001430 CGF.EmitOMPPrivateClause(S, SingleScope);
Alexey Bataev5521d782015-04-24 04:21:15 +00001431 (void)SingleScope.Privatize();
1432
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001433 CGF.EmitStmt(cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt());
1434 CGF.EnsureInsertPoint();
1435 };
1436 CGM.getOpenMPRuntime().emitSingleRegion(*this, CodeGen, S.getLocStart(),
Alexey Bataev420d45b2015-04-14 05:11:24 +00001437 CopyprivateVars, DestExprs, SrcExprs,
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001438 AssignmentOps);
Alexey Bataev5521d782015-04-24 04:21:15 +00001439 // Emit an implicit barrier at the end (to avoid data race on firstprivate
1440 // init or if no 'nowait' clause was specified and no 'copyprivate' clause).
1441 if ((!S.getSingleClause(OMPC_nowait) || HasFirstprivates) &&
1442 CopyprivateVars.empty()) {
1443 CGM.getOpenMPRuntime().emitBarrierCall(
1444 *this, S.getLocStart(),
1445 S.getSingleClause(OMPC_nowait) ? OMPD_unknown : OMPD_single);
Alexey Bataevf2685682015-03-30 04:30:22 +00001446 }
Alexey Bataevd1e40fb2014-06-26 12:05:45 +00001447}
1448
Alexey Bataev8d690652014-12-04 07:23:53 +00001449void CodeGenFunction::EmitOMPMasterDirective(const OMPMasterDirective &S) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001450 LexicalScope Scope(*this, S.getSourceRange());
1451 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
1452 CGF.EmitStmt(cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt());
1453 CGF.EnsureInsertPoint();
1454 };
1455 CGM.getOpenMPRuntime().emitMasterRegion(*this, CodeGen, S.getLocStart());
Alexander Musman80c22892014-07-17 08:54:58 +00001456}
1457
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +00001458void CodeGenFunction::EmitOMPCriticalDirective(const OMPCriticalDirective &S) {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001459 LexicalScope Scope(*this, S.getSourceRange());
1460 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
1461 CGF.EmitStmt(cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt());
1462 CGF.EnsureInsertPoint();
1463 };
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001464 CGM.getOpenMPRuntime().emitCriticalRegion(
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001465 *this, S.getDirectiveName().getAsString(), CodeGen, S.getLocStart());
Alexander Musmand9ed09f2014-07-21 09:42:05 +00001466}
1467
Alexey Bataev671605e2015-04-13 05:28:11 +00001468void CodeGenFunction::EmitOMPParallelForDirective(
1469 const OMPParallelForDirective &S) {
1470 // Emit directive as a combined directive that consists of two implicit
1471 // directives: 'parallel' with 'for' directive.
1472 LexicalScope Scope(*this, S.getSourceRange());
Alexey Bataev040d5402015-05-12 08:35:28 +00001473 (void)emitScheduleClause(*this, S, /*OuterRegion=*/true);
Alexey Bataev671605e2015-04-13 05:28:11 +00001474 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
1475 CGF.EmitOMPWorksharingLoop(S);
1476 // Emit implicit barrier at the end of parallel region, but this barrier
1477 // is at the end of 'for' directive, so emit it as the implicit barrier for
1478 // this 'for' directive.
1479 CGF.CGM.getOpenMPRuntime().emitBarrierCall(CGF, S.getLocStart(),
1480 OMPD_parallel);
1481 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001482 emitCommonOMPParallelDirective(*this, S, OMPD_for, CodeGen);
Alexey Bataev4acb8592014-07-07 13:01:15 +00001483}
1484
Alexander Musmane4e893b2014-09-23 09:33:00 +00001485void CodeGenFunction::EmitOMPParallelForSimdDirective(
Alexey Bataev3b5b5c42015-06-18 10:10:12 +00001486 const OMPParallelForSimdDirective &S) {
1487 // Emit directive as a combined directive that consists of two implicit
1488 // directives: 'parallel' with 'for' directive.
1489 LexicalScope Scope(*this, S.getSourceRange());
1490 (void)emitScheduleClause(*this, S, /*OuterRegion=*/true);
1491 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
1492 CGF.EmitOMPWorksharingLoop(S);
1493 // Emit implicit barrier at the end of parallel region, but this barrier
1494 // is at the end of 'for' directive, so emit it as the implicit barrier for
1495 // this 'for' directive.
1496 CGF.CGM.getOpenMPRuntime().emitBarrierCall(CGF, S.getLocStart(),
1497 OMPD_parallel);
1498 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001499 emitCommonOMPParallelDirective(*this, S, OMPD_simd, CodeGen);
Alexander Musmane4e893b2014-09-23 09:33:00 +00001500}
1501
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00001502void CodeGenFunction::EmitOMPParallelSectionsDirective(
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001503 const OMPParallelSectionsDirective &S) {
1504 // Emit directive as a combined directive that consists of two implicit
1505 // directives: 'parallel' with 'sections' directive.
1506 LexicalScope Scope(*this, S.getSourceRange());
1507 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
Alexey Bataev0f34da12015-07-02 04:17:07 +00001508 (void)CGF.EmitSections(S);
Alexey Bataev68adb7d2015-04-14 03:29:22 +00001509 // Emit implicit barrier at the end of parallel region.
1510 CGF.CGM.getOpenMPRuntime().emitBarrierCall(CGF, S.getLocStart(),
1511 OMPD_parallel);
1512 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001513 emitCommonOMPParallelDirective(*this, S, OMPD_sections, CodeGen);
Alexey Bataev84d0b3e2014-07-08 08:12:03 +00001514}
1515
Alexey Bataev62b63b12015-03-10 07:28:44 +00001516void CodeGenFunction::EmitOMPTaskDirective(const OMPTaskDirective &S) {
1517 // Emit outlined function for task construct.
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001518 LexicalScope Scope(*this, S.getSourceRange());
Alexey Bataev62b63b12015-03-10 07:28:44 +00001519 auto CS = cast<CapturedStmt>(S.getAssociatedStmt());
1520 auto CapturedStruct = GenerateCapturedStmtArgument(*CS);
1521 auto *I = CS->getCapturedDecl()->param_begin();
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001522 auto *PartId = std::next(I);
Alexey Bataev62b63b12015-03-10 07:28:44 +00001523 // The first function argument for tasks is a thread id, the second one is a
1524 // part id (0 for tied tasks, >=0 for untied task).
Alexey Bataev3ae88e22015-05-22 08:56:35 +00001525 llvm::DenseSet<const VarDecl *> EmittedAsPrivate;
1526 // Get list of private variables.
1527 llvm::SmallVector<const Expr *, 8> PrivateVars;
1528 llvm::SmallVector<const Expr *, 8> PrivateCopies;
1529 for (auto &&I = S.getClausesOfKind(OMPC_private); I; ++I) {
1530 auto *C = cast<OMPPrivateClause>(*I);
1531 auto IRef = C->varlist_begin();
1532 for (auto *IInit : C->private_copies()) {
1533 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
1534 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
1535 PrivateVars.push_back(*IRef);
1536 PrivateCopies.push_back(IInit);
1537 }
1538 ++IRef;
1539 }
1540 }
1541 EmittedAsPrivate.clear();
1542 // Get list of firstprivate variables.
1543 llvm::SmallVector<const Expr *, 8> FirstprivateVars;
1544 llvm::SmallVector<const Expr *, 8> FirstprivateCopies;
1545 llvm::SmallVector<const Expr *, 8> FirstprivateInits;
1546 for (auto &&I = S.getClausesOfKind(OMPC_firstprivate); I; ++I) {
1547 auto *C = cast<OMPFirstprivateClause>(*I);
1548 auto IRef = C->varlist_begin();
1549 auto IElemInitRef = C->inits().begin();
1550 for (auto *IInit : C->private_copies()) {
1551 auto *OrigVD = cast<VarDecl>(cast<DeclRefExpr>(*IRef)->getDecl());
1552 if (EmittedAsPrivate.insert(OrigVD->getCanonicalDecl()).second) {
1553 FirstprivateVars.push_back(*IRef);
1554 FirstprivateCopies.push_back(IInit);
1555 FirstprivateInits.push_back(*IElemInitRef);
1556 }
1557 ++IRef, ++IElemInitRef;
1558 }
1559 }
Alexey Bataev1d2353d2015-06-24 11:01:36 +00001560 // Build list of dependences.
1561 llvm::SmallVector<std::pair<OpenMPDependClauseKind, const Expr *>, 8>
1562 Dependences;
1563 for (auto &&I = S.getClausesOfKind(OMPC_depend); I; ++I) {
1564 auto *C = cast<OMPDependClause>(*I);
1565 for (auto *IRef : C->varlists()) {
1566 Dependences.push_back(std::make_pair(C->getDependencyKind(), IRef));
1567 }
1568 }
Alexey Bataev3ae88e22015-05-22 08:56:35 +00001569 auto &&CodeGen = [PartId, &S, &PrivateVars, &FirstprivateVars](
1570 CodeGenFunction &CGF) {
1571 // Set proper addresses for generated private copies.
1572 auto *CS = cast<CapturedStmt>(S.getAssociatedStmt());
1573 OMPPrivateScope Scope(CGF);
1574 if (!PrivateVars.empty() || !FirstprivateVars.empty()) {
1575 auto *CopyFn = CGF.Builder.CreateAlignedLoad(
1576 CGF.GetAddrOfLocalVar(CS->getCapturedDecl()->getParam(3)),
1577 CGF.PointerAlignInBytes);
1578 auto *PrivatesPtr = CGF.Builder.CreateAlignedLoad(
1579 CGF.GetAddrOfLocalVar(CS->getCapturedDecl()->getParam(2)),
1580 CGF.PointerAlignInBytes);
1581 // Map privates.
1582 llvm::SmallVector<std::pair<const VarDecl *, llvm::Value *>, 16>
1583 PrivatePtrs;
1584 llvm::SmallVector<llvm::Value *, 16> CallArgs;
1585 CallArgs.push_back(PrivatesPtr);
1586 for (auto *E : PrivateVars) {
1587 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
1588 auto *PrivatePtr =
1589 CGF.CreateMemTemp(CGF.getContext().getPointerType(E->getType()));
1590 PrivatePtrs.push_back(std::make_pair(VD, PrivatePtr));
1591 CallArgs.push_back(PrivatePtr);
1592 }
1593 for (auto *E : FirstprivateVars) {
1594 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
1595 auto *PrivatePtr =
1596 CGF.CreateMemTemp(CGF.getContext().getPointerType(E->getType()));
1597 PrivatePtrs.push_back(std::make_pair(VD, PrivatePtr));
1598 CallArgs.push_back(PrivatePtr);
1599 }
1600 CGF.EmitRuntimeCall(CopyFn, CallArgs);
1601 for (auto &&Pair : PrivatePtrs) {
1602 auto *Replacement =
1603 CGF.Builder.CreateAlignedLoad(Pair.second, CGF.PointerAlignInBytes);
1604 Scope.addPrivate(Pair.first, [Replacement]() { return Replacement; });
1605 }
1606 }
1607 (void)Scope.Privatize();
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001608 if (*PartId) {
1609 // TODO: emit code for untied tasks.
1610 }
Alexey Bataev3ae88e22015-05-22 08:56:35 +00001611 CGF.EmitStmt(CS->getCapturedStmt());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001612 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001613 auto OutlinedFn = CGM.getOpenMPRuntime().emitTaskOutlinedFunction(
1614 S, *I, OMPD_task, CodeGen);
Alexey Bataev62b63b12015-03-10 07:28:44 +00001615 // Check if we should emit tied or untied task.
1616 bool Tied = !S.getSingleClause(OMPC_untied);
1617 // Check if the task is final
1618 llvm::PointerIntPair<llvm::Value *, 1, bool> Final;
1619 if (auto *Clause = S.getSingleClause(OMPC_final)) {
1620 // If the condition constant folds and can be elided, try to avoid emitting
1621 // the condition and the dead arm of the if/else.
1622 auto *Cond = cast<OMPFinalClause>(Clause)->getCondition();
1623 bool CondConstant;
1624 if (ConstantFoldsToSimpleInteger(Cond, CondConstant))
1625 Final.setInt(CondConstant);
1626 else
1627 Final.setPointer(EvaluateExprAsBool(Cond));
1628 } else {
1629 // By default the task is not final.
1630 Final.setInt(/*IntVal=*/false);
1631 }
1632 auto SharedsTy = getContext().getRecordType(CS->getCapturedRecordDecl());
Alexey Bataev1d677132015-04-22 13:57:31 +00001633 const Expr *IfCond = nullptr;
1634 if (auto C = S.getSingleClause(OMPC_if)) {
1635 IfCond = cast<OMPIfClause>(C)->getCondition();
1636 }
Alexey Bataev9e034042015-05-05 04:05:12 +00001637 CGM.getOpenMPRuntime().emitTaskCall(
1638 *this, S.getLocStart(), S, Tied, Final, OutlinedFn, SharedsTy,
Alexey Bataev3ae88e22015-05-22 08:56:35 +00001639 CapturedStruct, IfCond, PrivateVars, PrivateCopies, FirstprivateVars,
Alexey Bataev1d2353d2015-06-24 11:01:36 +00001640 FirstprivateCopies, FirstprivateInits, Dependences);
Alexey Bataev9c2e8ee2014-07-11 11:25:16 +00001641}
1642
Alexey Bataev9f797f32015-02-05 05:57:51 +00001643void CodeGenFunction::EmitOMPTaskyieldDirective(
1644 const OMPTaskyieldDirective &S) {
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001645 CGM.getOpenMPRuntime().emitTaskyieldCall(*this, S.getLocStart());
Alexey Bataev68446b72014-07-18 07:47:19 +00001646}
1647
Alexey Bataev8f7c1b02014-12-05 04:09:23 +00001648void CodeGenFunction::EmitOMPBarrierDirective(const OMPBarrierDirective &S) {
Alexey Bataevf2685682015-03-30 04:30:22 +00001649 CGM.getOpenMPRuntime().emitBarrierCall(*this, S.getLocStart(), OMPD_barrier);
Alexey Bataev4d1dfea2014-07-18 09:11:51 +00001650}
1651
Alexey Bataev8b8e2022015-04-27 05:22:09 +00001652void CodeGenFunction::EmitOMPTaskwaitDirective(const OMPTaskwaitDirective &S) {
1653 CGM.getOpenMPRuntime().emitTaskwaitCall(*this, S.getLocStart());
Alexey Bataev2df347a2014-07-18 10:17:07 +00001654}
1655
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001656void CodeGenFunction::EmitOMPTaskgroupDirective(
1657 const OMPTaskgroupDirective &S) {
1658 LexicalScope Scope(*this, S.getSourceRange());
1659 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
1660 CGF.EmitStmt(cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt());
1661 CGF.EnsureInsertPoint();
1662 };
1663 CGM.getOpenMPRuntime().emitTaskgroupRegion(*this, CodeGen, S.getLocStart());
1664}
1665
Alexey Bataevcc37cc12014-11-20 04:34:54 +00001666void CodeGenFunction::EmitOMPFlushDirective(const OMPFlushDirective &S) {
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001667 CGM.getOpenMPRuntime().emitFlush(*this, [&]() -> ArrayRef<const Expr *> {
1668 if (auto C = S.getSingleClause(/*K*/ OMPC_flush)) {
1669 auto FlushClause = cast<OMPFlushClause>(C);
1670 return llvm::makeArrayRef(FlushClause->varlist_begin(),
1671 FlushClause->varlist_end());
1672 }
1673 return llvm::None;
1674 }(), S.getLocStart());
Alexey Bataev6125da92014-07-21 11:26:11 +00001675}
1676
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001677void CodeGenFunction::EmitOMPOrderedDirective(const OMPOrderedDirective &S) {
1678 LexicalScope Scope(*this, S.getSourceRange());
1679 auto &&CodeGen = [&S](CodeGenFunction &CGF) {
1680 CGF.EmitStmt(cast<CapturedStmt>(S.getAssociatedStmt())->getCapturedStmt());
1681 CGF.EnsureInsertPoint();
1682 };
1683 CGM.getOpenMPRuntime().emitOrderedRegion(*this, CodeGen, S.getLocStart());
Alexey Bataev9fb6e642014-07-22 06:45:04 +00001684}
1685
Alexey Bataevb57056f2015-01-22 06:17:56 +00001686static llvm::Value *convertToScalarValue(CodeGenFunction &CGF, RValue Val,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001687 QualType SrcType, QualType DestType,
1688 SourceLocation Loc) {
Alexey Bataevb57056f2015-01-22 06:17:56 +00001689 assert(CGF.hasScalarEvaluationKind(DestType) &&
1690 "DestType must have scalar evaluation kind.");
1691 assert(!Val.isAggregate() && "Must be a scalar or complex.");
1692 return Val.isScalar()
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001693 ? CGF.EmitScalarConversion(Val.getScalarVal(), SrcType, DestType,
1694 Loc)
Alexey Bataevb57056f2015-01-22 06:17:56 +00001695 : CGF.EmitComplexToScalarConversion(Val.getComplexVal(), SrcType,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001696 DestType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00001697}
1698
1699static CodeGenFunction::ComplexPairTy
1700convertToComplexValue(CodeGenFunction &CGF, RValue Val, QualType SrcType,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001701 QualType DestType, SourceLocation Loc) {
Alexey Bataevb57056f2015-01-22 06:17:56 +00001702 assert(CGF.getEvaluationKind(DestType) == TEK_Complex &&
1703 "DestType must have complex evaluation kind.");
1704 CodeGenFunction::ComplexPairTy ComplexVal;
1705 if (Val.isScalar()) {
1706 // Convert the input element to the element type of the complex.
1707 auto DestElementType = DestType->castAs<ComplexType>()->getElementType();
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001708 auto ScalarVal = CGF.EmitScalarConversion(Val.getScalarVal(), SrcType,
1709 DestElementType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00001710 ComplexVal = CodeGenFunction::ComplexPairTy(
1711 ScalarVal, llvm::Constant::getNullValue(ScalarVal->getType()));
1712 } else {
1713 assert(Val.isComplex() && "Must be a scalar or complex.");
1714 auto SrcElementType = SrcType->castAs<ComplexType>()->getElementType();
1715 auto DestElementType = DestType->castAs<ComplexType>()->getElementType();
1716 ComplexVal.first = CGF.EmitScalarConversion(
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001717 Val.getComplexVal().first, SrcElementType, DestElementType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00001718 ComplexVal.second = CGF.EmitScalarConversion(
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001719 Val.getComplexVal().second, SrcElementType, DestElementType, Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00001720 }
1721 return ComplexVal;
1722}
1723
Alexey Bataev5e018f92015-04-23 06:35:10 +00001724static void emitSimpleAtomicStore(CodeGenFunction &CGF, bool IsSeqCst,
1725 LValue LVal, RValue RVal) {
1726 if (LVal.isGlobalReg()) {
1727 CGF.EmitStoreThroughGlobalRegLValue(RVal, LVal);
1728 } else {
1729 CGF.EmitAtomicStore(RVal, LVal, IsSeqCst ? llvm::SequentiallyConsistent
1730 : llvm::Monotonic,
1731 LVal.isVolatile(), /*IsInit=*/false);
1732 }
1733}
1734
1735static void emitSimpleStore(CodeGenFunction &CGF, LValue LVal, RValue RVal,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001736 QualType RValTy, SourceLocation Loc) {
Alexey Bataev5e018f92015-04-23 06:35:10 +00001737 switch (CGF.getEvaluationKind(LVal.getType())) {
1738 case TEK_Scalar:
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001739 CGF.EmitStoreThroughLValue(RValue::get(convertToScalarValue(
1740 CGF, RVal, RValTy, LVal.getType(), Loc)),
1741 LVal);
Alexey Bataev5e018f92015-04-23 06:35:10 +00001742 break;
1743 case TEK_Complex:
1744 CGF.EmitStoreOfComplex(
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001745 convertToComplexValue(CGF, RVal, RValTy, LVal.getType(), Loc), LVal,
Alexey Bataev5e018f92015-04-23 06:35:10 +00001746 /*isInit=*/false);
1747 break;
1748 case TEK_Aggregate:
1749 llvm_unreachable("Must be a scalar or complex.");
1750 }
1751}
1752
Alexey Bataevb57056f2015-01-22 06:17:56 +00001753static void EmitOMPAtomicReadExpr(CodeGenFunction &CGF, bool IsSeqCst,
1754 const Expr *X, const Expr *V,
1755 SourceLocation Loc) {
1756 // v = x;
1757 assert(V->isLValue() && "V of 'omp atomic read' is not lvalue");
1758 assert(X->isLValue() && "X of 'omp atomic read' is not lvalue");
1759 LValue XLValue = CGF.EmitLValue(X);
1760 LValue VLValue = CGF.EmitLValue(V);
David Majnemera5b195a2015-02-14 01:35:12 +00001761 RValue Res = XLValue.isGlobalReg()
1762 ? CGF.EmitLoadOfLValue(XLValue, Loc)
1763 : CGF.EmitAtomicLoad(XLValue, Loc,
1764 IsSeqCst ? llvm::SequentiallyConsistent
Alexey Bataevb8329262015-02-27 06:33:30 +00001765 : llvm::Monotonic,
1766 XLValue.isVolatile());
Alexey Bataevb57056f2015-01-22 06:17:56 +00001767 // OpenMP, 2.12.6, atomic Construct
1768 // Any atomic construct with a seq_cst clause forces the atomically
1769 // performed operation to include an implicit flush operation without a
1770 // list.
1771 if (IsSeqCst)
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001772 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc);
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001773 emitSimpleStore(CGF, VLValue, Res, X->getType().getNonReferenceType(), Loc);
Alexey Bataevb57056f2015-01-22 06:17:56 +00001774}
1775
Alexey Bataevb8329262015-02-27 06:33:30 +00001776static void EmitOMPAtomicWriteExpr(CodeGenFunction &CGF, bool IsSeqCst,
1777 const Expr *X, const Expr *E,
1778 SourceLocation Loc) {
1779 // x = expr;
1780 assert(X->isLValue() && "X of 'omp atomic write' is not lvalue");
Alexey Bataev5e018f92015-04-23 06:35:10 +00001781 emitSimpleAtomicStore(CGF, IsSeqCst, CGF.EmitLValue(X), CGF.EmitAnyExpr(E));
Alexey Bataevb8329262015-02-27 06:33:30 +00001782 // OpenMP, 2.12.6, atomic Construct
1783 // Any atomic construct with a seq_cst clause forces the atomically
1784 // performed operation to include an implicit flush operation without a
1785 // list.
1786 if (IsSeqCst)
1787 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc);
1788}
1789
Benjamin Kramer439ee9d2015-05-01 13:59:53 +00001790static std::pair<bool, RValue> emitOMPAtomicRMW(CodeGenFunction &CGF, LValue X,
1791 RValue Update,
1792 BinaryOperatorKind BO,
1793 llvm::AtomicOrdering AO,
1794 bool IsXLHSInRHSPart) {
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001795 auto &Context = CGF.CGM.getContext();
1796 // Allow atomicrmw only if 'x' and 'update' are integer values, lvalue for 'x'
Alexey Bataevb4505a72015-03-30 05:20:59 +00001797 // expression is simple and atomic is allowed for the given type for the
1798 // target platform.
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001799 if (BO == BO_Comma || !Update.isScalar() ||
Alexey Bataev9d541a72015-05-08 11:47:16 +00001800 !Update.getScalarVal()->getType()->isIntegerTy() ||
1801 !X.isSimple() || (!isa<llvm::ConstantInt>(Update.getScalarVal()) &&
1802 (Update.getScalarVal()->getType() !=
1803 X.getAddress()->getType()->getPointerElementType())) ||
1804 !X.getAddress()->getType()->getPointerElementType()->isIntegerTy() ||
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001805 !Context.getTargetInfo().hasBuiltinAtomic(
1806 Context.getTypeSize(X.getType()), Context.toBits(X.getAlignment())))
Alexey Bataev5e018f92015-04-23 06:35:10 +00001807 return std::make_pair(false, RValue::get(nullptr));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001808
1809 llvm::AtomicRMWInst::BinOp RMWOp;
1810 switch (BO) {
1811 case BO_Add:
1812 RMWOp = llvm::AtomicRMWInst::Add;
1813 break;
1814 case BO_Sub:
1815 if (!IsXLHSInRHSPart)
Alexey Bataev5e018f92015-04-23 06:35:10 +00001816 return std::make_pair(false, RValue::get(nullptr));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001817 RMWOp = llvm::AtomicRMWInst::Sub;
1818 break;
1819 case BO_And:
1820 RMWOp = llvm::AtomicRMWInst::And;
1821 break;
1822 case BO_Or:
1823 RMWOp = llvm::AtomicRMWInst::Or;
1824 break;
1825 case BO_Xor:
1826 RMWOp = llvm::AtomicRMWInst::Xor;
1827 break;
1828 case BO_LT:
1829 RMWOp = X.getType()->hasSignedIntegerRepresentation()
1830 ? (IsXLHSInRHSPart ? llvm::AtomicRMWInst::Min
1831 : llvm::AtomicRMWInst::Max)
1832 : (IsXLHSInRHSPart ? llvm::AtomicRMWInst::UMin
1833 : llvm::AtomicRMWInst::UMax);
1834 break;
1835 case BO_GT:
1836 RMWOp = X.getType()->hasSignedIntegerRepresentation()
1837 ? (IsXLHSInRHSPart ? llvm::AtomicRMWInst::Max
1838 : llvm::AtomicRMWInst::Min)
1839 : (IsXLHSInRHSPart ? llvm::AtomicRMWInst::UMax
1840 : llvm::AtomicRMWInst::UMin);
1841 break;
Alexey Bataev5e018f92015-04-23 06:35:10 +00001842 case BO_Assign:
1843 RMWOp = llvm::AtomicRMWInst::Xchg;
1844 break;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001845 case BO_Mul:
1846 case BO_Div:
1847 case BO_Rem:
1848 case BO_Shl:
1849 case BO_Shr:
1850 case BO_LAnd:
1851 case BO_LOr:
Alexey Bataev5e018f92015-04-23 06:35:10 +00001852 return std::make_pair(false, RValue::get(nullptr));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001853 case BO_PtrMemD:
1854 case BO_PtrMemI:
1855 case BO_LE:
1856 case BO_GE:
1857 case BO_EQ:
1858 case BO_NE:
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001859 case BO_AddAssign:
1860 case BO_SubAssign:
1861 case BO_AndAssign:
1862 case BO_OrAssign:
1863 case BO_XorAssign:
1864 case BO_MulAssign:
1865 case BO_DivAssign:
1866 case BO_RemAssign:
1867 case BO_ShlAssign:
1868 case BO_ShrAssign:
1869 case BO_Comma:
1870 llvm_unreachable("Unsupported atomic update operation");
1871 }
1872 auto *UpdateVal = Update.getScalarVal();
1873 if (auto *IC = dyn_cast<llvm::ConstantInt>(UpdateVal)) {
1874 UpdateVal = CGF.Builder.CreateIntCast(
1875 IC, X.getAddress()->getType()->getPointerElementType(),
1876 X.getType()->hasSignedIntegerRepresentation());
1877 }
Alexey Bataev5e018f92015-04-23 06:35:10 +00001878 auto *Res = CGF.Builder.CreateAtomicRMW(RMWOp, X.getAddress(), UpdateVal, AO);
1879 return std::make_pair(true, RValue::get(Res));
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001880}
1881
Alexey Bataev5e018f92015-04-23 06:35:10 +00001882std::pair<bool, RValue> CodeGenFunction::EmitOMPAtomicSimpleUpdateExpr(
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001883 LValue X, RValue E, BinaryOperatorKind BO, bool IsXLHSInRHSPart,
1884 llvm::AtomicOrdering AO, SourceLocation Loc,
1885 const llvm::function_ref<RValue(RValue)> &CommonGen) {
1886 // Update expressions are allowed to have the following forms:
1887 // x binop= expr; -> xrval + expr;
1888 // x++, ++x -> xrval + 1;
1889 // x--, --x -> xrval - 1;
1890 // x = x binop expr; -> xrval binop expr
1891 // x = expr Op x; - > expr binop xrval;
Alexey Bataev5e018f92015-04-23 06:35:10 +00001892 auto Res = emitOMPAtomicRMW(*this, X, E, BO, AO, IsXLHSInRHSPart);
1893 if (!Res.first) {
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001894 if (X.isGlobalReg()) {
1895 // Emit an update expression: 'xrval' binop 'expr' or 'expr' binop
1896 // 'xrval'.
1897 EmitStoreThroughLValue(CommonGen(EmitLoadOfLValue(X, Loc)), X);
1898 } else {
1899 // Perform compare-and-swap procedure.
1900 EmitAtomicUpdate(X, AO, CommonGen, X.getType().isVolatileQualified());
Alexey Bataevb4505a72015-03-30 05:20:59 +00001901 }
1902 }
Alexey Bataev5e018f92015-04-23 06:35:10 +00001903 return Res;
Alexey Bataevb4505a72015-03-30 05:20:59 +00001904}
1905
1906static void EmitOMPAtomicUpdateExpr(CodeGenFunction &CGF, bool IsSeqCst,
1907 const Expr *X, const Expr *E,
1908 const Expr *UE, bool IsXLHSInRHSPart,
1909 SourceLocation Loc) {
1910 assert(isa<BinaryOperator>(UE->IgnoreImpCasts()) &&
1911 "Update expr in 'atomic update' must be a binary operator.");
1912 auto *BOUE = cast<BinaryOperator>(UE->IgnoreImpCasts());
1913 // Update expressions are allowed to have the following forms:
1914 // x binop= expr; -> xrval + expr;
1915 // x++, ++x -> xrval + 1;
1916 // x--, --x -> xrval - 1;
1917 // x = x binop expr; -> xrval binop expr
1918 // x = expr Op x; - > expr binop xrval;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001919 assert(X->isLValue() && "X of 'omp atomic update' is not lvalue");
Alexey Bataevb4505a72015-03-30 05:20:59 +00001920 LValue XLValue = CGF.EmitLValue(X);
1921 RValue ExprRValue = CGF.EmitAnyExpr(E);
Alexey Bataevb4505a72015-03-30 05:20:59 +00001922 auto AO = IsSeqCst ? llvm::SequentiallyConsistent : llvm::Monotonic;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00001923 auto *LHS = cast<OpaqueValueExpr>(BOUE->getLHS()->IgnoreImpCasts());
1924 auto *RHS = cast<OpaqueValueExpr>(BOUE->getRHS()->IgnoreImpCasts());
1925 auto *XRValExpr = IsXLHSInRHSPart ? LHS : RHS;
1926 auto *ERValExpr = IsXLHSInRHSPart ? RHS : LHS;
1927 auto Gen =
1928 [&CGF, UE, ExprRValue, XRValExpr, ERValExpr](RValue XRValue) -> RValue {
1929 CodeGenFunction::OpaqueValueMapping MapExpr(CGF, ERValExpr, ExprRValue);
1930 CodeGenFunction::OpaqueValueMapping MapX(CGF, XRValExpr, XRValue);
1931 return CGF.EmitAnyExpr(UE);
1932 };
Alexey Bataev5e018f92015-04-23 06:35:10 +00001933 (void)CGF.EmitOMPAtomicSimpleUpdateExpr(
1934 XLValue, ExprRValue, BOUE->getOpcode(), IsXLHSInRHSPart, AO, Loc, Gen);
1935 // OpenMP, 2.12.6, atomic Construct
1936 // Any atomic construct with a seq_cst clause forces the atomically
1937 // performed operation to include an implicit flush operation without a
1938 // list.
1939 if (IsSeqCst)
1940 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc);
1941}
1942
1943static RValue convertToType(CodeGenFunction &CGF, RValue Value,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001944 QualType SourceType, QualType ResType,
1945 SourceLocation Loc) {
Alexey Bataev5e018f92015-04-23 06:35:10 +00001946 switch (CGF.getEvaluationKind(ResType)) {
1947 case TEK_Scalar:
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001948 return RValue::get(
1949 convertToScalarValue(CGF, Value, SourceType, ResType, Loc));
Alexey Bataev5e018f92015-04-23 06:35:10 +00001950 case TEK_Complex: {
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001951 auto Res = convertToComplexValue(CGF, Value, SourceType, ResType, Loc);
Alexey Bataev5e018f92015-04-23 06:35:10 +00001952 return RValue::getComplex(Res.first, Res.second);
1953 }
1954 case TEK_Aggregate:
1955 break;
1956 }
1957 llvm_unreachable("Must be a scalar or complex.");
1958}
1959
1960static void EmitOMPAtomicCaptureExpr(CodeGenFunction &CGF, bool IsSeqCst,
1961 bool IsPostfixUpdate, const Expr *V,
1962 const Expr *X, const Expr *E,
1963 const Expr *UE, bool IsXLHSInRHSPart,
1964 SourceLocation Loc) {
1965 assert(X->isLValue() && "X of 'omp atomic capture' is not lvalue");
1966 assert(V->isLValue() && "V of 'omp atomic capture' is not lvalue");
1967 RValue NewVVal;
1968 LValue VLValue = CGF.EmitLValue(V);
1969 LValue XLValue = CGF.EmitLValue(X);
1970 RValue ExprRValue = CGF.EmitAnyExpr(E);
1971 auto AO = IsSeqCst ? llvm::SequentiallyConsistent : llvm::Monotonic;
1972 QualType NewVValType;
1973 if (UE) {
1974 // 'x' is updated with some additional value.
1975 assert(isa<BinaryOperator>(UE->IgnoreImpCasts()) &&
1976 "Update expr in 'atomic capture' must be a binary operator.");
1977 auto *BOUE = cast<BinaryOperator>(UE->IgnoreImpCasts());
1978 // Update expressions are allowed to have the following forms:
1979 // x binop= expr; -> xrval + expr;
1980 // x++, ++x -> xrval + 1;
1981 // x--, --x -> xrval - 1;
1982 // x = x binop expr; -> xrval binop expr
1983 // x = expr Op x; - > expr binop xrval;
1984 auto *LHS = cast<OpaqueValueExpr>(BOUE->getLHS()->IgnoreImpCasts());
1985 auto *RHS = cast<OpaqueValueExpr>(BOUE->getRHS()->IgnoreImpCasts());
1986 auto *XRValExpr = IsXLHSInRHSPart ? LHS : RHS;
1987 NewVValType = XRValExpr->getType();
1988 auto *ERValExpr = IsXLHSInRHSPart ? RHS : LHS;
1989 auto &&Gen = [&CGF, &NewVVal, UE, ExprRValue, XRValExpr, ERValExpr,
1990 IsSeqCst, IsPostfixUpdate](RValue XRValue) -> RValue {
1991 CodeGenFunction::OpaqueValueMapping MapExpr(CGF, ERValExpr, ExprRValue);
1992 CodeGenFunction::OpaqueValueMapping MapX(CGF, XRValExpr, XRValue);
1993 RValue Res = CGF.EmitAnyExpr(UE);
1994 NewVVal = IsPostfixUpdate ? XRValue : Res;
1995 return Res;
1996 };
1997 auto Res = CGF.EmitOMPAtomicSimpleUpdateExpr(
1998 XLValue, ExprRValue, BOUE->getOpcode(), IsXLHSInRHSPart, AO, Loc, Gen);
1999 if (Res.first) {
2000 // 'atomicrmw' instruction was generated.
2001 if (IsPostfixUpdate) {
2002 // Use old value from 'atomicrmw'.
2003 NewVVal = Res.second;
2004 } else {
2005 // 'atomicrmw' does not provide new value, so evaluate it using old
2006 // value of 'x'.
2007 CodeGenFunction::OpaqueValueMapping MapExpr(CGF, ERValExpr, ExprRValue);
2008 CodeGenFunction::OpaqueValueMapping MapX(CGF, XRValExpr, Res.second);
2009 NewVVal = CGF.EmitAnyExpr(UE);
2010 }
2011 }
2012 } else {
2013 // 'x' is simply rewritten with some 'expr'.
2014 NewVValType = X->getType().getNonReferenceType();
2015 ExprRValue = convertToType(CGF, ExprRValue, E->getType(),
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00002016 X->getType().getNonReferenceType(), Loc);
Alexey Bataev5e018f92015-04-23 06:35:10 +00002017 auto &&Gen = [&CGF, &NewVVal, ExprRValue](RValue XRValue) -> RValue {
2018 NewVVal = XRValue;
2019 return ExprRValue;
2020 };
2021 // Try to perform atomicrmw xchg, otherwise simple exchange.
2022 auto Res = CGF.EmitOMPAtomicSimpleUpdateExpr(
2023 XLValue, ExprRValue, /*BO=*/BO_Assign, /*IsXLHSInRHSPart=*/false, AO,
2024 Loc, Gen);
2025 if (Res.first) {
2026 // 'atomicrmw' instruction was generated.
2027 NewVVal = IsPostfixUpdate ? Res.second : ExprRValue;
2028 }
2029 }
2030 // Emit post-update store to 'v' of old/new 'x' value.
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00002031 emitSimpleStore(CGF, VLValue, NewVVal, NewVValType, Loc);
Alexey Bataevb4505a72015-03-30 05:20:59 +00002032 // OpenMP, 2.12.6, atomic Construct
2033 // Any atomic construct with a seq_cst clause forces the atomically
2034 // performed operation to include an implicit flush operation without a
2035 // list.
2036 if (IsSeqCst)
2037 CGF.CGM.getOpenMPRuntime().emitFlush(CGF, llvm::None, Loc);
2038}
2039
Alexey Bataevb57056f2015-01-22 06:17:56 +00002040static void EmitOMPAtomicExpr(CodeGenFunction &CGF, OpenMPClauseKind Kind,
Alexey Bataev5e018f92015-04-23 06:35:10 +00002041 bool IsSeqCst, bool IsPostfixUpdate,
2042 const Expr *X, const Expr *V, const Expr *E,
2043 const Expr *UE, bool IsXLHSInRHSPart,
2044 SourceLocation Loc) {
Alexey Bataevb57056f2015-01-22 06:17:56 +00002045 switch (Kind) {
2046 case OMPC_read:
2047 EmitOMPAtomicReadExpr(CGF, IsSeqCst, X, V, Loc);
2048 break;
2049 case OMPC_write:
Alexey Bataevb8329262015-02-27 06:33:30 +00002050 EmitOMPAtomicWriteExpr(CGF, IsSeqCst, X, E, Loc);
2051 break;
Alexey Bataevb4505a72015-03-30 05:20:59 +00002052 case OMPC_unknown:
Alexey Bataevb57056f2015-01-22 06:17:56 +00002053 case OMPC_update:
Alexey Bataevb4505a72015-03-30 05:20:59 +00002054 EmitOMPAtomicUpdateExpr(CGF, IsSeqCst, X, E, UE, IsXLHSInRHSPart, Loc);
2055 break;
Alexey Bataevb57056f2015-01-22 06:17:56 +00002056 case OMPC_capture:
Alexey Bataev5e018f92015-04-23 06:35:10 +00002057 EmitOMPAtomicCaptureExpr(CGF, IsSeqCst, IsPostfixUpdate, V, X, E, UE,
2058 IsXLHSInRHSPart, Loc);
2059 break;
Alexey Bataevb57056f2015-01-22 06:17:56 +00002060 case OMPC_if:
2061 case OMPC_final:
2062 case OMPC_num_threads:
2063 case OMPC_private:
2064 case OMPC_firstprivate:
2065 case OMPC_lastprivate:
2066 case OMPC_reduction:
2067 case OMPC_safelen:
2068 case OMPC_collapse:
2069 case OMPC_default:
2070 case OMPC_seq_cst:
2071 case OMPC_shared:
2072 case OMPC_linear:
2073 case OMPC_aligned:
2074 case OMPC_copyin:
2075 case OMPC_copyprivate:
2076 case OMPC_flush:
2077 case OMPC_proc_bind:
2078 case OMPC_schedule:
2079 case OMPC_ordered:
2080 case OMPC_nowait:
2081 case OMPC_untied:
2082 case OMPC_threadprivate:
Alexey Bataev1c2cfbc2015-06-23 14:25:19 +00002083 case OMPC_depend:
Alexey Bataevb57056f2015-01-22 06:17:56 +00002084 case OMPC_mergeable:
Michael Wonge710d542015-08-07 16:16:36 +00002085 case OMPC_device:
Alexey Bataevb57056f2015-01-22 06:17:56 +00002086 llvm_unreachable("Clause is not allowed in 'omp atomic'.");
2087 }
2088}
2089
2090void CodeGenFunction::EmitOMPAtomicDirective(const OMPAtomicDirective &S) {
2091 bool IsSeqCst = S.getSingleClause(/*K=*/OMPC_seq_cst);
2092 OpenMPClauseKind Kind = OMPC_unknown;
2093 for (auto *C : S.clauses()) {
2094 // Find first clause (skip seq_cst clause, if it is first).
2095 if (C->getClauseKind() != OMPC_seq_cst) {
2096 Kind = C->getClauseKind();
2097 break;
2098 }
2099 }
Alexey Bataev10fec572015-03-11 04:48:56 +00002100
2101 const auto *CS =
2102 S.getAssociatedStmt()->IgnoreContainers(/*IgnoreCaptured=*/true);
Alexey Bataev5e018f92015-04-23 06:35:10 +00002103 if (const auto *EWC = dyn_cast<ExprWithCleanups>(CS)) {
Alexey Bataev10fec572015-03-11 04:48:56 +00002104 enterFullExpression(EWC);
Alexey Bataev5e018f92015-04-23 06:35:10 +00002105 }
2106 // Processing for statements under 'atomic capture'.
2107 if (const auto *Compound = dyn_cast<CompoundStmt>(CS)) {
2108 for (const auto *C : Compound->body()) {
2109 if (const auto *EWC = dyn_cast<ExprWithCleanups>(C)) {
2110 enterFullExpression(EWC);
2111 }
2112 }
2113 }
Alexey Bataev10fec572015-03-11 04:48:56 +00002114
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00002115 LexicalScope Scope(*this, S.getSourceRange());
2116 auto &&CodeGen = [&S, Kind, IsSeqCst](CodeGenFunction &CGF) {
Alexey Bataev5e018f92015-04-23 06:35:10 +00002117 EmitOMPAtomicExpr(CGF, Kind, IsSeqCst, S.isPostfixUpdate(), S.getX(),
2118 S.getV(), S.getExpr(), S.getUpdateExpr(),
2119 S.isXLHSInRHSPart(), S.getLocStart());
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00002120 };
Alexey Bataev81c7ea02015-07-03 09:56:58 +00002121 CGM.getOpenMPRuntime().emitInlinedDirective(*this, OMPD_atomic, CodeGen);
Alexey Bataev0162e452014-07-22 10:10:35 +00002122}
2123
Alexey Bataev0bd520b2014-09-19 08:19:49 +00002124void CodeGenFunction::EmitOMPTargetDirective(const OMPTargetDirective &) {
2125 llvm_unreachable("CodeGen for 'omp target' is not supported yet.");
2126}
2127
Alexey Bataev13314bf2014-10-09 04:18:56 +00002128void CodeGenFunction::EmitOMPTeamsDirective(const OMPTeamsDirective &) {
2129 llvm_unreachable("CodeGen for 'omp teams' is not supported yet.");
2130}
Alexey Bataev6d4ed052015-07-01 06:57:41 +00002131
2132void CodeGenFunction::EmitOMPCancellationPointDirective(
2133 const OMPCancellationPointDirective &S) {
Alexey Bataev0f34da12015-07-02 04:17:07 +00002134 CGM.getOpenMPRuntime().emitCancellationPointCall(*this, S.getLocStart(),
2135 S.getCancelRegion());
Alexey Bataev6d4ed052015-07-01 06:57:41 +00002136}
2137
Alexey Bataev80909872015-07-02 11:25:17 +00002138void CodeGenFunction::EmitOMPCancelDirective(const OMPCancelDirective &S) {
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00002139 CGM.getOpenMPRuntime().emitCancelCall(*this, S.getLocStart(),
2140 S.getCancelRegion());
Alexey Bataev80909872015-07-02 11:25:17 +00002141}
2142
Alexey Bataev81c7ea02015-07-03 09:56:58 +00002143CodeGenFunction::JumpDest
2144CodeGenFunction::getOMPCancelDestination(OpenMPDirectiveKind Kind) {
2145 if (Kind == OMPD_parallel || Kind == OMPD_task)
2146 return ReturnBlock;
2147 else if (Kind == OMPD_for || Kind == OMPD_section || Kind == OMPD_sections)
2148 return BreakContinueStack.empty() ? JumpDest()
2149 : BreakContinueStack.back().BreakBlock;
2150 return JumpDest();
2151}
Michael Wong65f367f2015-07-21 13:44:28 +00002152
2153// Generate the instructions for '#pragma omp target data' directive.
2154void CodeGenFunction::EmitOMPTargetDataDirective(
2155 const OMPTargetDataDirective &S) {
2156
2157 // emit the code inside the construct for now
2158 auto CS = cast<CapturedStmt>(S.getAssociatedStmt());
2159 EmitStmt(CS->getCapturedStmt());
2160}