blob: e6bc99eaf569c644401ddff1e64831798d7cb4b2 [file] [log] [blame]
Alexey Bataev9959db52014-05-06 10:08:46 +00001//===----- CGOpenMPRuntime.cpp - Interface to OpenMP Runtimes -------------===//
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 provides a class for OpenMP runtime code generation.
11//
12//===----------------------------------------------------------------------===//
13
Samuel Antaoee8fb302016-01-06 13:42:12 +000014#include "CGCXXABI.h"
15#include "CGCleanup.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000016#include "CGOpenMPRuntime.h"
17#include "CodeGenFunction.h"
18#include "clang/AST/Decl.h"
Chandler Carruth0d9593d2015-01-14 11:29:14 +000019#include "clang/AST/StmtOpenMP.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000020#include "llvm/ADT/ArrayRef.h"
Samuel Antaoee8fb302016-01-06 13:42:12 +000021#include "llvm/Bitcode/ReaderWriter.h"
Alexey Bataevd74d0602014-10-13 06:02:40 +000022#include "llvm/IR/CallSite.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000023#include "llvm/IR/DerivedTypes.h"
24#include "llvm/IR/GlobalValue.h"
25#include "llvm/IR/Value.h"
Samuel Antaoee8fb302016-01-06 13:42:12 +000026#include "llvm/Support/Format.h"
Alexey Bataev9959db52014-05-06 10:08:46 +000027#include "llvm/Support/raw_ostream.h"
Alexey Bataev23b69422014-06-18 07:08:49 +000028#include <cassert>
Alexey Bataev9959db52014-05-06 10:08:46 +000029
30using namespace clang;
31using namespace CodeGen;
32
Benjamin Kramerc52193f2014-10-10 13:57:57 +000033namespace {
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000034/// \brief Base class for handling code generation inside OpenMP regions.
Alexey Bataev18095712014-10-10 12:19:54 +000035class CGOpenMPRegionInfo : public CodeGenFunction::CGCapturedStmtInfo {
36public:
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000037 /// \brief Kinds of OpenMP regions used in codegen.
38 enum CGOpenMPRegionKind {
39 /// \brief Region with outlined function for standalone 'parallel'
40 /// directive.
41 ParallelOutlinedRegion,
42 /// \brief Region with outlined function for standalone 'task' directive.
43 TaskOutlinedRegion,
44 /// \brief Region for constructs that do not require function outlining,
45 /// like 'for', 'sections', 'atomic' etc. directives.
46 InlinedRegion,
Samuel Antaobed3c462015-10-02 16:14:20 +000047 /// \brief Region with outlined function for standalone 'target' directive.
48 TargetRegion,
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000049 };
Alexey Bataev18095712014-10-10 12:19:54 +000050
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000051 CGOpenMPRegionInfo(const CapturedStmt &CS,
52 const CGOpenMPRegionKind RegionKind,
Alexey Bataev25e5b442015-09-15 12:52:43 +000053 const RegionCodeGenTy &CodeGen, OpenMPDirectiveKind Kind,
54 bool HasCancel)
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000055 : CGCapturedStmtInfo(CS, CR_OpenMP), RegionKind(RegionKind),
Alexey Bataev25e5b442015-09-15 12:52:43 +000056 CodeGen(CodeGen), Kind(Kind), HasCancel(HasCancel) {}
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000057
58 CGOpenMPRegionInfo(const CGOpenMPRegionKind RegionKind,
Alexey Bataev25e5b442015-09-15 12:52:43 +000059 const RegionCodeGenTy &CodeGen, OpenMPDirectiveKind Kind,
60 bool HasCancel)
Alexey Bataev81c7ea02015-07-03 09:56:58 +000061 : CGCapturedStmtInfo(CR_OpenMP), RegionKind(RegionKind), CodeGen(CodeGen),
Alexey Bataev25e5b442015-09-15 12:52:43 +000062 Kind(Kind), HasCancel(HasCancel) {}
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000063
64 /// \brief Get a variable or parameter for storing global thread id
Alexey Bataev18095712014-10-10 12:19:54 +000065 /// inside OpenMP construct.
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000066 virtual const VarDecl *getThreadIDVariable() const = 0;
Alexey Bataev18095712014-10-10 12:19:54 +000067
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000068 /// \brief Emit the captured statement body.
Hans Wennborg7eb54642015-09-10 17:07:54 +000069 void EmitBody(CodeGenFunction &CGF, const Stmt *S) override;
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000070
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000071 /// \brief Get an LValue for the current ThreadID variable.
Alexey Bataev62b63b12015-03-10 07:28:44 +000072 /// \return LValue for thread id variable. This LValue always has type int32*.
73 virtual LValue getThreadIDVariableLValue(CodeGenFunction &CGF);
Alexey Bataev18095712014-10-10 12:19:54 +000074
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000075 CGOpenMPRegionKind getRegionKind() const { return RegionKind; }
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000076
Alexey Bataev81c7ea02015-07-03 09:56:58 +000077 OpenMPDirectiveKind getDirectiveKind() const { return Kind; }
78
Alexey Bataev25e5b442015-09-15 12:52:43 +000079 bool hasCancel() const { return HasCancel; }
80
Alexey Bataev18095712014-10-10 12:19:54 +000081 static bool classof(const CGCapturedStmtInfo *Info) {
82 return Info->getKind() == CR_OpenMP;
83 }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000084
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000085protected:
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000086 CGOpenMPRegionKind RegionKind;
Hans Wennborg45c74392016-01-12 20:54:36 +000087 RegionCodeGenTy CodeGen;
Alexey Bataev81c7ea02015-07-03 09:56:58 +000088 OpenMPDirectiveKind Kind;
Alexey Bataev25e5b442015-09-15 12:52:43 +000089 bool HasCancel;
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000090};
Alexey Bataev18095712014-10-10 12:19:54 +000091
Alexey Bataev8cbe0a62015-02-26 10:27:34 +000092/// \brief API for captured statement code generation in OpenMP constructs.
93class CGOpenMPOutlinedRegionInfo : public CGOpenMPRegionInfo {
94public:
Alexey Bataev6f1ffc02015-04-10 04:50:10 +000095 CGOpenMPOutlinedRegionInfo(const CapturedStmt &CS, const VarDecl *ThreadIDVar,
Alexey Bataev81c7ea02015-07-03 09:56:58 +000096 const RegionCodeGenTy &CodeGen,
Alexey Bataev25e5b442015-09-15 12:52:43 +000097 OpenMPDirectiveKind Kind, bool HasCancel)
98 : CGOpenMPRegionInfo(CS, ParallelOutlinedRegion, CodeGen, Kind,
99 HasCancel),
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000100 ThreadIDVar(ThreadIDVar) {
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000101 assert(ThreadIDVar != nullptr && "No ThreadID in OpenMP region.");
102 }
103 /// \brief Get a variable or parameter for storing global thread id
104 /// inside OpenMP construct.
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000105 const VarDecl *getThreadIDVariable() const override { return ThreadIDVar; }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000106
Alexey Bataev18095712014-10-10 12:19:54 +0000107 /// \brief Get the name of the capture helper.
Benjamin Kramerc52193f2014-10-10 13:57:57 +0000108 StringRef getHelperName() const override { return ".omp_outlined."; }
Alexey Bataev18095712014-10-10 12:19:54 +0000109
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000110 static bool classof(const CGCapturedStmtInfo *Info) {
111 return CGOpenMPRegionInfo::classof(Info) &&
112 cast<CGOpenMPRegionInfo>(Info)->getRegionKind() ==
113 ParallelOutlinedRegion;
114 }
115
Alexey Bataev18095712014-10-10 12:19:54 +0000116private:
117 /// \brief A variable or parameter storing global thread id for OpenMP
118 /// constructs.
119 const VarDecl *ThreadIDVar;
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000120};
121
Alexey Bataev62b63b12015-03-10 07:28:44 +0000122/// \brief API for captured statement code generation in OpenMP constructs.
123class CGOpenMPTaskOutlinedRegionInfo : public CGOpenMPRegionInfo {
124public:
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000125 CGOpenMPTaskOutlinedRegionInfo(const CapturedStmt &CS,
Alexey Bataev62b63b12015-03-10 07:28:44 +0000126 const VarDecl *ThreadIDVar,
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000127 const RegionCodeGenTy &CodeGen,
Alexey Bataev25e5b442015-09-15 12:52:43 +0000128 OpenMPDirectiveKind Kind, bool HasCancel)
129 : CGOpenMPRegionInfo(CS, TaskOutlinedRegion, CodeGen, Kind, HasCancel),
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000130 ThreadIDVar(ThreadIDVar) {
Alexey Bataev62b63b12015-03-10 07:28:44 +0000131 assert(ThreadIDVar != nullptr && "No ThreadID in OpenMP region.");
132 }
133 /// \brief Get a variable or parameter for storing global thread id
134 /// inside OpenMP construct.
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000135 const VarDecl *getThreadIDVariable() const override { return ThreadIDVar; }
Alexey Bataev62b63b12015-03-10 07:28:44 +0000136
137 /// \brief Get an LValue for the current ThreadID variable.
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000138 LValue getThreadIDVariableLValue(CodeGenFunction &CGF) override;
Alexey Bataev62b63b12015-03-10 07:28:44 +0000139
Alexey Bataev62b63b12015-03-10 07:28:44 +0000140 /// \brief Get the name of the capture helper.
141 StringRef getHelperName() const override { return ".omp_outlined."; }
142
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000143 static bool classof(const CGCapturedStmtInfo *Info) {
144 return CGOpenMPRegionInfo::classof(Info) &&
145 cast<CGOpenMPRegionInfo>(Info)->getRegionKind() ==
146 TaskOutlinedRegion;
147 }
148
Alexey Bataev62b63b12015-03-10 07:28:44 +0000149private:
150 /// \brief A variable or parameter storing global thread id for OpenMP
151 /// constructs.
152 const VarDecl *ThreadIDVar;
Alexey Bataev62b63b12015-03-10 07:28:44 +0000153};
154
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000155/// \brief API for inlined captured statement code generation in OpenMP
156/// constructs.
157class CGOpenMPInlinedRegionInfo : public CGOpenMPRegionInfo {
158public:
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000159 CGOpenMPInlinedRegionInfo(CodeGenFunction::CGCapturedStmtInfo *OldCSI,
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000160 const RegionCodeGenTy &CodeGen,
Alexey Bataev25e5b442015-09-15 12:52:43 +0000161 OpenMPDirectiveKind Kind, bool HasCancel)
162 : CGOpenMPRegionInfo(InlinedRegion, CodeGen, Kind, HasCancel),
163 OldCSI(OldCSI),
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000164 OuterRegionInfo(dyn_cast_or_null<CGOpenMPRegionInfo>(OldCSI)) {}
165 // \brief Retrieve the value of the context parameter.
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000166 llvm::Value *getContextValue() const override {
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000167 if (OuterRegionInfo)
168 return OuterRegionInfo->getContextValue();
169 llvm_unreachable("No context value for inlined OpenMP region");
170 }
Hans Wennborg7eb54642015-09-10 17:07:54 +0000171 void setContextValue(llvm::Value *V) override {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000172 if (OuterRegionInfo) {
173 OuterRegionInfo->setContextValue(V);
174 return;
175 }
176 llvm_unreachable("No context value for inlined OpenMP region");
177 }
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000178 /// \brief Lookup the captured field decl for a variable.
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000179 const FieldDecl *lookup(const VarDecl *VD) const override {
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000180 if (OuterRegionInfo)
181 return OuterRegionInfo->lookup(VD);
Alexey Bataev69c62a92015-04-15 04:52:20 +0000182 // If there is no outer outlined region,no need to lookup in a list of
183 // captured variables, we can use the original one.
184 return nullptr;
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000185 }
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000186 FieldDecl *getThisFieldDecl() const override {
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000187 if (OuterRegionInfo)
188 return OuterRegionInfo->getThisFieldDecl();
189 return nullptr;
190 }
191 /// \brief Get a variable or parameter for storing global thread id
192 /// inside OpenMP construct.
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000193 const VarDecl *getThreadIDVariable() const override {
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000194 if (OuterRegionInfo)
195 return OuterRegionInfo->getThreadIDVariable();
196 return nullptr;
197 }
Alexey Bataev62b63b12015-03-10 07:28:44 +0000198
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000199 /// \brief Get the name of the capture helper.
Alexander Kornienko34eb2072015-04-11 02:00:23 +0000200 StringRef getHelperName() const override {
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000201 if (auto *OuterRegionInfo = getOldCSI())
202 return OuterRegionInfo->getHelperName();
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000203 llvm_unreachable("No helper name for inlined OpenMP construct");
204 }
205
206 CodeGenFunction::CGCapturedStmtInfo *getOldCSI() const { return OldCSI; }
207
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000208 static bool classof(const CGCapturedStmtInfo *Info) {
209 return CGOpenMPRegionInfo::classof(Info) &&
210 cast<CGOpenMPRegionInfo>(Info)->getRegionKind() == InlinedRegion;
211 }
212
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000213private:
214 /// \brief CodeGen info about outer OpenMP region.
215 CodeGenFunction::CGCapturedStmtInfo *OldCSI;
216 CGOpenMPRegionInfo *OuterRegionInfo;
Alexey Bataev18095712014-10-10 12:19:54 +0000217};
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000218
Samuel Antaobed3c462015-10-02 16:14:20 +0000219/// \brief API for captured statement code generation in OpenMP target
220/// constructs. For this captures, implicit parameters are used instead of the
Samuel Antaoee8fb302016-01-06 13:42:12 +0000221/// captured fields. The name of the target region has to be unique in a given
222/// application so it is provided by the client, because only the client has
223/// the information to generate that.
Samuel Antaobed3c462015-10-02 16:14:20 +0000224class CGOpenMPTargetRegionInfo : public CGOpenMPRegionInfo {
225public:
226 CGOpenMPTargetRegionInfo(const CapturedStmt &CS,
Samuel Antaoee8fb302016-01-06 13:42:12 +0000227 const RegionCodeGenTy &CodeGen, StringRef HelperName)
Samuel Antaobed3c462015-10-02 16:14:20 +0000228 : CGOpenMPRegionInfo(CS, TargetRegion, CodeGen, OMPD_target,
Samuel Antaoee8fb302016-01-06 13:42:12 +0000229 /*HasCancel=*/false),
230 HelperName(HelperName) {}
Samuel Antaobed3c462015-10-02 16:14:20 +0000231
232 /// \brief This is unused for target regions because each starts executing
233 /// with a single thread.
234 const VarDecl *getThreadIDVariable() const override { return nullptr; }
235
236 /// \brief Get the name of the capture helper.
Samuel Antaoee8fb302016-01-06 13:42:12 +0000237 StringRef getHelperName() const override { return HelperName; }
Samuel Antaobed3c462015-10-02 16:14:20 +0000238
239 static bool classof(const CGCapturedStmtInfo *Info) {
240 return CGOpenMPRegionInfo::classof(Info) &&
241 cast<CGOpenMPRegionInfo>(Info)->getRegionKind() == TargetRegion;
242 }
Samuel Antaoee8fb302016-01-06 13:42:12 +0000243
244private:
245 StringRef HelperName;
Samuel Antaobed3c462015-10-02 16:14:20 +0000246};
247
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000248/// \brief RAII for emitting code of OpenMP constructs.
249class InlinedOpenMPRegionRAII {
250 CodeGenFunction &CGF;
251
252public:
253 /// \brief Constructs region for combined constructs.
254 /// \param CodeGen Code generation sequence for combined directives. Includes
255 /// a list of functions used for code generation of implicitly inlined
256 /// regions.
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000257 InlinedOpenMPRegionRAII(CodeGenFunction &CGF, const RegionCodeGenTy &CodeGen,
Alexey Bataev25e5b442015-09-15 12:52:43 +0000258 OpenMPDirectiveKind Kind, bool HasCancel)
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000259 : CGF(CGF) {
260 // Start emission for the construct.
Alexey Bataev25e5b442015-09-15 12:52:43 +0000261 CGF.CapturedStmtInfo = new CGOpenMPInlinedRegionInfo(
262 CGF.CapturedStmtInfo, CodeGen, Kind, HasCancel);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000263 }
264 ~InlinedOpenMPRegionRAII() {
265 // Restore original CapturedStmtInfo only if we're done with code emission.
266 auto *OldCSI =
267 cast<CGOpenMPInlinedRegionInfo>(CGF.CapturedStmtInfo)->getOldCSI();
268 delete CGF.CapturedStmtInfo;
269 CGF.CapturedStmtInfo = OldCSI;
270 }
271};
272
Hans Wennborg7eb54642015-09-10 17:07:54 +0000273} // anonymous namespace
Alexey Bataev18095712014-10-10 12:19:54 +0000274
Alexey Bataev2377fe92015-09-10 08:12:02 +0000275static LValue emitLoadOfPointerLValue(CodeGenFunction &CGF, Address PtrAddr,
276 QualType Ty) {
277 AlignmentSource Source;
278 CharUnits Align = CGF.getNaturalPointeeTypeAlignment(Ty, &Source);
279 return CGF.MakeAddrLValue(Address(CGF.Builder.CreateLoad(PtrAddr), Align),
280 Ty->getPointeeType(), Source);
281}
282
Alexey Bataev18095712014-10-10 12:19:54 +0000283LValue CGOpenMPRegionInfo::getThreadIDVariableLValue(CodeGenFunction &CGF) {
Alexey Bataev2377fe92015-09-10 08:12:02 +0000284 return emitLoadOfPointerLValue(CGF,
285 CGF.GetAddrOfLocalVar(getThreadIDVariable()),
286 getThreadIDVariable()->getType());
Alexey Bataev18095712014-10-10 12:19:54 +0000287}
288
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000289void CGOpenMPRegionInfo::EmitBody(CodeGenFunction &CGF, const Stmt * /*S*/) {
Alexey Bataev8ef31412015-12-18 07:58:25 +0000290 if (!CGF.HaveInsertPoint())
291 return;
Alexey Bataev6f1ffc02015-04-10 04:50:10 +0000292 // 1.2.2 OpenMP Language Terminology
293 // Structured block - An executable statement with a single entry at the
294 // top and a single exit at the bottom.
295 // The point of exit cannot be a branch out of the structured block.
296 // longjmp() and throw() must not violate the entry/exit criteria.
297 CGF.EHStack.pushTerminate();
298 {
299 CodeGenFunction::RunCleanupsScope Scope(CGF);
300 CodeGen(CGF);
301 }
302 CGF.EHStack.popTerminate();
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000303}
304
Alexey Bataev62b63b12015-03-10 07:28:44 +0000305LValue CGOpenMPTaskOutlinedRegionInfo::getThreadIDVariableLValue(
306 CodeGenFunction &CGF) {
Alexey Bataev2377fe92015-09-10 08:12:02 +0000307 return CGF.MakeAddrLValue(CGF.GetAddrOfLocalVar(getThreadIDVariable()),
308 getThreadIDVariable()->getType(),
309 AlignmentSource::Decl);
Alexey Bataev62b63b12015-03-10 07:28:44 +0000310}
311
Alexey Bataev9959db52014-05-06 10:08:46 +0000312CGOpenMPRuntime::CGOpenMPRuntime(CodeGenModule &CGM)
Samuel Antaoee8fb302016-01-06 13:42:12 +0000313 : CGM(CGM), DefaultOpenMPPSource(nullptr), KmpRoutineEntryPtrTy(nullptr),
314 OffloadEntriesInfoManager(CGM) {
Alexey Bataev9959db52014-05-06 10:08:46 +0000315 IdentTy = llvm::StructType::create(
316 "ident_t", CGM.Int32Ty /* reserved_1 */, CGM.Int32Ty /* flags */,
317 CGM.Int32Ty /* reserved_2 */, CGM.Int32Ty /* reserved_3 */,
Alexander Musmanfdfa8552014-09-11 08:10:57 +0000318 CGM.Int8PtrTy /* psource */, nullptr);
Alexey Bataev9959db52014-05-06 10:08:46 +0000319 // Build void (*kmpc_micro)(kmp_int32 *global_tid, kmp_int32 *bound_tid,...)
Alexey Bataev23b69422014-06-18 07:08:49 +0000320 llvm::Type *MicroParams[] = {llvm::PointerType::getUnqual(CGM.Int32Ty),
321 llvm::PointerType::getUnqual(CGM.Int32Ty)};
Alexey Bataev9959db52014-05-06 10:08:46 +0000322 Kmpc_MicroTy = llvm::FunctionType::get(CGM.VoidTy, MicroParams, true);
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +0000323 KmpCriticalNameTy = llvm::ArrayType::get(CGM.Int32Ty, /*NumElements*/ 8);
Samuel Antaoee8fb302016-01-06 13:42:12 +0000324
325 loadOffloadInfoMetadata();
Alexey Bataev9959db52014-05-06 10:08:46 +0000326}
327
Alexey Bataev91797552015-03-18 04:13:55 +0000328void CGOpenMPRuntime::clear() {
329 InternalVars.clear();
330}
331
John McCall7f416cc2015-09-08 08:05:57 +0000332// Layout information for ident_t.
333static CharUnits getIdentAlign(CodeGenModule &CGM) {
334 return CGM.getPointerAlign();
335}
336static CharUnits getIdentSize(CodeGenModule &CGM) {
337 assert((4 * CGM.getPointerSize()).isMultipleOf(CGM.getPointerAlign()));
338 return CharUnits::fromQuantity(16) + CGM.getPointerSize();
339}
340static CharUnits getOffsetOfIdentField(CGOpenMPRuntime::IdentFieldIndex Field) {
341 // All the fields except the last are i32, so this works beautifully.
342 return unsigned(Field) * CharUnits::fromQuantity(4);
343}
344static Address createIdentFieldGEP(CodeGenFunction &CGF, Address Addr,
345 CGOpenMPRuntime::IdentFieldIndex Field,
346 const llvm::Twine &Name = "") {
347 auto Offset = getOffsetOfIdentField(Field);
348 return CGF.Builder.CreateStructGEP(Addr, Field, Offset, Name);
349}
350
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000351llvm::Value *CGOpenMPRuntime::emitParallelOutlinedFunction(
352 const OMPExecutableDirective &D, const VarDecl *ThreadIDVar,
353 OpenMPDirectiveKind InnermostKind, const RegionCodeGenTy &CodeGen) {
Alexey Bataev62b63b12015-03-10 07:28:44 +0000354 assert(ThreadIDVar->getType()->isPointerType() &&
355 "thread id variable must be of type kmp_int32 *");
Alexey Bataev18095712014-10-10 12:19:54 +0000356 const CapturedStmt *CS = cast<CapturedStmt>(D.getAssociatedStmt());
357 CodeGenFunction CGF(CGM, true);
Alexey Bataev25e5b442015-09-15 12:52:43 +0000358 bool HasCancel = false;
359 if (auto *OPD = dyn_cast<OMPParallelDirective>(&D))
360 HasCancel = OPD->hasCancel();
361 else if (auto *OPSD = dyn_cast<OMPParallelSectionsDirective>(&D))
362 HasCancel = OPSD->hasCancel();
363 else if (auto *OPFD = dyn_cast<OMPParallelForDirective>(&D))
364 HasCancel = OPFD->hasCancel();
365 CGOpenMPOutlinedRegionInfo CGInfo(*CS, ThreadIDVar, CodeGen, InnermostKind,
366 HasCancel);
Alexey Bataevd157d472015-06-24 03:35:38 +0000367 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(CGF, &CGInfo);
Alexey Bataev2377fe92015-09-10 08:12:02 +0000368 return CGF.GenerateOpenMPCapturedStmtFunction(*CS);
Alexey Bataev18095712014-10-10 12:19:54 +0000369}
370
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000371llvm::Value *CGOpenMPRuntime::emitTaskOutlinedFunction(
372 const OMPExecutableDirective &D, const VarDecl *ThreadIDVar,
373 OpenMPDirectiveKind InnermostKind, const RegionCodeGenTy &CodeGen) {
Alexey Bataev62b63b12015-03-10 07:28:44 +0000374 assert(!ThreadIDVar->getType()->isPointerType() &&
375 "thread id variable must be of type kmp_int32 for tasks");
376 auto *CS = cast<CapturedStmt>(D.getAssociatedStmt());
377 CodeGenFunction CGF(CGM, true);
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000378 CGOpenMPTaskOutlinedRegionInfo CGInfo(*CS, ThreadIDVar, CodeGen,
Alexey Bataev25e5b442015-09-15 12:52:43 +0000379 InnermostKind,
380 cast<OMPTaskDirective>(D).hasCancel());
Alexey Bataevd157d472015-06-24 03:35:38 +0000381 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(CGF, &CGInfo);
Alexey Bataev62b63b12015-03-10 07:28:44 +0000382 return CGF.GenerateCapturedStmtFunction(*CS);
383}
384
John McCall7f416cc2015-09-08 08:05:57 +0000385Address CGOpenMPRuntime::getOrCreateDefaultLocation(OpenMPLocationFlags Flags) {
386 CharUnits Align = getIdentAlign(CGM);
Alexey Bataev15007ba2014-05-07 06:18:01 +0000387 llvm::Value *Entry = OpenMPDefaultLocMap.lookup(Flags);
Alexey Bataev9959db52014-05-06 10:08:46 +0000388 if (!Entry) {
389 if (!DefaultOpenMPPSource) {
390 // Initialize default location for psource field of ident_t structure of
391 // all ident_t objects. Format is ";file;function;line;column;;".
392 // Taken from
393 // http://llvm.org/svn/llvm-project/openmp/trunk/runtime/src/kmp_str.c
394 DefaultOpenMPPSource =
John McCall7f416cc2015-09-08 08:05:57 +0000395 CGM.GetAddrOfConstantCString(";unknown;unknown;0;0;;").getPointer();
Alexey Bataev9959db52014-05-06 10:08:46 +0000396 DefaultOpenMPPSource =
397 llvm::ConstantExpr::getBitCast(DefaultOpenMPPSource, CGM.Int8PtrTy);
398 }
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000399 auto DefaultOpenMPLocation = new llvm::GlobalVariable(
400 CGM.getModule(), IdentTy, /*isConstant*/ true,
401 llvm::GlobalValue::PrivateLinkage, /*Initializer*/ nullptr);
Alexey Bataev9959db52014-05-06 10:08:46 +0000402 DefaultOpenMPLocation->setUnnamedAddr(true);
John McCall7f416cc2015-09-08 08:05:57 +0000403 DefaultOpenMPLocation->setAlignment(Align.getQuantity());
Alexey Bataev9959db52014-05-06 10:08:46 +0000404
405 llvm::Constant *Zero = llvm::ConstantInt::get(CGM.Int32Ty, 0, true);
Alexey Bataev23b69422014-06-18 07:08:49 +0000406 llvm::Constant *Values[] = {Zero,
407 llvm::ConstantInt::get(CGM.Int32Ty, Flags),
408 Zero, Zero, DefaultOpenMPPSource};
Alexey Bataev9959db52014-05-06 10:08:46 +0000409 llvm::Constant *Init = llvm::ConstantStruct::get(IdentTy, Values);
410 DefaultOpenMPLocation->setInitializer(Init);
John McCall7f416cc2015-09-08 08:05:57 +0000411 OpenMPDefaultLocMap[Flags] = Entry = DefaultOpenMPLocation;
Alexey Bataev9959db52014-05-06 10:08:46 +0000412 }
John McCall7f416cc2015-09-08 08:05:57 +0000413 return Address(Entry, Align);
Alexey Bataev9959db52014-05-06 10:08:46 +0000414}
415
Alexey Bataev3eff5f42015-02-25 08:32:46 +0000416llvm::Value *CGOpenMPRuntime::emitUpdateLocation(CodeGenFunction &CGF,
417 SourceLocation Loc,
418 OpenMPLocationFlags Flags) {
Alexey Bataev9959db52014-05-06 10:08:46 +0000419 // If no debug info is generated - return global default location.
420 if (CGM.getCodeGenOpts().getDebugInfo() == CodeGenOptions::NoDebugInfo ||
421 Loc.isInvalid())
John McCall7f416cc2015-09-08 08:05:57 +0000422 return getOrCreateDefaultLocation(Flags).getPointer();
Alexey Bataev9959db52014-05-06 10:08:46 +0000423
424 assert(CGF.CurFn && "No function in current CodeGenFunction.");
425
John McCall7f416cc2015-09-08 08:05:57 +0000426 Address LocValue = Address::invalid();
Alexey Bataev1e4b7132014-12-03 12:11:24 +0000427 auto I = OpenMPLocThreadIDMap.find(CGF.CurFn);
428 if (I != OpenMPLocThreadIDMap.end())
John McCall7f416cc2015-09-08 08:05:57 +0000429 LocValue = Address(I->second.DebugLoc, getIdentAlign(CGF.CGM));
430
Alexander Musmanc6388682014-12-15 07:07:06 +0000431 // OpenMPLocThreadIDMap may have null DebugLoc and non-null ThreadID, if
432 // GetOpenMPThreadID was called before this routine.
John McCall7f416cc2015-09-08 08:05:57 +0000433 if (!LocValue.isValid()) {
Alexey Bataev15007ba2014-05-07 06:18:01 +0000434 // Generate "ident_t .kmpc_loc.addr;"
John McCall7f416cc2015-09-08 08:05:57 +0000435 Address AI = CGF.CreateTempAlloca(IdentTy, getIdentAlign(CGF.CGM),
436 ".kmpc_loc.addr");
Alexey Bataev18095712014-10-10 12:19:54 +0000437 auto &Elem = OpenMPLocThreadIDMap.FindAndConstruct(CGF.CurFn);
John McCall7f416cc2015-09-08 08:05:57 +0000438 Elem.second.DebugLoc = AI.getPointer();
Alexey Bataev9959db52014-05-06 10:08:46 +0000439 LocValue = AI;
440
441 CGBuilderTy::InsertPointGuard IPG(CGF.Builder);
442 CGF.Builder.SetInsertPoint(CGF.AllocaInsertPt);
Alexey Bataev3eff5f42015-02-25 08:32:46 +0000443 CGF.Builder.CreateMemCpy(LocValue, getOrCreateDefaultLocation(Flags),
John McCall7f416cc2015-09-08 08:05:57 +0000444 CGM.getSize(getIdentSize(CGF.CGM)));
Alexey Bataev9959db52014-05-06 10:08:46 +0000445 }
446
447 // char **psource = &.kmpc_loc_<flags>.addr.psource;
John McCall7f416cc2015-09-08 08:05:57 +0000448 Address PSource = createIdentFieldGEP(CGF, LocValue, IdentField_PSource);
Alexey Bataev9959db52014-05-06 10:08:46 +0000449
Alexey Bataevf002aca2014-05-30 05:48:40 +0000450 auto OMPDebugLoc = OpenMPDebugLocMap.lookup(Loc.getRawEncoding());
451 if (OMPDebugLoc == nullptr) {
452 SmallString<128> Buffer2;
453 llvm::raw_svector_ostream OS2(Buffer2);
454 // Build debug location
455 PresumedLoc PLoc = CGF.getContext().getSourceManager().getPresumedLoc(Loc);
456 OS2 << ";" << PLoc.getFilename() << ";";
457 if (const FunctionDecl *FD =
458 dyn_cast_or_null<FunctionDecl>(CGF.CurFuncDecl)) {
459 OS2 << FD->getQualifiedNameAsString();
460 }
461 OS2 << ";" << PLoc.getLine() << ";" << PLoc.getColumn() << ";;";
462 OMPDebugLoc = CGF.Builder.CreateGlobalStringPtr(OS2.str());
463 OpenMPDebugLocMap[Loc.getRawEncoding()] = OMPDebugLoc;
Alexey Bataev9959db52014-05-06 10:08:46 +0000464 }
Alexey Bataev9959db52014-05-06 10:08:46 +0000465 // *psource = ";<File>;<Function>;<Line>;<Column>;;";
Alexey Bataevf002aca2014-05-30 05:48:40 +0000466 CGF.Builder.CreateStore(OMPDebugLoc, PSource);
467
John McCall7f416cc2015-09-08 08:05:57 +0000468 // Our callers always pass this to a runtime function, so for
469 // convenience, go ahead and return a naked pointer.
470 return LocValue.getPointer();
Alexey Bataev9959db52014-05-06 10:08:46 +0000471}
472
Alexey Bataev3eff5f42015-02-25 08:32:46 +0000473llvm::Value *CGOpenMPRuntime::getThreadID(CodeGenFunction &CGF,
474 SourceLocation Loc) {
Alexey Bataev9959db52014-05-06 10:08:46 +0000475 assert(CGF.CurFn && "No function in current CodeGenFunction.");
476
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000477 llvm::Value *ThreadID = nullptr;
Alexey Bataev18095712014-10-10 12:19:54 +0000478 // Check whether we've already cached a load of the thread id in this
479 // function.
Alexey Bataev1e4b7132014-12-03 12:11:24 +0000480 auto I = OpenMPLocThreadIDMap.find(CGF.CurFn);
Alexey Bataev18095712014-10-10 12:19:54 +0000481 if (I != OpenMPLocThreadIDMap.end()) {
482 ThreadID = I->second.ThreadID;
Alexey Bataev03b340a2014-10-21 03:16:40 +0000483 if (ThreadID != nullptr)
484 return ThreadID;
485 }
Alexey Bataev3015bcc2016-01-22 08:56:50 +0000486 if (auto *OMPRegionInfo =
Alexey Bataev1e4b7132014-12-03 12:11:24 +0000487 dyn_cast_or_null<CGOpenMPRegionInfo>(CGF.CapturedStmtInfo)) {
Alexey Bataev62b63b12015-03-10 07:28:44 +0000488 if (OMPRegionInfo->getThreadIDVariable()) {
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000489 // Check if this an outlined function with thread id passed as argument.
490 auto LVal = OMPRegionInfo->getThreadIDVariableLValue(CGF);
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000491 ThreadID = CGF.EmitLoadOfLValue(LVal, Loc).getScalarVal();
492 // If value loaded in entry block, cache it and use it everywhere in
493 // function.
494 if (CGF.Builder.GetInsertBlock() == CGF.AllocaInsertPt->getParent()) {
495 auto &Elem = OpenMPLocThreadIDMap.FindAndConstruct(CGF.CurFn);
496 Elem.second.ThreadID = ThreadID;
497 }
498 return ThreadID;
Alexey Bataevd6c57552014-07-25 07:55:17 +0000499 }
Alexey Bataev9959db52014-05-06 10:08:46 +0000500 }
Alexey Bataev8cbe0a62015-02-26 10:27:34 +0000501
502 // This is not an outlined function region - need to call __kmpc_int32
503 // kmpc_global_thread_num(ident_t *loc).
504 // Generate thread id value and cache this value for use across the
505 // function.
506 CGBuilderTy::InsertPointGuard IPG(CGF.Builder);
507 CGF.Builder.SetInsertPoint(CGF.AllocaInsertPt);
508 ThreadID =
509 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_global_thread_num),
510 emitUpdateLocation(CGF, Loc));
511 auto &Elem = OpenMPLocThreadIDMap.FindAndConstruct(CGF.CurFn);
512 Elem.second.ThreadID = ThreadID;
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000513 return ThreadID;
Alexey Bataev9959db52014-05-06 10:08:46 +0000514}
515
Alexey Bataev3eff5f42015-02-25 08:32:46 +0000516void CGOpenMPRuntime::functionFinished(CodeGenFunction &CGF) {
Alexey Bataev9959db52014-05-06 10:08:46 +0000517 assert(CGF.CurFn && "No function in current CodeGenFunction.");
Alexey Bataev03b340a2014-10-21 03:16:40 +0000518 if (OpenMPLocThreadIDMap.count(CGF.CurFn))
519 OpenMPLocThreadIDMap.erase(CGF.CurFn);
Alexey Bataev9959db52014-05-06 10:08:46 +0000520}
521
522llvm::Type *CGOpenMPRuntime::getIdentTyPointerTy() {
523 return llvm::PointerType::getUnqual(IdentTy);
524}
525
526llvm::Type *CGOpenMPRuntime::getKmpc_MicroPointerTy() {
527 return llvm::PointerType::getUnqual(Kmpc_MicroTy);
528}
529
530llvm::Constant *
Alexey Bataev3eff5f42015-02-25 08:32:46 +0000531CGOpenMPRuntime::createRuntimeFunction(OpenMPRTLFunction Function) {
Alexey Bataev9959db52014-05-06 10:08:46 +0000532 llvm::Constant *RTLFn = nullptr;
533 switch (Function) {
534 case OMPRTL__kmpc_fork_call: {
535 // Build void __kmpc_fork_call(ident_t *loc, kmp_int32 argc, kmpc_micro
536 // microtask, ...);
Alexey Bataev23b69422014-06-18 07:08:49 +0000537 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
538 getKmpc_MicroPointerTy()};
Alexey Bataev9959db52014-05-06 10:08:46 +0000539 llvm::FunctionType *FnTy =
Alexey Bataevd74d0602014-10-13 06:02:40 +0000540 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ true);
Alexey Bataev9959db52014-05-06 10:08:46 +0000541 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_fork_call");
542 break;
543 }
544 case OMPRTL__kmpc_global_thread_num: {
545 // Build kmp_int32 __kmpc_global_thread_num(ident_t *loc);
Alexey Bataev23b69422014-06-18 07:08:49 +0000546 llvm::Type *TypeParams[] = {getIdentTyPointerTy()};
Alexey Bataev9959db52014-05-06 10:08:46 +0000547 llvm::FunctionType *FnTy =
Alexey Bataevd74d0602014-10-13 06:02:40 +0000548 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
Alexey Bataev9959db52014-05-06 10:08:46 +0000549 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_global_thread_num");
550 break;
551 }
Alexey Bataev97720002014-11-11 04:05:39 +0000552 case OMPRTL__kmpc_threadprivate_cached: {
553 // Build void *__kmpc_threadprivate_cached(ident_t *loc,
554 // kmp_int32 global_tid, void *data, size_t size, void ***cache);
555 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
556 CGM.VoidPtrTy, CGM.SizeTy,
557 CGM.VoidPtrTy->getPointerTo()->getPointerTo()};
558 llvm::FunctionType *FnTy =
559 llvm::FunctionType::get(CGM.VoidPtrTy, TypeParams, /*isVarArg*/ false);
560 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_threadprivate_cached");
561 break;
562 }
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +0000563 case OMPRTL__kmpc_critical: {
Alexey Bataevf9472182014-09-22 12:32:31 +0000564 // Build void __kmpc_critical(ident_t *loc, kmp_int32 global_tid,
565 // kmp_critical_name *crit);
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +0000566 llvm::Type *TypeParams[] = {
567 getIdentTyPointerTy(), CGM.Int32Ty,
568 llvm::PointerType::getUnqual(KmpCriticalNameTy)};
569 llvm::FunctionType *FnTy =
570 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
571 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_critical");
572 break;
573 }
Alexey Bataevfc57d162015-12-15 10:55:09 +0000574 case OMPRTL__kmpc_critical_with_hint: {
575 // Build void __kmpc_critical_with_hint(ident_t *loc, kmp_int32 global_tid,
576 // kmp_critical_name *crit, uintptr_t hint);
577 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
578 llvm::PointerType::getUnqual(KmpCriticalNameTy),
579 CGM.IntPtrTy};
580 llvm::FunctionType *FnTy =
581 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
582 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_critical_with_hint");
583 break;
584 }
Alexey Bataev97720002014-11-11 04:05:39 +0000585 case OMPRTL__kmpc_threadprivate_register: {
586 // Build void __kmpc_threadprivate_register(ident_t *, void *data,
587 // kmpc_ctor ctor, kmpc_cctor cctor, kmpc_dtor dtor);
588 // typedef void *(*kmpc_ctor)(void *);
589 auto KmpcCtorTy =
590 llvm::FunctionType::get(CGM.VoidPtrTy, CGM.VoidPtrTy,
591 /*isVarArg*/ false)->getPointerTo();
592 // typedef void *(*kmpc_cctor)(void *, void *);
593 llvm::Type *KmpcCopyCtorTyArgs[] = {CGM.VoidPtrTy, CGM.VoidPtrTy};
594 auto KmpcCopyCtorTy =
595 llvm::FunctionType::get(CGM.VoidPtrTy, KmpcCopyCtorTyArgs,
596 /*isVarArg*/ false)->getPointerTo();
597 // typedef void (*kmpc_dtor)(void *);
598 auto KmpcDtorTy =
599 llvm::FunctionType::get(CGM.VoidTy, CGM.VoidPtrTy, /*isVarArg*/ false)
600 ->getPointerTo();
601 llvm::Type *FnTyArgs[] = {getIdentTyPointerTy(), CGM.VoidPtrTy, KmpcCtorTy,
602 KmpcCopyCtorTy, KmpcDtorTy};
603 auto FnTy = llvm::FunctionType::get(CGM.VoidTy, FnTyArgs,
604 /*isVarArg*/ false);
605 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_threadprivate_register");
606 break;
607 }
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +0000608 case OMPRTL__kmpc_end_critical: {
Alexey Bataevf9472182014-09-22 12:32:31 +0000609 // Build void __kmpc_end_critical(ident_t *loc, kmp_int32 global_tid,
610 // kmp_critical_name *crit);
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +0000611 llvm::Type *TypeParams[] = {
612 getIdentTyPointerTy(), CGM.Int32Ty,
613 llvm::PointerType::getUnqual(KmpCriticalNameTy)};
614 llvm::FunctionType *FnTy =
615 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
616 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_end_critical");
617 break;
618 }
Alexey Bataev8f7c1b02014-12-05 04:09:23 +0000619 case OMPRTL__kmpc_cancel_barrier: {
620 // Build kmp_int32 __kmpc_cancel_barrier(ident_t *loc, kmp_int32
621 // global_tid);
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000622 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
623 llvm::FunctionType *FnTy =
Alexey Bataev8f7c1b02014-12-05 04:09:23 +0000624 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
625 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name*/ "__kmpc_cancel_barrier");
Alexey Bataev4a5bb772014-10-08 14:01:46 +0000626 break;
627 }
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000628 case OMPRTL__kmpc_barrier: {
Alexey Bataev7d5d33e2015-07-06 05:50:32 +0000629 // Build void __kmpc_barrier(ident_t *loc, kmp_int32 global_tid);
Alexey Bataev81c7ea02015-07-03 09:56:58 +0000630 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
631 llvm::FunctionType *FnTy =
632 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
633 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name*/ "__kmpc_barrier");
634 break;
635 }
Alexander Musmanc6388682014-12-15 07:07:06 +0000636 case OMPRTL__kmpc_for_static_fini: {
637 // Build void __kmpc_for_static_fini(ident_t *loc, kmp_int32 global_tid);
638 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
639 llvm::FunctionType *FnTy =
640 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
641 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_for_static_fini");
642 break;
643 }
Alexey Bataevb2059782014-10-13 08:23:51 +0000644 case OMPRTL__kmpc_push_num_threads: {
645 // Build void __kmpc_push_num_threads(ident_t *loc, kmp_int32 global_tid,
646 // kmp_int32 num_threads)
647 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
648 CGM.Int32Ty};
649 llvm::FunctionType *FnTy =
650 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
651 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_push_num_threads");
652 break;
653 }
Alexey Bataevd74d0602014-10-13 06:02:40 +0000654 case OMPRTL__kmpc_serialized_parallel: {
655 // Build void __kmpc_serialized_parallel(ident_t *loc, kmp_int32
656 // global_tid);
657 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
658 llvm::FunctionType *FnTy =
659 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
660 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_serialized_parallel");
661 break;
662 }
663 case OMPRTL__kmpc_end_serialized_parallel: {
664 // Build void __kmpc_end_serialized_parallel(ident_t *loc, kmp_int32
665 // global_tid);
666 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
667 llvm::FunctionType *FnTy =
668 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
669 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_end_serialized_parallel");
670 break;
671 }
Alexey Bataevcc37cc12014-11-20 04:34:54 +0000672 case OMPRTL__kmpc_flush: {
Alexey Bataevd76df6d2015-02-24 12:55:09 +0000673 // Build void __kmpc_flush(ident_t *loc);
Alexey Bataevcc37cc12014-11-20 04:34:54 +0000674 llvm::Type *TypeParams[] = {getIdentTyPointerTy()};
675 llvm::FunctionType *FnTy =
Alexey Bataevd76df6d2015-02-24 12:55:09 +0000676 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
Alexey Bataevcc37cc12014-11-20 04:34:54 +0000677 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_flush");
678 break;
679 }
Alexey Bataev8d690652014-12-04 07:23:53 +0000680 case OMPRTL__kmpc_master: {
681 // Build kmp_int32 __kmpc_master(ident_t *loc, kmp_int32 global_tid);
682 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
683 llvm::FunctionType *FnTy =
684 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
685 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_master");
686 break;
687 }
688 case OMPRTL__kmpc_end_master: {
689 // Build void __kmpc_end_master(ident_t *loc, kmp_int32 global_tid);
690 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
691 llvm::FunctionType *FnTy =
692 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
693 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_end_master");
694 break;
695 }
Alexey Bataev9f797f32015-02-05 05:57:51 +0000696 case OMPRTL__kmpc_omp_taskyield: {
697 // Build kmp_int32 __kmpc_omp_taskyield(ident_t *, kmp_int32 global_tid,
698 // int end_part);
699 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty, CGM.IntTy};
700 llvm::FunctionType *FnTy =
701 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
702 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_omp_taskyield");
703 break;
704 }
Alexey Bataev6956e2e2015-02-05 06:35:41 +0000705 case OMPRTL__kmpc_single: {
706 // Build kmp_int32 __kmpc_single(ident_t *loc, kmp_int32 global_tid);
707 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
708 llvm::FunctionType *FnTy =
709 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
710 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_single");
711 break;
712 }
713 case OMPRTL__kmpc_end_single: {
714 // Build void __kmpc_end_single(ident_t *loc, kmp_int32 global_tid);
715 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
716 llvm::FunctionType *FnTy =
717 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
718 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_end_single");
719 break;
720 }
Alexey Bataev62b63b12015-03-10 07:28:44 +0000721 case OMPRTL__kmpc_omp_task_alloc: {
722 // Build kmp_task_t *__kmpc_omp_task_alloc(ident_t *, kmp_int32 gtid,
723 // kmp_int32 flags, size_t sizeof_kmp_task_t, size_t sizeof_shareds,
724 // kmp_routine_entry_t *task_entry);
725 assert(KmpRoutineEntryPtrTy != nullptr &&
726 "Type kmp_routine_entry_t must be created.");
727 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty, CGM.Int32Ty,
728 CGM.SizeTy, CGM.SizeTy, KmpRoutineEntryPtrTy};
729 // Return void * and then cast to particular kmp_task_t type.
730 llvm::FunctionType *FnTy =
731 llvm::FunctionType::get(CGM.VoidPtrTy, TypeParams, /*isVarArg=*/false);
732 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_omp_task_alloc");
733 break;
734 }
735 case OMPRTL__kmpc_omp_task: {
736 // Build kmp_int32 __kmpc_omp_task(ident_t *, kmp_int32 gtid, kmp_task_t
737 // *new_task);
738 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
739 CGM.VoidPtrTy};
740 llvm::FunctionType *FnTy =
741 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
742 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_omp_task");
743 break;
744 }
Alexey Bataeva63048e2015-03-23 06:18:07 +0000745 case OMPRTL__kmpc_copyprivate: {
746 // Build void __kmpc_copyprivate(ident_t *loc, kmp_int32 global_tid,
Alexey Bataev66beaa92015-04-30 03:47:32 +0000747 // size_t cpy_size, void *cpy_data, void(*cpy_func)(void *, void *),
Alexey Bataeva63048e2015-03-23 06:18:07 +0000748 // kmp_int32 didit);
749 llvm::Type *CpyTypeParams[] = {CGM.VoidPtrTy, CGM.VoidPtrTy};
750 auto *CpyFnTy =
751 llvm::FunctionType::get(CGM.VoidTy, CpyTypeParams, /*isVarArg=*/false);
Alexey Bataev66beaa92015-04-30 03:47:32 +0000752 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty, CGM.SizeTy,
Alexey Bataeva63048e2015-03-23 06:18:07 +0000753 CGM.VoidPtrTy, CpyFnTy->getPointerTo(),
754 CGM.Int32Ty};
755 llvm::FunctionType *FnTy =
756 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
757 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_copyprivate");
758 break;
759 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +0000760 case OMPRTL__kmpc_reduce: {
761 // Build kmp_int32 __kmpc_reduce(ident_t *loc, kmp_int32 global_tid,
762 // kmp_int32 num_vars, size_t reduce_size, void *reduce_data, void
763 // (*reduce_func)(void *lhs_data, void *rhs_data), kmp_critical_name *lck);
764 llvm::Type *ReduceTypeParams[] = {CGM.VoidPtrTy, CGM.VoidPtrTy};
765 auto *ReduceFnTy = llvm::FunctionType::get(CGM.VoidTy, ReduceTypeParams,
766 /*isVarArg=*/false);
767 llvm::Type *TypeParams[] = {
768 getIdentTyPointerTy(), CGM.Int32Ty, CGM.Int32Ty, CGM.SizeTy,
769 CGM.VoidPtrTy, ReduceFnTy->getPointerTo(),
770 llvm::PointerType::getUnqual(KmpCriticalNameTy)};
771 llvm::FunctionType *FnTy =
772 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
773 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_reduce");
774 break;
775 }
776 case OMPRTL__kmpc_reduce_nowait: {
777 // Build kmp_int32 __kmpc_reduce_nowait(ident_t *loc, kmp_int32
778 // global_tid, kmp_int32 num_vars, size_t reduce_size, void *reduce_data,
779 // void (*reduce_func)(void *lhs_data, void *rhs_data), kmp_critical_name
780 // *lck);
781 llvm::Type *ReduceTypeParams[] = {CGM.VoidPtrTy, CGM.VoidPtrTy};
782 auto *ReduceFnTy = llvm::FunctionType::get(CGM.VoidTy, ReduceTypeParams,
783 /*isVarArg=*/false);
784 llvm::Type *TypeParams[] = {
785 getIdentTyPointerTy(), CGM.Int32Ty, CGM.Int32Ty, CGM.SizeTy,
786 CGM.VoidPtrTy, ReduceFnTy->getPointerTo(),
787 llvm::PointerType::getUnqual(KmpCriticalNameTy)};
788 llvm::FunctionType *FnTy =
789 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
790 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_reduce_nowait");
791 break;
792 }
793 case OMPRTL__kmpc_end_reduce: {
794 // Build void __kmpc_end_reduce(ident_t *loc, kmp_int32 global_tid,
795 // kmp_critical_name *lck);
796 llvm::Type *TypeParams[] = {
797 getIdentTyPointerTy(), CGM.Int32Ty,
798 llvm::PointerType::getUnqual(KmpCriticalNameTy)};
799 llvm::FunctionType *FnTy =
800 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
801 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_end_reduce");
802 break;
803 }
804 case OMPRTL__kmpc_end_reduce_nowait: {
805 // Build __kmpc_end_reduce_nowait(ident_t *loc, kmp_int32 global_tid,
806 // kmp_critical_name *lck);
807 llvm::Type *TypeParams[] = {
808 getIdentTyPointerTy(), CGM.Int32Ty,
809 llvm::PointerType::getUnqual(KmpCriticalNameTy)};
810 llvm::FunctionType *FnTy =
811 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
812 RTLFn =
813 CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_end_reduce_nowait");
814 break;
815 }
Alexey Bataev1d677132015-04-22 13:57:31 +0000816 case OMPRTL__kmpc_omp_task_begin_if0: {
817 // Build void __kmpc_omp_task(ident_t *, kmp_int32 gtid, kmp_task_t
818 // *new_task);
819 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
820 CGM.VoidPtrTy};
821 llvm::FunctionType *FnTy =
822 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
823 RTLFn =
824 CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_omp_task_begin_if0");
825 break;
826 }
827 case OMPRTL__kmpc_omp_task_complete_if0: {
828 // Build void __kmpc_omp_task(ident_t *, kmp_int32 gtid, kmp_task_t
829 // *new_task);
830 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
831 CGM.VoidPtrTy};
832 llvm::FunctionType *FnTy =
833 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
834 RTLFn = CGM.CreateRuntimeFunction(FnTy,
835 /*Name=*/"__kmpc_omp_task_complete_if0");
836 break;
837 }
Alexey Bataev98eb6e32015-04-22 11:15:40 +0000838 case OMPRTL__kmpc_ordered: {
839 // Build void __kmpc_ordered(ident_t *loc, kmp_int32 global_tid);
840 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
841 llvm::FunctionType *FnTy =
842 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
843 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_ordered");
844 break;
845 }
846 case OMPRTL__kmpc_end_ordered: {
Alexey Bataevc30dd2d2015-06-18 12:14:09 +0000847 // Build void __kmpc_end_ordered(ident_t *loc, kmp_int32 global_tid);
Alexey Bataev98eb6e32015-04-22 11:15:40 +0000848 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
849 llvm::FunctionType *FnTy =
850 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
851 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_end_ordered");
852 break;
853 }
Alexey Bataev8b8e2022015-04-27 05:22:09 +0000854 case OMPRTL__kmpc_omp_taskwait: {
855 // Build kmp_int32 __kmpc_omp_taskwait(ident_t *loc, kmp_int32 global_tid);
856 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
857 llvm::FunctionType *FnTy =
858 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
859 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_omp_taskwait");
860 break;
861 }
Alexey Bataevc30dd2d2015-06-18 12:14:09 +0000862 case OMPRTL__kmpc_taskgroup: {
863 // Build void __kmpc_taskgroup(ident_t *loc, kmp_int32 global_tid);
864 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
865 llvm::FunctionType *FnTy =
866 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
867 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_taskgroup");
868 break;
869 }
870 case OMPRTL__kmpc_end_taskgroup: {
871 // Build void __kmpc_end_taskgroup(ident_t *loc, kmp_int32 global_tid);
872 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty};
873 llvm::FunctionType *FnTy =
874 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
875 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_end_taskgroup");
876 break;
877 }
Alexey Bataev7f210c62015-06-18 13:40:03 +0000878 case OMPRTL__kmpc_push_proc_bind: {
879 // Build void __kmpc_push_proc_bind(ident_t *loc, kmp_int32 global_tid,
880 // int proc_bind)
881 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty, CGM.IntTy};
882 llvm::FunctionType *FnTy =
883 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
884 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_push_proc_bind");
885 break;
886 }
Alexey Bataev1d2353d2015-06-24 11:01:36 +0000887 case OMPRTL__kmpc_omp_task_with_deps: {
888 // Build kmp_int32 __kmpc_omp_task_with_deps(ident_t *, kmp_int32 gtid,
889 // kmp_task_t *new_task, kmp_int32 ndeps, kmp_depend_info_t *dep_list,
890 // kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list);
891 llvm::Type *TypeParams[] = {
892 getIdentTyPointerTy(), CGM.Int32Ty, CGM.VoidPtrTy, CGM.Int32Ty,
893 CGM.VoidPtrTy, CGM.Int32Ty, CGM.VoidPtrTy};
894 llvm::FunctionType *FnTy =
895 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg=*/false);
896 RTLFn =
897 CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_omp_task_with_deps");
898 break;
899 }
900 case OMPRTL__kmpc_omp_wait_deps: {
901 // Build void __kmpc_omp_wait_deps(ident_t *, kmp_int32 gtid,
902 // kmp_int32 ndeps, kmp_depend_info_t *dep_list, kmp_int32 ndeps_noalias,
903 // kmp_depend_info_t *noalias_dep_list);
904 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty,
905 CGM.Int32Ty, CGM.VoidPtrTy,
906 CGM.Int32Ty, CGM.VoidPtrTy};
907 llvm::FunctionType *FnTy =
908 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
909 RTLFn = CGM.CreateRuntimeFunction(FnTy, /*Name=*/"__kmpc_omp_wait_deps");
910 break;
911 }
Alexey Bataev0f34da12015-07-02 04:17:07 +0000912 case OMPRTL__kmpc_cancellationpoint: {
913 // Build kmp_int32 __kmpc_cancellationpoint(ident_t *loc, kmp_int32
914 // global_tid, kmp_int32 cncl_kind)
915 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty, CGM.IntTy};
916 llvm::FunctionType *FnTy =
917 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
918 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_cancellationpoint");
919 break;
920 }
Alexey Bataev7d5d33e2015-07-06 05:50:32 +0000921 case OMPRTL__kmpc_cancel: {
922 // Build kmp_int32 __kmpc_cancel(ident_t *loc, kmp_int32 global_tid,
923 // kmp_int32 cncl_kind)
924 llvm::Type *TypeParams[] = {getIdentTyPointerTy(), CGM.Int32Ty, CGM.IntTy};
925 llvm::FunctionType *FnTy =
926 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
927 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__kmpc_cancel");
928 break;
929 }
Samuel Antaobed3c462015-10-02 16:14:20 +0000930 case OMPRTL__tgt_target: {
931 // Build int32_t __tgt_target(int32_t device_id, void *host_ptr, int32_t
932 // arg_num, void** args_base, void **args, size_t *arg_sizes, int32_t
933 // *arg_types);
934 llvm::Type *TypeParams[] = {CGM.Int32Ty,
935 CGM.VoidPtrTy,
936 CGM.Int32Ty,
937 CGM.VoidPtrPtrTy,
938 CGM.VoidPtrPtrTy,
939 CGM.SizeTy->getPointerTo(),
940 CGM.Int32Ty->getPointerTo()};
941 llvm::FunctionType *FnTy =
942 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
943 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__tgt_target");
944 break;
945 }
Samuel Antaoee8fb302016-01-06 13:42:12 +0000946 case OMPRTL__tgt_register_lib: {
947 // Build void __tgt_register_lib(__tgt_bin_desc *desc);
948 QualType ParamTy =
949 CGM.getContext().getPointerType(getTgtBinaryDescriptorQTy());
950 llvm::Type *TypeParams[] = {CGM.getTypes().ConvertTypeForMem(ParamTy)};
951 llvm::FunctionType *FnTy =
952 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
953 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__tgt_register_lib");
954 break;
955 }
956 case OMPRTL__tgt_unregister_lib: {
957 // Build void __tgt_unregister_lib(__tgt_bin_desc *desc);
958 QualType ParamTy =
959 CGM.getContext().getPointerType(getTgtBinaryDescriptorQTy());
960 llvm::Type *TypeParams[] = {CGM.getTypes().ConvertTypeForMem(ParamTy)};
961 llvm::FunctionType *FnTy =
962 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
963 RTLFn = CGM.CreateRuntimeFunction(FnTy, "__tgt_unregister_lib");
964 break;
965 }
Alexey Bataev9959db52014-05-06 10:08:46 +0000966 }
967 return RTLFn;
968}
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +0000969
Alexander Musman21212e42015-03-13 10:38:23 +0000970llvm::Constant *CGOpenMPRuntime::createForStaticInitFunction(unsigned IVSize,
971 bool IVSigned) {
972 assert((IVSize == 32 || IVSize == 64) &&
973 "IV size is not compatible with the omp runtime");
974 auto Name = IVSize == 32 ? (IVSigned ? "__kmpc_for_static_init_4"
975 : "__kmpc_for_static_init_4u")
976 : (IVSigned ? "__kmpc_for_static_init_8"
977 : "__kmpc_for_static_init_8u");
978 auto ITy = IVSize == 32 ? CGM.Int32Ty : CGM.Int64Ty;
979 auto PtrTy = llvm::PointerType::getUnqual(ITy);
980 llvm::Type *TypeParams[] = {
981 getIdentTyPointerTy(), // loc
982 CGM.Int32Ty, // tid
983 CGM.Int32Ty, // schedtype
984 llvm::PointerType::getUnqual(CGM.Int32Ty), // p_lastiter
985 PtrTy, // p_lower
986 PtrTy, // p_upper
987 PtrTy, // p_stride
988 ITy, // incr
989 ITy // chunk
990 };
991 llvm::FunctionType *FnTy =
992 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
993 return CGM.CreateRuntimeFunction(FnTy, Name);
994}
995
Alexander Musman92bdaab2015-03-12 13:37:50 +0000996llvm::Constant *CGOpenMPRuntime::createDispatchInitFunction(unsigned IVSize,
997 bool IVSigned) {
998 assert((IVSize == 32 || IVSize == 64) &&
999 "IV size is not compatible with the omp runtime");
1000 auto Name =
1001 IVSize == 32
1002 ? (IVSigned ? "__kmpc_dispatch_init_4" : "__kmpc_dispatch_init_4u")
1003 : (IVSigned ? "__kmpc_dispatch_init_8" : "__kmpc_dispatch_init_8u");
1004 auto ITy = IVSize == 32 ? CGM.Int32Ty : CGM.Int64Ty;
1005 llvm::Type *TypeParams[] = { getIdentTyPointerTy(), // loc
1006 CGM.Int32Ty, // tid
1007 CGM.Int32Ty, // schedtype
1008 ITy, // lower
1009 ITy, // upper
1010 ITy, // stride
1011 ITy // chunk
1012 };
1013 llvm::FunctionType *FnTy =
1014 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg*/ false);
1015 return CGM.CreateRuntimeFunction(FnTy, Name);
1016}
1017
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001018llvm::Constant *CGOpenMPRuntime::createDispatchFiniFunction(unsigned IVSize,
1019 bool IVSigned) {
1020 assert((IVSize == 32 || IVSize == 64) &&
1021 "IV size is not compatible with the omp runtime");
1022 auto Name =
1023 IVSize == 32
1024 ? (IVSigned ? "__kmpc_dispatch_fini_4" : "__kmpc_dispatch_fini_4u")
1025 : (IVSigned ? "__kmpc_dispatch_fini_8" : "__kmpc_dispatch_fini_8u");
1026 llvm::Type *TypeParams[] = {
1027 getIdentTyPointerTy(), // loc
1028 CGM.Int32Ty, // tid
1029 };
1030 llvm::FunctionType *FnTy =
1031 llvm::FunctionType::get(CGM.VoidTy, TypeParams, /*isVarArg=*/false);
1032 return CGM.CreateRuntimeFunction(FnTy, Name);
1033}
1034
Alexander Musman92bdaab2015-03-12 13:37:50 +00001035llvm::Constant *CGOpenMPRuntime::createDispatchNextFunction(unsigned IVSize,
1036 bool IVSigned) {
1037 assert((IVSize == 32 || IVSize == 64) &&
1038 "IV size is not compatible with the omp runtime");
1039 auto Name =
1040 IVSize == 32
1041 ? (IVSigned ? "__kmpc_dispatch_next_4" : "__kmpc_dispatch_next_4u")
1042 : (IVSigned ? "__kmpc_dispatch_next_8" : "__kmpc_dispatch_next_8u");
1043 auto ITy = IVSize == 32 ? CGM.Int32Ty : CGM.Int64Ty;
1044 auto PtrTy = llvm::PointerType::getUnqual(ITy);
1045 llvm::Type *TypeParams[] = {
1046 getIdentTyPointerTy(), // loc
1047 CGM.Int32Ty, // tid
1048 llvm::PointerType::getUnqual(CGM.Int32Ty), // p_lastiter
1049 PtrTy, // p_lower
1050 PtrTy, // p_upper
1051 PtrTy // p_stride
1052 };
1053 llvm::FunctionType *FnTy =
1054 llvm::FunctionType::get(CGM.Int32Ty, TypeParams, /*isVarArg*/ false);
1055 return CGM.CreateRuntimeFunction(FnTy, Name);
1056}
1057
Alexey Bataev97720002014-11-11 04:05:39 +00001058llvm::Constant *
1059CGOpenMPRuntime::getOrCreateThreadPrivateCache(const VarDecl *VD) {
Samuel Antaof8b50122015-07-13 22:54:53 +00001060 assert(!CGM.getLangOpts().OpenMPUseTLS ||
1061 !CGM.getContext().getTargetInfo().isTLSSupported());
Alexey Bataev97720002014-11-11 04:05:39 +00001062 // Lookup the entry, lazily creating it if necessary.
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001063 return getOrCreateInternalVariable(CGM.Int8PtrPtrTy,
Alexey Bataev97720002014-11-11 04:05:39 +00001064 Twine(CGM.getMangledName(VD)) + ".cache.");
1065}
1066
John McCall7f416cc2015-09-08 08:05:57 +00001067Address CGOpenMPRuntime::getAddrOfThreadPrivate(CodeGenFunction &CGF,
1068 const VarDecl *VD,
1069 Address VDAddr,
1070 SourceLocation Loc) {
Samuel Antaof8b50122015-07-13 22:54:53 +00001071 if (CGM.getLangOpts().OpenMPUseTLS &&
1072 CGM.getContext().getTargetInfo().isTLSSupported())
1073 return VDAddr;
1074
John McCall7f416cc2015-09-08 08:05:57 +00001075 auto VarTy = VDAddr.getElementType();
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001076 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc),
John McCall7f416cc2015-09-08 08:05:57 +00001077 CGF.Builder.CreatePointerCast(VDAddr.getPointer(),
1078 CGM.Int8PtrTy),
Alexey Bataev97720002014-11-11 04:05:39 +00001079 CGM.getSize(CGM.GetTargetTypeStoreSize(VarTy)),
1080 getOrCreateThreadPrivateCache(VD)};
John McCall7f416cc2015-09-08 08:05:57 +00001081 return Address(CGF.EmitRuntimeCall(
1082 createRuntimeFunction(OMPRTL__kmpc_threadprivate_cached), Args),
1083 VDAddr.getAlignment());
Alexey Bataev97720002014-11-11 04:05:39 +00001084}
1085
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001086void CGOpenMPRuntime::emitThreadPrivateVarInit(
John McCall7f416cc2015-09-08 08:05:57 +00001087 CodeGenFunction &CGF, Address VDAddr, llvm::Value *Ctor,
Alexey Bataev97720002014-11-11 04:05:39 +00001088 llvm::Value *CopyCtor, llvm::Value *Dtor, SourceLocation Loc) {
1089 // Call kmp_int32 __kmpc_global_thread_num(&loc) to init OpenMP runtime
1090 // library.
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001091 auto OMPLoc = emitUpdateLocation(CGF, Loc);
1092 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_global_thread_num),
Alexey Bataev97720002014-11-11 04:05:39 +00001093 OMPLoc);
1094 // Call __kmpc_threadprivate_register(&loc, &var, ctor, cctor/*NULL*/, dtor)
1095 // to register constructor/destructor for variable.
1096 llvm::Value *Args[] = {OMPLoc,
John McCall7f416cc2015-09-08 08:05:57 +00001097 CGF.Builder.CreatePointerCast(VDAddr.getPointer(),
1098 CGM.VoidPtrTy),
Alexey Bataev97720002014-11-11 04:05:39 +00001099 Ctor, CopyCtor, Dtor};
Alexey Bataev1e4b7132014-12-03 12:11:24 +00001100 CGF.EmitRuntimeCall(
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001101 createRuntimeFunction(OMPRTL__kmpc_threadprivate_register), Args);
Alexey Bataev97720002014-11-11 04:05:39 +00001102}
1103
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001104llvm::Function *CGOpenMPRuntime::emitThreadPrivateVarDefinition(
John McCall7f416cc2015-09-08 08:05:57 +00001105 const VarDecl *VD, Address VDAddr, SourceLocation Loc,
Alexey Bataev97720002014-11-11 04:05:39 +00001106 bool PerformInit, CodeGenFunction *CGF) {
Samuel Antaof8b50122015-07-13 22:54:53 +00001107 if (CGM.getLangOpts().OpenMPUseTLS &&
1108 CGM.getContext().getTargetInfo().isTLSSupported())
1109 return nullptr;
1110
Alexey Bataev97720002014-11-11 04:05:39 +00001111 VD = VD->getDefinition(CGM.getContext());
1112 if (VD && ThreadPrivateWithDefinition.count(VD) == 0) {
1113 ThreadPrivateWithDefinition.insert(VD);
1114 QualType ASTTy = VD->getType();
1115
1116 llvm::Value *Ctor = nullptr, *CopyCtor = nullptr, *Dtor = nullptr;
1117 auto Init = VD->getAnyInitializer();
1118 if (CGM.getLangOpts().CPlusPlus && PerformInit) {
1119 // Generate function that re-emits the declaration's initializer into the
1120 // threadprivate copy of the variable VD
1121 CodeGenFunction CtorCGF(CGM);
1122 FunctionArgList Args;
1123 ImplicitParamDecl Dst(CGM.getContext(), /*DC=*/nullptr, SourceLocation(),
1124 /*Id=*/nullptr, CGM.getContext().VoidPtrTy);
1125 Args.push_back(&Dst);
1126
1127 auto &FI = CGM.getTypes().arrangeFreeFunctionDeclaration(
1128 CGM.getContext().VoidPtrTy, Args, FunctionType::ExtInfo(),
1129 /*isVariadic=*/false);
1130 auto FTy = CGM.getTypes().GetFunctionType(FI);
1131 auto Fn = CGM.CreateGlobalInitOrDestructFunction(
Akira Hatanaka7791f1a42015-10-31 01:28:07 +00001132 FTy, ".__kmpc_global_ctor_.", FI, Loc);
Alexey Bataev97720002014-11-11 04:05:39 +00001133 CtorCGF.StartFunction(GlobalDecl(), CGM.getContext().VoidPtrTy, Fn, FI,
1134 Args, SourceLocation());
1135 auto ArgVal = CtorCGF.EmitLoadOfScalar(
John McCall7f416cc2015-09-08 08:05:57 +00001136 CtorCGF.GetAddrOfLocalVar(&Dst), /*Volatile=*/false,
Alexey Bataev97720002014-11-11 04:05:39 +00001137 CGM.getContext().VoidPtrTy, Dst.getLocation());
John McCall7f416cc2015-09-08 08:05:57 +00001138 Address Arg = Address(ArgVal, VDAddr.getAlignment());
1139 Arg = CtorCGF.Builder.CreateElementBitCast(Arg,
1140 CtorCGF.ConvertTypeForMem(ASTTy));
Alexey Bataev97720002014-11-11 04:05:39 +00001141 CtorCGF.EmitAnyExprToMem(Init, Arg, Init->getType().getQualifiers(),
1142 /*IsInitializer=*/true);
1143 ArgVal = CtorCGF.EmitLoadOfScalar(
John McCall7f416cc2015-09-08 08:05:57 +00001144 CtorCGF.GetAddrOfLocalVar(&Dst), /*Volatile=*/false,
Alexey Bataev97720002014-11-11 04:05:39 +00001145 CGM.getContext().VoidPtrTy, Dst.getLocation());
1146 CtorCGF.Builder.CreateStore(ArgVal, CtorCGF.ReturnValue);
1147 CtorCGF.FinishFunction();
1148 Ctor = Fn;
1149 }
1150 if (VD->getType().isDestructedType() != QualType::DK_none) {
1151 // Generate function that emits destructor call for the threadprivate copy
1152 // of the variable VD
1153 CodeGenFunction DtorCGF(CGM);
1154 FunctionArgList Args;
1155 ImplicitParamDecl Dst(CGM.getContext(), /*DC=*/nullptr, SourceLocation(),
1156 /*Id=*/nullptr, CGM.getContext().VoidPtrTy);
1157 Args.push_back(&Dst);
1158
1159 auto &FI = CGM.getTypes().arrangeFreeFunctionDeclaration(
1160 CGM.getContext().VoidTy, Args, FunctionType::ExtInfo(),
1161 /*isVariadic=*/false);
1162 auto FTy = CGM.getTypes().GetFunctionType(FI);
1163 auto Fn = CGM.CreateGlobalInitOrDestructFunction(
Akira Hatanaka7791f1a42015-10-31 01:28:07 +00001164 FTy, ".__kmpc_global_dtor_.", FI, Loc);
Alexey Bataev97720002014-11-11 04:05:39 +00001165 DtorCGF.StartFunction(GlobalDecl(), CGM.getContext().VoidTy, Fn, FI, Args,
1166 SourceLocation());
1167 auto ArgVal = DtorCGF.EmitLoadOfScalar(
1168 DtorCGF.GetAddrOfLocalVar(&Dst),
John McCall7f416cc2015-09-08 08:05:57 +00001169 /*Volatile=*/false, CGM.getContext().VoidPtrTy, Dst.getLocation());
1170 DtorCGF.emitDestroy(Address(ArgVal, VDAddr.getAlignment()), ASTTy,
Alexey Bataev97720002014-11-11 04:05:39 +00001171 DtorCGF.getDestroyer(ASTTy.isDestructedType()),
1172 DtorCGF.needsEHCleanup(ASTTy.isDestructedType()));
1173 DtorCGF.FinishFunction();
1174 Dtor = Fn;
1175 }
1176 // Do not emit init function if it is not required.
1177 if (!Ctor && !Dtor)
1178 return nullptr;
1179
1180 llvm::Type *CopyCtorTyArgs[] = {CGM.VoidPtrTy, CGM.VoidPtrTy};
1181 auto CopyCtorTy =
1182 llvm::FunctionType::get(CGM.VoidPtrTy, CopyCtorTyArgs,
1183 /*isVarArg=*/false)->getPointerTo();
1184 // Copying constructor for the threadprivate variable.
1185 // Must be NULL - reserved by runtime, but currently it requires that this
1186 // parameter is always NULL. Otherwise it fires assertion.
1187 CopyCtor = llvm::Constant::getNullValue(CopyCtorTy);
1188 if (Ctor == nullptr) {
1189 auto CtorTy = llvm::FunctionType::get(CGM.VoidPtrTy, CGM.VoidPtrTy,
1190 /*isVarArg=*/false)->getPointerTo();
1191 Ctor = llvm::Constant::getNullValue(CtorTy);
1192 }
1193 if (Dtor == nullptr) {
1194 auto DtorTy = llvm::FunctionType::get(CGM.VoidTy, CGM.VoidPtrTy,
1195 /*isVarArg=*/false)->getPointerTo();
1196 Dtor = llvm::Constant::getNullValue(DtorTy);
1197 }
1198 if (!CGF) {
1199 auto InitFunctionTy =
1200 llvm::FunctionType::get(CGM.VoidTy, /*isVarArg*/ false);
1201 auto InitFunction = CGM.CreateGlobalInitOrDestructFunction(
Akira Hatanaka7791f1a42015-10-31 01:28:07 +00001202 InitFunctionTy, ".__omp_threadprivate_init_.",
1203 CGM.getTypes().arrangeNullaryFunction());
Alexey Bataev97720002014-11-11 04:05:39 +00001204 CodeGenFunction InitCGF(CGM);
1205 FunctionArgList ArgList;
1206 InitCGF.StartFunction(GlobalDecl(), CGM.getContext().VoidTy, InitFunction,
1207 CGM.getTypes().arrangeNullaryFunction(), ArgList,
1208 Loc);
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001209 emitThreadPrivateVarInit(InitCGF, VDAddr, Ctor, CopyCtor, Dtor, Loc);
Alexey Bataev97720002014-11-11 04:05:39 +00001210 InitCGF.FinishFunction();
1211 return InitFunction;
1212 }
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001213 emitThreadPrivateVarInit(*CGF, VDAddr, Ctor, CopyCtor, Dtor, Loc);
Alexey Bataev97720002014-11-11 04:05:39 +00001214 }
1215 return nullptr;
1216}
1217
Alexey Bataev1d677132015-04-22 13:57:31 +00001218/// \brief Emits code for OpenMP 'if' clause using specified \a CodeGen
1219/// function. Here is the logic:
1220/// if (Cond) {
1221/// ThenGen();
1222/// } else {
1223/// ElseGen();
1224/// }
1225static void emitOMPIfClause(CodeGenFunction &CGF, const Expr *Cond,
1226 const RegionCodeGenTy &ThenGen,
1227 const RegionCodeGenTy &ElseGen) {
1228 CodeGenFunction::LexicalScope ConditionScope(CGF, Cond->getSourceRange());
1229
1230 // If the condition constant folds and can be elided, try to avoid emitting
1231 // the condition and the dead arm of the if/else.
1232 bool CondConstant;
1233 if (CGF.ConstantFoldsToSimpleInteger(Cond, CondConstant)) {
1234 CodeGenFunction::RunCleanupsScope Scope(CGF);
1235 if (CondConstant) {
1236 ThenGen(CGF);
1237 } else {
1238 ElseGen(CGF);
1239 }
1240 return;
1241 }
1242
1243 // Otherwise, the condition did not fold, or we couldn't elide it. Just
1244 // emit the conditional branch.
1245 auto ThenBlock = CGF.createBasicBlock("omp_if.then");
1246 auto ElseBlock = CGF.createBasicBlock("omp_if.else");
1247 auto ContBlock = CGF.createBasicBlock("omp_if.end");
1248 CGF.EmitBranchOnBoolExpr(Cond, ThenBlock, ElseBlock, /*TrueCount=*/0);
1249
1250 // Emit the 'then' code.
1251 CGF.EmitBlock(ThenBlock);
1252 {
1253 CodeGenFunction::RunCleanupsScope ThenScope(CGF);
1254 ThenGen(CGF);
1255 }
1256 CGF.EmitBranch(ContBlock);
1257 // Emit the 'else' code if present.
1258 {
1259 // There is no need to emit line number for unconditional branch.
1260 auto NL = ApplyDebugLocation::CreateEmpty(CGF);
1261 CGF.EmitBlock(ElseBlock);
1262 }
1263 {
1264 CodeGenFunction::RunCleanupsScope ThenScope(CGF);
1265 ElseGen(CGF);
1266 }
1267 {
1268 // There is no need to emit line number for unconditional branch.
1269 auto NL = ApplyDebugLocation::CreateEmpty(CGF);
1270 CGF.EmitBranch(ContBlock);
1271 }
1272 // Emit the continuation block for code after the if.
1273 CGF.EmitBlock(ContBlock, /*IsFinished=*/true);
Alexey Bataev4a5bb772014-10-08 14:01:46 +00001274}
1275
Alexey Bataev1d677132015-04-22 13:57:31 +00001276void CGOpenMPRuntime::emitParallelCall(CodeGenFunction &CGF, SourceLocation Loc,
1277 llvm::Value *OutlinedFn,
Alexey Bataev2377fe92015-09-10 08:12:02 +00001278 ArrayRef<llvm::Value *> CapturedVars,
Alexey Bataev1d677132015-04-22 13:57:31 +00001279 const Expr *IfCond) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001280 if (!CGF.HaveInsertPoint())
1281 return;
Alexey Bataev1d677132015-04-22 13:57:31 +00001282 auto *RTLoc = emitUpdateLocation(CGF, Loc);
Alexey Bataev2377fe92015-09-10 08:12:02 +00001283 auto &&ThenGen = [this, OutlinedFn, CapturedVars,
1284 RTLoc](CodeGenFunction &CGF) {
1285 // Build call __kmpc_fork_call(loc, n, microtask, var1, .., varn);
1286 llvm::Value *Args[] = {
1287 RTLoc,
1288 CGF.Builder.getInt32(CapturedVars.size()), // Number of captured vars
1289 CGF.Builder.CreateBitCast(OutlinedFn, getKmpc_MicroPointerTy())};
1290 llvm::SmallVector<llvm::Value *, 16> RealArgs;
1291 RealArgs.append(std::begin(Args), std::end(Args));
1292 RealArgs.append(CapturedVars.begin(), CapturedVars.end());
1293
1294 auto RTLFn = createRuntimeFunction(OMPRTL__kmpc_fork_call);
1295 CGF.EmitRuntimeCall(RTLFn, RealArgs);
1296 };
1297 auto &&ElseGen = [this, OutlinedFn, CapturedVars, RTLoc,
1298 Loc](CodeGenFunction &CGF) {
Alexey Bataev1d677132015-04-22 13:57:31 +00001299 auto ThreadID = getThreadID(CGF, Loc);
1300 // Build calls:
1301 // __kmpc_serialized_parallel(&Loc, GTid);
1302 llvm::Value *Args[] = {RTLoc, ThreadID};
1303 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_serialized_parallel),
1304 Args);
Alexey Bataevd74d0602014-10-13 06:02:40 +00001305
Alexey Bataev1d677132015-04-22 13:57:31 +00001306 // OutlinedFn(&GTid, &zero, CapturedStruct);
1307 auto ThreadIDAddr = emitThreadIDAddress(CGF, Loc);
John McCall7f416cc2015-09-08 08:05:57 +00001308 Address ZeroAddr =
1309 CGF.CreateTempAlloca(CGF.Int32Ty, CharUnits::fromQuantity(4),
1310 /*Name*/ ".zero.addr");
Alexey Bataev1d677132015-04-22 13:57:31 +00001311 CGF.InitTempAlloca(ZeroAddr, CGF.Builder.getInt32(/*C*/ 0));
Alexey Bataev2377fe92015-09-10 08:12:02 +00001312 llvm::SmallVector<llvm::Value *, 16> OutlinedFnArgs;
1313 OutlinedFnArgs.push_back(ThreadIDAddr.getPointer());
1314 OutlinedFnArgs.push_back(ZeroAddr.getPointer());
1315 OutlinedFnArgs.append(CapturedVars.begin(), CapturedVars.end());
Alexey Bataev1d677132015-04-22 13:57:31 +00001316 CGF.EmitCallOrInvoke(OutlinedFn, OutlinedFnArgs);
Alexey Bataevd74d0602014-10-13 06:02:40 +00001317
Alexey Bataev1d677132015-04-22 13:57:31 +00001318 // __kmpc_end_serialized_parallel(&Loc, GTid);
1319 llvm::Value *EndArgs[] = {emitUpdateLocation(CGF, Loc), ThreadID};
1320 CGF.EmitRuntimeCall(
1321 createRuntimeFunction(OMPRTL__kmpc_end_serialized_parallel), EndArgs);
1322 };
1323 if (IfCond) {
1324 emitOMPIfClause(CGF, IfCond, ThenGen, ElseGen);
1325 } else {
1326 CodeGenFunction::RunCleanupsScope Scope(CGF);
1327 ThenGen(CGF);
1328 }
Alexey Bataevd74d0602014-10-13 06:02:40 +00001329}
1330
NAKAMURA Takumi59c74b222014-10-27 08:08:18 +00001331// If we're inside an (outlined) parallel region, use the region info's
Alexey Bataevd74d0602014-10-13 06:02:40 +00001332// thread-ID variable (it is passed in a first argument of the outlined function
1333// as "kmp_int32 *gtid"). Otherwise, if we're not inside parallel region, but in
1334// regular serial code region, get thread ID by calling kmp_int32
1335// kmpc_global_thread_num(ident_t *loc), stash this thread ID in a temporary and
1336// return the address of that temp.
John McCall7f416cc2015-09-08 08:05:57 +00001337Address CGOpenMPRuntime::emitThreadIDAddress(CodeGenFunction &CGF,
1338 SourceLocation Loc) {
Alexey Bataev3015bcc2016-01-22 08:56:50 +00001339 if (auto *OMPRegionInfo =
Alexey Bataevd74d0602014-10-13 06:02:40 +00001340 dyn_cast_or_null<CGOpenMPRegionInfo>(CGF.CapturedStmtInfo))
Alexey Bataev8cbe0a62015-02-26 10:27:34 +00001341 if (OMPRegionInfo->getThreadIDVariable())
Alexey Bataev62b63b12015-03-10 07:28:44 +00001342 return OMPRegionInfo->getThreadIDVariableLValue(CGF).getAddress();
Alexey Bataev8cbe0a62015-02-26 10:27:34 +00001343
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001344 auto ThreadID = getThreadID(CGF, Loc);
Alexey Bataevd74d0602014-10-13 06:02:40 +00001345 auto Int32Ty =
1346 CGF.getContext().getIntTypeForBitwidth(/*DestWidth*/ 32, /*Signed*/ true);
1347 auto ThreadIDTemp = CGF.CreateMemTemp(Int32Ty, /*Name*/ ".threadid_temp.");
1348 CGF.EmitStoreOfScalar(ThreadID,
John McCall7f416cc2015-09-08 08:05:57 +00001349 CGF.MakeAddrLValue(ThreadIDTemp, Int32Ty));
Alexey Bataevd74d0602014-10-13 06:02:40 +00001350
1351 return ThreadIDTemp;
1352}
1353
Alexey Bataev97720002014-11-11 04:05:39 +00001354llvm::Constant *
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001355CGOpenMPRuntime::getOrCreateInternalVariable(llvm::Type *Ty,
Alexey Bataev97720002014-11-11 04:05:39 +00001356 const llvm::Twine &Name) {
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +00001357 SmallString<256> Buffer;
1358 llvm::raw_svector_ostream Out(Buffer);
Alexey Bataev97720002014-11-11 04:05:39 +00001359 Out << Name;
1360 auto RuntimeName = Out.str();
David Blaikie13156b62014-11-19 03:06:06 +00001361 auto &Elem = *InternalVars.insert(std::make_pair(RuntimeName, nullptr)).first;
1362 if (Elem.second) {
1363 assert(Elem.second->getType()->getPointerElementType() == Ty &&
Alexey Bataev97720002014-11-11 04:05:39 +00001364 "OMP internal variable has different type than requested");
David Blaikie13156b62014-11-19 03:06:06 +00001365 return &*Elem.second;
Alexey Bataev97720002014-11-11 04:05:39 +00001366 }
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +00001367
David Blaikie13156b62014-11-19 03:06:06 +00001368 return Elem.second = new llvm::GlobalVariable(
1369 CGM.getModule(), Ty, /*IsConstant*/ false,
1370 llvm::GlobalValue::CommonLinkage, llvm::Constant::getNullValue(Ty),
1371 Elem.first());
Alexey Bataev97720002014-11-11 04:05:39 +00001372}
1373
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001374llvm::Value *CGOpenMPRuntime::getCriticalRegionLock(StringRef CriticalName) {
Alexey Bataev97720002014-11-11 04:05:39 +00001375 llvm::Twine Name(".gomp_critical_user_", CriticalName);
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001376 return getOrCreateInternalVariable(KmpCriticalNameTy, Name.concat(".var"));
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +00001377}
1378
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001379namespace {
David Blaikie7e70d682015-08-18 22:40:54 +00001380template <size_t N> class CallEndCleanup final : public EHScopeStack::Cleanup {
Alexey Bataev3e6124b2015-04-10 07:48:12 +00001381 llvm::Value *Callee;
Alexey Bataeva744ff52015-05-05 09:24:37 +00001382 llvm::Value *Args[N];
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001383
1384public:
Alexey Bataeva744ff52015-05-05 09:24:37 +00001385 CallEndCleanup(llvm::Value *Callee, ArrayRef<llvm::Value *> CleanupArgs)
1386 : Callee(Callee) {
1387 assert(CleanupArgs.size() == N);
1388 std::copy(CleanupArgs.begin(), CleanupArgs.end(), std::begin(Args));
1389 }
Alexey Bataev3e6124b2015-04-10 07:48:12 +00001390 void Emit(CodeGenFunction &CGF, Flags /*flags*/) override {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001391 if (!CGF.HaveInsertPoint())
1392 return;
Alexey Bataev3e6124b2015-04-10 07:48:12 +00001393 CGF.EmitRuntimeCall(Callee, Args);
1394 }
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001395};
Hans Wennborg7eb54642015-09-10 17:07:54 +00001396} // anonymous namespace
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001397
1398void CGOpenMPRuntime::emitCriticalRegion(CodeGenFunction &CGF,
1399 StringRef CriticalName,
1400 const RegionCodeGenTy &CriticalOpGen,
Alexey Bataevfc57d162015-12-15 10:55:09 +00001401 SourceLocation Loc, const Expr *Hint) {
1402 // __kmpc_critical[_with_hint](ident_t *, gtid, Lock[, hint]);
Alexey Bataev75ddfab2014-12-01 11:32:38 +00001403 // CriticalOpGen();
1404 // __kmpc_end_critical(ident_t *, gtid, Lock);
1405 // Prepare arguments and build a call to __kmpc_critical
Alexey Bataev8ef31412015-12-18 07:58:25 +00001406 if (!CGF.HaveInsertPoint())
1407 return;
Alexey Bataevfc57d162015-12-15 10:55:09 +00001408 CodeGenFunction::RunCleanupsScope Scope(CGF);
1409 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc),
1410 getCriticalRegionLock(CriticalName)};
1411 if (Hint) {
1412 llvm::SmallVector<llvm::Value *, 8> ArgsWithHint(std::begin(Args),
1413 std::end(Args));
1414 auto *HintVal = CGF.EmitScalarExpr(Hint);
1415 ArgsWithHint.push_back(
1416 CGF.Builder.CreateIntCast(HintVal, CGM.IntPtrTy, /*isSigned=*/false));
1417 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_critical_with_hint),
1418 ArgsWithHint);
1419 } else
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001420 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_critical), Args);
Alexey Bataevfc57d162015-12-15 10:55:09 +00001421 // Build a call to __kmpc_end_critical
1422 CGF.EHStack.pushCleanup<CallEndCleanup<std::extent<decltype(Args)>::value>>(
1423 NormalAndEHCleanup, createRuntimeFunction(OMPRTL__kmpc_end_critical),
1424 llvm::makeArrayRef(Args));
1425 emitInlinedDirective(CGF, OMPD_critical, CriticalOpGen);
Alexey Bataev3a3bf0b2014-09-22 10:01:53 +00001426}
Alexey Bataev4a5bb772014-10-08 14:01:46 +00001427
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001428static void emitIfStmt(CodeGenFunction &CGF, llvm::Value *IfCond,
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001429 OpenMPDirectiveKind Kind, SourceLocation Loc,
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001430 const RegionCodeGenTy &BodyOpGen) {
Alexey Bataev8d690652014-12-04 07:23:53 +00001431 llvm::Value *CallBool = CGF.EmitScalarConversion(
1432 IfCond,
1433 CGF.getContext().getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/true),
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001434 CGF.getContext().BoolTy, Loc);
Alexey Bataev8d690652014-12-04 07:23:53 +00001435
1436 auto *ThenBlock = CGF.createBasicBlock("omp_if.then");
1437 auto *ContBlock = CGF.createBasicBlock("omp_if.end");
1438 // Generate the branch (If-stmt)
1439 CGF.Builder.CreateCondBr(CallBool, ThenBlock, ContBlock);
1440 CGF.EmitBlock(ThenBlock);
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001441 CGF.CGM.getOpenMPRuntime().emitInlinedDirective(CGF, Kind, BodyOpGen);
Alexey Bataev8d690652014-12-04 07:23:53 +00001442 // Emit the rest of bblocks/branches
1443 CGF.EmitBranch(ContBlock);
1444 CGF.EmitBlock(ContBlock, true);
1445}
1446
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001447void CGOpenMPRuntime::emitMasterRegion(CodeGenFunction &CGF,
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001448 const RegionCodeGenTy &MasterOpGen,
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001449 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001450 if (!CGF.HaveInsertPoint())
1451 return;
Alexey Bataev8d690652014-12-04 07:23:53 +00001452 // if(__kmpc_master(ident_t *, gtid)) {
1453 // MasterOpGen();
1454 // __kmpc_end_master(ident_t *, gtid);
1455 // }
1456 // Prepare arguments and build a call to __kmpc_master
Alexey Bataevd7614fb2015-04-10 06:33:45 +00001457 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc)};
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001458 auto *IsMaster =
1459 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_master), Args);
Alexey Bataeva744ff52015-05-05 09:24:37 +00001460 typedef CallEndCleanup<std::extent<decltype(Args)>::value>
1461 MasterCallEndCleanup;
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001462 emitIfStmt(
1463 CGF, IsMaster, OMPD_master, Loc, [&](CodeGenFunction &CGF) -> void {
1464 CodeGenFunction::RunCleanupsScope Scope(CGF);
1465 CGF.EHStack.pushCleanup<MasterCallEndCleanup>(
1466 NormalAndEHCleanup, createRuntimeFunction(OMPRTL__kmpc_end_master),
1467 llvm::makeArrayRef(Args));
1468 MasterOpGen(CGF);
1469 });
Alexey Bataev8d690652014-12-04 07:23:53 +00001470}
1471
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001472void CGOpenMPRuntime::emitTaskyieldCall(CodeGenFunction &CGF,
1473 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001474 if (!CGF.HaveInsertPoint())
1475 return;
Alexey Bataev9f797f32015-02-05 05:57:51 +00001476 // Build call __kmpc_omp_taskyield(loc, thread_id, 0);
1477 llvm::Value *Args[] = {
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001478 emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc),
Alexey Bataev9f797f32015-02-05 05:57:51 +00001479 llvm::ConstantInt::get(CGM.IntTy, /*V=*/0, /*isSigned=*/true)};
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001480 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_omp_taskyield), Args);
Alexey Bataev9f797f32015-02-05 05:57:51 +00001481}
1482
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001483void CGOpenMPRuntime::emitTaskgroupRegion(CodeGenFunction &CGF,
1484 const RegionCodeGenTy &TaskgroupOpGen,
1485 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001486 if (!CGF.HaveInsertPoint())
1487 return;
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001488 // __kmpc_taskgroup(ident_t *, gtid);
1489 // TaskgroupOpGen();
1490 // __kmpc_end_taskgroup(ident_t *, gtid);
1491 // Prepare arguments and build a call to __kmpc_taskgroup
1492 {
1493 CodeGenFunction::RunCleanupsScope Scope(CGF);
1494 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc)};
1495 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_taskgroup), Args);
1496 // Build a call to __kmpc_end_taskgroup
1497 CGF.EHStack.pushCleanup<CallEndCleanup<std::extent<decltype(Args)>::value>>(
1498 NormalAndEHCleanup, createRuntimeFunction(OMPRTL__kmpc_end_taskgroup),
1499 llvm::makeArrayRef(Args));
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001500 emitInlinedDirective(CGF, OMPD_taskgroup, TaskgroupOpGen);
Alexey Bataevc30dd2d2015-06-18 12:14:09 +00001501 }
1502}
1503
John McCall7f416cc2015-09-08 08:05:57 +00001504/// Given an array of pointers to variables, project the address of a
1505/// given variable.
Alexey Bataevf24e7b12015-10-08 09:10:53 +00001506static Address emitAddrOfVarFromArray(CodeGenFunction &CGF, Address Array,
1507 unsigned Index, const VarDecl *Var) {
John McCall7f416cc2015-09-08 08:05:57 +00001508 // Pull out the pointer to the variable.
1509 Address PtrAddr =
Alexey Bataevf24e7b12015-10-08 09:10:53 +00001510 CGF.Builder.CreateConstArrayGEP(Array, Index, CGF.getPointerSize());
John McCall7f416cc2015-09-08 08:05:57 +00001511 llvm::Value *Ptr = CGF.Builder.CreateLoad(PtrAddr);
1512
1513 Address Addr = Address(Ptr, CGF.getContext().getDeclAlign(Var));
Alexey Bataevf24e7b12015-10-08 09:10:53 +00001514 Addr = CGF.Builder.CreateElementBitCast(
1515 Addr, CGF.ConvertTypeForMem(Var->getType()));
John McCall7f416cc2015-09-08 08:05:57 +00001516 return Addr;
1517}
1518
Alexey Bataeva63048e2015-03-23 06:18:07 +00001519static llvm::Value *emitCopyprivateCopyFunction(
Alexey Bataev420d45b2015-04-14 05:11:24 +00001520 CodeGenModule &CGM, llvm::Type *ArgsType,
1521 ArrayRef<const Expr *> CopyprivateVars, ArrayRef<const Expr *> DestExprs,
1522 ArrayRef<const Expr *> SrcExprs, ArrayRef<const Expr *> AssignmentOps) {
Alexey Bataeva63048e2015-03-23 06:18:07 +00001523 auto &C = CGM.getContext();
1524 // void copy_func(void *LHSArg, void *RHSArg);
1525 FunctionArgList Args;
1526 ImplicitParamDecl LHSArg(C, /*DC=*/nullptr, SourceLocation(), /*Id=*/nullptr,
1527 C.VoidPtrTy);
1528 ImplicitParamDecl RHSArg(C, /*DC=*/nullptr, SourceLocation(), /*Id=*/nullptr,
1529 C.VoidPtrTy);
1530 Args.push_back(&LHSArg);
1531 Args.push_back(&RHSArg);
1532 FunctionType::ExtInfo EI;
1533 auto &CGFI = CGM.getTypes().arrangeFreeFunctionDeclaration(
1534 C.VoidTy, Args, EI, /*isVariadic=*/false);
1535 auto *Fn = llvm::Function::Create(
1536 CGM.getTypes().GetFunctionType(CGFI), llvm::GlobalValue::InternalLinkage,
1537 ".omp.copyprivate.copy_func", &CGM.getModule());
Akira Hatanaka44a59f82015-10-28 02:30:47 +00001538 CGM.SetInternalFunctionAttributes(/*D=*/nullptr, Fn, CGFI);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001539 CodeGenFunction CGF(CGM);
1540 CGF.StartFunction(GlobalDecl(), C.VoidTy, Fn, CGFI, Args);
Alexey Bataev420d45b2015-04-14 05:11:24 +00001541 // Dest = (void*[n])(LHSArg);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001542 // Src = (void*[n])(RHSArg);
John McCall7f416cc2015-09-08 08:05:57 +00001543 Address LHS(CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
1544 CGF.Builder.CreateLoad(CGF.GetAddrOfLocalVar(&LHSArg)),
1545 ArgsType), CGF.getPointerAlign());
1546 Address RHS(CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
1547 CGF.Builder.CreateLoad(CGF.GetAddrOfLocalVar(&RHSArg)),
1548 ArgsType), CGF.getPointerAlign());
Alexey Bataeva63048e2015-03-23 06:18:07 +00001549 // *(Type0*)Dst[0] = *(Type0*)Src[0];
1550 // *(Type1*)Dst[1] = *(Type1*)Src[1];
1551 // ...
1552 // *(Typen*)Dst[n] = *(Typen*)Src[n];
Alexey Bataeva63048e2015-03-23 06:18:07 +00001553 for (unsigned I = 0, E = AssignmentOps.size(); I < E; ++I) {
John McCall7f416cc2015-09-08 08:05:57 +00001554 auto DestVar = cast<VarDecl>(cast<DeclRefExpr>(DestExprs[I])->getDecl());
1555 Address DestAddr = emitAddrOfVarFromArray(CGF, LHS, I, DestVar);
1556
1557 auto SrcVar = cast<VarDecl>(cast<DeclRefExpr>(SrcExprs[I])->getDecl());
1558 Address SrcAddr = emitAddrOfVarFromArray(CGF, RHS, I, SrcVar);
1559
Alexey Bataev1d9c15c2015-05-19 12:31:28 +00001560 auto *VD = cast<DeclRefExpr>(CopyprivateVars[I])->getDecl();
1561 QualType Type = VD->getType();
John McCall7f416cc2015-09-08 08:05:57 +00001562 CGF.EmitOMPCopy(Type, DestAddr, SrcAddr, DestVar, SrcVar, AssignmentOps[I]);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001563 }
Alexey Bataeva63048e2015-03-23 06:18:07 +00001564 CGF.FinishFunction();
1565 return Fn;
1566}
1567
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001568void CGOpenMPRuntime::emitSingleRegion(CodeGenFunction &CGF,
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00001569 const RegionCodeGenTy &SingleOpGen,
Alexey Bataeva63048e2015-03-23 06:18:07 +00001570 SourceLocation Loc,
1571 ArrayRef<const Expr *> CopyprivateVars,
1572 ArrayRef<const Expr *> SrcExprs,
1573 ArrayRef<const Expr *> DstExprs,
1574 ArrayRef<const Expr *> AssignmentOps) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001575 if (!CGF.HaveInsertPoint())
1576 return;
Alexey Bataeva63048e2015-03-23 06:18:07 +00001577 assert(CopyprivateVars.size() == SrcExprs.size() &&
1578 CopyprivateVars.size() == DstExprs.size() &&
1579 CopyprivateVars.size() == AssignmentOps.size());
1580 auto &C = CGM.getContext();
1581 // int32 did_it = 0;
Alexey Bataev6956e2e2015-02-05 06:35:41 +00001582 // if(__kmpc_single(ident_t *, gtid)) {
1583 // SingleOpGen();
1584 // __kmpc_end_single(ident_t *, gtid);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001585 // did_it = 1;
Alexey Bataev6956e2e2015-02-05 06:35:41 +00001586 // }
Alexey Bataeva63048e2015-03-23 06:18:07 +00001587 // call __kmpc_copyprivate(ident_t *, gtid, <buf_size>, <copyprivate list>,
1588 // <copy_func>, did_it);
1589
John McCall7f416cc2015-09-08 08:05:57 +00001590 Address DidIt = Address::invalid();
Alexey Bataeva63048e2015-03-23 06:18:07 +00001591 if (!CopyprivateVars.empty()) {
1592 // int32 did_it = 0;
1593 auto KmpInt32Ty = C.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/1);
1594 DidIt = CGF.CreateMemTemp(KmpInt32Ty, ".omp.copyprivate.did_it");
John McCall7f416cc2015-09-08 08:05:57 +00001595 CGF.Builder.CreateStore(CGF.Builder.getInt32(0), DidIt);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001596 }
Alexey Bataev6956e2e2015-02-05 06:35:41 +00001597 // Prepare arguments and build a call to __kmpc_single
Alexey Bataevd7614fb2015-04-10 06:33:45 +00001598 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc)};
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001599 auto *IsSingle =
1600 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_single), Args);
Alexey Bataeva744ff52015-05-05 09:24:37 +00001601 typedef CallEndCleanup<std::extent<decltype(Args)>::value>
1602 SingleCallEndCleanup;
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001603 emitIfStmt(
1604 CGF, IsSingle, OMPD_single, Loc, [&](CodeGenFunction &CGF) -> void {
1605 CodeGenFunction::RunCleanupsScope Scope(CGF);
1606 CGF.EHStack.pushCleanup<SingleCallEndCleanup>(
1607 NormalAndEHCleanup, createRuntimeFunction(OMPRTL__kmpc_end_single),
1608 llvm::makeArrayRef(Args));
1609 SingleOpGen(CGF);
John McCall7f416cc2015-09-08 08:05:57 +00001610 if (DidIt.isValid()) {
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001611 // did_it = 1;
John McCall7f416cc2015-09-08 08:05:57 +00001612 CGF.Builder.CreateStore(CGF.Builder.getInt32(1), DidIt);
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001613 }
1614 });
Alexey Bataeva63048e2015-03-23 06:18:07 +00001615 // call __kmpc_copyprivate(ident_t *, gtid, <buf_size>, <copyprivate list>,
1616 // <copy_func>, did_it);
John McCall7f416cc2015-09-08 08:05:57 +00001617 if (DidIt.isValid()) {
Alexey Bataeva63048e2015-03-23 06:18:07 +00001618 llvm::APInt ArraySize(/*unsigned int numBits=*/32, CopyprivateVars.size());
1619 auto CopyprivateArrayTy =
1620 C.getConstantArrayType(C.VoidPtrTy, ArraySize, ArrayType::Normal,
1621 /*IndexTypeQuals=*/0);
1622 // Create a list of all private variables for copyprivate.
John McCall7f416cc2015-09-08 08:05:57 +00001623 Address CopyprivateList =
Alexey Bataeva63048e2015-03-23 06:18:07 +00001624 CGF.CreateMemTemp(CopyprivateArrayTy, ".omp.copyprivate.cpr_list");
1625 for (unsigned I = 0, E = CopyprivateVars.size(); I < E; ++I) {
John McCall7f416cc2015-09-08 08:05:57 +00001626 Address Elem = CGF.Builder.CreateConstArrayGEP(
1627 CopyprivateList, I, CGF.getPointerSize());
1628 CGF.Builder.CreateStore(
Alexey Bataeva63048e2015-03-23 06:18:07 +00001629 CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
John McCall7f416cc2015-09-08 08:05:57 +00001630 CGF.EmitLValue(CopyprivateVars[I]).getPointer(), CGF.VoidPtrTy),
1631 Elem);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001632 }
1633 // Build function that copies private values from single region to all other
1634 // threads in the corresponding parallel region.
1635 auto *CpyFn = emitCopyprivateCopyFunction(
1636 CGM, CGF.ConvertTypeForMem(CopyprivateArrayTy)->getPointerTo(),
Alexey Bataev420d45b2015-04-14 05:11:24 +00001637 CopyprivateVars, SrcExprs, DstExprs, AssignmentOps);
Alexey Bataev1189bd02016-01-26 12:20:39 +00001638 auto *BufSize = CGF.getTypeSize(CopyprivateArrayTy);
John McCall7f416cc2015-09-08 08:05:57 +00001639 Address CL =
1640 CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(CopyprivateList,
1641 CGF.VoidPtrTy);
1642 auto *DidItVal = CGF.Builder.CreateLoad(DidIt);
Alexey Bataeva63048e2015-03-23 06:18:07 +00001643 llvm::Value *Args[] = {
1644 emitUpdateLocation(CGF, Loc), // ident_t *<loc>
1645 getThreadID(CGF, Loc), // i32 <gtid>
Alexey Bataev66beaa92015-04-30 03:47:32 +00001646 BufSize, // size_t <buf_size>
John McCall7f416cc2015-09-08 08:05:57 +00001647 CL.getPointer(), // void *<copyprivate list>
Alexey Bataeva63048e2015-03-23 06:18:07 +00001648 CpyFn, // void (*) (void *, void *) <copy_func>
1649 DidItVal // i32 did_it
1650 };
1651 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_copyprivate), Args);
1652 }
Alexey Bataev6956e2e2015-02-05 06:35:41 +00001653}
1654
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001655void CGOpenMPRuntime::emitOrderedRegion(CodeGenFunction &CGF,
1656 const RegionCodeGenTy &OrderedOpGen,
Alexey Bataev5f600d62015-09-29 03:48:57 +00001657 SourceLocation Loc, bool IsThreads) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001658 if (!CGF.HaveInsertPoint())
1659 return;
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001660 // __kmpc_ordered(ident_t *, gtid);
1661 // OrderedOpGen();
1662 // __kmpc_end_ordered(ident_t *, gtid);
1663 // Prepare arguments and build a call to __kmpc_ordered
Alexey Bataev5f600d62015-09-29 03:48:57 +00001664 CodeGenFunction::RunCleanupsScope Scope(CGF);
1665 if (IsThreads) {
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001666 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc)};
1667 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_ordered), Args);
1668 // Build a call to __kmpc_end_ordered
Alexey Bataeva744ff52015-05-05 09:24:37 +00001669 CGF.EHStack.pushCleanup<CallEndCleanup<std::extent<decltype(Args)>::value>>(
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001670 NormalAndEHCleanup, createRuntimeFunction(OMPRTL__kmpc_end_ordered),
1671 llvm::makeArrayRef(Args));
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001672 }
Alexey Bataev5f600d62015-09-29 03:48:57 +00001673 emitInlinedDirective(CGF, OMPD_ordered, OrderedOpGen);
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001674}
1675
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001676void CGOpenMPRuntime::emitBarrierCall(CodeGenFunction &CGF, SourceLocation Loc,
Alexey Bataev25e5b442015-09-15 12:52:43 +00001677 OpenMPDirectiveKind Kind, bool EmitChecks,
1678 bool ForceSimpleCall) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001679 if (!CGF.HaveInsertPoint())
1680 return;
Alexey Bataev8f7c1b02014-12-05 04:09:23 +00001681 // Build call __kmpc_cancel_barrier(loc, thread_id);
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001682 // Build call __kmpc_barrier(loc, thread_id);
Alexey Bataevf2685682015-03-30 04:30:22 +00001683 OpenMPLocationFlags Flags = OMP_IDENT_KMPC;
1684 if (Kind == OMPD_for) {
1685 Flags =
1686 static_cast<OpenMPLocationFlags>(Flags | OMP_IDENT_BARRIER_IMPL_FOR);
1687 } else if (Kind == OMPD_sections) {
1688 Flags = static_cast<OpenMPLocationFlags>(Flags |
1689 OMP_IDENT_BARRIER_IMPL_SECTIONS);
1690 } else if (Kind == OMPD_single) {
1691 Flags =
1692 static_cast<OpenMPLocationFlags>(Flags | OMP_IDENT_BARRIER_IMPL_SINGLE);
1693 } else if (Kind == OMPD_barrier) {
1694 Flags = static_cast<OpenMPLocationFlags>(Flags | OMP_IDENT_BARRIER_EXPL);
1695 } else {
1696 Flags = static_cast<OpenMPLocationFlags>(Flags | OMP_IDENT_BARRIER_IMPL);
1697 }
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001698 // Build call __kmpc_cancel_barrier(loc, thread_id) or __kmpc_barrier(loc,
1699 // thread_id);
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001700 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc, Flags),
1701 getThreadID(CGF, Loc)};
Alexey Bataev3015bcc2016-01-22 08:56:50 +00001702 if (auto *OMPRegionInfo =
1703 dyn_cast_or_null<CGOpenMPRegionInfo>(CGF.CapturedStmtInfo)) {
Alexey Bataev25e5b442015-09-15 12:52:43 +00001704 if (!ForceSimpleCall && OMPRegionInfo->hasCancel()) {
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001705 auto *Result = CGF.EmitRuntimeCall(
1706 createRuntimeFunction(OMPRTL__kmpc_cancel_barrier), Args);
Alexey Bataev25e5b442015-09-15 12:52:43 +00001707 if (EmitChecks) {
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001708 // if (__kmpc_cancel_barrier()) {
1709 // exit from construct;
1710 // }
1711 auto *ExitBB = CGF.createBasicBlock(".cancel.exit");
1712 auto *ContBB = CGF.createBasicBlock(".cancel.continue");
1713 auto *Cmp = CGF.Builder.CreateIsNotNull(Result);
1714 CGF.Builder.CreateCondBr(Cmp, ExitBB, ContBB);
1715 CGF.EmitBlock(ExitBB);
1716 // exit from construct;
Alexey Bataev25e5b442015-09-15 12:52:43 +00001717 auto CancelDestination =
1718 CGF.getOMPCancelDestination(OMPRegionInfo->getDirectiveKind());
Alexey Bataev81c7ea02015-07-03 09:56:58 +00001719 CGF.EmitBranchThroughCleanup(CancelDestination);
1720 CGF.EmitBlock(ContBB, /*IsFinished=*/true);
1721 }
1722 return;
1723 }
1724 }
1725 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_barrier), Args);
Alexey Bataev4a5bb772014-10-08 14:01:46 +00001726}
1727
Alexander Musmanc6388682014-12-15 07:07:06 +00001728/// \brief Schedule types for 'omp for' loops (these enumerators are taken from
1729/// the enum sched_type in kmp.h).
1730enum OpenMPSchedType {
1731 /// \brief Lower bound for default (unordered) versions.
1732 OMP_sch_lower = 32,
1733 OMP_sch_static_chunked = 33,
1734 OMP_sch_static = 34,
1735 OMP_sch_dynamic_chunked = 35,
1736 OMP_sch_guided_chunked = 36,
1737 OMP_sch_runtime = 37,
1738 OMP_sch_auto = 38,
1739 /// \brief Lower bound for 'ordered' versions.
1740 OMP_ord_lower = 64,
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001741 OMP_ord_static_chunked = 65,
1742 OMP_ord_static = 66,
1743 OMP_ord_dynamic_chunked = 67,
1744 OMP_ord_guided_chunked = 68,
1745 OMP_ord_runtime = 69,
1746 OMP_ord_auto = 70,
1747 OMP_sch_default = OMP_sch_static,
Alexander Musmanc6388682014-12-15 07:07:06 +00001748};
1749
1750/// \brief Map the OpenMP loop schedule to the runtime enumeration.
1751static OpenMPSchedType getRuntimeSchedule(OpenMPScheduleClauseKind ScheduleKind,
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001752 bool Chunked, bool Ordered) {
Alexander Musmanc6388682014-12-15 07:07:06 +00001753 switch (ScheduleKind) {
1754 case OMPC_SCHEDULE_static:
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001755 return Chunked ? (Ordered ? OMP_ord_static_chunked : OMP_sch_static_chunked)
1756 : (Ordered ? OMP_ord_static : OMP_sch_static);
Alexander Musmanc6388682014-12-15 07:07:06 +00001757 case OMPC_SCHEDULE_dynamic:
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001758 return Ordered ? OMP_ord_dynamic_chunked : OMP_sch_dynamic_chunked;
Alexander Musmanc6388682014-12-15 07:07:06 +00001759 case OMPC_SCHEDULE_guided:
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001760 return Ordered ? OMP_ord_guided_chunked : OMP_sch_guided_chunked;
Alexander Musmanc6388682014-12-15 07:07:06 +00001761 case OMPC_SCHEDULE_runtime:
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001762 return Ordered ? OMP_ord_runtime : OMP_sch_runtime;
1763 case OMPC_SCHEDULE_auto:
1764 return Ordered ? OMP_ord_auto : OMP_sch_auto;
Alexander Musmanc6388682014-12-15 07:07:06 +00001765 case OMPC_SCHEDULE_unknown:
1766 assert(!Chunked && "chunk was specified but schedule kind not known");
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001767 return Ordered ? OMP_ord_static : OMP_sch_static;
Alexander Musmanc6388682014-12-15 07:07:06 +00001768 }
1769 llvm_unreachable("Unexpected runtime schedule");
1770}
1771
1772bool CGOpenMPRuntime::isStaticNonchunked(OpenMPScheduleClauseKind ScheduleKind,
1773 bool Chunked) const {
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001774 auto Schedule = getRuntimeSchedule(ScheduleKind, Chunked, /*Ordered=*/false);
Alexander Musmanc6388682014-12-15 07:07:06 +00001775 return Schedule == OMP_sch_static;
1776}
1777
Alexander Musmandf7a8e22015-01-22 08:49:35 +00001778bool CGOpenMPRuntime::isDynamic(OpenMPScheduleClauseKind ScheduleKind) const {
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001779 auto Schedule =
1780 getRuntimeSchedule(ScheduleKind, /*Chunked=*/false, /*Ordered=*/false);
Alexander Musmandf7a8e22015-01-22 08:49:35 +00001781 assert(Schedule != OMP_sch_static_chunked && "cannot be chunked here");
1782 return Schedule != OMP_sch_static;
1783}
1784
John McCall7f416cc2015-09-08 08:05:57 +00001785void CGOpenMPRuntime::emitForDispatchInit(CodeGenFunction &CGF,
1786 SourceLocation Loc,
1787 OpenMPScheduleClauseKind ScheduleKind,
1788 unsigned IVSize, bool IVSigned,
1789 bool Ordered, llvm::Value *UB,
1790 llvm::Value *Chunk) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001791 if (!CGF.HaveInsertPoint())
1792 return;
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001793 OpenMPSchedType Schedule =
1794 getRuntimeSchedule(ScheduleKind, Chunk != nullptr, Ordered);
John McCall7f416cc2015-09-08 08:05:57 +00001795 assert(Ordered ||
1796 (Schedule != OMP_sch_static && Schedule != OMP_sch_static_chunked &&
1797 Schedule != OMP_ord_static && Schedule != OMP_ord_static_chunked));
1798 // Call __kmpc_dispatch_init(
1799 // ident_t *loc, kmp_int32 tid, kmp_int32 schedule,
1800 // kmp_int[32|64] lower, kmp_int[32|64] upper,
1801 // kmp_int[32|64] stride, kmp_int[32|64] chunk);
Alexander Musmanc6388682014-12-15 07:07:06 +00001802
John McCall7f416cc2015-09-08 08:05:57 +00001803 // If the Chunk was not specified in the clause - use default value 1.
1804 if (Chunk == nullptr)
1805 Chunk = CGF.Builder.getIntN(IVSize, 1);
1806 llvm::Value *Args[] = {
1807 emitUpdateLocation(CGF, Loc, OMP_IDENT_KMPC),
1808 getThreadID(CGF, Loc),
1809 CGF.Builder.getInt32(Schedule), // Schedule type
1810 CGF.Builder.getIntN(IVSize, 0), // Lower
1811 UB, // Upper
1812 CGF.Builder.getIntN(IVSize, 1), // Stride
1813 Chunk // Chunk
1814 };
1815 CGF.EmitRuntimeCall(createDispatchInitFunction(IVSize, IVSigned), Args);
1816}
1817
1818void CGOpenMPRuntime::emitForStaticInit(CodeGenFunction &CGF,
1819 SourceLocation Loc,
1820 OpenMPScheduleClauseKind ScheduleKind,
1821 unsigned IVSize, bool IVSigned,
1822 bool Ordered, Address IL, Address LB,
1823 Address UB, Address ST,
1824 llvm::Value *Chunk) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001825 if (!CGF.HaveInsertPoint())
1826 return;
John McCall7f416cc2015-09-08 08:05:57 +00001827 OpenMPSchedType Schedule =
1828 getRuntimeSchedule(ScheduleKind, Chunk != nullptr, Ordered);
1829 assert(!Ordered);
1830 assert(Schedule == OMP_sch_static || Schedule == OMP_sch_static_chunked ||
1831 Schedule == OMP_ord_static || Schedule == OMP_ord_static_chunked);
1832
1833 // Call __kmpc_for_static_init(
1834 // ident_t *loc, kmp_int32 tid, kmp_int32 schedtype,
1835 // kmp_int32 *p_lastiter, kmp_int[32|64] *p_lower,
1836 // kmp_int[32|64] *p_upper, kmp_int[32|64] *p_stride,
1837 // kmp_int[32|64] incr, kmp_int[32|64] chunk);
1838 if (Chunk == nullptr) {
1839 assert((Schedule == OMP_sch_static || Schedule == OMP_ord_static) &&
1840 "expected static non-chunked schedule");
Alexander Musman92bdaab2015-03-12 13:37:50 +00001841 // If the Chunk was not specified in the clause - use default value 1.
Alexander Musman92bdaab2015-03-12 13:37:50 +00001842 Chunk = CGF.Builder.getIntN(IVSize, 1);
Alexander Musman92bdaab2015-03-12 13:37:50 +00001843 } else {
John McCall7f416cc2015-09-08 08:05:57 +00001844 assert((Schedule == OMP_sch_static_chunked ||
1845 Schedule == OMP_ord_static_chunked) &&
1846 "expected static chunked schedule");
Alexander Musman92bdaab2015-03-12 13:37:50 +00001847 }
John McCall7f416cc2015-09-08 08:05:57 +00001848 llvm::Value *Args[] = {
1849 emitUpdateLocation(CGF, Loc, OMP_IDENT_KMPC),
1850 getThreadID(CGF, Loc),
1851 CGF.Builder.getInt32(Schedule), // Schedule type
1852 IL.getPointer(), // &isLastIter
1853 LB.getPointer(), // &LB
1854 UB.getPointer(), // &UB
1855 ST.getPointer(), // &Stride
1856 CGF.Builder.getIntN(IVSize, 1), // Incr
1857 Chunk // Chunk
1858 };
1859 CGF.EmitRuntimeCall(createForStaticInitFunction(IVSize, IVSigned), Args);
Alexander Musmanc6388682014-12-15 07:07:06 +00001860}
1861
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001862void CGOpenMPRuntime::emitForStaticFinish(CodeGenFunction &CGF,
1863 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001864 if (!CGF.HaveInsertPoint())
1865 return;
Alexander Musmanc6388682014-12-15 07:07:06 +00001866 // Call __kmpc_for_static_fini(ident_t *loc, kmp_int32 tid);
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001867 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc, OMP_IDENT_KMPC),
1868 getThreadID(CGF, Loc)};
1869 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_for_static_fini),
1870 Args);
Alexander Musmanc6388682014-12-15 07:07:06 +00001871}
1872
Alexey Bataevd7589ffe2015-05-20 13:12:48 +00001873void CGOpenMPRuntime::emitForOrderedIterationEnd(CodeGenFunction &CGF,
1874 SourceLocation Loc,
1875 unsigned IVSize,
1876 bool IVSigned) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001877 if (!CGF.HaveInsertPoint())
1878 return;
Alexey Bataev98eb6e32015-04-22 11:15:40 +00001879 // Call __kmpc_for_dynamic_fini_(4|8)[u](ident_t *loc, kmp_int32 tid);
1880 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc, OMP_IDENT_KMPC),
1881 getThreadID(CGF, Loc)};
1882 CGF.EmitRuntimeCall(createDispatchFiniFunction(IVSize, IVSigned), Args);
1883}
1884
Alexander Musman92bdaab2015-03-12 13:37:50 +00001885llvm::Value *CGOpenMPRuntime::emitForNext(CodeGenFunction &CGF,
1886 SourceLocation Loc, unsigned IVSize,
John McCall7f416cc2015-09-08 08:05:57 +00001887 bool IVSigned, Address IL,
1888 Address LB, Address UB,
1889 Address ST) {
Alexander Musman92bdaab2015-03-12 13:37:50 +00001890 // Call __kmpc_dispatch_next(
1891 // ident_t *loc, kmp_int32 tid, kmp_int32 *p_lastiter,
1892 // kmp_int[32|64] *p_lower, kmp_int[32|64] *p_upper,
1893 // kmp_int[32|64] *p_stride);
1894 llvm::Value *Args[] = {
1895 emitUpdateLocation(CGF, Loc, OMP_IDENT_KMPC), getThreadID(CGF, Loc),
John McCall7f416cc2015-09-08 08:05:57 +00001896 IL.getPointer(), // &isLastIter
1897 LB.getPointer(), // &Lower
1898 UB.getPointer(), // &Upper
1899 ST.getPointer() // &Stride
Alexander Musman92bdaab2015-03-12 13:37:50 +00001900 };
1901 llvm::Value *Call =
1902 CGF.EmitRuntimeCall(createDispatchNextFunction(IVSize, IVSigned), Args);
1903 return CGF.EmitScalarConversion(
1904 Call, CGF.getContext().getIntTypeForBitwidth(32, /* Signed */ true),
Filipe Cabecinhas7af183d2015-08-11 04:19:28 +00001905 CGF.getContext().BoolTy, Loc);
Alexander Musman92bdaab2015-03-12 13:37:50 +00001906}
1907
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001908void CGOpenMPRuntime::emitNumThreadsClause(CodeGenFunction &CGF,
1909 llvm::Value *NumThreads,
1910 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001911 if (!CGF.HaveInsertPoint())
1912 return;
Alexey Bataevb2059782014-10-13 08:23:51 +00001913 // Build call __kmpc_push_num_threads(&loc, global_tid, num_threads)
1914 llvm::Value *Args[] = {
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001915 emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc),
Alexey Bataevb2059782014-10-13 08:23:51 +00001916 CGF.Builder.CreateIntCast(NumThreads, CGF.Int32Ty, /*isSigned*/ true)};
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001917 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_push_num_threads),
1918 Args);
Alexey Bataevb2059782014-10-13 08:23:51 +00001919}
1920
Alexey Bataev7f210c62015-06-18 13:40:03 +00001921void CGOpenMPRuntime::emitProcBindClause(CodeGenFunction &CGF,
1922 OpenMPProcBindClauseKind ProcBind,
1923 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001924 if (!CGF.HaveInsertPoint())
1925 return;
Alexey Bataev7f210c62015-06-18 13:40:03 +00001926 // Constants for proc bind value accepted by the runtime.
1927 enum ProcBindTy {
1928 ProcBindFalse = 0,
1929 ProcBindTrue,
1930 ProcBindMaster,
1931 ProcBindClose,
1932 ProcBindSpread,
1933 ProcBindIntel,
1934 ProcBindDefault
1935 } RuntimeProcBind;
1936 switch (ProcBind) {
1937 case OMPC_PROC_BIND_master:
1938 RuntimeProcBind = ProcBindMaster;
1939 break;
1940 case OMPC_PROC_BIND_close:
1941 RuntimeProcBind = ProcBindClose;
1942 break;
1943 case OMPC_PROC_BIND_spread:
1944 RuntimeProcBind = ProcBindSpread;
1945 break;
1946 case OMPC_PROC_BIND_unknown:
1947 llvm_unreachable("Unsupported proc_bind value.");
1948 }
1949 // Build call __kmpc_push_proc_bind(&loc, global_tid, proc_bind)
1950 llvm::Value *Args[] = {
1951 emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc),
1952 llvm::ConstantInt::get(CGM.IntTy, RuntimeProcBind, /*isSigned=*/true)};
1953 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_push_proc_bind), Args);
1954}
1955
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001956void CGOpenMPRuntime::emitFlush(CodeGenFunction &CGF, ArrayRef<const Expr *>,
1957 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00001958 if (!CGF.HaveInsertPoint())
1959 return;
Alexey Bataevd76df6d2015-02-24 12:55:09 +00001960 // Build call void __kmpc_flush(ident_t *loc)
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001961 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_flush),
1962 emitUpdateLocation(CGF, Loc));
Alexey Bataevcc37cc12014-11-20 04:34:54 +00001963}
Alexey Bataev3eff5f42015-02-25 08:32:46 +00001964
Alexey Bataev62b63b12015-03-10 07:28:44 +00001965namespace {
1966/// \brief Indexes of fields for type kmp_task_t.
1967enum KmpTaskTFields {
1968 /// \brief List of shared variables.
1969 KmpTaskTShareds,
1970 /// \brief Task routine.
1971 KmpTaskTRoutine,
1972 /// \brief Partition id for the untied tasks.
1973 KmpTaskTPartId,
1974 /// \brief Function with call of destructors for private variables.
1975 KmpTaskTDestructors,
1976};
Hans Wennborg7eb54642015-09-10 17:07:54 +00001977} // anonymous namespace
Alexey Bataev62b63b12015-03-10 07:28:44 +00001978
Samuel Antaoee8fb302016-01-06 13:42:12 +00001979bool CGOpenMPRuntime::OffloadEntriesInfoManagerTy::empty() const {
1980 // FIXME: Add other entries type when they become supported.
1981 return OffloadEntriesTargetRegion.empty();
1982}
1983
1984/// \brief Initialize target region entry.
1985void CGOpenMPRuntime::OffloadEntriesInfoManagerTy::
1986 initializeTargetRegionEntryInfo(unsigned DeviceID, unsigned FileID,
1987 StringRef ParentName, unsigned LineNum,
1988 unsigned ColNum, unsigned Order) {
1989 assert(CGM.getLangOpts().OpenMPIsDevice && "Initialization of entries is "
1990 "only required for the device "
1991 "code generation.");
1992 OffloadEntriesTargetRegion[DeviceID][FileID][ParentName][LineNum][ColNum] =
1993 OffloadEntryInfoTargetRegion(Order, /*Addr=*/nullptr, /*ID=*/nullptr);
1994 ++OffloadingEntriesNum;
1995}
1996
1997void CGOpenMPRuntime::OffloadEntriesInfoManagerTy::
1998 registerTargetRegionEntryInfo(unsigned DeviceID, unsigned FileID,
1999 StringRef ParentName, unsigned LineNum,
2000 unsigned ColNum, llvm::Constant *Addr,
2001 llvm::Constant *ID) {
2002 // If we are emitting code for a target, the entry is already initialized,
2003 // only has to be registered.
2004 if (CGM.getLangOpts().OpenMPIsDevice) {
2005 assert(hasTargetRegionEntryInfo(DeviceID, FileID, ParentName, LineNum,
2006 ColNum) &&
2007 "Entry must exist.");
2008 auto &Entry = OffloadEntriesTargetRegion[DeviceID][FileID][ParentName]
2009 [LineNum][ColNum];
2010 assert(Entry.isValid() && "Entry not initialized!");
2011 Entry.setAddress(Addr);
2012 Entry.setID(ID);
2013 return;
2014 } else {
2015 OffloadEntryInfoTargetRegion Entry(OffloadingEntriesNum++, Addr, ID);
2016 OffloadEntriesTargetRegion[DeviceID][FileID][ParentName][LineNum][ColNum] =
2017 Entry;
2018 }
2019}
2020
2021bool CGOpenMPRuntime::OffloadEntriesInfoManagerTy::hasTargetRegionEntryInfo(
2022 unsigned DeviceID, unsigned FileID, StringRef ParentName, unsigned LineNum,
2023 unsigned ColNum) const {
2024 auto PerDevice = OffloadEntriesTargetRegion.find(DeviceID);
2025 if (PerDevice == OffloadEntriesTargetRegion.end())
2026 return false;
2027 auto PerFile = PerDevice->second.find(FileID);
2028 if (PerFile == PerDevice->second.end())
2029 return false;
2030 auto PerParentName = PerFile->second.find(ParentName);
2031 if (PerParentName == PerFile->second.end())
2032 return false;
2033 auto PerLine = PerParentName->second.find(LineNum);
2034 if (PerLine == PerParentName->second.end())
2035 return false;
2036 auto PerColumn = PerLine->second.find(ColNum);
2037 if (PerColumn == PerLine->second.end())
2038 return false;
2039 // Fail if this entry is already registered.
2040 if (PerColumn->second.getAddress() || PerColumn->second.getID())
2041 return false;
2042 return true;
2043}
2044
2045void CGOpenMPRuntime::OffloadEntriesInfoManagerTy::actOnTargetRegionEntriesInfo(
2046 const OffloadTargetRegionEntryInfoActTy &Action) {
2047 // Scan all target region entries and perform the provided action.
2048 for (auto &D : OffloadEntriesTargetRegion)
2049 for (auto &F : D.second)
2050 for (auto &P : F.second)
2051 for (auto &L : P.second)
2052 for (auto &C : L.second)
2053 Action(D.first, F.first, P.first(), L.first, C.first, C.second);
2054}
2055
2056/// \brief Create a Ctor/Dtor-like function whose body is emitted through
2057/// \a Codegen. This is used to emit the two functions that register and
2058/// unregister the descriptor of the current compilation unit.
2059static llvm::Function *
2060createOffloadingBinaryDescriptorFunction(CodeGenModule &CGM, StringRef Name,
2061 const RegionCodeGenTy &Codegen) {
2062 auto &C = CGM.getContext();
2063 FunctionArgList Args;
2064 ImplicitParamDecl DummyPtr(C, /*DC=*/nullptr, SourceLocation(),
2065 /*Id=*/nullptr, C.VoidPtrTy);
2066 Args.push_back(&DummyPtr);
2067
2068 CodeGenFunction CGF(CGM);
2069 GlobalDecl();
2070 auto &FI = CGM.getTypes().arrangeFreeFunctionDeclaration(
2071 C.VoidTy, Args, FunctionType::ExtInfo(),
2072 /*isVariadic=*/false);
2073 auto FTy = CGM.getTypes().GetFunctionType(FI);
2074 auto *Fn =
2075 CGM.CreateGlobalInitOrDestructFunction(FTy, Name, FI, SourceLocation());
2076 CGF.StartFunction(GlobalDecl(), C.VoidTy, Fn, FI, Args, SourceLocation());
2077 Codegen(CGF);
2078 CGF.FinishFunction();
2079 return Fn;
2080}
2081
2082llvm::Function *
2083CGOpenMPRuntime::createOffloadingBinaryDescriptorRegistration() {
2084
2085 // If we don't have entries or if we are emitting code for the device, we
2086 // don't need to do anything.
2087 if (CGM.getLangOpts().OpenMPIsDevice || OffloadEntriesInfoManager.empty())
2088 return nullptr;
2089
2090 auto &M = CGM.getModule();
2091 auto &C = CGM.getContext();
2092
2093 // Get list of devices we care about
2094 auto &Devices = CGM.getLangOpts().OMPTargetTriples;
2095
2096 // We should be creating an offloading descriptor only if there are devices
2097 // specified.
2098 assert(!Devices.empty() && "No OpenMP offloading devices??");
2099
2100 // Create the external variables that will point to the begin and end of the
2101 // host entries section. These will be defined by the linker.
2102 auto *OffloadEntryTy =
2103 CGM.getTypes().ConvertTypeForMem(getTgtOffloadEntryQTy());
2104 llvm::GlobalVariable *HostEntriesBegin = new llvm::GlobalVariable(
2105 M, OffloadEntryTy, /*isConstant=*/true,
Eugene Zelenko1660a5d2016-01-26 19:01:06 +00002106 llvm::GlobalValue::ExternalLinkage, /*Initializer=*/nullptr,
Samuel Antaoee8fb302016-01-06 13:42:12 +00002107 ".omp_offloading.entries_begin");
2108 llvm::GlobalVariable *HostEntriesEnd = new llvm::GlobalVariable(
2109 M, OffloadEntryTy, /*isConstant=*/true,
Eugene Zelenko1660a5d2016-01-26 19:01:06 +00002110 llvm::GlobalValue::ExternalLinkage, /*Initializer=*/nullptr,
Samuel Antaoee8fb302016-01-06 13:42:12 +00002111 ".omp_offloading.entries_end");
2112
2113 // Create all device images
2114 llvm::SmallVector<llvm::Constant *, 4> DeviceImagesEntires;
2115 auto *DeviceImageTy = cast<llvm::StructType>(
2116 CGM.getTypes().ConvertTypeForMem(getTgtDeviceImageQTy()));
2117
2118 for (unsigned i = 0; i < Devices.size(); ++i) {
2119 StringRef T = Devices[i].getTriple();
2120 auto *ImgBegin = new llvm::GlobalVariable(
2121 M, CGM.Int8Ty, /*isConstant=*/true, llvm::GlobalValue::ExternalLinkage,
Eugene Zelenko1660a5d2016-01-26 19:01:06 +00002122 /*Initializer=*/nullptr,
2123 Twine(".omp_offloading.img_start.") + Twine(T));
Samuel Antaoee8fb302016-01-06 13:42:12 +00002124 auto *ImgEnd = new llvm::GlobalVariable(
2125 M, CGM.Int8Ty, /*isConstant=*/true, llvm::GlobalValue::ExternalLinkage,
Eugene Zelenko1660a5d2016-01-26 19:01:06 +00002126 /*Initializer=*/nullptr, Twine(".omp_offloading.img_end.") + Twine(T));
Samuel Antaoee8fb302016-01-06 13:42:12 +00002127
2128 llvm::Constant *Dev =
2129 llvm::ConstantStruct::get(DeviceImageTy, ImgBegin, ImgEnd,
2130 HostEntriesBegin, HostEntriesEnd, nullptr);
2131 DeviceImagesEntires.push_back(Dev);
2132 }
2133
2134 // Create device images global array.
2135 llvm::ArrayType *DeviceImagesInitTy =
2136 llvm::ArrayType::get(DeviceImageTy, DeviceImagesEntires.size());
2137 llvm::Constant *DeviceImagesInit =
2138 llvm::ConstantArray::get(DeviceImagesInitTy, DeviceImagesEntires);
2139
2140 llvm::GlobalVariable *DeviceImages = new llvm::GlobalVariable(
2141 M, DeviceImagesInitTy, /*isConstant=*/true,
2142 llvm::GlobalValue::InternalLinkage, DeviceImagesInit,
2143 ".omp_offloading.device_images");
2144 DeviceImages->setUnnamedAddr(true);
2145
2146 // This is a Zero array to be used in the creation of the constant expressions
2147 llvm::Constant *Index[] = {llvm::Constant::getNullValue(CGM.Int32Ty),
2148 llvm::Constant::getNullValue(CGM.Int32Ty)};
2149
2150 // Create the target region descriptor.
2151 auto *BinaryDescriptorTy = cast<llvm::StructType>(
2152 CGM.getTypes().ConvertTypeForMem(getTgtBinaryDescriptorQTy()));
2153 llvm::Constant *TargetRegionsDescriptorInit = llvm::ConstantStruct::get(
2154 BinaryDescriptorTy, llvm::ConstantInt::get(CGM.Int32Ty, Devices.size()),
2155 llvm::ConstantExpr::getGetElementPtr(DeviceImagesInitTy, DeviceImages,
2156 Index),
2157 HostEntriesBegin, HostEntriesEnd, nullptr);
2158
2159 auto *Desc = new llvm::GlobalVariable(
2160 M, BinaryDescriptorTy, /*isConstant=*/true,
2161 llvm::GlobalValue::InternalLinkage, TargetRegionsDescriptorInit,
2162 ".omp_offloading.descriptor");
2163
2164 // Emit code to register or unregister the descriptor at execution
2165 // startup or closing, respectively.
2166
2167 // Create a variable to drive the registration and unregistration of the
2168 // descriptor, so we can reuse the logic that emits Ctors and Dtors.
2169 auto *IdentInfo = &C.Idents.get(".omp_offloading.reg_unreg_var");
2170 ImplicitParamDecl RegUnregVar(C, C.getTranslationUnitDecl(), SourceLocation(),
2171 IdentInfo, C.CharTy);
2172
2173 auto *UnRegFn = createOffloadingBinaryDescriptorFunction(
2174 CGM, ".omp_offloading.descriptor_unreg", [&](CodeGenFunction &CGF) {
2175 CGF.EmitCallOrInvoke(createRuntimeFunction(OMPRTL__tgt_unregister_lib),
2176 Desc);
2177 });
2178 auto *RegFn = createOffloadingBinaryDescriptorFunction(
2179 CGM, ".omp_offloading.descriptor_reg", [&](CodeGenFunction &CGF) {
2180 CGF.EmitCallOrInvoke(createRuntimeFunction(OMPRTL__tgt_register_lib),
2181 Desc);
2182 CGM.getCXXABI().registerGlobalDtor(CGF, RegUnregVar, UnRegFn, Desc);
2183 });
2184 return RegFn;
2185}
2186
2187void CGOpenMPRuntime::createOffloadEntry(llvm::Constant *Addr, StringRef Name,
2188 uint64_t Size) {
2189 auto *TgtOffloadEntryType = cast<llvm::StructType>(
2190 CGM.getTypes().ConvertTypeForMem(getTgtOffloadEntryQTy()));
2191 llvm::LLVMContext &C = CGM.getModule().getContext();
2192 llvm::Module &M = CGM.getModule();
2193
2194 // Make sure the address has the right type.
2195 llvm::Constant *AddrPtr = llvm::ConstantExpr::getBitCast(Addr, CGM.VoidPtrTy);
2196
2197 // Create constant string with the name.
2198 llvm::Constant *StrPtrInit = llvm::ConstantDataArray::getString(C, Name);
2199
2200 llvm::GlobalVariable *Str =
2201 new llvm::GlobalVariable(M, StrPtrInit->getType(), /*isConstant=*/true,
2202 llvm::GlobalValue::InternalLinkage, StrPtrInit,
2203 ".omp_offloading.entry_name");
2204 Str->setUnnamedAddr(true);
2205 llvm::Constant *StrPtr = llvm::ConstantExpr::getBitCast(Str, CGM.Int8PtrTy);
2206
2207 // Create the entry struct.
2208 llvm::Constant *EntryInit = llvm::ConstantStruct::get(
2209 TgtOffloadEntryType, AddrPtr, StrPtr,
2210 llvm::ConstantInt::get(CGM.SizeTy, Size), nullptr);
2211 llvm::GlobalVariable *Entry = new llvm::GlobalVariable(
2212 M, TgtOffloadEntryType, true, llvm::GlobalValue::ExternalLinkage,
2213 EntryInit, ".omp_offloading.entry");
2214
2215 // The entry has to be created in the section the linker expects it to be.
2216 Entry->setSection(".omp_offloading.entries");
2217 // We can't have any padding between symbols, so we need to have 1-byte
2218 // alignment.
2219 Entry->setAlignment(1);
2220 return;
2221}
2222
2223void CGOpenMPRuntime::createOffloadEntriesAndInfoMetadata() {
2224 // Emit the offloading entries and metadata so that the device codegen side
2225 // can
2226 // easily figure out what to emit. The produced metadata looks like this:
2227 //
2228 // !omp_offload.info = !{!1, ...}
2229 //
2230 // Right now we only generate metadata for function that contain target
2231 // regions.
2232
2233 // If we do not have entries, we dont need to do anything.
2234 if (OffloadEntriesInfoManager.empty())
2235 return;
2236
2237 llvm::Module &M = CGM.getModule();
2238 llvm::LLVMContext &C = M.getContext();
2239 SmallVector<OffloadEntriesInfoManagerTy::OffloadEntryInfo *, 16>
2240 OrderedEntries(OffloadEntriesInfoManager.size());
2241
2242 // Create the offloading info metadata node.
2243 llvm::NamedMDNode *MD = M.getOrInsertNamedMetadata("omp_offload.info");
2244
2245 // Auxiliar methods to create metadata values and strings.
2246 auto getMDInt = [&](unsigned v) {
2247 return llvm::ConstantAsMetadata::get(
2248 llvm::ConstantInt::get(llvm::Type::getInt32Ty(C), v));
2249 };
2250
2251 auto getMDString = [&](StringRef v) { return llvm::MDString::get(C, v); };
2252
2253 // Create function that emits metadata for each target region entry;
2254 auto &&TargetRegionMetadataEmitter = [&](
2255 unsigned DeviceID, unsigned FileID, StringRef ParentName, unsigned Line,
2256 unsigned Column,
2257 OffloadEntriesInfoManagerTy::OffloadEntryInfoTargetRegion &E) {
2258 llvm::SmallVector<llvm::Metadata *, 32> Ops;
2259 // Generate metadata for target regions. Each entry of this metadata
2260 // contains:
2261 // - Entry 0 -> Kind of this type of metadata (0).
2262 // - Entry 1 -> Device ID of the file where the entry was identified.
2263 // - Entry 2 -> File ID of the file where the entry was identified.
2264 // - Entry 3 -> Mangled name of the function where the entry was identified.
2265 // - Entry 4 -> Line in the file where the entry was identified.
2266 // - Entry 5 -> Column in the file where the entry was identified.
2267 // - Entry 6 -> Order the entry was created.
2268 // The first element of the metadata node is the kind.
2269 Ops.push_back(getMDInt(E.getKind()));
2270 Ops.push_back(getMDInt(DeviceID));
2271 Ops.push_back(getMDInt(FileID));
2272 Ops.push_back(getMDString(ParentName));
2273 Ops.push_back(getMDInt(Line));
2274 Ops.push_back(getMDInt(Column));
2275 Ops.push_back(getMDInt(E.getOrder()));
2276
2277 // Save this entry in the right position of the ordered entries array.
2278 OrderedEntries[E.getOrder()] = &E;
2279
2280 // Add metadata to the named metadata node.
2281 MD->addOperand(llvm::MDNode::get(C, Ops));
2282 };
2283
2284 OffloadEntriesInfoManager.actOnTargetRegionEntriesInfo(
2285 TargetRegionMetadataEmitter);
2286
2287 for (auto *E : OrderedEntries) {
2288 assert(E && "All ordered entries must exist!");
2289 if (auto *CE =
2290 dyn_cast<OffloadEntriesInfoManagerTy::OffloadEntryInfoTargetRegion>(
2291 E)) {
2292 assert(CE->getID() && CE->getAddress() &&
2293 "Entry ID and Addr are invalid!");
2294 createOffloadEntry(CE->getID(), CE->getAddress()->getName(), /*Size=*/0);
2295 } else
2296 llvm_unreachable("Unsupported entry kind.");
2297 }
2298}
2299
2300/// \brief Loads all the offload entries information from the host IR
2301/// metadata.
2302void CGOpenMPRuntime::loadOffloadInfoMetadata() {
2303 // If we are in target mode, load the metadata from the host IR. This code has
2304 // to match the metadaata creation in createOffloadEntriesAndInfoMetadata().
2305
2306 if (!CGM.getLangOpts().OpenMPIsDevice)
2307 return;
2308
2309 if (CGM.getLangOpts().OMPHostIRFile.empty())
2310 return;
2311
2312 auto Buf = llvm::MemoryBuffer::getFile(CGM.getLangOpts().OMPHostIRFile);
2313 if (Buf.getError())
2314 return;
2315
2316 llvm::LLVMContext C;
2317 auto ME = llvm::parseBitcodeFile(Buf.get()->getMemBufferRef(), C);
2318
2319 if (ME.getError())
2320 return;
2321
2322 llvm::NamedMDNode *MD = ME.get()->getNamedMetadata("omp_offload.info");
2323 if (!MD)
2324 return;
2325
2326 for (auto I : MD->operands()) {
2327 llvm::MDNode *MN = cast<llvm::MDNode>(I);
2328
2329 auto getMDInt = [&](unsigned Idx) {
2330 llvm::ConstantAsMetadata *V =
2331 cast<llvm::ConstantAsMetadata>(MN->getOperand(Idx));
2332 return cast<llvm::ConstantInt>(V->getValue())->getZExtValue();
2333 };
2334
2335 auto getMDString = [&](unsigned Idx) {
2336 llvm::MDString *V = cast<llvm::MDString>(MN->getOperand(Idx));
2337 return V->getString();
2338 };
2339
2340 switch (getMDInt(0)) {
2341 default:
2342 llvm_unreachable("Unexpected metadata!");
2343 break;
2344 case OffloadEntriesInfoManagerTy::OffloadEntryInfo::
2345 OFFLOAD_ENTRY_INFO_TARGET_REGION:
2346 OffloadEntriesInfoManager.initializeTargetRegionEntryInfo(
2347 /*DeviceID=*/getMDInt(1), /*FileID=*/getMDInt(2),
2348 /*ParentName=*/getMDString(3), /*Line=*/getMDInt(4),
2349 /*Column=*/getMDInt(5), /*Order=*/getMDInt(6));
2350 break;
2351 }
2352 }
2353}
2354
Alexey Bataev62b63b12015-03-10 07:28:44 +00002355void CGOpenMPRuntime::emitKmpRoutineEntryT(QualType KmpInt32Ty) {
2356 if (!KmpRoutineEntryPtrTy) {
2357 // Build typedef kmp_int32 (* kmp_routine_entry_t)(kmp_int32, void *); type.
2358 auto &C = CGM.getContext();
2359 QualType KmpRoutineEntryTyArgs[] = {KmpInt32Ty, C.VoidPtrTy};
2360 FunctionProtoType::ExtProtoInfo EPI;
2361 KmpRoutineEntryPtrQTy = C.getPointerType(
2362 C.getFunctionType(KmpInt32Ty, KmpRoutineEntryTyArgs, EPI));
2363 KmpRoutineEntryPtrTy = CGM.getTypes().ConvertType(KmpRoutineEntryPtrQTy);
2364 }
2365}
2366
Alexey Bataevc71a4092015-09-11 10:29:41 +00002367static FieldDecl *addFieldToRecordDecl(ASTContext &C, DeclContext *DC,
2368 QualType FieldTy) {
Alexey Bataev62b63b12015-03-10 07:28:44 +00002369 auto *Field = FieldDecl::Create(
2370 C, DC, SourceLocation(), SourceLocation(), /*Id=*/nullptr, FieldTy,
2371 C.getTrivialTypeSourceInfo(FieldTy, SourceLocation()),
2372 /*BW=*/nullptr, /*Mutable=*/false, /*InitStyle=*/ICIS_NoInit);
2373 Field->setAccess(AS_public);
2374 DC->addDecl(Field);
Alexey Bataevc71a4092015-09-11 10:29:41 +00002375 return Field;
Alexey Bataev62b63b12015-03-10 07:28:44 +00002376}
2377
Samuel Antaoee8fb302016-01-06 13:42:12 +00002378QualType CGOpenMPRuntime::getTgtOffloadEntryQTy() {
2379
2380 // Make sure the type of the entry is already created. This is the type we
2381 // have to create:
2382 // struct __tgt_offload_entry{
2383 // void *addr; // Pointer to the offload entry info.
2384 // // (function or global)
2385 // char *name; // Name of the function or global.
2386 // size_t size; // Size of the entry info (0 if it a function).
2387 // };
2388 if (TgtOffloadEntryQTy.isNull()) {
2389 ASTContext &C = CGM.getContext();
2390 auto *RD = C.buildImplicitRecord("__tgt_offload_entry");
2391 RD->startDefinition();
2392 addFieldToRecordDecl(C, RD, C.VoidPtrTy);
2393 addFieldToRecordDecl(C, RD, C.getPointerType(C.CharTy));
2394 addFieldToRecordDecl(C, RD, C.getSizeType());
2395 RD->completeDefinition();
2396 TgtOffloadEntryQTy = C.getRecordType(RD);
2397 }
2398 return TgtOffloadEntryQTy;
2399}
2400
2401QualType CGOpenMPRuntime::getTgtDeviceImageQTy() {
2402 // These are the types we need to build:
2403 // struct __tgt_device_image{
2404 // void *ImageStart; // Pointer to the target code start.
2405 // void *ImageEnd; // Pointer to the target code end.
2406 // // We also add the host entries to the device image, as it may be useful
2407 // // for the target runtime to have access to that information.
2408 // __tgt_offload_entry *EntriesBegin; // Begin of the table with all
2409 // // the entries.
2410 // __tgt_offload_entry *EntriesEnd; // End of the table with all the
2411 // // entries (non inclusive).
2412 // };
2413 if (TgtDeviceImageQTy.isNull()) {
2414 ASTContext &C = CGM.getContext();
2415 auto *RD = C.buildImplicitRecord("__tgt_device_image");
2416 RD->startDefinition();
2417 addFieldToRecordDecl(C, RD, C.VoidPtrTy);
2418 addFieldToRecordDecl(C, RD, C.VoidPtrTy);
2419 addFieldToRecordDecl(C, RD, C.getPointerType(getTgtOffloadEntryQTy()));
2420 addFieldToRecordDecl(C, RD, C.getPointerType(getTgtOffloadEntryQTy()));
2421 RD->completeDefinition();
2422 TgtDeviceImageQTy = C.getRecordType(RD);
2423 }
2424 return TgtDeviceImageQTy;
2425}
2426
2427QualType CGOpenMPRuntime::getTgtBinaryDescriptorQTy() {
2428 // struct __tgt_bin_desc{
2429 // int32_t NumDevices; // Number of devices supported.
2430 // __tgt_device_image *DeviceImages; // Arrays of device images
2431 // // (one per device).
2432 // __tgt_offload_entry *EntriesBegin; // Begin of the table with all the
2433 // // entries.
2434 // __tgt_offload_entry *EntriesEnd; // End of the table with all the
2435 // // entries (non inclusive).
2436 // };
2437 if (TgtBinaryDescriptorQTy.isNull()) {
2438 ASTContext &C = CGM.getContext();
2439 auto *RD = C.buildImplicitRecord("__tgt_bin_desc");
2440 RD->startDefinition();
2441 addFieldToRecordDecl(
2442 C, RD, C.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/true));
2443 addFieldToRecordDecl(C, RD, C.getPointerType(getTgtDeviceImageQTy()));
2444 addFieldToRecordDecl(C, RD, C.getPointerType(getTgtOffloadEntryQTy()));
2445 addFieldToRecordDecl(C, RD, C.getPointerType(getTgtOffloadEntryQTy()));
2446 RD->completeDefinition();
2447 TgtBinaryDescriptorQTy = C.getRecordType(RD);
2448 }
2449 return TgtBinaryDescriptorQTy;
2450}
2451
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002452namespace {
Alexey Bataev9e034042015-05-05 04:05:12 +00002453struct PrivateHelpersTy {
2454 PrivateHelpersTy(const VarDecl *Original, const VarDecl *PrivateCopy,
2455 const VarDecl *PrivateElemInit)
2456 : Original(Original), PrivateCopy(PrivateCopy),
2457 PrivateElemInit(PrivateElemInit) {}
2458 const VarDecl *Original;
2459 const VarDecl *PrivateCopy;
2460 const VarDecl *PrivateElemInit;
2461};
2462typedef std::pair<CharUnits /*Align*/, PrivateHelpersTy> PrivateDataTy;
Hans Wennborg7eb54642015-09-10 17:07:54 +00002463} // anonymous namespace
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002464
Alexey Bataev9e034042015-05-05 04:05:12 +00002465static RecordDecl *
Craig Topper8674c5c2015-09-29 04:30:07 +00002466createPrivatesRecordDecl(CodeGenModule &CGM, ArrayRef<PrivateDataTy> Privates) {
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002467 if (!Privates.empty()) {
2468 auto &C = CGM.getContext();
2469 // Build struct .kmp_privates_t. {
2470 // /* private vars */
2471 // };
2472 auto *RD = C.buildImplicitRecord(".kmp_privates.t");
2473 RD->startDefinition();
2474 for (auto &&Pair : Privates) {
Alexey Bataevc71a4092015-09-11 10:29:41 +00002475 auto *VD = Pair.second.Original;
2476 auto Type = VD->getType();
Alexey Bataev1d9c15c2015-05-19 12:31:28 +00002477 Type = Type.getNonReferenceType();
Alexey Bataevc71a4092015-09-11 10:29:41 +00002478 auto *FD = addFieldToRecordDecl(C, RD, Type);
2479 if (VD->hasAttrs()) {
2480 for (specific_attr_iterator<AlignedAttr> I(VD->getAttrs().begin()),
2481 E(VD->getAttrs().end());
2482 I != E; ++I)
2483 FD->addAttr(*I);
2484 }
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002485 }
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002486 RD->completeDefinition();
2487 return RD;
2488 }
2489 return nullptr;
2490}
2491
Alexey Bataev9e034042015-05-05 04:05:12 +00002492static RecordDecl *
2493createKmpTaskTRecordDecl(CodeGenModule &CGM, QualType KmpInt32Ty,
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002494 QualType KmpRoutineEntryPointerQTy) {
Alexey Bataev62b63b12015-03-10 07:28:44 +00002495 auto &C = CGM.getContext();
2496 // Build struct kmp_task_t {
2497 // void * shareds;
2498 // kmp_routine_entry_t routine;
2499 // kmp_int32 part_id;
2500 // kmp_routine_entry_t destructors;
Alexey Bataev62b63b12015-03-10 07:28:44 +00002501 // };
2502 auto *RD = C.buildImplicitRecord("kmp_task_t");
2503 RD->startDefinition();
2504 addFieldToRecordDecl(C, RD, C.VoidPtrTy);
2505 addFieldToRecordDecl(C, RD, KmpRoutineEntryPointerQTy);
2506 addFieldToRecordDecl(C, RD, KmpInt32Ty);
2507 addFieldToRecordDecl(C, RD, KmpRoutineEntryPointerQTy);
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002508 RD->completeDefinition();
2509 return RD;
2510}
2511
2512static RecordDecl *
2513createKmpTaskTWithPrivatesRecordDecl(CodeGenModule &CGM, QualType KmpTaskTQTy,
Craig Topper8674c5c2015-09-29 04:30:07 +00002514 ArrayRef<PrivateDataTy> Privates) {
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002515 auto &C = CGM.getContext();
2516 // Build struct kmp_task_t_with_privates {
2517 // kmp_task_t task_data;
2518 // .kmp_privates_t. privates;
2519 // };
2520 auto *RD = C.buildImplicitRecord("kmp_task_t_with_privates");
2521 RD->startDefinition();
2522 addFieldToRecordDecl(C, RD, KmpTaskTQTy);
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002523 if (auto *PrivateRD = createPrivatesRecordDecl(CGM, Privates)) {
2524 addFieldToRecordDecl(C, RD, C.getRecordType(PrivateRD));
2525 }
Alexey Bataev62b63b12015-03-10 07:28:44 +00002526 RD->completeDefinition();
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002527 return RD;
Alexey Bataev62b63b12015-03-10 07:28:44 +00002528}
2529
2530/// \brief Emit a proxy function which accepts kmp_task_t as the second
2531/// argument.
2532/// \code
2533/// kmp_int32 .omp_task_entry.(kmp_int32 gtid, kmp_task_t *tt) {
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002534/// TaskFunction(gtid, tt->part_id, &tt->privates, task_privates_map,
2535/// tt->shareds);
Alexey Bataev62b63b12015-03-10 07:28:44 +00002536/// return 0;
2537/// }
2538/// \endcode
2539static llvm::Value *
2540emitProxyTaskFunction(CodeGenModule &CGM, SourceLocation Loc,
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002541 QualType KmpInt32Ty, QualType KmpTaskTWithPrivatesPtrQTy,
2542 QualType KmpTaskTWithPrivatesQTy, QualType KmpTaskTQTy,
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002543 QualType SharedsPtrTy, llvm::Value *TaskFunction,
2544 llvm::Value *TaskPrivatesMap) {
Alexey Bataev62b63b12015-03-10 07:28:44 +00002545 auto &C = CGM.getContext();
2546 FunctionArgList Args;
2547 ImplicitParamDecl GtidArg(C, /*DC=*/nullptr, Loc, /*Id=*/nullptr, KmpInt32Ty);
2548 ImplicitParamDecl TaskTypeArg(C, /*DC=*/nullptr, Loc,
Alexey Bataev2377fe92015-09-10 08:12:02 +00002549 /*Id=*/nullptr,
2550 KmpTaskTWithPrivatesPtrQTy.withRestrict());
Alexey Bataev62b63b12015-03-10 07:28:44 +00002551 Args.push_back(&GtidArg);
2552 Args.push_back(&TaskTypeArg);
2553 FunctionType::ExtInfo Info;
2554 auto &TaskEntryFnInfo =
2555 CGM.getTypes().arrangeFreeFunctionDeclaration(KmpInt32Ty, Args, Info,
2556 /*isVariadic=*/false);
2557 auto *TaskEntryTy = CGM.getTypes().GetFunctionType(TaskEntryFnInfo);
2558 auto *TaskEntry =
2559 llvm::Function::Create(TaskEntryTy, llvm::GlobalValue::InternalLinkage,
2560 ".omp_task_entry.", &CGM.getModule());
Akira Hatanaka44a59f82015-10-28 02:30:47 +00002561 CGM.SetInternalFunctionAttributes(/*D=*/nullptr, TaskEntry, TaskEntryFnInfo);
Alexey Bataev62b63b12015-03-10 07:28:44 +00002562 CodeGenFunction CGF(CGM);
2563 CGF.disableDebugInfo();
2564 CGF.StartFunction(GlobalDecl(), KmpInt32Ty, TaskEntry, TaskEntryFnInfo, Args);
2565
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002566 // TaskFunction(gtid, tt->task_data.part_id, &tt->privates, task_privates_map,
2567 // tt->task_data.shareds);
Alexey Bataev62b63b12015-03-10 07:28:44 +00002568 auto *GtidParam = CGF.EmitLoadOfScalar(
John McCall7f416cc2015-09-08 08:05:57 +00002569 CGF.GetAddrOfLocalVar(&GtidArg), /*Volatile=*/false, KmpInt32Ty, Loc);
Alexey Bataev2377fe92015-09-10 08:12:02 +00002570 LValue TDBase = emitLoadOfPointerLValue(
2571 CGF, CGF.GetAddrOfLocalVar(&TaskTypeArg), KmpTaskTWithPrivatesPtrQTy);
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002572 auto *KmpTaskTWithPrivatesQTyRD =
2573 cast<RecordDecl>(KmpTaskTWithPrivatesQTy->getAsTagDecl());
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002574 LValue Base =
2575 CGF.EmitLValueForField(TDBase, *KmpTaskTWithPrivatesQTyRD->field_begin());
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002576 auto *KmpTaskTQTyRD = cast<RecordDecl>(KmpTaskTQTy->getAsTagDecl());
2577 auto PartIdFI = std::next(KmpTaskTQTyRD->field_begin(), KmpTaskTPartId);
2578 auto PartIdLVal = CGF.EmitLValueForField(Base, *PartIdFI);
2579 auto *PartidParam = CGF.EmitLoadOfLValue(PartIdLVal, Loc).getScalarVal();
2580
2581 auto SharedsFI = std::next(KmpTaskTQTyRD->field_begin(), KmpTaskTShareds);
2582 auto SharedsLVal = CGF.EmitLValueForField(Base, *SharedsFI);
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002583 auto *SharedsParam = CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002584 CGF.EmitLoadOfLValue(SharedsLVal, Loc).getScalarVal(),
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002585 CGF.ConvertTypeForMem(SharedsPtrTy));
2586
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002587 auto PrivatesFI = std::next(KmpTaskTWithPrivatesQTyRD->field_begin(), 1);
2588 llvm::Value *PrivatesParam;
2589 if (PrivatesFI != KmpTaskTWithPrivatesQTyRD->field_end()) {
2590 auto PrivatesLVal = CGF.EmitLValueForField(TDBase, *PrivatesFI);
2591 PrivatesParam = CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
John McCall7f416cc2015-09-08 08:05:57 +00002592 PrivatesLVal.getPointer(), CGF.VoidPtrTy);
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002593 } else {
2594 PrivatesParam = llvm::ConstantPointerNull::get(CGF.VoidPtrTy);
2595 }
2596
2597 llvm::Value *CallArgs[] = {GtidParam, PartidParam, PrivatesParam,
2598 TaskPrivatesMap, SharedsParam};
Alexey Bataev62b63b12015-03-10 07:28:44 +00002599 CGF.EmitCallOrInvoke(TaskFunction, CallArgs);
2600 CGF.EmitStoreThroughLValue(
2601 RValue::get(CGF.Builder.getInt32(/*C=*/0)),
John McCall7f416cc2015-09-08 08:05:57 +00002602 CGF.MakeAddrLValue(CGF.ReturnValue, KmpInt32Ty));
Alexey Bataev62b63b12015-03-10 07:28:44 +00002603 CGF.FinishFunction();
2604 return TaskEntry;
2605}
2606
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002607static llvm::Value *emitDestructorsFunction(CodeGenModule &CGM,
2608 SourceLocation Loc,
2609 QualType KmpInt32Ty,
2610 QualType KmpTaskTWithPrivatesPtrQTy,
2611 QualType KmpTaskTWithPrivatesQTy) {
Alexey Bataev62b63b12015-03-10 07:28:44 +00002612 auto &C = CGM.getContext();
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002613 FunctionArgList Args;
2614 ImplicitParamDecl GtidArg(C, /*DC=*/nullptr, Loc, /*Id=*/nullptr, KmpInt32Ty);
2615 ImplicitParamDecl TaskTypeArg(C, /*DC=*/nullptr, Loc,
Alexey Bataev2377fe92015-09-10 08:12:02 +00002616 /*Id=*/nullptr,
2617 KmpTaskTWithPrivatesPtrQTy.withRestrict());
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002618 Args.push_back(&GtidArg);
2619 Args.push_back(&TaskTypeArg);
2620 FunctionType::ExtInfo Info;
2621 auto &DestructorFnInfo =
2622 CGM.getTypes().arrangeFreeFunctionDeclaration(KmpInt32Ty, Args, Info,
2623 /*isVariadic=*/false);
2624 auto *DestructorFnTy = CGM.getTypes().GetFunctionType(DestructorFnInfo);
2625 auto *DestructorFn =
2626 llvm::Function::Create(DestructorFnTy, llvm::GlobalValue::InternalLinkage,
2627 ".omp_task_destructor.", &CGM.getModule());
Akira Hatanaka44a59f82015-10-28 02:30:47 +00002628 CGM.SetInternalFunctionAttributes(/*D=*/nullptr, DestructorFn,
2629 DestructorFnInfo);
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002630 CodeGenFunction CGF(CGM);
2631 CGF.disableDebugInfo();
2632 CGF.StartFunction(GlobalDecl(), KmpInt32Ty, DestructorFn, DestructorFnInfo,
2633 Args);
2634
Alexey Bataev2377fe92015-09-10 08:12:02 +00002635 LValue Base = emitLoadOfPointerLValue(
2636 CGF, CGF.GetAddrOfLocalVar(&TaskTypeArg), KmpTaskTWithPrivatesPtrQTy);
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002637 auto *KmpTaskTWithPrivatesQTyRD =
2638 cast<RecordDecl>(KmpTaskTWithPrivatesQTy->getAsTagDecl());
2639 auto FI = std::next(KmpTaskTWithPrivatesQTyRD->field_begin());
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002640 Base = CGF.EmitLValueForField(Base, *FI);
2641 for (auto *Field :
2642 cast<RecordDecl>(FI->getType()->getAsTagDecl())->fields()) {
2643 if (auto DtorKind = Field->getType().isDestructedType()) {
2644 auto FieldLValue = CGF.EmitLValueForField(Base, Field);
2645 CGF.pushDestroy(DtorKind, FieldLValue.getAddress(), Field->getType());
2646 }
2647 }
2648 CGF.FinishFunction();
2649 return DestructorFn;
2650}
2651
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002652/// \brief Emit a privates mapping function for correct handling of private and
2653/// firstprivate variables.
2654/// \code
2655/// void .omp_task_privates_map.(const .privates. *noalias privs, <ty1>
2656/// **noalias priv1,..., <tyn> **noalias privn) {
2657/// *priv1 = &.privates.priv1;
2658/// ...;
2659/// *privn = &.privates.privn;
2660/// }
2661/// \endcode
2662static llvm::Value *
2663emitTaskPrivateMappingFunction(CodeGenModule &CGM, SourceLocation Loc,
Craig Topper8674c5c2015-09-29 04:30:07 +00002664 ArrayRef<const Expr *> PrivateVars,
2665 ArrayRef<const Expr *> FirstprivateVars,
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002666 QualType PrivatesQTy,
Craig Topper8674c5c2015-09-29 04:30:07 +00002667 ArrayRef<PrivateDataTy> Privates) {
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002668 auto &C = CGM.getContext();
2669 FunctionArgList Args;
2670 ImplicitParamDecl TaskPrivatesArg(
2671 C, /*DC=*/nullptr, Loc, /*Id=*/nullptr,
2672 C.getPointerType(PrivatesQTy).withConst().withRestrict());
2673 Args.push_back(&TaskPrivatesArg);
2674 llvm::DenseMap<const VarDecl *, unsigned> PrivateVarsPos;
2675 unsigned Counter = 1;
2676 for (auto *E: PrivateVars) {
2677 Args.push_back(ImplicitParamDecl::Create(
2678 C, /*DC=*/nullptr, Loc,
2679 /*Id=*/nullptr, C.getPointerType(C.getPointerType(E->getType()))
2680 .withConst()
2681 .withRestrict()));
2682 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
2683 PrivateVarsPos[VD] = Counter;
2684 ++Counter;
2685 }
2686 for (auto *E : FirstprivateVars) {
2687 Args.push_back(ImplicitParamDecl::Create(
2688 C, /*DC=*/nullptr, Loc,
2689 /*Id=*/nullptr, C.getPointerType(C.getPointerType(E->getType()))
2690 .withConst()
2691 .withRestrict()));
2692 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
2693 PrivateVarsPos[VD] = Counter;
2694 ++Counter;
2695 }
2696 FunctionType::ExtInfo Info;
2697 auto &TaskPrivatesMapFnInfo =
2698 CGM.getTypes().arrangeFreeFunctionDeclaration(C.VoidTy, Args, Info,
2699 /*isVariadic=*/false);
2700 auto *TaskPrivatesMapTy =
2701 CGM.getTypes().GetFunctionType(TaskPrivatesMapFnInfo);
2702 auto *TaskPrivatesMap = llvm::Function::Create(
2703 TaskPrivatesMapTy, llvm::GlobalValue::InternalLinkage,
2704 ".omp_task_privates_map.", &CGM.getModule());
Akira Hatanaka44a59f82015-10-28 02:30:47 +00002705 CGM.SetInternalFunctionAttributes(/*D=*/nullptr, TaskPrivatesMap,
2706 TaskPrivatesMapFnInfo);
Evgeniy Stepanov6b2a61d2015-09-14 21:35:16 +00002707 TaskPrivatesMap->addFnAttr(llvm::Attribute::AlwaysInline);
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002708 CodeGenFunction CGF(CGM);
2709 CGF.disableDebugInfo();
2710 CGF.StartFunction(GlobalDecl(), C.VoidTy, TaskPrivatesMap,
2711 TaskPrivatesMapFnInfo, Args);
2712
2713 // *privi = &.privates.privi;
Alexey Bataev2377fe92015-09-10 08:12:02 +00002714 LValue Base = emitLoadOfPointerLValue(
2715 CGF, CGF.GetAddrOfLocalVar(&TaskPrivatesArg), TaskPrivatesArg.getType());
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002716 auto *PrivatesQTyRD = cast<RecordDecl>(PrivatesQTy->getAsTagDecl());
2717 Counter = 0;
2718 for (auto *Field : PrivatesQTyRD->fields()) {
2719 auto FieldLVal = CGF.EmitLValueForField(Base, Field);
2720 auto *VD = Args[PrivateVarsPos[Privates[Counter].second.Original]];
John McCall7f416cc2015-09-08 08:05:57 +00002721 auto RefLVal = CGF.MakeAddrLValue(CGF.GetAddrOfLocalVar(VD), VD->getType());
Alexey Bataev2377fe92015-09-10 08:12:02 +00002722 auto RefLoadLVal =
2723 emitLoadOfPointerLValue(CGF, RefLVal.getAddress(), RefLVal.getType());
2724 CGF.EmitStoreOfScalar(FieldLVal.getPointer(), RefLoadLVal);
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002725 ++Counter;
2726 }
2727 CGF.FinishFunction();
2728 return TaskPrivatesMap;
2729}
2730
Alexey Bataev9e034042015-05-05 04:05:12 +00002731static int array_pod_sort_comparator(const PrivateDataTy *P1,
2732 const PrivateDataTy *P2) {
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002733 return P1->first < P2->first ? 1 : (P2->first < P1->first ? -1 : 0);
2734}
2735
2736void CGOpenMPRuntime::emitTaskCall(
2737 CodeGenFunction &CGF, SourceLocation Loc, const OMPExecutableDirective &D,
2738 bool Tied, llvm::PointerIntPair<llvm::Value *, 1, bool> Final,
John McCall7f416cc2015-09-08 08:05:57 +00002739 llvm::Value *TaskFunction, QualType SharedsTy, Address Shareds,
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002740 const Expr *IfCond, ArrayRef<const Expr *> PrivateVars,
2741 ArrayRef<const Expr *> PrivateCopies,
2742 ArrayRef<const Expr *> FirstprivateVars,
2743 ArrayRef<const Expr *> FirstprivateCopies,
2744 ArrayRef<const Expr *> FirstprivateInits,
2745 ArrayRef<std::pair<OpenMPDependClauseKind, const Expr *>> Dependences) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00002746 if (!CGF.HaveInsertPoint())
2747 return;
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002748 auto &C = CGM.getContext();
Alexey Bataev9e034042015-05-05 04:05:12 +00002749 llvm::SmallVector<PrivateDataTy, 8> Privates;
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002750 // Aggregate privates and sort them by the alignment.
Alexey Bataev9e034042015-05-05 04:05:12 +00002751 auto I = PrivateCopies.begin();
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002752 for (auto *E : PrivateVars) {
2753 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
2754 Privates.push_back(std::make_pair(
Alexey Bataevc71a4092015-09-11 10:29:41 +00002755 C.getDeclAlign(VD),
Alexey Bataev9e034042015-05-05 04:05:12 +00002756 PrivateHelpersTy(VD, cast<VarDecl>(cast<DeclRefExpr>(*I)->getDecl()),
2757 /*PrivateElemInit=*/nullptr)));
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002758 ++I;
2759 }
Alexey Bataev9e034042015-05-05 04:05:12 +00002760 I = FirstprivateCopies.begin();
2761 auto IElemInitRef = FirstprivateInits.begin();
2762 for (auto *E : FirstprivateVars) {
2763 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(E)->getDecl());
2764 Privates.push_back(std::make_pair(
Alexey Bataevc71a4092015-09-11 10:29:41 +00002765 C.getDeclAlign(VD),
Alexey Bataev9e034042015-05-05 04:05:12 +00002766 PrivateHelpersTy(
2767 VD, cast<VarDecl>(cast<DeclRefExpr>(*I)->getDecl()),
2768 cast<VarDecl>(cast<DeclRefExpr>(*IElemInitRef)->getDecl()))));
2769 ++I, ++IElemInitRef;
2770 }
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002771 llvm::array_pod_sort(Privates.begin(), Privates.end(),
2772 array_pod_sort_comparator);
Alexey Bataev62b63b12015-03-10 07:28:44 +00002773 auto KmpInt32Ty = C.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/1);
2774 // Build type kmp_routine_entry_t (if not built yet).
2775 emitKmpRoutineEntryT(KmpInt32Ty);
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002776 // Build type kmp_task_t (if not built yet).
2777 if (KmpTaskTQTy.isNull()) {
2778 KmpTaskTQTy = C.getRecordType(
2779 createKmpTaskTRecordDecl(CGM, KmpInt32Ty, KmpRoutineEntryPtrQTy));
2780 }
2781 auto *KmpTaskTQTyRD = cast<RecordDecl>(KmpTaskTQTy->getAsTagDecl());
Alexey Bataev62b63b12015-03-10 07:28:44 +00002782 // Build particular struct kmp_task_t for the given task.
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002783 auto *KmpTaskTWithPrivatesQTyRD =
2784 createKmpTaskTWithPrivatesRecordDecl(CGM, KmpTaskTQTy, Privates);
2785 auto KmpTaskTWithPrivatesQTy = C.getRecordType(KmpTaskTWithPrivatesQTyRD);
2786 QualType KmpTaskTWithPrivatesPtrQTy =
2787 C.getPointerType(KmpTaskTWithPrivatesQTy);
2788 auto *KmpTaskTWithPrivatesTy = CGF.ConvertType(KmpTaskTWithPrivatesQTy);
2789 auto *KmpTaskTWithPrivatesPtrTy = KmpTaskTWithPrivatesTy->getPointerTo();
Alexey Bataev1189bd02016-01-26 12:20:39 +00002790 auto *KmpTaskTWithPrivatesTySize = CGF.getTypeSize(KmpTaskTWithPrivatesQTy);
Alexey Bataev62b63b12015-03-10 07:28:44 +00002791 QualType SharedsPtrTy = C.getPointerType(SharedsTy);
2792
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002793 // Emit initial values for private copies (if any).
2794 llvm::Value *TaskPrivatesMap = nullptr;
2795 auto *TaskPrivatesMapTy =
2796 std::next(cast<llvm::Function>(TaskFunction)->getArgumentList().begin(),
2797 3)
2798 ->getType();
2799 if (!Privates.empty()) {
2800 auto FI = std::next(KmpTaskTWithPrivatesQTyRD->field_begin());
2801 TaskPrivatesMap = emitTaskPrivateMappingFunction(
2802 CGM, Loc, PrivateVars, FirstprivateVars, FI->getType(), Privates);
2803 TaskPrivatesMap = CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
2804 TaskPrivatesMap, TaskPrivatesMapTy);
2805 } else {
2806 TaskPrivatesMap = llvm::ConstantPointerNull::get(
2807 cast<llvm::PointerType>(TaskPrivatesMapTy));
2808 }
Alexey Bataev62b63b12015-03-10 07:28:44 +00002809 // Build a proxy function kmp_int32 .omp_task_entry.(kmp_int32 gtid,
2810 // kmp_task_t *tt);
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002811 auto *TaskEntry = emitProxyTaskFunction(
2812 CGM, Loc, KmpInt32Ty, KmpTaskTWithPrivatesPtrQTy, KmpTaskTWithPrivatesQTy,
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002813 KmpTaskTQTy, SharedsPtrTy, TaskFunction, TaskPrivatesMap);
Alexey Bataev62b63b12015-03-10 07:28:44 +00002814
2815 // Build call kmp_task_t * __kmpc_omp_task_alloc(ident_t *, kmp_int32 gtid,
2816 // kmp_int32 flags, size_t sizeof_kmp_task_t, size_t sizeof_shareds,
2817 // kmp_routine_entry_t *task_entry);
2818 // Task flags. Format is taken from
2819 // http://llvm.org/svn/llvm-project/openmp/trunk/runtime/src/kmp.h,
2820 // description of kmp_tasking_flags struct.
2821 const unsigned TiedFlag = 0x1;
2822 const unsigned FinalFlag = 0x2;
2823 unsigned Flags = Tied ? TiedFlag : 0;
2824 auto *TaskFlags =
2825 Final.getPointer()
2826 ? CGF.Builder.CreateSelect(Final.getPointer(),
2827 CGF.Builder.getInt32(FinalFlag),
2828 CGF.Builder.getInt32(/*C=*/0))
2829 : CGF.Builder.getInt32(Final.getInt() ? FinalFlag : 0);
2830 TaskFlags = CGF.Builder.CreateOr(TaskFlags, CGF.Builder.getInt32(Flags));
Alexey Bataev40e36f12015-11-24 13:01:44 +00002831 auto *SharedsSize = CGM.getSize(C.getTypeSizeInChars(SharedsTy));
Alexey Bataevf24e7b12015-10-08 09:10:53 +00002832 llvm::Value *AllocArgs[] = {emitUpdateLocation(CGF, Loc),
2833 getThreadID(CGF, Loc), TaskFlags,
2834 KmpTaskTWithPrivatesTySize, SharedsSize,
2835 CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
2836 TaskEntry, KmpRoutineEntryPtrTy)};
Alexey Bataev62b63b12015-03-10 07:28:44 +00002837 auto *NewTask = CGF.EmitRuntimeCall(
2838 createRuntimeFunction(OMPRTL__kmpc_omp_task_alloc), AllocArgs);
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002839 auto *NewTaskNewTaskTTy = CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
2840 NewTask, KmpTaskTWithPrivatesPtrTy);
2841 LValue Base = CGF.MakeNaturalAlignAddrLValue(NewTaskNewTaskTTy,
2842 KmpTaskTWithPrivatesQTy);
2843 LValue TDBase =
2844 CGF.EmitLValueForField(Base, *KmpTaskTWithPrivatesQTyRD->field_begin());
Alexey Bataev62b63b12015-03-10 07:28:44 +00002845 // Fill the data in the resulting kmp_task_t record.
2846 // Copy shareds if there are any.
John McCall7f416cc2015-09-08 08:05:57 +00002847 Address KmpTaskSharedsPtr = Address::invalid();
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002848 if (!SharedsTy->getAsStructureType()->getDecl()->field_empty()) {
Alexey Bataev2377fe92015-09-10 08:12:02 +00002849 KmpTaskSharedsPtr =
2850 Address(CGF.EmitLoadOfScalar(
2851 CGF.EmitLValueForField(
2852 TDBase, *std::next(KmpTaskTQTyRD->field_begin(),
2853 KmpTaskTShareds)),
2854 Loc),
2855 CGF.getNaturalTypeAlignment(SharedsTy));
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002856 CGF.EmitAggregateCopy(KmpTaskSharedsPtr, Shareds, SharedsTy);
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002857 }
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002858 // Emit initial values for private copies (if any).
2859 bool NeedsCleanup = false;
2860 if (!Privates.empty()) {
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002861 auto FI = std::next(KmpTaskTWithPrivatesQTyRD->field_begin());
2862 auto PrivatesBase = CGF.EmitLValueForField(Base, *FI);
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002863 FI = cast<RecordDecl>(FI->getType()->getAsTagDecl())->field_begin();
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002864 LValue SharedsBase;
2865 if (!FirstprivateVars.empty()) {
John McCall7f416cc2015-09-08 08:05:57 +00002866 SharedsBase = CGF.MakeAddrLValue(
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002867 CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
2868 KmpTaskSharedsPtr, CGF.ConvertTypeForMem(SharedsPtrTy)),
2869 SharedsTy);
2870 }
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002871 CodeGenFunction::CGCapturedStmtInfo CapturesInfo(
2872 cast<CapturedStmt>(*D.getAssociatedStmt()));
2873 for (auto &&Pair : Privates) {
Alexey Bataev9e034042015-05-05 04:05:12 +00002874 auto *VD = Pair.second.PrivateCopy;
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002875 auto *Init = VD->getAnyInitializer();
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002876 LValue PrivateLValue = CGF.EmitLValueForField(PrivatesBase, *FI);
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002877 if (Init) {
Alexey Bataev9e034042015-05-05 04:05:12 +00002878 if (auto *Elem = Pair.second.PrivateElemInit) {
2879 auto *OriginalVD = Pair.second.Original;
2880 auto *SharedField = CapturesInfo.lookup(OriginalVD);
2881 auto SharedRefLValue =
2882 CGF.EmitLValueForField(SharedsBase, SharedField);
Alexey Bataevc71a4092015-09-11 10:29:41 +00002883 SharedRefLValue = CGF.MakeAddrLValue(
2884 Address(SharedRefLValue.getPointer(), C.getDeclAlign(OriginalVD)),
2885 SharedRefLValue.getType(), AlignmentSource::Decl);
Alexey Bataev1d9c15c2015-05-19 12:31:28 +00002886 QualType Type = OriginalVD->getType();
Alexey Bataev1d9c15c2015-05-19 12:31:28 +00002887 if (Type->isArrayType()) {
Alexey Bataev9e034042015-05-05 04:05:12 +00002888 // Initialize firstprivate array.
2889 if (!isa<CXXConstructExpr>(Init) ||
2890 CGF.isTrivialInitializer(Init)) {
2891 // Perform simple memcpy.
2892 CGF.EmitAggregateAssign(PrivateLValue.getAddress(),
Alexey Bataev1d9c15c2015-05-19 12:31:28 +00002893 SharedRefLValue.getAddress(), Type);
Alexey Bataev9e034042015-05-05 04:05:12 +00002894 } else {
2895 // Initialize firstprivate array using element-by-element
2896 // intialization.
2897 CGF.EmitOMPAggregateAssign(
2898 PrivateLValue.getAddress(), SharedRefLValue.getAddress(),
Alexey Bataev1d9c15c2015-05-19 12:31:28 +00002899 Type, [&CGF, Elem, Init, &CapturesInfo](
John McCall7f416cc2015-09-08 08:05:57 +00002900 Address DestElement, Address SrcElement) {
Alexey Bataev9e034042015-05-05 04:05:12 +00002901 // Clean up any temporaries needed by the initialization.
2902 CodeGenFunction::OMPPrivateScope InitScope(CGF);
John McCall7f416cc2015-09-08 08:05:57 +00002903 InitScope.addPrivate(Elem, [SrcElement]() -> Address {
Alexey Bataev9e034042015-05-05 04:05:12 +00002904 return SrcElement;
2905 });
2906 (void)InitScope.Privatize();
2907 // Emit initialization for single element.
Alexey Bataevd157d472015-06-24 03:35:38 +00002908 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(
2909 CGF, &CapturesInfo);
Alexey Bataev9e034042015-05-05 04:05:12 +00002910 CGF.EmitAnyExprToMem(Init, DestElement,
2911 Init->getType().getQualifiers(),
2912 /*IsInitializer=*/false);
Alexey Bataev9e034042015-05-05 04:05:12 +00002913 });
2914 }
2915 } else {
2916 CodeGenFunction::OMPPrivateScope InitScope(CGF);
John McCall7f416cc2015-09-08 08:05:57 +00002917 InitScope.addPrivate(Elem, [SharedRefLValue]() -> Address {
Alexey Bataev9e034042015-05-05 04:05:12 +00002918 return SharedRefLValue.getAddress();
2919 });
2920 (void)InitScope.Privatize();
Alexey Bataevd157d472015-06-24 03:35:38 +00002921 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(CGF, &CapturesInfo);
Alexey Bataev9e034042015-05-05 04:05:12 +00002922 CGF.EmitExprAsInit(Init, VD, PrivateLValue,
2923 /*capturedByInit=*/false);
Alexey Bataev9e034042015-05-05 04:05:12 +00002924 }
2925 } else {
2926 CGF.EmitExprAsInit(Init, VD, PrivateLValue, /*capturedByInit=*/false);
2927 }
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002928 }
2929 NeedsCleanup = NeedsCleanup || FI->getType().isDestructedType();
Alexey Bataev3ae88e22015-05-22 08:56:35 +00002930 ++FI;
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002931 }
2932 }
Alexey Bataev62b63b12015-03-10 07:28:44 +00002933 // Provide pointer to function with destructors for privates.
Alexey Bataev36c1eb92015-04-30 06:51:57 +00002934 llvm::Value *DestructorFn =
Alexey Bataev8fc69dc2015-05-18 07:54:53 +00002935 NeedsCleanup ? emitDestructorsFunction(CGM, Loc, KmpInt32Ty,
2936 KmpTaskTWithPrivatesPtrQTy,
2937 KmpTaskTWithPrivatesQTy)
2938 : llvm::ConstantPointerNull::get(
2939 cast<llvm::PointerType>(KmpRoutineEntryPtrTy));
2940 LValue Destructor = CGF.EmitLValueForField(
2941 TDBase, *std::next(KmpTaskTQTyRD->field_begin(), KmpTaskTDestructors));
2942 CGF.EmitStoreOfScalar(CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
2943 DestructorFn, KmpRoutineEntryPtrTy),
2944 Destructor);
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002945
2946 // Process list of dependences.
John McCall7f416cc2015-09-08 08:05:57 +00002947 Address DependenciesArray = Address::invalid();
2948 unsigned NumDependencies = Dependences.size();
2949 if (NumDependencies) {
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002950 // Dependence kind for RTL.
Alexey Bataev92e82f92015-11-23 13:33:42 +00002951 enum RTLDependenceKindTy { DepIn = 0x01, DepInOut = 0x3 };
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002952 enum RTLDependInfoFieldsTy { BaseAddr, Len, Flags };
2953 RecordDecl *KmpDependInfoRD;
Alexey Bataevf24e7b12015-10-08 09:10:53 +00002954 QualType FlagsTy =
2955 C.getIntTypeForBitwidth(C.getTypeSize(C.BoolTy), /*Signed=*/false);
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002956 llvm::Type *LLVMFlagsTy = CGF.ConvertTypeForMem(FlagsTy);
2957 if (KmpDependInfoTy.isNull()) {
2958 KmpDependInfoRD = C.buildImplicitRecord("kmp_depend_info");
2959 KmpDependInfoRD->startDefinition();
2960 addFieldToRecordDecl(C, KmpDependInfoRD, C.getIntPtrType());
2961 addFieldToRecordDecl(C, KmpDependInfoRD, C.getSizeType());
2962 addFieldToRecordDecl(C, KmpDependInfoRD, FlagsTy);
2963 KmpDependInfoRD->completeDefinition();
2964 KmpDependInfoTy = C.getRecordType(KmpDependInfoRD);
2965 } else {
2966 KmpDependInfoRD = cast<RecordDecl>(KmpDependInfoTy->getAsTagDecl());
2967 }
John McCall7f416cc2015-09-08 08:05:57 +00002968 CharUnits DependencySize = C.getTypeSizeInChars(KmpDependInfoTy);
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002969 // Define type kmp_depend_info[<Dependences.size()>];
2970 QualType KmpDependInfoArrayTy = C.getConstantArrayType(
John McCall7f416cc2015-09-08 08:05:57 +00002971 KmpDependInfoTy, llvm::APInt(/*numBits=*/64, NumDependencies),
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002972 ArrayType::Normal, /*IndexTypeQuals=*/0);
2973 // kmp_depend_info[<Dependences.size()>] deps;
John McCall7f416cc2015-09-08 08:05:57 +00002974 DependenciesArray = CGF.CreateMemTemp(KmpDependInfoArrayTy);
2975 for (unsigned i = 0; i < NumDependencies; ++i) {
2976 const Expr *E = Dependences[i].second;
2977 auto Addr = CGF.EmitLValue(E);
Alexey Bataevd6fdc8b2015-08-31 07:32:19 +00002978 llvm::Value *Size;
2979 QualType Ty = E->getType();
Alexey Bataevd6fdc8b2015-08-31 07:32:19 +00002980 if (auto *ASE = dyn_cast<OMPArraySectionExpr>(E->IgnoreParenImpCasts())) {
2981 LValue UpAddrLVal =
2982 CGF.EmitOMPArraySectionExpr(ASE, /*LowerBound=*/false);
2983 llvm::Value *UpAddr =
John McCall7f416cc2015-09-08 08:05:57 +00002984 CGF.Builder.CreateConstGEP1_32(UpAddrLVal.getPointer(), /*Idx0=*/1);
Alexey Bataevd6fdc8b2015-08-31 07:32:19 +00002985 llvm::Value *LowIntPtr =
John McCall7f416cc2015-09-08 08:05:57 +00002986 CGF.Builder.CreatePtrToInt(Addr.getPointer(), CGM.SizeTy);
Alexey Bataevd6fdc8b2015-08-31 07:32:19 +00002987 llvm::Value *UpIntPtr = CGF.Builder.CreatePtrToInt(UpAddr, CGM.SizeTy);
2988 Size = CGF.Builder.CreateNUWSub(UpIntPtr, LowIntPtr);
Alexey Bataevf24e7b12015-10-08 09:10:53 +00002989 } else
Alexey Bataev1189bd02016-01-26 12:20:39 +00002990 Size = CGF.getTypeSize(Ty);
John McCall7f416cc2015-09-08 08:05:57 +00002991 auto Base = CGF.MakeAddrLValue(
2992 CGF.Builder.CreateConstArrayGEP(DependenciesArray, i, DependencySize),
Alexey Bataev1d2353d2015-06-24 11:01:36 +00002993 KmpDependInfoTy);
2994 // deps[i].base_addr = &<Dependences[i].second>;
2995 auto BaseAddrLVal = CGF.EmitLValueForField(
2996 Base, *std::next(KmpDependInfoRD->field_begin(), BaseAddr));
John McCall7f416cc2015-09-08 08:05:57 +00002997 CGF.EmitStoreOfScalar(
2998 CGF.Builder.CreatePtrToInt(Addr.getPointer(), CGF.IntPtrTy),
2999 BaseAddrLVal);
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003000 // deps[i].len = sizeof(<Dependences[i].second>);
3001 auto LenLVal = CGF.EmitLValueForField(
3002 Base, *std::next(KmpDependInfoRD->field_begin(), Len));
3003 CGF.EmitStoreOfScalar(Size, LenLVal);
3004 // deps[i].flags = <Dependences[i].first>;
3005 RTLDependenceKindTy DepKind;
3006 switch (Dependences[i].first) {
3007 case OMPC_DEPEND_in:
3008 DepKind = DepIn;
3009 break;
Alexey Bataev92e82f92015-11-23 13:33:42 +00003010 // Out and InOut dependencies must use the same code.
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003011 case OMPC_DEPEND_out:
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003012 case OMPC_DEPEND_inout:
3013 DepKind = DepInOut;
3014 break;
Alexey Bataeveb482352015-12-18 05:05:56 +00003015 case OMPC_DEPEND_source:
Alexey Bataeva636c7f2015-12-23 10:27:45 +00003016 case OMPC_DEPEND_sink:
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003017 case OMPC_DEPEND_unknown:
3018 llvm_unreachable("Unknown task dependence type");
3019 }
3020 auto FlagsLVal = CGF.EmitLValueForField(
3021 Base, *std::next(KmpDependInfoRD->field_begin(), Flags));
3022 CGF.EmitStoreOfScalar(llvm::ConstantInt::get(LLVMFlagsTy, DepKind),
3023 FlagsLVal);
3024 }
John McCall7f416cc2015-09-08 08:05:57 +00003025 DependenciesArray = CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
3026 CGF.Builder.CreateStructGEP(DependenciesArray, 0, CharUnits::Zero()),
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003027 CGF.VoidPtrTy);
3028 }
3029
Alexey Bataev62b63b12015-03-10 07:28:44 +00003030 // NOTE: routine and part_id fields are intialized by __kmpc_omp_task_alloc()
3031 // libcall.
3032 // Build kmp_int32 __kmpc_omp_task(ident_t *, kmp_int32 gtid, kmp_task_t
3033 // *new_task);
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003034 // Build kmp_int32 __kmpc_omp_task_with_deps(ident_t *, kmp_int32 gtid,
3035 // kmp_task_t *new_task, kmp_int32 ndeps, kmp_depend_info_t *dep_list,
3036 // kmp_int32 ndeps_noalias, kmp_depend_info_t *noalias_dep_list) if dependence
3037 // list is not empty
Alexey Bataev1d677132015-04-22 13:57:31 +00003038 auto *ThreadID = getThreadID(CGF, Loc);
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003039 auto *UpLoc = emitUpdateLocation(CGF, Loc);
John McCall7f416cc2015-09-08 08:05:57 +00003040 llvm::Value *TaskArgs[] = { UpLoc, ThreadID, NewTask };
3041 llvm::Value *DepTaskArgs[7];
3042 if (NumDependencies) {
3043 DepTaskArgs[0] = UpLoc;
3044 DepTaskArgs[1] = ThreadID;
3045 DepTaskArgs[2] = NewTask;
3046 DepTaskArgs[3] = CGF.Builder.getInt32(NumDependencies);
3047 DepTaskArgs[4] = DependenciesArray.getPointer();
3048 DepTaskArgs[5] = CGF.Builder.getInt32(0);
3049 DepTaskArgs[6] = llvm::ConstantPointerNull::get(CGF.VoidPtrTy);
3050 }
3051 auto &&ThenCodeGen = [this, NumDependencies,
3052 &TaskArgs, &DepTaskArgs](CodeGenFunction &CGF) {
3053 // TODO: add check for untied tasks.
3054 if (NumDependencies) {
3055 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_omp_task_with_deps),
3056 DepTaskArgs);
3057 } else {
3058 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_omp_task),
3059 TaskArgs);
3060 }
Alexey Bataev1d677132015-04-22 13:57:31 +00003061 };
Alexey Bataeva744ff52015-05-05 09:24:37 +00003062 typedef CallEndCleanup<std::extent<decltype(TaskArgs)>::value>
3063 IfCallEndCleanup;
John McCall7f416cc2015-09-08 08:05:57 +00003064
3065 llvm::Value *DepWaitTaskArgs[6];
3066 if (NumDependencies) {
3067 DepWaitTaskArgs[0] = UpLoc;
3068 DepWaitTaskArgs[1] = ThreadID;
3069 DepWaitTaskArgs[2] = CGF.Builder.getInt32(NumDependencies);
3070 DepWaitTaskArgs[3] = DependenciesArray.getPointer();
3071 DepWaitTaskArgs[4] = CGF.Builder.getInt32(0);
3072 DepWaitTaskArgs[5] = llvm::ConstantPointerNull::get(CGF.VoidPtrTy);
3073 }
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003074 auto &&ElseCodeGen = [this, &TaskArgs, ThreadID, NewTaskNewTaskTTy, TaskEntry,
John McCall7f416cc2015-09-08 08:05:57 +00003075 NumDependencies, &DepWaitTaskArgs](CodeGenFunction &CGF) {
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003076 CodeGenFunction::RunCleanupsScope LocalScope(CGF);
3077 // Build void __kmpc_omp_wait_deps(ident_t *, kmp_int32 gtid,
3078 // kmp_int32 ndeps, kmp_depend_info_t *dep_list, kmp_int32
3079 // ndeps_noalias, kmp_depend_info_t *noalias_dep_list); if dependence info
3080 // is specified.
John McCall7f416cc2015-09-08 08:05:57 +00003081 if (NumDependencies)
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003082 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_omp_wait_deps),
3083 DepWaitTaskArgs);
3084 // Build void __kmpc_omp_task_begin_if0(ident_t *, kmp_int32 gtid,
3085 // kmp_task_t *new_task);
3086 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_omp_task_begin_if0),
3087 TaskArgs);
3088 // Build void __kmpc_omp_task_complete_if0(ident_t *, kmp_int32 gtid,
3089 // kmp_task_t *new_task);
3090 CGF.EHStack.pushCleanup<IfCallEndCleanup>(
3091 NormalAndEHCleanup,
3092 createRuntimeFunction(OMPRTL__kmpc_omp_task_complete_if0),
3093 llvm::makeArrayRef(TaskArgs));
Alexey Bataev1d677132015-04-22 13:57:31 +00003094
Alexey Bataev1d2353d2015-06-24 11:01:36 +00003095 // Call proxy_task_entry(gtid, new_task);
3096 llvm::Value *OutlinedFnArgs[] = {ThreadID, NewTaskNewTaskTTy};
3097 CGF.EmitCallOrInvoke(TaskEntry, OutlinedFnArgs);
3098 };
John McCall7f416cc2015-09-08 08:05:57 +00003099
Alexey Bataev1d677132015-04-22 13:57:31 +00003100 if (IfCond) {
3101 emitOMPIfClause(CGF, IfCond, ThenCodeGen, ElseCodeGen);
3102 } else {
3103 CodeGenFunction::RunCleanupsScope Scope(CGF);
3104 ThenCodeGen(CGF);
3105 }
Alexey Bataev62b63b12015-03-10 07:28:44 +00003106}
3107
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003108/// \brief Emit reduction operation for each element of array (required for
3109/// array sections) LHS op = RHS.
3110/// \param Type Type of array.
3111/// \param LHSVar Variable on the left side of the reduction operation
3112/// (references element of array in original variable).
3113/// \param RHSVar Variable on the right side of the reduction operation
3114/// (references element of array in original variable).
3115/// \param RedOpGen Generator of reduction operation with use of LHSVar and
3116/// RHSVar.
Benjamin Kramere003ca22015-10-28 13:54:16 +00003117static void EmitOMPAggregateReduction(
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003118 CodeGenFunction &CGF, QualType Type, const VarDecl *LHSVar,
3119 const VarDecl *RHSVar,
3120 const llvm::function_ref<void(CodeGenFunction &CGF, const Expr *,
3121 const Expr *, const Expr *)> &RedOpGen,
3122 const Expr *XExpr = nullptr, const Expr *EExpr = nullptr,
3123 const Expr *UpExpr = nullptr) {
3124 // Perform element-by-element initialization.
3125 QualType ElementTy;
3126 Address LHSAddr = CGF.GetAddrOfLocalVar(LHSVar);
3127 Address RHSAddr = CGF.GetAddrOfLocalVar(RHSVar);
3128
3129 // Drill down to the base element type on both arrays.
3130 auto ArrayTy = Type->getAsArrayTypeUnsafe();
3131 auto NumElements = CGF.emitArrayLength(ArrayTy, ElementTy, LHSAddr);
3132
3133 auto RHSBegin = RHSAddr.getPointer();
3134 auto LHSBegin = LHSAddr.getPointer();
3135 // Cast from pointer to array type to pointer to single element.
3136 auto LHSEnd = CGF.Builder.CreateGEP(LHSBegin, NumElements);
3137 // The basic structure here is a while-do loop.
3138 auto BodyBB = CGF.createBasicBlock("omp.arraycpy.body");
3139 auto DoneBB = CGF.createBasicBlock("omp.arraycpy.done");
3140 auto IsEmpty =
3141 CGF.Builder.CreateICmpEQ(LHSBegin, LHSEnd, "omp.arraycpy.isempty");
3142 CGF.Builder.CreateCondBr(IsEmpty, DoneBB, BodyBB);
3143
3144 // Enter the loop body, making that address the current address.
3145 auto EntryBB = CGF.Builder.GetInsertBlock();
3146 CGF.EmitBlock(BodyBB);
3147
3148 CharUnits ElementSize = CGF.getContext().getTypeSizeInChars(ElementTy);
3149
3150 llvm::PHINode *RHSElementPHI = CGF.Builder.CreatePHI(
3151 RHSBegin->getType(), 2, "omp.arraycpy.srcElementPast");
3152 RHSElementPHI->addIncoming(RHSBegin, EntryBB);
3153 Address RHSElementCurrent =
3154 Address(RHSElementPHI,
3155 RHSAddr.getAlignment().alignmentOfArrayElement(ElementSize));
3156
3157 llvm::PHINode *LHSElementPHI = CGF.Builder.CreatePHI(
3158 LHSBegin->getType(), 2, "omp.arraycpy.destElementPast");
3159 LHSElementPHI->addIncoming(LHSBegin, EntryBB);
3160 Address LHSElementCurrent =
3161 Address(LHSElementPHI,
3162 LHSAddr.getAlignment().alignmentOfArrayElement(ElementSize));
3163
3164 // Emit copy.
3165 CodeGenFunction::OMPPrivateScope Scope(CGF);
3166 Scope.addPrivate(LHSVar, [=]() -> Address { return LHSElementCurrent; });
3167 Scope.addPrivate(RHSVar, [=]() -> Address { return RHSElementCurrent; });
3168 Scope.Privatize();
3169 RedOpGen(CGF, XExpr, EExpr, UpExpr);
3170 Scope.ForceCleanup();
3171
3172 // Shift the address forward by one element.
3173 auto LHSElementNext = CGF.Builder.CreateConstGEP1_32(
3174 LHSElementPHI, /*Idx0=*/1, "omp.arraycpy.dest.element");
3175 auto RHSElementNext = CGF.Builder.CreateConstGEP1_32(
3176 RHSElementPHI, /*Idx0=*/1, "omp.arraycpy.src.element");
3177 // Check whether we've reached the end.
3178 auto Done =
3179 CGF.Builder.CreateICmpEQ(LHSElementNext, LHSEnd, "omp.arraycpy.done");
3180 CGF.Builder.CreateCondBr(Done, DoneBB, BodyBB);
3181 LHSElementPHI->addIncoming(LHSElementNext, CGF.Builder.GetInsertBlock());
3182 RHSElementPHI->addIncoming(RHSElementNext, CGF.Builder.GetInsertBlock());
3183
3184 // Done.
3185 CGF.EmitBlock(DoneBB, /*IsFinished=*/true);
3186}
3187
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003188static llvm::Value *emitReductionFunction(CodeGenModule &CGM,
3189 llvm::Type *ArgsType,
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003190 ArrayRef<const Expr *> Privates,
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003191 ArrayRef<const Expr *> LHSExprs,
3192 ArrayRef<const Expr *> RHSExprs,
3193 ArrayRef<const Expr *> ReductionOps) {
3194 auto &C = CGM.getContext();
3195
3196 // void reduction_func(void *LHSArg, void *RHSArg);
3197 FunctionArgList Args;
3198 ImplicitParamDecl LHSArg(C, /*DC=*/nullptr, SourceLocation(), /*Id=*/nullptr,
3199 C.VoidPtrTy);
3200 ImplicitParamDecl RHSArg(C, /*DC=*/nullptr, SourceLocation(), /*Id=*/nullptr,
3201 C.VoidPtrTy);
3202 Args.push_back(&LHSArg);
3203 Args.push_back(&RHSArg);
3204 FunctionType::ExtInfo EI;
3205 auto &CGFI = CGM.getTypes().arrangeFreeFunctionDeclaration(
3206 C.VoidTy, Args, EI, /*isVariadic=*/false);
3207 auto *Fn = llvm::Function::Create(
3208 CGM.getTypes().GetFunctionType(CGFI), llvm::GlobalValue::InternalLinkage,
3209 ".omp.reduction.reduction_func", &CGM.getModule());
Akira Hatanaka44a59f82015-10-28 02:30:47 +00003210 CGM.SetInternalFunctionAttributes(/*D=*/nullptr, Fn, CGFI);
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003211 CodeGenFunction CGF(CGM);
3212 CGF.StartFunction(GlobalDecl(), C.VoidTy, Fn, CGFI, Args);
3213
3214 // Dst = (void*[n])(LHSArg);
3215 // Src = (void*[n])(RHSArg);
John McCall7f416cc2015-09-08 08:05:57 +00003216 Address LHS(CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
3217 CGF.Builder.CreateLoad(CGF.GetAddrOfLocalVar(&LHSArg)),
3218 ArgsType), CGF.getPointerAlign());
3219 Address RHS(CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
3220 CGF.Builder.CreateLoad(CGF.GetAddrOfLocalVar(&RHSArg)),
3221 ArgsType), CGF.getPointerAlign());
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003222
3223 // ...
3224 // *(Type<i>*)lhs[i] = RedOp<i>(*(Type<i>*)lhs[i], *(Type<i>*)rhs[i]);
3225 // ...
3226 CodeGenFunction::OMPPrivateScope Scope(CGF);
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003227 auto IPriv = Privates.begin();
3228 unsigned Idx = 0;
3229 for (unsigned I = 0, E = ReductionOps.size(); I < E; ++I, ++IPriv, ++Idx) {
John McCall7f416cc2015-09-08 08:05:57 +00003230 auto RHSVar = cast<VarDecl>(cast<DeclRefExpr>(RHSExprs[I])->getDecl());
3231 Scope.addPrivate(RHSVar, [&]() -> Address {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003232 return emitAddrOfVarFromArray(CGF, RHS, Idx, RHSVar);
John McCall7f416cc2015-09-08 08:05:57 +00003233 });
3234 auto LHSVar = cast<VarDecl>(cast<DeclRefExpr>(LHSExprs[I])->getDecl());
3235 Scope.addPrivate(LHSVar, [&]() -> Address {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003236 return emitAddrOfVarFromArray(CGF, LHS, Idx, LHSVar);
John McCall7f416cc2015-09-08 08:05:57 +00003237 });
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003238 QualType PrivTy = (*IPriv)->getType();
Alexey Bataev1189bd02016-01-26 12:20:39 +00003239 if (PrivTy->isVariablyModifiedType()) {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003240 // Get array size and emit VLA type.
3241 ++Idx;
3242 Address Elem =
3243 CGF.Builder.CreateConstArrayGEP(LHS, Idx, CGF.getPointerSize());
3244 llvm::Value *Ptr = CGF.Builder.CreateLoad(Elem);
Alexey Bataev1189bd02016-01-26 12:20:39 +00003245 auto *VLA = CGF.getContext().getAsVariableArrayType(PrivTy);
3246 auto *OVE = cast<OpaqueValueExpr>(VLA->getSizeExpr());
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003247 CodeGenFunction::OpaqueValueMapping OpaqueMap(
Alexey Bataev1189bd02016-01-26 12:20:39 +00003248 CGF, OVE, RValue::get(CGF.Builder.CreatePtrToInt(Ptr, CGF.SizeTy)));
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003249 CGF.EmitVariablyModifiedType(PrivTy);
3250 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003251 }
3252 Scope.Privatize();
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003253 IPriv = Privates.begin();
3254 auto ILHS = LHSExprs.begin();
3255 auto IRHS = RHSExprs.begin();
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003256 for (auto *E : ReductionOps) {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003257 if ((*IPriv)->getType()->isArrayType()) {
3258 // Emit reduction for array section.
3259 auto *LHSVar = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
3260 auto *RHSVar = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
3261 EmitOMPAggregateReduction(CGF, (*IPriv)->getType(), LHSVar, RHSVar,
3262 [=](CodeGenFunction &CGF, const Expr *,
3263 const Expr *,
3264 const Expr *) { CGF.EmitIgnoredExpr(E); });
3265 } else
3266 // Emit reduction for array subscript or single variable.
3267 CGF.EmitIgnoredExpr(E);
3268 ++IPriv, ++ILHS, ++IRHS;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003269 }
3270 Scope.ForceCleanup();
3271 CGF.FinishFunction();
3272 return Fn;
3273}
3274
3275void CGOpenMPRuntime::emitReduction(CodeGenFunction &CGF, SourceLocation Loc,
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003276 ArrayRef<const Expr *> Privates,
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003277 ArrayRef<const Expr *> LHSExprs,
3278 ArrayRef<const Expr *> RHSExprs,
3279 ArrayRef<const Expr *> ReductionOps,
Alexey Bataev89e7e8e2015-06-17 06:21:39 +00003280 bool WithNowait, bool SimpleReduction) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00003281 if (!CGF.HaveInsertPoint())
3282 return;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003283 // Next code should be emitted for reduction:
3284 //
3285 // static kmp_critical_name lock = { 0 };
3286 //
3287 // void reduce_func(void *lhs[<n>], void *rhs[<n>]) {
3288 // *(Type0*)lhs[0] = ReductionOperation0(*(Type0*)lhs[0], *(Type0*)rhs[0]);
3289 // ...
3290 // *(Type<n>-1*)lhs[<n>-1] = ReductionOperation<n>-1(*(Type<n>-1*)lhs[<n>-1],
3291 // *(Type<n>-1*)rhs[<n>-1]);
3292 // }
3293 //
3294 // ...
3295 // void *RedList[<n>] = {&<RHSExprs>[0], ..., &<RHSExprs>[<n>-1]};
3296 // switch (__kmpc_reduce{_nowait}(<loc>, <gtid>, <n>, sizeof(RedList),
3297 // RedList, reduce_func, &<lock>)) {
3298 // case 1:
3299 // ...
3300 // <LHSExprs>[i] = RedOp<i>(*<LHSExprs>[i], *<RHSExprs>[i]);
3301 // ...
3302 // __kmpc_end_reduce{_nowait}(<loc>, <gtid>, &<lock>);
3303 // break;
3304 // case 2:
3305 // ...
3306 // Atomic(<LHSExprs>[i] = RedOp<i>(*<LHSExprs>[i], *<RHSExprs>[i]));
3307 // ...
Alexey Bataev69a47792015-05-07 03:54:03 +00003308 // [__kmpc_end_reduce(<loc>, <gtid>, &<lock>);]
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003309 // break;
3310 // default:;
3311 // }
Alexey Bataev89e7e8e2015-06-17 06:21:39 +00003312 //
3313 // if SimpleReduction is true, only the next code is generated:
3314 // ...
3315 // <LHSExprs>[i] = RedOp<i>(*<LHSExprs>[i], *<RHSExprs>[i]);
3316 // ...
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003317
3318 auto &C = CGM.getContext();
3319
Alexey Bataev89e7e8e2015-06-17 06:21:39 +00003320 if (SimpleReduction) {
3321 CodeGenFunction::RunCleanupsScope Scope(CGF);
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003322 auto IPriv = Privates.begin();
3323 auto ILHS = LHSExprs.begin();
3324 auto IRHS = RHSExprs.begin();
Alexey Bataev89e7e8e2015-06-17 06:21:39 +00003325 for (auto *E : ReductionOps) {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003326 if ((*IPriv)->getType()->isArrayType()) {
3327 auto *LHSVar = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
3328 auto *RHSVar = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
3329 EmitOMPAggregateReduction(
3330 CGF, (*IPriv)->getType(), LHSVar, RHSVar,
3331 [=](CodeGenFunction &CGF, const Expr *, const Expr *,
3332 const Expr *) { CGF.EmitIgnoredExpr(E); });
3333 } else
3334 CGF.EmitIgnoredExpr(E);
3335 ++IPriv, ++ILHS, ++IRHS;
Alexey Bataev89e7e8e2015-06-17 06:21:39 +00003336 }
3337 return;
3338 }
3339
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003340 // 1. Build a list of reduction variables.
3341 // void *RedList[<n>] = {<ReductionVars>[0], ..., <ReductionVars>[<n>-1]};
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003342 auto Size = RHSExprs.size();
3343 for (auto *E : Privates) {
Alexey Bataev1189bd02016-01-26 12:20:39 +00003344 if (E->getType()->isVariablyModifiedType())
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003345 // Reserve place for array size.
3346 ++Size;
3347 }
3348 llvm::APInt ArraySize(/*unsigned int numBits=*/32, Size);
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003349 QualType ReductionArrayTy =
3350 C.getConstantArrayType(C.VoidPtrTy, ArraySize, ArrayType::Normal,
3351 /*IndexTypeQuals=*/0);
John McCall7f416cc2015-09-08 08:05:57 +00003352 Address ReductionList =
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003353 CGF.CreateMemTemp(ReductionArrayTy, ".omp.reduction.red_list");
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003354 auto IPriv = Privates.begin();
3355 unsigned Idx = 0;
3356 for (unsigned I = 0, E = RHSExprs.size(); I < E; ++I, ++IPriv, ++Idx) {
John McCall7f416cc2015-09-08 08:05:57 +00003357 Address Elem =
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003358 CGF.Builder.CreateConstArrayGEP(ReductionList, Idx, CGF.getPointerSize());
John McCall7f416cc2015-09-08 08:05:57 +00003359 CGF.Builder.CreateStore(
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003360 CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(
John McCall7f416cc2015-09-08 08:05:57 +00003361 CGF.EmitLValue(RHSExprs[I]).getPointer(), CGF.VoidPtrTy),
3362 Elem);
Alexey Bataev1189bd02016-01-26 12:20:39 +00003363 if ((*IPriv)->getType()->isVariablyModifiedType()) {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003364 // Store array size.
3365 ++Idx;
3366 Elem = CGF.Builder.CreateConstArrayGEP(ReductionList, Idx,
3367 CGF.getPointerSize());
Alexey Bataev1189bd02016-01-26 12:20:39 +00003368 llvm::Value *Size = CGF.Builder.CreateIntCast(
3369 CGF.getVLASize(
3370 CGF.getContext().getAsVariableArrayType((*IPriv)->getType()))
3371 .first,
3372 CGF.SizeTy, /*isSigned=*/false);
3373 CGF.Builder.CreateStore(CGF.Builder.CreateIntToPtr(Size, CGF.VoidPtrTy),
3374 Elem);
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003375 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003376 }
3377
3378 // 2. Emit reduce_func().
3379 auto *ReductionFn = emitReductionFunction(
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003380 CGM, CGF.ConvertTypeForMem(ReductionArrayTy)->getPointerTo(), Privates,
3381 LHSExprs, RHSExprs, ReductionOps);
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003382
3383 // 3. Create static kmp_critical_name lock = { 0 };
3384 auto *Lock = getCriticalRegionLock(".reduction");
3385
3386 // 4. Build res = __kmpc_reduce{_nowait}(<loc>, <gtid>, <n>, sizeof(RedList),
3387 // RedList, reduce_func, &<lock>);
3388 auto *IdentTLoc = emitUpdateLocation(
3389 CGF, Loc,
3390 static_cast<OpenMPLocationFlags>(OMP_IDENT_KMPC | OMP_ATOMIC_REDUCE));
3391 auto *ThreadId = getThreadID(CGF, Loc);
Alexey Bataev1189bd02016-01-26 12:20:39 +00003392 auto *ReductionArrayTySize = CGF.getTypeSize(ReductionArrayTy);
John McCall7f416cc2015-09-08 08:05:57 +00003393 auto *RL =
3394 CGF.Builder.CreatePointerBitCastOrAddrSpaceCast(ReductionList.getPointer(),
3395 CGF.VoidPtrTy);
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003396 llvm::Value *Args[] = {
3397 IdentTLoc, // ident_t *<loc>
3398 ThreadId, // i32 <gtid>
3399 CGF.Builder.getInt32(RHSExprs.size()), // i32 <n>
3400 ReductionArrayTySize, // size_type sizeof(RedList)
3401 RL, // void *RedList
3402 ReductionFn, // void (*) (void *, void *) <reduce_func>
3403 Lock // kmp_critical_name *&<lock>
3404 };
3405 auto Res = CGF.EmitRuntimeCall(
3406 createRuntimeFunction(WithNowait ? OMPRTL__kmpc_reduce_nowait
3407 : OMPRTL__kmpc_reduce),
3408 Args);
3409
3410 // 5. Build switch(res)
3411 auto *DefaultBB = CGF.createBasicBlock(".omp.reduction.default");
3412 auto *SwInst = CGF.Builder.CreateSwitch(Res, DefaultBB, /*NumCases=*/2);
3413
3414 // 6. Build case 1:
3415 // ...
3416 // <LHSExprs>[i] = RedOp<i>(*<LHSExprs>[i], *<RHSExprs>[i]);
3417 // ...
3418 // __kmpc_end_reduce{_nowait}(<loc>, <gtid>, &<lock>);
3419 // break;
3420 auto *Case1BB = CGF.createBasicBlock(".omp.reduction.case1");
3421 SwInst->addCase(CGF.Builder.getInt32(1), Case1BB);
3422 CGF.EmitBlock(Case1BB);
3423
3424 {
3425 CodeGenFunction::RunCleanupsScope Scope(CGF);
3426 // Add emission of __kmpc_end_reduce{_nowait}(<loc>, <gtid>, &<lock>);
3427 llvm::Value *EndArgs[] = {
3428 IdentTLoc, // ident_t *<loc>
3429 ThreadId, // i32 <gtid>
3430 Lock // kmp_critical_name *&<lock>
3431 };
Alexey Bataeva744ff52015-05-05 09:24:37 +00003432 CGF.EHStack
3433 .pushCleanup<CallEndCleanup<std::extent<decltype(EndArgs)>::value>>(
3434 NormalAndEHCleanup,
3435 createRuntimeFunction(WithNowait ? OMPRTL__kmpc_end_reduce_nowait
3436 : OMPRTL__kmpc_end_reduce),
3437 llvm::makeArrayRef(EndArgs));
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003438 auto IPriv = Privates.begin();
3439 auto ILHS = LHSExprs.begin();
3440 auto IRHS = RHSExprs.begin();
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003441 for (auto *E : ReductionOps) {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003442 if ((*IPriv)->getType()->isArrayType()) {
3443 // Emit reduction for array section.
3444 auto *LHSVar = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
3445 auto *RHSVar = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
3446 EmitOMPAggregateReduction(
3447 CGF, (*IPriv)->getType(), LHSVar, RHSVar,
3448 [=](CodeGenFunction &CGF, const Expr *, const Expr *,
3449 const Expr *) { CGF.EmitIgnoredExpr(E); });
3450 } else
3451 // Emit reduction for array subscript or single variable.
3452 CGF.EmitIgnoredExpr(E);
3453 ++IPriv, ++ILHS, ++IRHS;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003454 }
3455 }
3456
3457 CGF.EmitBranch(DefaultBB);
3458
3459 // 7. Build case 2:
3460 // ...
3461 // Atomic(<LHSExprs>[i] = RedOp<i>(*<LHSExprs>[i], *<RHSExprs>[i]));
3462 // ...
3463 // break;
3464 auto *Case2BB = CGF.createBasicBlock(".omp.reduction.case2");
3465 SwInst->addCase(CGF.Builder.getInt32(2), Case2BB);
3466 CGF.EmitBlock(Case2BB);
3467
3468 {
3469 CodeGenFunction::RunCleanupsScope Scope(CGF);
Alexey Bataev69a47792015-05-07 03:54:03 +00003470 if (!WithNowait) {
3471 // Add emission of __kmpc_end_reduce(<loc>, <gtid>, &<lock>);
3472 llvm::Value *EndArgs[] = {
3473 IdentTLoc, // ident_t *<loc>
3474 ThreadId, // i32 <gtid>
3475 Lock // kmp_critical_name *&<lock>
3476 };
3477 CGF.EHStack
3478 .pushCleanup<CallEndCleanup<std::extent<decltype(EndArgs)>::value>>(
3479 NormalAndEHCleanup,
3480 createRuntimeFunction(OMPRTL__kmpc_end_reduce),
3481 llvm::makeArrayRef(EndArgs));
3482 }
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003483 auto ILHS = LHSExprs.begin();
3484 auto IRHS = RHSExprs.begin();
3485 auto IPriv = Privates.begin();
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003486 for (auto *E : ReductionOps) {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003487 const Expr *XExpr = nullptr;
3488 const Expr *EExpr = nullptr;
3489 const Expr *UpExpr = nullptr;
3490 BinaryOperatorKind BO = BO_Comma;
3491 if (auto *BO = dyn_cast<BinaryOperator>(E)) {
3492 if (BO->getOpcode() == BO_Assign) {
3493 XExpr = BO->getLHS();
3494 UpExpr = BO->getRHS();
3495 }
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003496 }
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003497 // Try to emit update expression as a simple atomic.
3498 auto *RHSExpr = UpExpr;
3499 if (RHSExpr) {
3500 // Analyze RHS part of the whole expression.
3501 if (auto *ACO = dyn_cast<AbstractConditionalOperator>(
3502 RHSExpr->IgnoreParenImpCasts())) {
3503 // If this is a conditional operator, analyze its condition for
3504 // min/max reduction operator.
3505 RHSExpr = ACO->getCond();
3506 }
3507 if (auto *BORHS =
3508 dyn_cast<BinaryOperator>(RHSExpr->IgnoreParenImpCasts())) {
3509 EExpr = BORHS->getRHS();
3510 BO = BORHS->getOpcode();
3511 }
Alexey Bataev69a47792015-05-07 03:54:03 +00003512 }
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003513 if (XExpr) {
3514 auto *VD = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
3515 auto &&AtomicRedGen = [this, BO, VD, IPriv,
3516 Loc](CodeGenFunction &CGF, const Expr *XExpr,
3517 const Expr *EExpr, const Expr *UpExpr) {
3518 LValue X = CGF.EmitLValue(XExpr);
3519 RValue E;
3520 if (EExpr)
3521 E = CGF.EmitAnyExpr(EExpr);
3522 CGF.EmitOMPAtomicSimpleUpdateExpr(
3523 X, E, BO, /*IsXLHSInRHSPart=*/true, llvm::Monotonic, Loc,
Alexey Bataev8524d152016-01-21 12:35:58 +00003524 [&CGF, UpExpr, VD, IPriv, Loc](RValue XRValue) {
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003525 CodeGenFunction::OMPPrivateScope PrivateScope(CGF);
Alexey Bataev8524d152016-01-21 12:35:58 +00003526 PrivateScope.addPrivate(
3527 VD, [&CGF, VD, XRValue, Loc]() -> Address {
3528 Address LHSTemp = CGF.CreateMemTemp(VD->getType());
3529 CGF.emitOMPSimpleStore(
3530 CGF.MakeAddrLValue(LHSTemp, VD->getType()), XRValue,
3531 VD->getType().getNonReferenceType(), Loc);
3532 return LHSTemp;
3533 });
Alexey Bataevf24e7b12015-10-08 09:10:53 +00003534 (void)PrivateScope.Privatize();
3535 return CGF.EmitAnyExpr(UpExpr);
3536 });
3537 };
3538 if ((*IPriv)->getType()->isArrayType()) {
3539 // Emit atomic reduction for array section.
3540 auto *RHSVar = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
3541 EmitOMPAggregateReduction(CGF, (*IPriv)->getType(), VD, RHSVar,
3542 AtomicRedGen, XExpr, EExpr, UpExpr);
3543 } else
3544 // Emit atomic reduction for array subscript or single variable.
3545 AtomicRedGen(CGF, XExpr, EExpr, UpExpr);
3546 } else {
3547 // Emit as a critical region.
3548 auto &&CritRedGen = [this, E, Loc](CodeGenFunction &CGF, const Expr *,
3549 const Expr *, const Expr *) {
3550 emitCriticalRegion(
3551 CGF, ".atomic_reduction",
3552 [E](CodeGenFunction &CGF) { CGF.EmitIgnoredExpr(E); }, Loc);
3553 };
3554 if ((*IPriv)->getType()->isArrayType()) {
3555 auto *LHSVar = cast<VarDecl>(cast<DeclRefExpr>(*ILHS)->getDecl());
3556 auto *RHSVar = cast<VarDecl>(cast<DeclRefExpr>(*IRHS)->getDecl());
3557 EmitOMPAggregateReduction(CGF, (*IPriv)->getType(), LHSVar, RHSVar,
3558 CritRedGen);
3559 } else
3560 CritRedGen(CGF, nullptr, nullptr, nullptr);
3561 }
3562 ++ILHS, ++IRHS, ++IPriv;
Alexey Bataev794ba0d2015-04-10 10:43:45 +00003563 }
3564 }
3565
3566 CGF.EmitBranch(DefaultBB);
3567 CGF.EmitBlock(DefaultBB, /*IsFinished=*/true);
3568}
3569
Alexey Bataev8b8e2022015-04-27 05:22:09 +00003570void CGOpenMPRuntime::emitTaskwaitCall(CodeGenFunction &CGF,
3571 SourceLocation Loc) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00003572 if (!CGF.HaveInsertPoint())
3573 return;
Alexey Bataev8b8e2022015-04-27 05:22:09 +00003574 // Build call kmp_int32 __kmpc_omp_taskwait(ident_t *loc, kmp_int32
3575 // global_tid);
3576 llvm::Value *Args[] = {emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc)};
3577 // Ignore return result until untied tasks are supported.
3578 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_omp_taskwait), Args);
3579}
3580
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003581void CGOpenMPRuntime::emitInlinedDirective(CodeGenFunction &CGF,
Alexey Bataev81c7ea02015-07-03 09:56:58 +00003582 OpenMPDirectiveKind InnerKind,
Alexey Bataev25e5b442015-09-15 12:52:43 +00003583 const RegionCodeGenTy &CodeGen,
3584 bool HasCancel) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00003585 if (!CGF.HaveInsertPoint())
3586 return;
Alexey Bataev25e5b442015-09-15 12:52:43 +00003587 InlinedOpenMPRegionRAII Region(CGF, CodeGen, InnerKind, HasCancel);
Alexey Bataev6f1ffc02015-04-10 04:50:10 +00003588 CGF.CapturedStmtInfo->EmitBody(CGF, /*S=*/nullptr);
Alexey Bataev8cbe0a62015-02-26 10:27:34 +00003589}
3590
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003591namespace {
3592enum RTCancelKind {
3593 CancelNoreq = 0,
3594 CancelParallel = 1,
3595 CancelLoop = 2,
3596 CancelSections = 3,
3597 CancelTaskgroup = 4
3598};
3599}
3600
3601static RTCancelKind getCancellationKind(OpenMPDirectiveKind CancelRegion) {
3602 RTCancelKind CancelKind = CancelNoreq;
Alexey Bataev0f34da12015-07-02 04:17:07 +00003603 if (CancelRegion == OMPD_parallel)
3604 CancelKind = CancelParallel;
3605 else if (CancelRegion == OMPD_for)
3606 CancelKind = CancelLoop;
3607 else if (CancelRegion == OMPD_sections)
3608 CancelKind = CancelSections;
3609 else {
3610 assert(CancelRegion == OMPD_taskgroup);
3611 CancelKind = CancelTaskgroup;
3612 }
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003613 return CancelKind;
3614}
3615
3616void CGOpenMPRuntime::emitCancellationPointCall(
3617 CodeGenFunction &CGF, SourceLocation Loc,
3618 OpenMPDirectiveKind CancelRegion) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00003619 if (!CGF.HaveInsertPoint())
3620 return;
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003621 // Build call kmp_int32 __kmpc_cancellationpoint(ident_t *loc, kmp_int32
3622 // global_tid, kmp_int32 cncl_kind);
Alexey Bataev81c7ea02015-07-03 09:56:58 +00003623 if (auto *OMPRegionInfo =
3624 dyn_cast_or_null<CGOpenMPRegionInfo>(CGF.CapturedStmtInfo)) {
Alexey Bataev25e5b442015-09-15 12:52:43 +00003625 if (OMPRegionInfo->hasCancel()) {
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003626 llvm::Value *Args[] = {
3627 emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc),
3628 CGF.Builder.getInt32(getCancellationKind(CancelRegion))};
Alexey Bataev81c7ea02015-07-03 09:56:58 +00003629 // Ignore return result until untied tasks are supported.
3630 auto *Result = CGF.EmitRuntimeCall(
3631 createRuntimeFunction(OMPRTL__kmpc_cancellationpoint), Args);
3632 // if (__kmpc_cancellationpoint()) {
3633 // __kmpc_cancel_barrier();
3634 // exit from construct;
3635 // }
3636 auto *ExitBB = CGF.createBasicBlock(".cancel.exit");
3637 auto *ContBB = CGF.createBasicBlock(".cancel.continue");
3638 auto *Cmp = CGF.Builder.CreateIsNotNull(Result);
3639 CGF.Builder.CreateCondBr(Cmp, ExitBB, ContBB);
3640 CGF.EmitBlock(ExitBB);
3641 // __kmpc_cancel_barrier();
Alexey Bataev25e5b442015-09-15 12:52:43 +00003642 emitBarrierCall(CGF, Loc, OMPD_unknown, /*EmitChecks=*/false);
Alexey Bataev81c7ea02015-07-03 09:56:58 +00003643 // exit from construct;
Alexey Bataev25e5b442015-09-15 12:52:43 +00003644 auto CancelDest =
3645 CGF.getOMPCancelDestination(OMPRegionInfo->getDirectiveKind());
Alexey Bataev81c7ea02015-07-03 09:56:58 +00003646 CGF.EmitBranchThroughCleanup(CancelDest);
3647 CGF.EmitBlock(ContBB, /*IsFinished=*/true);
3648 }
Alexey Bataev0f34da12015-07-02 04:17:07 +00003649 }
Alexey Bataev0f34da12015-07-02 04:17:07 +00003650}
3651
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003652void CGOpenMPRuntime::emitCancelCall(CodeGenFunction &CGF, SourceLocation Loc,
Alexey Bataev87933c72015-09-18 08:07:34 +00003653 const Expr *IfCond,
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003654 OpenMPDirectiveKind CancelRegion) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00003655 if (!CGF.HaveInsertPoint())
3656 return;
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003657 // Build call kmp_int32 __kmpc_cancel(ident_t *loc, kmp_int32 global_tid,
3658 // kmp_int32 cncl_kind);
3659 if (auto *OMPRegionInfo =
3660 dyn_cast_or_null<CGOpenMPRegionInfo>(CGF.CapturedStmtInfo)) {
Alexey Bataev87933c72015-09-18 08:07:34 +00003661 auto &&ThenGen = [this, Loc, CancelRegion,
3662 OMPRegionInfo](CodeGenFunction &CGF) {
3663 llvm::Value *Args[] = {
3664 emitUpdateLocation(CGF, Loc), getThreadID(CGF, Loc),
3665 CGF.Builder.getInt32(getCancellationKind(CancelRegion))};
3666 // Ignore return result until untied tasks are supported.
3667 auto *Result =
3668 CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__kmpc_cancel), Args);
3669 // if (__kmpc_cancel()) {
3670 // __kmpc_cancel_barrier();
3671 // exit from construct;
3672 // }
3673 auto *ExitBB = CGF.createBasicBlock(".cancel.exit");
3674 auto *ContBB = CGF.createBasicBlock(".cancel.continue");
3675 auto *Cmp = CGF.Builder.CreateIsNotNull(Result);
3676 CGF.Builder.CreateCondBr(Cmp, ExitBB, ContBB);
3677 CGF.EmitBlock(ExitBB);
3678 // __kmpc_cancel_barrier();
3679 emitBarrierCall(CGF, Loc, OMPD_unknown, /*EmitChecks=*/false);
3680 // exit from construct;
3681 auto CancelDest =
3682 CGF.getOMPCancelDestination(OMPRegionInfo->getDirectiveKind());
3683 CGF.EmitBranchThroughCleanup(CancelDest);
3684 CGF.EmitBlock(ContBB, /*IsFinished=*/true);
3685 };
3686 if (IfCond)
3687 emitOMPIfClause(CGF, IfCond, ThenGen, [](CodeGenFunction &) {});
3688 else
3689 ThenGen(CGF);
Alexey Bataev7d5d33e2015-07-06 05:50:32 +00003690 }
3691}
Samuel Antaobed3c462015-10-02 16:14:20 +00003692
Samuel Antaoee8fb302016-01-06 13:42:12 +00003693/// \brief Obtain information that uniquely identifies a target entry. This
3694/// consists of the file and device IDs as well as line and column numbers
3695/// associated with the relevant entry source location.
3696static void getTargetEntryUniqueInfo(ASTContext &C, SourceLocation Loc,
3697 unsigned &DeviceID, unsigned &FileID,
3698 unsigned &LineNum, unsigned &ColumnNum) {
3699
3700 auto &SM = C.getSourceManager();
3701
3702 // The loc should be always valid and have a file ID (the user cannot use
3703 // #pragma directives in macros)
3704
3705 assert(Loc.isValid() && "Source location is expected to be always valid.");
3706 assert(Loc.isFileID() && "Source location is expected to refer to a file.");
3707
3708 PresumedLoc PLoc = SM.getPresumedLoc(Loc);
3709 assert(PLoc.isValid() && "Source location is expected to be always valid.");
3710
3711 llvm::sys::fs::UniqueID ID;
3712 if (llvm::sys::fs::getUniqueID(PLoc.getFilename(), ID))
3713 llvm_unreachable("Source file with target region no longer exists!");
3714
3715 DeviceID = ID.getDevice();
3716 FileID = ID.getFile();
3717 LineNum = PLoc.getLine();
3718 ColumnNum = PLoc.getColumn();
3719 return;
3720}
3721
3722void CGOpenMPRuntime::emitTargetOutlinedFunction(
3723 const OMPExecutableDirective &D, StringRef ParentName,
3724 llvm::Function *&OutlinedFn, llvm::Constant *&OutlinedFnID,
3725 bool IsOffloadEntry) {
3726
3727 assert(!ParentName.empty() && "Invalid target region parent name!");
3728
Samuel Antaobed3c462015-10-02 16:14:20 +00003729 const CapturedStmt &CS = *cast<CapturedStmt>(D.getAssociatedStmt());
3730
Samuel Antaoee8fb302016-01-06 13:42:12 +00003731 // Emit target region as a standalone region.
3732 auto &&CodeGen = [&CS](CodeGenFunction &CGF) {
3733 CGF.EmitStmt(CS.getCapturedStmt());
3734 };
3735
3736 // Create a unique name for the proxy/entry function that using the source
3737 // location information of the current target region. The name will be
3738 // something like:
3739 //
3740 // .omp_offloading.DD_FFFF.PP.lBB.cCC
3741 //
3742 // where DD_FFFF is an ID unique to the file (device and file IDs), PP is the
3743 // mangled name of the function that encloses the target region, BB is the
3744 // line number of the target region, and CC is the column number of the target
3745 // region.
3746
3747 unsigned DeviceID;
3748 unsigned FileID;
3749 unsigned Line;
3750 unsigned Column;
3751 getTargetEntryUniqueInfo(CGM.getContext(), D.getLocStart(), DeviceID, FileID,
3752 Line, Column);
3753 SmallString<64> EntryFnName;
3754 {
3755 llvm::raw_svector_ostream OS(EntryFnName);
3756 OS << ".omp_offloading" << llvm::format(".%x", DeviceID)
3757 << llvm::format(".%x.", FileID) << ParentName << ".l" << Line << ".c"
3758 << Column;
3759 }
3760
Samuel Antaobed3c462015-10-02 16:14:20 +00003761 CodeGenFunction CGF(CGM, true);
Samuel Antaoee8fb302016-01-06 13:42:12 +00003762 CGOpenMPTargetRegionInfo CGInfo(CS, CodeGen, EntryFnName);
Samuel Antaobed3c462015-10-02 16:14:20 +00003763 CodeGenFunction::CGCapturedStmtRAII CapInfoRAII(CGF, &CGInfo);
Samuel Antaoee8fb302016-01-06 13:42:12 +00003764
3765 OutlinedFn = CGF.GenerateOpenMPCapturedStmtFunction(CS);
3766
3767 // If this target outline function is not an offload entry, we don't need to
3768 // register it.
3769 if (!IsOffloadEntry)
3770 return;
3771
3772 // The target region ID is used by the runtime library to identify the current
3773 // target region, so it only has to be unique and not necessarily point to
3774 // anything. It could be the pointer to the outlined function that implements
3775 // the target region, but we aren't using that so that the compiler doesn't
3776 // need to keep that, and could therefore inline the host function if proven
3777 // worthwhile during optimization. In the other hand, if emitting code for the
3778 // device, the ID has to be the function address so that it can retrieved from
3779 // the offloading entry and launched by the runtime library. We also mark the
3780 // outlined function to have external linkage in case we are emitting code for
3781 // the device, because these functions will be entry points to the device.
3782
3783 if (CGM.getLangOpts().OpenMPIsDevice) {
3784 OutlinedFnID = llvm::ConstantExpr::getBitCast(OutlinedFn, CGM.Int8PtrTy);
3785 OutlinedFn->setLinkage(llvm::GlobalValue::ExternalLinkage);
3786 } else
3787 OutlinedFnID = new llvm::GlobalVariable(
3788 CGM.getModule(), CGM.Int8Ty, /*isConstant=*/true,
3789 llvm::GlobalValue::PrivateLinkage,
3790 llvm::Constant::getNullValue(CGM.Int8Ty), ".omp_offload.region_id");
3791
3792 // Register the information for the entry associated with this target region.
3793 OffloadEntriesInfoManager.registerTargetRegionEntryInfo(
3794 DeviceID, FileID, ParentName, Line, Column, OutlinedFn, OutlinedFnID);
3795 return;
Samuel Antaobed3c462015-10-02 16:14:20 +00003796}
3797
3798void CGOpenMPRuntime::emitTargetCall(CodeGenFunction &CGF,
3799 const OMPExecutableDirective &D,
3800 llvm::Value *OutlinedFn,
Samuel Antaoee8fb302016-01-06 13:42:12 +00003801 llvm::Value *OutlinedFnID,
Samuel Antaobed3c462015-10-02 16:14:20 +00003802 const Expr *IfCond, const Expr *Device,
3803 ArrayRef<llvm::Value *> CapturedVars) {
Alexey Bataev8ef31412015-12-18 07:58:25 +00003804 if (!CGF.HaveInsertPoint())
3805 return;
Samuel Antaobed3c462015-10-02 16:14:20 +00003806 /// \brief Values for bit flags used to specify the mapping type for
3807 /// offloading.
3808 enum OpenMPOffloadMappingFlags {
3809 /// \brief Allocate memory on the device and move data from host to device.
3810 OMP_MAP_TO = 0x01,
3811 /// \brief Allocate memory on the device and move data from device to host.
3812 OMP_MAP_FROM = 0x02,
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003813 /// \brief The element passed to the device is a pointer.
3814 OMP_MAP_PTR = 0x20,
3815 /// \brief Pass the element to the device by value.
3816 OMP_MAP_BYCOPY = 0x80,
Samuel Antaobed3c462015-10-02 16:14:20 +00003817 };
3818
3819 enum OpenMPOffloadingReservedDeviceIDs {
3820 /// \brief Device ID if the device was not defined, runtime should get it
3821 /// from environment variables in the spec.
3822 OMP_DEVICEID_UNDEF = -1,
3823 };
3824
Samuel Antaoee8fb302016-01-06 13:42:12 +00003825 assert(OutlinedFn && "Invalid outlined function!");
3826
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003827 auto &Ctx = CGF.getContext();
3828
Samuel Antaobed3c462015-10-02 16:14:20 +00003829 // Fill up the arrays with the all the captured variables.
3830 SmallVector<llvm::Value *, 16> BasePointers;
3831 SmallVector<llvm::Value *, 16> Pointers;
3832 SmallVector<llvm::Value *, 16> Sizes;
3833 SmallVector<unsigned, 16> MapTypes;
3834
3835 bool hasVLACaptures = false;
3836
3837 const CapturedStmt &CS = *cast<CapturedStmt>(D.getAssociatedStmt());
3838 auto RI = CS.getCapturedRecordDecl()->field_begin();
3839 // auto II = CS.capture_init_begin();
3840 auto CV = CapturedVars.begin();
3841 for (CapturedStmt::const_capture_iterator CI = CS.capture_begin(),
3842 CE = CS.capture_end();
3843 CI != CE; ++CI, ++RI, ++CV) {
3844 StringRef Name;
3845 QualType Ty;
3846 llvm::Value *BasePointer;
3847 llvm::Value *Pointer;
3848 llvm::Value *Size;
3849 unsigned MapType;
3850
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003851 // VLA sizes are passed to the outlined region by copy.
Samuel Antaobed3c462015-10-02 16:14:20 +00003852 if (CI->capturesVariableArrayType()) {
3853 BasePointer = Pointer = *CV;
Alexey Bataev1189bd02016-01-26 12:20:39 +00003854 Size = CGF.getTypeSize(RI->getType());
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003855 // Copy to the device as an argument. No need to retrieve it.
3856 MapType = OMP_MAP_BYCOPY;
Samuel Antaobed3c462015-10-02 16:14:20 +00003857 hasVLACaptures = true;
Samuel Antaobed3c462015-10-02 16:14:20 +00003858 } else if (CI->capturesThis()) {
3859 BasePointer = Pointer = *CV;
3860 const PointerType *PtrTy = cast<PointerType>(RI->getType().getTypePtr());
Alexey Bataev1189bd02016-01-26 12:20:39 +00003861 Size = CGF.getTypeSize(PtrTy->getPointeeType());
Samuel Antaobed3c462015-10-02 16:14:20 +00003862 // Default map type.
3863 MapType = OMP_MAP_TO | OMP_MAP_FROM;
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003864 } else if (CI->capturesVariableByCopy()) {
3865 MapType = OMP_MAP_BYCOPY;
3866 if (!RI->getType()->isAnyPointerType()) {
3867 // If the field is not a pointer, we need to save the actual value and
3868 // load it as a void pointer.
3869 auto DstAddr = CGF.CreateMemTemp(
3870 Ctx.getUIntPtrType(),
3871 Twine(CI->getCapturedVar()->getName()) + ".casted");
3872 LValue DstLV = CGF.MakeAddrLValue(DstAddr, Ctx.getUIntPtrType());
3873
3874 auto *SrcAddrVal = CGF.EmitScalarConversion(
3875 DstAddr.getPointer(), Ctx.getPointerType(Ctx.getUIntPtrType()),
3876 Ctx.getPointerType(RI->getType()), SourceLocation());
3877 LValue SrcLV =
3878 CGF.MakeNaturalAlignAddrLValue(SrcAddrVal, RI->getType());
3879
3880 // Store the value using the source type pointer.
3881 CGF.EmitStoreThroughLValue(RValue::get(*CV), SrcLV);
3882
3883 // Load the value using the destination type pointer.
3884 BasePointer = Pointer =
3885 CGF.EmitLoadOfLValue(DstLV, SourceLocation()).getScalarVal();
3886 } else {
3887 MapType |= OMP_MAP_PTR;
3888 BasePointer = Pointer = *CV;
3889 }
Alexey Bataev1189bd02016-01-26 12:20:39 +00003890 Size = CGF.getTypeSize(RI->getType());
Samuel Antaobed3c462015-10-02 16:14:20 +00003891 } else {
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003892 assert(CI->capturesVariable() && "Expected captured reference.");
Samuel Antaobed3c462015-10-02 16:14:20 +00003893 BasePointer = Pointer = *CV;
3894
3895 const ReferenceType *PtrTy =
3896 cast<ReferenceType>(RI->getType().getTypePtr());
3897 QualType ElementType = PtrTy->getPointeeType();
Alexey Bataev1189bd02016-01-26 12:20:39 +00003898 Size = CGF.getTypeSize(ElementType);
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003899 // The default map type for a scalar/complex type is 'to' because by
3900 // default the value doesn't have to be retrieved. For an aggregate type,
3901 // the default is 'tofrom'.
3902 MapType = ElementType->isAggregateType() ? (OMP_MAP_TO | OMP_MAP_FROM)
3903 : OMP_MAP_TO;
3904 if (ElementType->isAnyPointerType())
3905 MapType |= OMP_MAP_PTR;
Samuel Antaobed3c462015-10-02 16:14:20 +00003906 }
3907
3908 BasePointers.push_back(BasePointer);
3909 Pointers.push_back(Pointer);
3910 Sizes.push_back(Size);
3911 MapTypes.push_back(MapType);
3912 }
3913
3914 // Keep track on whether the host function has to be executed.
3915 auto OffloadErrorQType =
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003916 Ctx.getIntTypeForBitwidth(/*DestWidth=*/32, /*Signed=*/true);
Samuel Antaobed3c462015-10-02 16:14:20 +00003917 auto OffloadError = CGF.MakeAddrLValue(
3918 CGF.CreateMemTemp(OffloadErrorQType, ".run_host_version"),
3919 OffloadErrorQType);
3920 CGF.EmitStoreOfScalar(llvm::Constant::getNullValue(CGM.Int32Ty),
3921 OffloadError);
3922
3923 // Fill up the pointer arrays and transfer execution to the device.
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003924 auto &&ThenGen = [this, &Ctx, &BasePointers, &Pointers, &Sizes, &MapTypes,
Samuel Antaoee8fb302016-01-06 13:42:12 +00003925 hasVLACaptures, Device, OutlinedFnID, OffloadError,
Samuel Antaobed3c462015-10-02 16:14:20 +00003926 OffloadErrorQType](CodeGenFunction &CGF) {
3927 unsigned PointerNumVal = BasePointers.size();
3928 llvm::Value *PointerNum = CGF.Builder.getInt32(PointerNumVal);
3929 llvm::Value *BasePointersArray;
3930 llvm::Value *PointersArray;
3931 llvm::Value *SizesArray;
3932 llvm::Value *MapTypesArray;
3933
3934 if (PointerNumVal) {
3935 llvm::APInt PointerNumAP(32, PointerNumVal, /*isSigned=*/true);
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003936 QualType PointerArrayType = Ctx.getConstantArrayType(
3937 Ctx.VoidPtrTy, PointerNumAP, ArrayType::Normal,
Samuel Antaobed3c462015-10-02 16:14:20 +00003938 /*IndexTypeQuals=*/0);
3939
3940 BasePointersArray =
3941 CGF.CreateMemTemp(PointerArrayType, ".offload_baseptrs").getPointer();
3942 PointersArray =
3943 CGF.CreateMemTemp(PointerArrayType, ".offload_ptrs").getPointer();
3944
3945 // If we don't have any VLA types, we can use a constant array for the map
3946 // sizes, otherwise we need to fill up the arrays as we do for the
3947 // pointers.
3948 if (hasVLACaptures) {
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003949 QualType SizeArrayType = Ctx.getConstantArrayType(
3950 Ctx.getSizeType(), PointerNumAP, ArrayType::Normal,
Samuel Antaobed3c462015-10-02 16:14:20 +00003951 /*IndexTypeQuals=*/0);
3952 SizesArray =
3953 CGF.CreateMemTemp(SizeArrayType, ".offload_sizes").getPointer();
3954 } else {
3955 // We expect all the sizes to be constant, so we collect them to create
3956 // a constant array.
3957 SmallVector<llvm::Constant *, 16> ConstSizes;
3958 for (auto S : Sizes)
3959 ConstSizes.push_back(cast<llvm::Constant>(S));
3960
3961 auto *SizesArrayInit = llvm::ConstantArray::get(
3962 llvm::ArrayType::get(CGM.SizeTy, ConstSizes.size()), ConstSizes);
3963 auto *SizesArrayGbl = new llvm::GlobalVariable(
3964 CGM.getModule(), SizesArrayInit->getType(),
3965 /*isConstant=*/true, llvm::GlobalValue::PrivateLinkage,
3966 SizesArrayInit, ".offload_sizes");
3967 SizesArrayGbl->setUnnamedAddr(true);
3968 SizesArray = SizesArrayGbl;
3969 }
3970
3971 // The map types are always constant so we don't need to generate code to
3972 // fill arrays. Instead, we create an array constant.
3973 llvm::Constant *MapTypesArrayInit =
3974 llvm::ConstantDataArray::get(CGF.Builder.getContext(), MapTypes);
3975 auto *MapTypesArrayGbl = new llvm::GlobalVariable(
3976 CGM.getModule(), MapTypesArrayInit->getType(),
3977 /*isConstant=*/true, llvm::GlobalValue::PrivateLinkage,
3978 MapTypesArrayInit, ".offload_maptypes");
3979 MapTypesArrayGbl->setUnnamedAddr(true);
3980 MapTypesArray = MapTypesArrayGbl;
3981
3982 for (unsigned i = 0; i < PointerNumVal; ++i) {
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003983
3984 llvm::Value *BPVal = BasePointers[i];
3985 if (BPVal->getType()->isPointerTy())
3986 BPVal = CGF.Builder.CreateBitCast(BPVal, CGM.VoidPtrTy);
3987 else {
3988 assert(BPVal->getType()->isIntegerTy() &&
3989 "If not a pointer, the value type must be an integer.");
3990 BPVal = CGF.Builder.CreateIntToPtr(BPVal, CGM.VoidPtrTy);
3991 }
Samuel Antaobed3c462015-10-02 16:14:20 +00003992 llvm::Value *BP = CGF.Builder.CreateConstInBoundsGEP2_32(
3993 llvm::ArrayType::get(CGM.VoidPtrTy, PointerNumVal),
3994 BasePointersArray, 0, i);
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003995 Address BPAddr(BP, Ctx.getTypeAlignInChars(Ctx.VoidPtrTy));
3996 CGF.Builder.CreateStore(BPVal, BPAddr);
Samuel Antaobed3c462015-10-02 16:14:20 +00003997
Samuel Antao4af1b7b2015-12-02 17:44:43 +00003998 llvm::Value *PVal = Pointers[i];
3999 if (PVal->getType()->isPointerTy())
4000 PVal = CGF.Builder.CreateBitCast(PVal, CGM.VoidPtrTy);
4001 else {
4002 assert(PVal->getType()->isIntegerTy() &&
4003 "If not a pointer, the value type must be an integer.");
4004 PVal = CGF.Builder.CreateIntToPtr(PVal, CGM.VoidPtrTy);
4005 }
Samuel Antaobed3c462015-10-02 16:14:20 +00004006 llvm::Value *P = CGF.Builder.CreateConstInBoundsGEP2_32(
4007 llvm::ArrayType::get(CGM.VoidPtrTy, PointerNumVal), PointersArray,
4008 0, i);
Samuel Antao4af1b7b2015-12-02 17:44:43 +00004009 Address PAddr(P, Ctx.getTypeAlignInChars(Ctx.VoidPtrTy));
4010 CGF.Builder.CreateStore(PVal, PAddr);
Samuel Antaobed3c462015-10-02 16:14:20 +00004011
4012 if (hasVLACaptures) {
4013 llvm::Value *S = CGF.Builder.CreateConstInBoundsGEP2_32(
4014 llvm::ArrayType::get(CGM.SizeTy, PointerNumVal), SizesArray,
4015 /*Idx0=*/0,
4016 /*Idx1=*/i);
Samuel Antao4af1b7b2015-12-02 17:44:43 +00004017 Address SAddr(S, Ctx.getTypeAlignInChars(Ctx.getSizeType()));
Samuel Antaobed3c462015-10-02 16:14:20 +00004018 CGF.Builder.CreateStore(CGF.Builder.CreateIntCast(
4019 Sizes[i], CGM.SizeTy, /*isSigned=*/true),
4020 SAddr);
4021 }
4022 }
4023
4024 BasePointersArray = CGF.Builder.CreateConstInBoundsGEP2_32(
4025 llvm::ArrayType::get(CGM.VoidPtrTy, PointerNumVal), BasePointersArray,
4026 /*Idx0=*/0, /*Idx1=*/0);
4027 PointersArray = CGF.Builder.CreateConstInBoundsGEP2_32(
4028 llvm::ArrayType::get(CGM.VoidPtrTy, PointerNumVal), PointersArray,
4029 /*Idx0=*/0,
4030 /*Idx1=*/0);
4031 SizesArray = CGF.Builder.CreateConstInBoundsGEP2_32(
4032 llvm::ArrayType::get(CGM.SizeTy, PointerNumVal), SizesArray,
4033 /*Idx0=*/0, /*Idx1=*/0);
4034 MapTypesArray = CGF.Builder.CreateConstInBoundsGEP2_32(
4035 llvm::ArrayType::get(CGM.Int32Ty, PointerNumVal), MapTypesArray,
4036 /*Idx0=*/0,
4037 /*Idx1=*/0);
4038
4039 } else {
4040 BasePointersArray = llvm::ConstantPointerNull::get(CGM.VoidPtrPtrTy);
4041 PointersArray = llvm::ConstantPointerNull::get(CGM.VoidPtrPtrTy);
4042 SizesArray = llvm::ConstantPointerNull::get(CGM.SizeTy->getPointerTo());
4043 MapTypesArray =
4044 llvm::ConstantPointerNull::get(CGM.Int32Ty->getPointerTo());
4045 }
4046
4047 // On top of the arrays that were filled up, the target offloading call
4048 // takes as arguments the device id as well as the host pointer. The host
4049 // pointer is used by the runtime library to identify the current target
4050 // region, so it only has to be unique and not necessarily point to
4051 // anything. It could be the pointer to the outlined function that
4052 // implements the target region, but we aren't using that so that the
4053 // compiler doesn't need to keep that, and could therefore inline the host
4054 // function if proven worthwhile during optimization.
4055
Samuel Antaoee8fb302016-01-06 13:42:12 +00004056 // From this point on, we need to have an ID of the target region defined.
4057 assert(OutlinedFnID && "Invalid outlined function ID!");
Samuel Antaobed3c462015-10-02 16:14:20 +00004058
4059 // Emit device ID if any.
4060 llvm::Value *DeviceID;
4061 if (Device)
4062 DeviceID = CGF.Builder.CreateIntCast(CGF.EmitScalarExpr(Device),
4063 CGM.Int32Ty, /*isSigned=*/true);
4064 else
4065 DeviceID = CGF.Builder.getInt32(OMP_DEVICEID_UNDEF);
4066
4067 llvm::Value *OffloadingArgs[] = {
Samuel Antaoee8fb302016-01-06 13:42:12 +00004068 DeviceID, OutlinedFnID, PointerNum, BasePointersArray,
4069 PointersArray, SizesArray, MapTypesArray};
Samuel Antaobed3c462015-10-02 16:14:20 +00004070 auto Return = CGF.EmitRuntimeCall(createRuntimeFunction(OMPRTL__tgt_target),
4071 OffloadingArgs);
4072
4073 CGF.EmitStoreOfScalar(Return, OffloadError);
4074 };
4075
Samuel Antaoee8fb302016-01-06 13:42:12 +00004076 // Notify that the host version must be executed.
4077 auto &&ElseGen = [this, OffloadError,
4078 OffloadErrorQType](CodeGenFunction &CGF) {
4079 CGF.EmitStoreOfScalar(llvm::ConstantInt::get(CGM.Int32Ty, /*V=*/-1u),
4080 OffloadError);
4081 };
4082
4083 // If we have a target function ID it means that we need to support
4084 // offloading, otherwise, just execute on the host. We need to execute on host
4085 // regardless of the conditional in the if clause if, e.g., the user do not
4086 // specify target triples.
4087 if (OutlinedFnID) {
4088 if (IfCond) {
4089 emitOMPIfClause(CGF, IfCond, ThenGen, ElseGen);
4090 } else {
4091 CodeGenFunction::RunCleanupsScope Scope(CGF);
4092 ThenGen(CGF);
4093 }
Samuel Antaobed3c462015-10-02 16:14:20 +00004094 } else {
4095 CodeGenFunction::RunCleanupsScope Scope(CGF);
Samuel Antaoee8fb302016-01-06 13:42:12 +00004096 ElseGen(CGF);
Samuel Antaobed3c462015-10-02 16:14:20 +00004097 }
4098
4099 // Check the error code and execute the host version if required.
4100 auto OffloadFailedBlock = CGF.createBasicBlock("omp_offload.failed");
4101 auto OffloadContBlock = CGF.createBasicBlock("omp_offload.cont");
4102 auto OffloadErrorVal = CGF.EmitLoadOfScalar(OffloadError, SourceLocation());
4103 auto Failed = CGF.Builder.CreateIsNotNull(OffloadErrorVal);
4104 CGF.Builder.CreateCondBr(Failed, OffloadFailedBlock, OffloadContBlock);
4105
4106 CGF.EmitBlock(OffloadFailedBlock);
4107 CGF.Builder.CreateCall(OutlinedFn, BasePointers);
4108 CGF.EmitBranch(OffloadContBlock);
4109
4110 CGF.EmitBlock(OffloadContBlock, /*IsFinished=*/true);
4111 return;
4112}
Samuel Antaoee8fb302016-01-06 13:42:12 +00004113
4114void CGOpenMPRuntime::scanForTargetRegionsFunctions(const Stmt *S,
4115 StringRef ParentName) {
4116 if (!S)
4117 return;
4118
4119 // If we find a OMP target directive, codegen the outline function and
4120 // register the result.
4121 // FIXME: Add other directives with target when they become supported.
4122 bool isTargetDirective = isa<OMPTargetDirective>(S);
4123
4124 if (isTargetDirective) {
4125 auto *E = cast<OMPExecutableDirective>(S);
4126 unsigned DeviceID;
4127 unsigned FileID;
4128 unsigned Line;
4129 unsigned Column;
4130 getTargetEntryUniqueInfo(CGM.getContext(), E->getLocStart(), DeviceID,
4131 FileID, Line, Column);
4132
4133 // Is this a target region that should not be emitted as an entry point? If
4134 // so just signal we are done with this target region.
4135 if (!OffloadEntriesInfoManager.hasTargetRegionEntryInfo(
4136 DeviceID, FileID, ParentName, Line, Column))
4137 return;
4138
4139 llvm::Function *Fn;
4140 llvm::Constant *Addr;
4141 emitTargetOutlinedFunction(*E, ParentName, Fn, Addr,
4142 /*isOffloadEntry=*/true);
4143 assert(Fn && Addr && "Target region emission failed.");
4144 return;
4145 }
4146
4147 if (const OMPExecutableDirective *E = dyn_cast<OMPExecutableDirective>(S)) {
4148 if (!E->getAssociatedStmt())
4149 return;
4150
4151 scanForTargetRegionsFunctions(
4152 cast<CapturedStmt>(E->getAssociatedStmt())->getCapturedStmt(),
4153 ParentName);
4154 return;
4155 }
4156
4157 // If this is a lambda function, look into its body.
4158 if (auto *L = dyn_cast<LambdaExpr>(S))
4159 S = L->getBody();
4160
4161 // Keep looking for target regions recursively.
4162 for (auto *II : S->children())
4163 scanForTargetRegionsFunctions(II, ParentName);
4164
4165 return;
4166}
4167
4168bool CGOpenMPRuntime::emitTargetFunctions(GlobalDecl GD) {
4169 auto &FD = *cast<FunctionDecl>(GD.getDecl());
4170
4171 // If emitting code for the host, we do not process FD here. Instead we do
4172 // the normal code generation.
4173 if (!CGM.getLangOpts().OpenMPIsDevice)
4174 return false;
4175
4176 // Try to detect target regions in the function.
4177 scanForTargetRegionsFunctions(FD.getBody(), CGM.getMangledName(GD));
4178
4179 // We should not emit any function othen that the ones created during the
4180 // scanning. Therefore, we signal that this function is completely dealt
4181 // with.
4182 return true;
4183}
4184
4185bool CGOpenMPRuntime::emitTargetGlobalVariable(GlobalDecl GD) {
4186 if (!CGM.getLangOpts().OpenMPIsDevice)
4187 return false;
4188
4189 // Check if there are Ctors/Dtors in this declaration and look for target
4190 // regions in it. We use the complete variant to produce the kernel name
4191 // mangling.
4192 QualType RDTy = cast<VarDecl>(GD.getDecl())->getType();
4193 if (auto *RD = RDTy->getBaseElementTypeUnsafe()->getAsCXXRecordDecl()) {
4194 for (auto *Ctor : RD->ctors()) {
4195 StringRef ParentName =
4196 CGM.getMangledName(GlobalDecl(Ctor, Ctor_Complete));
4197 scanForTargetRegionsFunctions(Ctor->getBody(), ParentName);
4198 }
4199 auto *Dtor = RD->getDestructor();
4200 if (Dtor) {
4201 StringRef ParentName =
4202 CGM.getMangledName(GlobalDecl(Dtor, Dtor_Complete));
4203 scanForTargetRegionsFunctions(Dtor->getBody(), ParentName);
4204 }
4205 }
4206
4207 // If we are in target mode we do not emit any global (declare target is not
4208 // implemented yet). Therefore we signal that GD was processed in this case.
4209 return true;
4210}
4211
4212bool CGOpenMPRuntime::emitTargetGlobal(GlobalDecl GD) {
4213 auto *VD = GD.getDecl();
4214 if (isa<FunctionDecl>(VD))
4215 return emitTargetFunctions(GD);
4216
4217 return emitTargetGlobalVariable(GD);
4218}
4219
4220llvm::Function *CGOpenMPRuntime::emitRegistrationFunction() {
4221 // If we have offloading in the current module, we need to emit the entries
4222 // now and register the offloading descriptor.
4223 createOffloadEntriesAndInfoMetadata();
4224
4225 // Create and register the offloading binary descriptors. This is the main
4226 // entity that captures all the information about offloading in the current
4227 // compilation unit.
4228 return createOffloadingBinaryDescriptorRegistration();
4229}