LLVM 24.0.0git
OMPIRBuilder.h
Go to the documentation of this file.
1//===- IR/OpenMPIRBuilder.h - OpenMP encoding builder for LLVM IR - C++ -*-===//
2//
3// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4// See https://llvm.org/LICENSE.txt for license information.
5// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6//
7//===----------------------------------------------------------------------===//
8//
9// This file defines the OpenMPIRBuilder class and helpers used as a convenient
10// way to create LLVM instructions for OpenMP directives.
11//
12//===----------------------------------------------------------------------===//
13
14#ifndef LLVM_FRONTEND_OPENMP_OMPIRBUILDER_H
15#define LLVM_FRONTEND_OPENMP_OMPIRBUILDER_H
16
17#include "llvm/ADT/APSInt.h"
18#include "llvm/ADT/SetVector.h"
22#include "llvm/IR/CallingConv.h"
23#include "llvm/IR/DebugLoc.h"
24#include "llvm/IR/IRBuilder.h"
25#include "llvm/IR/Module.h"
26#include "llvm/IR/ValueMap.h"
29#include "llvm/Support/Error.h"
32#include <forward_list>
33#include <map>
34#include <optional>
35
36namespace llvm {
38class CodeExtractor;
39class ScanInfo;
42class OpenMPIRBuilder;
43class Loop;
44class LoopAnalysis;
45class LoopInfo;
46
47/// Move the instruction after an InsertPoint to the beginning of another
48/// BasicBlock.
49///
50/// The instructions after \p IP are moved to the beginning of \p New which must
51/// not have any PHINodes. If \p CreateBranch is true, a branch instruction to
52/// \p New will be added such that there is no semantic change. Otherwise, the
53/// \p IP insert block remains degenerate and it is up to the caller to insert a
54/// terminator. \p DL is used as the debug location for the branch instruction
55/// if one is created.
57 bool CreateBranch, DebugLoc DL);
58
59/// Splice a BasicBlock at an IRBuilder's current insertion point. Its new
60/// insert location will stick to after the instruction before the insertion
61/// point (instead of moving with the instruction the InsertPoint stores
62/// internally).
63LLVM_ABI void spliceBB(IRBuilder<> &Builder, BasicBlock *New,
64 bool CreateBranch);
65
66/// Split a BasicBlock at an InsertPoint, even if the block is degenerate
67/// (missing the terminator).
68///
69/// llvm::SplitBasicBlock and BasicBlock::splitBasicBlock require a well-formed
70/// BasicBlock. \p Name is used for the new successor block. If \p CreateBranch
71/// is true, a branch to the new successor will new created such that
72/// semantically there is no change; otherwise the block of the insertion point
73/// remains degenerate and it is the caller's responsibility to insert a
74/// terminator. \p DL is used as the debug location for the branch instruction
75/// if one is created. Returns the new successor block.
77 DebugLoc DL, llvm::Twine Name = {});
78
79/// Split a BasicBlock at \p Builder's insertion point, even if the block is
80/// degenerate (missing the terminator). Its new insert location will stick to
81/// after the instruction before the insertion point (instead of moving with the
82/// instruction the InsertPoint stores internally).
83LLVM_ABI BasicBlock *splitBB(IRBuilderBase &Builder, bool CreateBranch,
84 llvm::Twine Name = {});
85
86/// Split a BasicBlock at \p Builder's insertion point, even if the block is
87/// degenerate (missing the terminator). Its new insert location will stick to
88/// after the instruction before the insertion point (instead of moving with the
89/// instruction the InsertPoint stores internally).
90LLVM_ABI BasicBlock *splitBB(IRBuilder<> &Builder, bool CreateBranch,
91 llvm::Twine Name);
92
93/// Like splitBB, but reuses the current block's name for the new name.
95 bool CreateBranch,
96 llvm::Twine Suffix = ".split");
97
98/// Captures attributes that affect generating LLVM-IR using the
99/// OpenMPIRBuilder and related classes. Note that not all attributes are
100/// required for all classes or functions. In some use cases the configuration
101/// is not necessary at all, because because the only functions that are called
102/// are ones that are not dependent on the configuration.
104public:
105 /// Flag to define whether to generate code for the role of the OpenMP host
106 /// (if set to false) or device (if set to true) in an offloading context. It
107 /// is set when the -fopenmp-is-target-device compiler frontend option is
108 /// specified.
109 std::optional<bool> IsTargetDevice;
110
111 /// Flag for specifying if the compilation is done for an accelerator. It is
112 /// set according to the architecture of the target triple and currently only
113 /// true when targeting AMDGPU or NVPTX. Today, these targets can only perform
114 /// the role of an OpenMP target device, so `IsTargetDevice` must also be true
115 /// if `IsGPU` is true. This restriction might be lifted if an accelerator-
116 /// like target with the ability to work as the OpenMP host is added, or if
117 /// the capabilities of the currently supported GPU architectures are
118 /// expanded.
119 std::optional<bool> IsGPU;
120
121 /// Flag for specifying if LLVMUsed information should be emitted.
122 std::optional<bool> EmitLLVMUsedMetaInfo;
123
124 /// Flag for specifying if offloading is mandatory.
125 std::optional<bool> OpenMPOffloadMandatory;
126
127 /// First separator used between the initial two parts of a name.
128 std::optional<StringRef> FirstSeparator;
129 /// Separator used between all of the rest consecutive parts of s name.
130 std::optional<StringRef> Separator;
131
132 /// Flag for specifying whether the no-signed-wrap (nsw) flag should be added
133 /// to loop induction variable arithmetic. Set when the frontend guarantees
134 /// that signed integer overflow is undefined (with -fno-wrapv).
135 std::optional<bool> NoSignedWrap;
136
137 // Grid Value for the GPU target.
138 std::optional<omp::GV> GridValue;
139
140 /// When compilation is being done for the OpenMP host (i.e. `IsTargetDevice =
141 /// false`), this contains the list of offloading triples associated, if any.
143
144 // Default address space for the target.
145 unsigned DefaultTargetAS = 0;
146
148
152 bool HasRequiresReverseOffload,
153 bool HasRequiresUnifiedAddress,
154 bool HasRequiresUnifiedSharedMemory,
155 bool HasRequiresDynamicAllocators);
156
157 // Getters functions that assert if the required values are not present.
158 bool isTargetDevice() const {
159 assert(IsTargetDevice.has_value() && "IsTargetDevice is not set");
160 return *IsTargetDevice;
161 }
162
163 bool isGPU() const {
164 assert(IsGPU.has_value() && "IsGPU is not set");
165 return *IsGPU;
166 }
167
169 assert(OpenMPOffloadMandatory.has_value() &&
170 "OpenMPOffloadMandatory is not set");
172 }
173
175 assert(GridValue.has_value() && "GridValue is not set");
176 return *GridValue;
177 }
178
179 unsigned getDefaultTargetAS() const { return DefaultTargetAS; }
180
181 bool hasNoSignedWrap() const { return NoSignedWrap.value_or(false); }
183
185
186 bool hasRequiresFlags() const { return RequiresFlags; }
191
192 /// Returns requires directive clauses as flags compatible with those expected
193 /// by libomptarget.
194 LLVM_ABI int64_t getRequiresFlags() const;
195
196 // Returns the FirstSeparator if set, otherwise use the default separator
197 // depending on isGPU
199 if (FirstSeparator.has_value())
200 return *FirstSeparator;
201 if (isGPU())
202 return "_";
203 return ".";
204 }
205
206 // Returns the Separator if set, otherwise use the default separator depending
207 // on isGPU
209 if (Separator.has_value())
210 return *Separator;
211 if (isGPU())
212 return "$";
213 return ".";
214 }
215
217 void setIsGPU(bool Value) { IsGPU = Value; }
223 void setDefaultTargetAS(unsigned AS) { DefaultTargetAS = AS; }
225
230
231private:
232 /// Flags for specifying which requires directive clauses are present.
233 int64_t RequiresFlags;
234};
235
236/// Data structure to contain the information needed to uniquely identify
237/// a target entry.
239 /// The prefix used for kernel names.
240 static constexpr const char *KernelNamePrefix = "__omp_offloading_";
241
242 std::string ParentName;
243 unsigned DeviceID;
244 unsigned FileID;
245 unsigned Line;
246 unsigned Count;
247
250 unsigned FileID, unsigned Line, unsigned Count = 0)
252 Count(Count) {}
253
254 LLVM_ABI static void
256 unsigned DeviceID, unsigned FileID, unsigned Line,
257 unsigned Count);
258
260 return std::make_tuple(ParentName, DeviceID, FileID, Line, Count) <
261 std::make_tuple(RHS.ParentName, RHS.DeviceID, RHS.FileID, RHS.Line,
262 RHS.Count);
263 }
264};
265
266/// Class that manages information about offload code regions and data
268 /// Number of entries registered so far.
269 OpenMPIRBuilder *OMPBuilder;
270 unsigned OffloadingEntriesNum = 0;
271
272public:
273 /// Base class of the entries info.
275 public:
276 /// Kind of a given entry.
277 enum OffloadingEntryInfoKinds : unsigned {
278 /// Entry is a target region.
280 /// Entry is a declare target variable.
282 /// Invalid entry info.
284 };
285
286 protected:
288 explicit OffloadEntryInfo(OffloadingEntryInfoKinds Kind) : Kind(Kind) {}
289 explicit OffloadEntryInfo(OffloadingEntryInfoKinds Kind, unsigned Order,
290 uint32_t Flags)
291 : Flags(Flags), Order(Order), Kind(Kind) {}
292 ~OffloadEntryInfo() = default;
293
294 public:
295 bool isValid() const { return Order != ~0u; }
296 unsigned getOrder() const { return Order; }
297 OffloadingEntryInfoKinds getKind() const { return Kind; }
298 uint32_t getFlags() const { return Flags; }
299 void setFlags(uint32_t NewFlags) { Flags = NewFlags; }
300 Constant *getAddress() const { return cast_or_null<Constant>(Addr); }
302 assert(!Addr.pointsToAliveValue() && "Address has been set before!");
303 Addr = V;
304 }
305 static bool classof(const OffloadEntryInfo *Info) { return true; }
306
307 private:
308 /// Address of the entity that has to be mapped for offloading.
309 WeakTrackingVH Addr;
310
311 /// Flags associated with the device global.
312 uint32_t Flags = 0u;
313
314 /// Order this entry was emitted.
315 unsigned Order = ~0u;
316
317 OffloadingEntryInfoKinds Kind = OffloadingEntryInfoInvalid;
318 };
319
320 /// Return true if a there are no entries defined.
321 LLVM_ABI bool empty() const;
322 /// Return number of entries defined so far.
323 unsigned size() const { return OffloadingEntriesNum; }
324
325 OffloadEntriesInfoManager(OpenMPIRBuilder *builder) : OMPBuilder(builder) {}
326
327 //
328 // Target region entries related.
329 //
330
331 /// Kind of the target registry entry.
333 /// Mark the entry as target region.
335 };
336
337 /// Target region entries info.
339 /// Address that can be used as the ID of the entry.
340 Constant *ID = nullptr;
341
342 public:
345 explicit OffloadEntryInfoTargetRegion(unsigned Order, Constant *Addr,
346 Constant *ID,
349 ID(ID) {
350 setAddress(Addr);
351 }
352
353 Constant *getID() const { return ID; }
354 void setID(Constant *V) {
355 assert(!ID && "ID has been set before!");
356 ID = V;
357 }
358 static bool classof(const OffloadEntryInfo *Info) {
359 return Info->getKind() == OffloadingEntryInfoTargetRegion;
360 }
361 };
362
363 /// Initialize target region entry.
364 /// This is ONLY needed for DEVICE compilation.
365 LLVM_ABI void
367 unsigned Order);
368 /// Register target region entry.
370 Constant *Addr, Constant *ID,
372 /// Return true if a target region entry with the provided information
373 /// exists.
375 bool IgnoreAddressId = false) const;
376
377 // Return the Name based on \a EntryInfo using the next available Count.
378 LLVM_ABI void
380 const TargetRegionEntryInfo &EntryInfo);
381
382 /// brief Applies action \a Action on all registered entries.
383 typedef function_ref<void(const TargetRegionEntryInfo &EntryInfo,
384 const OffloadEntryInfoTargetRegion &)>
386 LLVM_ABI void
388
389 //
390 // Device global variable entries related.
391 //
392
393 /// Kind of the global variable entry..
395 /// Mark the entry as a to declare target.
397 /// Mark the entry as a to declare target link.
399 /// Mark the entry as a declare target enter.
401 /// Mark the entry as having no declare target entry kind.
403 /// Mark the entry as a declare target indirect global.
405 /// Mark the entry as a register requires global.
407 /// Mark the entry as a declare target indirect vtable.
409 };
410
411 /// Kind of device clause for declare target variables
412 /// and functions
413 /// NOTE: Currently not used as a part of a variable entry
414 /// used for Flang and Clang to interface with the variable
415 /// related registration functions
417 /// The target is marked for all devices
419 /// The target is marked for non-host devices
421 /// The target is marked for host devices
423 /// The target is marked as having no clause
425 };
426
427 /// Device global variable entries info.
429 /// Type of the global variable.
430 int64_t VarSize;
432 const std::string VarName;
433
434 public:
440 explicit OffloadEntryInfoDeviceGlobalVar(unsigned Order, Constant *Addr,
441 int64_t VarSize,
444 const std::string &VarName)
446 VarSize(VarSize), Linkage(Linkage), VarName(VarName) {
447 setAddress(Addr);
448 }
449
450 int64_t getVarSize() const { return VarSize; }
451 StringRef getVarName() const { return VarName; }
452 void setVarSize(int64_t Size) { VarSize = Size; }
453 GlobalValue::LinkageTypes getLinkage() const { return Linkage; }
454 void setLinkage(GlobalValue::LinkageTypes LT) { Linkage = LT; }
455 static bool classof(const OffloadEntryInfo *Info) {
456 return Info->getKind() == OffloadingEntryInfoDeviceGlobalVar;
457 }
458 };
459
460 /// Initialize device global variable entry.
461 /// This is ONLY used for DEVICE compilation.
463 StringRef Name, OMPTargetGlobalVarEntryKind Flags, unsigned Order);
464
465 /// Register device global variable entry.
467 StringRef VarName, Constant *Addr, int64_t VarSize,
469 /// Checks if the variable with the given name has been registered already.
471 return OffloadEntriesDeviceGlobalVar.count(VarName) > 0;
472 }
473 /// Applies action \a Action on all registered entries.
474 typedef function_ref<void(StringRef, const OffloadEntryInfoDeviceGlobalVar &)>
478
479private:
480 /// Return the count of entries at a particular source location.
481 unsigned
482 getTargetRegionEntryInfoCount(const TargetRegionEntryInfo &EntryInfo) const;
483
484 /// Update the count of entries at a particular source location.
485 void
486 incrementTargetRegionEntryInfoCount(const TargetRegionEntryInfo &EntryInfo);
487
489 getTargetRegionEntryCountKey(const TargetRegionEntryInfo &EntryInfo) {
490 return TargetRegionEntryInfo(EntryInfo.ParentName, EntryInfo.DeviceID,
491 EntryInfo.FileID, EntryInfo.Line, 0);
492 }
493
494 // Count of entries at a location.
495 std::map<TargetRegionEntryInfo, unsigned> OffloadEntriesTargetRegionCount;
496
497 // Storage for target region entries kind.
498 typedef std::map<TargetRegionEntryInfo, OffloadEntryInfoTargetRegion>
499 OffloadEntriesTargetRegionTy;
500 OffloadEntriesTargetRegionTy OffloadEntriesTargetRegion;
501 /// Storage for device global variable entries kind. The storage is to be
502 /// indexed by mangled name.
504 OffloadEntriesDeviceGlobalVarTy;
505 OffloadEntriesDeviceGlobalVarTy OffloadEntriesDeviceGlobalVar;
506};
507
508/// An interface to create LLVM-IR for OpenMP directives.
509///
510/// Each OpenMP directive has a corresponding public generator method.
512public:
513 /// Create a new OpenMPIRBuilder operating on the given module \p M. This will
514 /// not have an effect on \p M (see initialize)
517 T(M.getTargetTriple()), IsFinalized(false) {}
519
521 llvm::Value *AtomicVar;
522
523 public:
531
532 llvm::Value *getAtomicPointer() const override { return AtomicVar; }
535 const llvm::Twine &Name) const override {
536 llvm::AllocaInst *allocaInst = Builder->CreateAlloca(Ty);
537 allocaInst->setName(Name);
538 return allocaInst;
539 }
540 };
541 /// Initialize the internal state, this will put structures types and
542 /// potentially other helpers into the underlying module. Must be called
543 /// before any other method and only once! This internal state includes types
544 /// used in the OpenMPIRBuilder generated from OMPKinds.def.
545 LLVM_ABI void initialize();
546
548
549 /// Finalize the underlying module, e.g., by outlining regions.
550 /// \param Fn The function to be finalized. If not used,
551 /// all functions are finalized.
552 LLVM_ABI void finalize(Function *Fn = nullptr);
553
554 /// Check whether the finalize function has already run
555 /// \return true if the finalize function has already run
556 LLVM_ABI bool isFinalized();
557
558 /// Add attributes known for \p FnID to \p Fn.
560
561 /// Type used throughout for insertion points.
563
564 /// Type used to represent an insertion point or an error value.
566
567 /// Get the create a name using the platform specific separators.
568 /// \param Parts parts of the final name that needs separation
569 /// The created name has a first separator between the first and second part
570 /// and a second separator between all other parts.
571 /// E.g. with FirstSeparator "$" and Separator "." and
572 /// parts: "p1", "p2", "p3", "p4"
573 /// The resulting name is "p1$p2.p3.p4"
574 /// The separators are retrieved from the OpenMPIRBuilderConfig.
575 LLVM_ABI std::string
577
578 /// Callback type for variable finalization (think destructors).
579 ///
580 /// \param CodeGenIP is the insertion point at which the finalization code
581 /// should be placed.
582 ///
583 /// A finalize callback knows about all objects that need finalization, e.g.
584 /// destruction, when the scope of the currently generated construct is left
585 /// at the time, and location, the callback is invoked.
586 using FinalizeCallbackTy = std::function<Error(InsertPointTy CodeGenIP)>;
587
589 FinalizationInfo(FinalizeCallbackTy FiniCB, omp::Directive DK,
590 bool IsCancellable)
591 : DK(DK), IsCancellable(IsCancellable), FiniCB(std::move(FiniCB)) {}
592 /// The directive kind of the innermost directive that has an associated
593 /// region which might require finalization when it is left.
594 const omp::Directive DK;
595
596 /// Flag to indicate if the directive is cancellable.
597 const bool IsCancellable;
598
599 /// The basic block to which control should be transferred to
600 /// implement the FiniCB. Memoized to avoid generating finalization
601 /// multiple times.
603
604 /// For cases where there is an unavoidable existing finalization block
605 /// (e.g. loop finialization after omp sections). The existing finalization
606 /// block must not contain any non-finalization code.
608 BasicBlock *ExistingFiniBB);
609
610 private:
611 /// Access via getFiniBB.
612 BasicBlock *FiniBB = nullptr;
613
614 /// The finalization callback provided by the last in-flight invocation of
615 /// createXXXX for the directive of kind DK.
616 FinalizeCallbackTy FiniCB;
617 };
618
619 /// Push a finalization callback on the finalization stack.
620 ///
621 /// NOTE: Temporary solution until Clang CG is gone.
623 FinalizationStack.push_back(FI);
624 }
625
626 /// Pop the last finalization callback from the finalization stack.
627 ///
628 /// NOTE: Temporary solution until Clang CG is gone.
630
631 /// Callback type for body (=inner region) code generation
632 ///
633 /// The callback takes code locations as arguments, each describing a
634 /// location where additional instructions can be inserted.
635 ///
636 /// The CodeGenIP may be in the middle of a basic block or point to the end of
637 /// it. The basic block may have a terminator or be degenerate. The callback
638 /// function may just insert instructions at that position, but also split the
639 /// block (without the Before argument of BasicBlock::splitBasicBlock such
640 /// that the identify of the split predecessor block is preserved) and insert
641 /// additional control flow, including branches that do not lead back to what
642 /// follows the CodeGenIP. Note that since the callback is allowed to split
643 /// the block, callers must assume that InsertPoints to positions in the
644 /// BasicBlock after CodeGenIP including CodeGenIP itself are invalidated. If
645 /// such InsertPoints need to be preserved, it can split the block itself
646 /// before calling the callback.
647 ///
648 /// AllocaIP and CodeGenIP must not point to the same position.
649 ///
650 /// \param AllocaIP is the insertion point at which new allocations should
651 /// be placed. The BasicBlock it is pointing to must not be
652 /// split.
653 /// \param CodeGenIP is the insertion point at which the body code should be
654 /// placed.
655 /// \param DeallocBlocks is the list of insertion blocks where explicit
656 /// deallocations, if needed, should be placed.
657 /// \return an error, if any were triggered during execution.
659 function_ref<Error(InsertPointTy AllocaIP, InsertPointTy CodeGenIP,
660 ArrayRef<BasicBlock *> DeallocBlocks)>;
661
662 /// Callback type for task duplication function code generation. This is the
663 /// task duplication function passed to __kmpc_taskloop. It is expected that
664 /// this function will set up (first)private variables in the duplicated task
665 /// which have non-trivial (copy-)constructors. Insertion points are handled
666 /// the same way as for BodyGenCallbackTy.
667 ///
668 /// \ref createTaskloop lays out the task's auxiliary data structure as:
669 /// `{ lower bound, upper bound, step, data... }`. DestPtr and SrcPtr point
670 /// to this data.
671 ///
672 /// It is acceptable for the callback to be set to nullptr. In that case no
673 /// function will be generated and nullptr will be passed as the task
674 /// duplication function to __kmpc_taskloop.
675 ///
676 /// \param AllocaIP is the insertion point at which new alloca instructions
677 /// should be placed. The BasicBlock it is pointing to must
678 /// not be split.
679 /// \param CodeGenIP is the insertion point at which the body code should be
680 /// placed.
681 /// \param DestPtr This is a pointer to data inside the newly duplicated
682 /// task's auxiliary data structure (allocated after the task
683 /// descriptor.)
684 /// \param SrcPtr This is a pointer to data inside the original task's
685 /// auxiliary data structure (allocated after the task
686 /// descriptor.)
687 ///
688 /// \return The insertion point immediately after the generated code, or an
689 /// error if any occured.
691 InsertPointTy AllocaIP, InsertPointTy CodeGenIP, Value *DestPtr,
692 Value *SrcPtr)>;
693
694 // This is created primarily for sections construct as llvm::function_ref
695 // (BodyGenCallbackTy) is not storable (as described in the comments of
696 // function_ref class - function_ref contains non-ownable reference
697 // to the callable.
698 ///
699 /// \return an error, if any were triggered during execution.
701 std::function<Error(InsertPointTy AllocaIP, InsertPointTy CodeGenIP,
702 ArrayRef<BasicBlock *> DeallocBlocks)>;
703
704 /// Callback type for loop body code generation.
705 ///
706 /// \param CodeGenIP is the insertion point where the loop's body code must be
707 /// placed. This will be a dedicated BasicBlock with a
708 /// conditional branch from the loop condition check and
709 /// terminated with an unconditional branch to the loop
710 /// latch.
711 /// \param IndVar is the induction variable usable at the insertion point.
712 ///
713 /// \return an error, if any were triggered during execution.
715 function_ref<Error(InsertPointTy CodeGenIP, Value *IndVar)>;
716
717 /// Callback type for variable privatization (think copy & default
718 /// constructor).
719 ///
720 /// \param AllocaIP is the insertion point at which new alloca instructions
721 /// should be placed.
722 /// \param CodeGenIP is the insertion point at which the privatization code
723 /// should be placed.
724 /// \param Original The value being copied/created, should not be used in the
725 /// generated IR.
726 /// \param Inner The equivalent of \p Original that should be used in the
727 /// generated IR; this is equal to \p Original if the value is
728 /// a pointer and can thus be passed directly, otherwise it is
729 /// an equivalent but different value.
730 /// \param ReplVal The replacement value, thus a copy or new created version
731 /// of \p Inner.
732 ///
733 /// \returns The new insertion point where code generation continues and
734 /// \p ReplVal the replacement value.
736 InsertPointTy AllocaIP, InsertPointTy CodeGenIP, Value &Original,
737 Value &Inner, Value *&ReplVal)>;
738
739 /// Description of a LLVM-IR insertion point (IP) and a debug/source location
740 /// (filename, line, column, ...).
743 : IP(IRB.saveIP()), DL(IRB.getCurrentDebugLocation()) {}
745 : IP(IP), DL(DL) {}
748 };
749
750 /// Emitter methods for OpenMP directives.
751 ///
752 ///{
753
754 /// Generator for '#omp barrier'
755 ///
756 /// \param Loc The location where the barrier directive was encountered.
757 /// \param Kind The kind of directive that caused the barrier.
758 /// \param ForceSimpleCall Flag to force a simple (=non-cancellation) barrier.
759 /// \param CheckCancelFlag Flag to indicate a cancel barrier return value
760 /// should be checked and acted upon.
761 /// \param ThreadID Optional parameter to pass in any existing ThreadID value.
762 ///
763 /// \returns The insertion point after the barrier.
765 omp::Directive Kind,
766 bool ForceSimpleCall = false,
767 bool CheckCancelFlag = true);
768
769 /// Generator for '#omp cancel'
770 ///
771 /// \param Loc The location where the directive was encountered.
772 /// \param IfCondition The evaluated 'if' clause expression, if any.
773 /// \param CanceledDirective The kind of directive that is cancled.
774 ///
775 /// \returns The insertion point after the barrier.
777 Value *IfCondition,
778 omp::Directive CanceledDirective);
779
780 /// Generator for '#omp cancellation point'
781 ///
782 /// \param Loc The location where the directive was encountered.
783 /// \param CanceledDirective The kind of directive that is cancled.
784 ///
785 /// \returns The insertion point after the barrier.
787 const LocationDescription &Loc, omp::Directive CanceledDirective);
788
789 /// Creates a ScanInfo object, allocates and returns the pointer.
791
792 /// Generator for '#omp parallel'
793 ///
794 /// \param Loc The insert and source location description.
795 /// \param AllocaIP The insertion point to be used for allocations.
796 /// \param DeallocBlocks The insertion blocks to be used for explicit
797 /// deallocations, if needed.
798 /// \param BodyGenCB Callback that will generate the region code.
799 /// \param PrivCB Callback to copy a given variable (think copy constructor).
800 /// \param FiniCB Callback to finalize variable copies.
801 /// \param IfCondition The evaluated 'if' clause expression, if any.
802 /// \param NumThreads The evaluated 'num_threads' clause expression, if any.
803 /// \param ProcBind The value of the 'proc_bind' clause (see ProcBindKind).
804 /// \param IsCancellable Flag to indicate a cancellable parallel region.
805 ///
806 /// \returns The insertion position *after* the parallel.
808 const LocationDescription &Loc, InsertPointTy AllocaIP,
809 ArrayRef<BasicBlock *> DeallocBlocks, BodyGenCallbackTy BodyGenCB,
810 PrivatizeCallbackTy PrivCB, FinalizeCallbackTy FiniCB, Value *IfCondition,
811 Value *NumThreads, omp::ProcBindKind ProcBind, bool IsCancellable);
812
813 /// Generator for the control flow structure of an OpenMP canonical loop.
814 ///
815 /// This generator operates on the logical iteration space of the loop, i.e.
816 /// the caller only has to provide a loop trip count of the loop as defined by
817 /// base language semantics. The trip count is interpreted as an unsigned
818 /// integer. The induction variable passed to \p BodyGenCB will be of the same
819 /// type and run from 0 to \p TripCount - 1. It is up to the callback to
820 /// convert the logical iteration variable to the loop counter variable in the
821 /// loop body.
822 ///
823 /// \param Loc The insert and source location description. The insert
824 /// location can be between two instructions or the end of a
825 /// degenerate block (e.g. a BB under construction).
826 /// \param BodyGenCB Callback that will generate the loop body code.
827 /// \param TripCount Number of iterations the loop body is executed.
828 /// \param Name Base name used to derive BB and instruction names.
829 ///
830 /// \returns An object representing the created control flow structure which
831 /// can be used for loop-associated directives.
834 LoopBodyGenCallbackTy BodyGenCB, Value *TripCount,
835 const Twine &Name = "loop");
836
837 /// Generator for the control flow structure of an OpenMP canonical loops if
838 /// the parent directive has an `inscan` modifier specified.
839 /// If the `inscan` modifier is specified, the region of the parent is
840 /// expected to have a `scan` directive. Based on the clauses in
841 /// scan directive, the body of the loop is split into two loops: Input loop
842 /// and Scan Loop. Input loop contains the code generated for input phase of
843 /// scan and Scan loop contains the code generated for scan phase of scan.
844 /// From the bodyGen callback of these loops, `createScan` would be called
845 /// when a scan directive is encountered from the loop body. `createScan`
846 /// based on whether 1. inclusive or exclusive scan is specified and, 2. input
847 /// loop or scan loop is generated, lowers the body of the for loop
848 /// accordingly.
849 ///
850 /// \param Loc The insert and source location description.
851 /// \param BodyGenCB Callback that will generate the loop body code.
852 /// \param Start Value of the loop counter for the first iterations.
853 /// \param Stop Loop counter values past this will stop the loop.
854 /// \param Step Loop counter increment after each iteration; negative
855 /// means counting down.
856 /// \param IsSigned Whether Start, Stop and Step are signed integers.
857 /// \param InclusiveStop Whether \p Stop itself is a valid value for the loop
858 /// counter.
859 /// \param ComputeIP Insertion point for instructions computing the trip
860 /// count. Can be used to ensure the trip count is available
861 /// at the outermost loop of a loop nest. If not set,
862 /// defaults to the preheader of the generated loop.
863 /// \param Name Base name used to derive BB and instruction names.
864 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
865 /// `ScanInfoInitialize`.
866 ///
867 /// \returns A vector containing Loop Info of Input Loop and Scan Loop.
870 LoopBodyGenCallbackTy BodyGenCB, Value *Start,
871 Value *Stop, Value *Step, bool IsSigned,
872 bool InclusiveStop, InsertPointTy ComputeIP,
873 const Twine &Name, ScanInfo *ScanRedInfo);
874
875 /// Calculate the trip count of a canonical loop.
876 ///
877 /// This allows specifying user-defined loop counter values using increment,
878 /// upper- and lower bounds. To disambiguate the terminology when counting
879 /// downwards, instead of lower bounds we use \p Start for the loop counter
880 /// value in the first body iteration.
881 ///
882 /// Consider the following limitations:
883 ///
884 /// * A loop counter space over all integer values of its bit-width cannot be
885 /// represented. E.g using uint8_t, its loop trip count of 256 cannot be
886 /// stored into an 8 bit integer):
887 ///
888 /// DO I = 0, 255, 1
889 ///
890 /// * Unsigned wrapping is only supported when wrapping only "once"; E.g.
891 /// effectively counting downwards:
892 ///
893 /// for (uint8_t i = 100u; i > 0; i += 127u)
894 ///
895 ///
896 /// TODO: May need to add additional parameters to represent:
897 ///
898 /// * Allow representing downcounting with unsigned integers.
899 ///
900 /// * Sign of the step and the comparison operator might disagree:
901 ///
902 /// for (int i = 0; i < 42; i -= 1u)
903 ///
904 /// \param Loc The insert and source location description.
905 /// \param Start Value of the loop counter for the first iterations.
906 /// \param Stop Loop counter values past this will stop the loop.
907 /// \param Step Loop counter increment after each iteration; negative
908 /// means counting down.
909 /// \param IsSigned Whether Start, Stop and Step are signed integers.
910 /// \param InclusiveStop Whether \p Stop itself is a valid value for the loop
911 /// counter.
912 /// \param Name Base name used to derive instruction names.
913 ///
914 /// \returns The value holding the calculated trip count.
916 const LocationDescription &Loc, Value *Start, Value *Stop, Value *Step,
917 bool IsSigned, bool InclusiveStop, const Twine &Name = "loop");
918
919 /// Generator for the control flow structure of an OpenMP canonical loop.
920 ///
921 /// Instead of a logical iteration space, this allows specifying user-defined
922 /// loop counter values using increment, upper- and lower bounds. To
923 /// disambiguate the terminology when counting downwards, instead of lower
924 /// bounds we use \p Start for the loop counter value in the first body
925 ///
926 /// It calls \see calculateCanonicalLoopTripCount for trip count calculations,
927 /// so limitations of that method apply here as well.
928 ///
929 /// \param Loc The insert and source location description.
930 /// \param BodyGenCB Callback that will generate the loop body code.
931 /// \param Start Value of the loop counter for the first iterations.
932 /// \param Stop Loop counter values past this will stop the loop.
933 /// \param Step Loop counter increment after each iteration; negative
934 /// means counting down.
935 /// \param IsSigned Whether Start, Stop and Step are signed integers.
936 /// \param InclusiveStop Whether \p Stop itself is a valid value for the loop
937 /// counter.
938 /// \param ComputeIP Insertion point for instructions computing the trip
939 /// count. Can be used to ensure the trip count is available
940 /// at the outermost loop of a loop nest. If not set,
941 /// defaults to the preheader of the generated loop.
942 /// \param Name Base name used to derive BB and instruction names.
943 /// \param InScan Whether loop has a scan reduction specified.
944 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
945 /// `ScanInfoInitialize`.
946 ///
947 /// \returns An object representing the created control flow structure which
948 /// can be used for loop-associated directives.
951 Value *Start, Value *Stop, Value *Step, bool IsSigned, bool InclusiveStop,
952 InsertPointTy ComputeIP = {}, const Twine &Name = "loop",
953 bool InScan = false, ScanInfo *ScanRedInfo = nullptr);
954
955 /// Collapse a loop nest into a single loop.
956 ///
957 /// Merges loops of a loop nest into a single CanonicalLoopNest representation
958 /// that has the same number of innermost loop iterations as the origin loop
959 /// nest. The induction variables of the input loops are derived from the
960 /// collapsed loop's induction variable. This is intended to be used to
961 /// implement OpenMP's collapse clause. Before applying a directive,
962 /// collapseLoops normalizes a loop nest to contain only a single loop and the
963 /// directive's implementation does not need to handle multiple loops itself.
964 /// This does not remove the need to handle all loop nest handling by
965 /// directives, such as the ordered(<n>) clause or the simd schedule-clause
966 /// modifier of the worksharing-loop directive.
967 ///
968 /// Example:
969 /// \code
970 /// for (int i = 0; i < 7; ++i) // Canonical loop "i"
971 /// for (int j = 0; j < 9; ++j) // Canonical loop "j"
972 /// body(i, j);
973 /// \endcode
974 ///
975 /// After collapsing with Loops={i,j}, the loop is changed to
976 /// \code
977 /// for (int ij = 0; ij < 63; ++ij) {
978 /// int i = ij / 9;
979 /// int j = ij % 9;
980 /// body(i, j);
981 /// }
982 /// \endcode
983 ///
984 /// In the current implementation, the following limitations apply:
985 ///
986 /// * All input loops have an induction variable of the same type.
987 ///
988 /// * The collapsed loop will have the same trip count integer type as the
989 /// input loops. Therefore it is possible that the collapsed loop cannot
990 /// represent all iterations of the input loops. For instance, assuming a
991 /// 32 bit integer type, and two input loops both iterating 2^16 times, the
992 /// theoretical trip count of the collapsed loop would be 2^32 iteration,
993 /// which cannot be represented in an 32-bit integer. Behavior is undefined
994 /// in this case.
995 ///
996 /// * The trip counts of every input loop must be available at \p ComputeIP.
997 /// Non-rectangular loops are not yet supported.
998 ///
999 /// * At each nest level, code between a surrounding loop and its nested loop
1000 /// is hoisted into the loop body, and such code will be executed more
1001 /// often than before collapsing (or not at all if any inner loop iteration
1002 /// has a trip count of 0). This is permitted by the OpenMP specification.
1003 ///
1004 /// \param DL Debug location for instructions added for collapsing,
1005 /// such as instructions to compute/derive the input loop's
1006 /// induction variables.
1007 /// \param Loops Loops in the loop nest to collapse. Loops are specified
1008 /// from outermost-to-innermost and every control flow of a
1009 /// loop's body must pass through its directly nested loop.
1010 /// \param ComputeIP Where additional instruction that compute the collapsed
1011 /// trip count. If not set, defaults to before the generated
1012 /// loop.
1013 ///
1014 /// \returns The CanonicalLoopInfo object representing the collapsed loop.
1017 InsertPointTy ComputeIP);
1018
1019 /// Get the default alignment value for given target
1020 ///
1021 /// \param TargetTriple Target triple
1022 /// \param Features StringMap which describes extra CPU features
1023 LLVM_ABI static unsigned
1024 getOpenMPDefaultSimdAlign(const Triple &TargetTriple,
1025 const StringMap<bool> &Features);
1026
1027 /// Retrieve (or create if non-existent) the address of a declare
1028 /// target variable, used in conjunction with registerTargetGlobalVariable
1029 /// to create declare target global variables.
1030 ///
1031 /// \param CaptureClause - enumerator corresponding to the OpenMP capture
1032 /// clause used in conjunction with the variable being registered (link,
1033 /// to, enter).
1034 /// \param DeviceClause - enumerator corresponding to the OpenMP capture
1035 /// clause used in conjunction with the variable being registered (nohost,
1036 /// host, any)
1037 /// \param IsDeclaration - boolean stating if the variable being registered
1038 /// is a declaration-only and not a definition
1039 /// \param IsExternallyVisible - boolean stating if the variable is externally
1040 /// visible
1041 /// \param EntryInfo - Unique entry information for the value generated
1042 /// using getTargetEntryUniqueInfo, used to name generated pointer references
1043 /// to the declare target variable
1044 /// \param MangledName - the mangled name of the variable being registered
1045 /// \param GeneratedRefs - references generated by invocations of
1046 /// registerTargetGlobalVariable invoked from getAddrOfDeclareTargetVar,
1047 /// these are required by Clang for book keeping.
1048 /// \param OpenMPSIMD - if OpenMP SIMD mode is currently enabled
1049 /// \param TargetTriple - The OpenMP device target triple we are compiling
1050 /// for
1051 /// \param LlvmPtrTy - The type of the variable we are generating or
1052 /// retrieving an address for
1053 /// \param GlobalInitializer - a lambda function which creates a constant
1054 /// used for initializing a pointer reference to the variable in certain
1055 /// cases. If a nullptr is passed, it will default to utilising the original
1056 /// variable to initialize the pointer reference.
1057 /// \param VariableLinkage - a lambda function which returns the variables
1058 /// linkage type, if unspecified and a nullptr is given, it will instead
1059 /// utilise the linkage stored on the existing global variable in the
1060 /// LLVMModule.
1064 bool IsDeclaration, bool IsExternallyVisible,
1065 TargetRegionEntryInfo EntryInfo, StringRef MangledName,
1066 std::vector<GlobalVariable *> &GeneratedRefs, bool OpenMPSIMD,
1067 std::vector<Triple> TargetTriple, Type *LlvmPtrTy,
1068 std::function<Constant *()> GlobalInitializer,
1069 std::function<GlobalValue::LinkageTypes()> VariableLinkage);
1070
1071 /// Registers a target variable for device or host.
1072 ///
1073 /// \param CaptureClause - enumerator corresponding to the OpenMP capture
1074 /// clause used in conjunction with the variable being registered (link,
1075 /// to, enter).
1076 /// \param DeviceClause - enumerator corresponding to the OpenMP capture
1077 /// clause used in conjunction with the variable being registered (nohost,
1078 /// host, any)
1079 /// \param IsDeclaration - boolean stating if the variable being registered
1080 /// is a declaration-only and not a definition
1081 /// \param IsExternallyVisible - boolean stating if the variable is externally
1082 /// visible
1083 /// \param EntryInfo - Unique entry information for the value generated
1084 /// using getTargetEntryUniqueInfo, used to name generated pointer references
1085 /// to the declare target variable
1086 /// \param MangledName - the mangled name of the variable being registered
1087 /// \param GeneratedRefs - references generated by invocations of
1088 /// registerTargetGlobalVariable these are required by Clang for book
1089 /// keeping.
1090 /// \param OpenMPSIMD - if OpenMP SIMD mode is currently enabled
1091 /// \param TargetTriple - The OpenMP device target triple we are compiling
1092 /// for
1093 /// \param GlobalInitializer - a lambda function which creates a constant
1094 /// used for initializing a pointer reference to the variable in certain
1095 /// cases. If a nullptr is passed, it will default to utilising the original
1096 /// variable to initialize the pointer reference.
1097 /// \param VariableLinkage - a lambda function which returns the variables
1098 /// linkage type, if unspecified and a nullptr is given, it will instead
1099 /// utilise the linkage stored on the existing global variable in the
1100 /// LLVMModule.
1101 /// \param LlvmPtrTy - The type of the variable we are generating or
1102 /// retrieving an address for
1103 /// \param Addr - the original llvm value (addr) of the variable to be
1104 /// registered
1108 bool IsDeclaration, bool IsExternallyVisible,
1109 TargetRegionEntryInfo EntryInfo, StringRef MangledName,
1110 std::vector<GlobalVariable *> &GeneratedRefs, bool OpenMPSIMD,
1111 std::vector<Triple> TargetTriple,
1112 std::function<Constant *()> GlobalInitializer,
1113 std::function<GlobalValue::LinkageTypes()> VariableLinkage,
1114 Type *LlvmPtrTy, Constant *Addr);
1115
1116 /// Register a module-scope replacement of a declare target global variable.
1117 /// During lowering new globals are generated for certain combinations of
1118 /// declare target input, and these new globals require substitution with
1119 /// the originals. This replacement occurs during finalization where uses
1120 /// of \p Original are rewritten to reference \p Replacement. This is
1121 /// predominantly required for lowering through the LLVM-IR + OpenMP
1122 /// dialect infrastructure where the lowering pattern to LLVM-IR prevents
1123 /// immediate use rewrites.
1124 ///
1125 /// This infrastructure is utilised for the device pass only currently,
1126 /// and host passes will skip the finalization process even if replacements
1127 /// are registered.
1128 ///
1129 /// \param Original - The original global variable that will be replaced.
1130 /// \param Replacement - The replacement reference pointer generated by the
1131 /// declare target infrastructure (declare target link or unified shared
1132 /// memory globals).
1133 LLVM_ABI void
1135 GlobalValue *Replacement);
1136
1137 /// Get the offset of the OMP_MAP_MEMBER_OF field.
1138 LLVM_ABI unsigned getFlagMemberOffset();
1139
1140 /// Get OMP_MAP_MEMBER_OF flag with extra bits reserved based on
1141 /// the position given.
1142 /// \param Position - A value indicating the position of the parent
1143 /// of the member in the kernel argument structure, often retrieved
1144 /// by the parents position in the combined information vectors used
1145 /// to generate the structure itself. Multiple children (member's of)
1146 /// with the same parent will use the same returned member flag.
1148
1149 /// Given an initial flag set, this function modifies it to contain
1150 /// the passed in MemberOfFlag generated from the getMemberOfFlag
1151 /// function. The results are dependent on the existing flag bits
1152 /// set in the original flag set.
1153 /// \param Flags - The original set of flags to be modified with the
1154 /// passed in MemberOfFlag.
1155 /// \param MemberOfFlag - A modified OMP_MAP_MEMBER_OF flag, adjusted
1156 /// slightly based on the getMemberOfFlag which adjusts the flag bits
1157 /// based on the members position in its parent.
1158 LLVM_ABI void
1160 omp::OpenMPOffloadMappingFlags MemberOfFlag);
1161
1162private:
1163 /// Modifies the canonical loop to be a statically-scheduled workshare loop
1164 /// which is executed on the device
1165 ///
1166 /// This takes a \p CLI representing a canonical loop, such as the one
1167 /// created by \see createCanonicalLoop and emits additional instructions to
1168 /// turn it into a workshare loop. In particular, it calls to an OpenMP
1169 /// runtime function in the preheader to call OpenMP device rtl function
1170 /// which handles worksharing of loop body interations.
1171 ///
1172 /// \param DL Debug location for instructions added for the
1173 /// workshare-loop construct itself.
1174 /// \param CLI A descriptor of the canonical loop to workshare.
1175 /// \param AllocaIP An insertion point for Alloca instructions usable in the
1176 /// preheader of the loop.
1177 /// \param LoopType Information about type of loop worksharing.
1178 /// It corresponds to type of loop workshare OpenMP pragma.
1179 /// \param NoLoop If true, no-loop code is generated.
1180 ///
1181 /// \returns Point where to insert code after the workshare construct.
1182 InsertPointTy applyWorkshareLoopTarget(DebugLoc DL, CanonicalLoopInfo *CLI,
1183 InsertPointTy AllocaIP,
1184 omp::WorksharingLoopType LoopType,
1185 bool NoLoop);
1186
1187 /// Modifies the canonical loop to be a statically-scheduled workshare loop.
1188 ///
1189 /// This takes a \p LoopInfo representing a canonical loop, such as the one
1190 /// created by \p createCanonicalLoop and emits additional instructions to
1191 /// turn it into a workshare loop. In particular, it calls to an OpenMP
1192 /// runtime function in the preheader to obtain the loop bounds to be used in
1193 /// the current thread, updates the relevant instructions in the canonical
1194 /// loop and calls to an OpenMP runtime finalization function after the loop.
1195 ///
1196 /// \param DL Debug location for instructions added for the
1197 /// workshare-loop construct itself.
1198 /// \param CLI A descriptor of the canonical loop to workshare.
1199 /// \param AllocaIP An insertion point for Alloca instructions usable in the
1200 /// preheader of the loop.
1201 /// \param NeedsBarrier Indicates whether a barrier must be inserted after
1202 /// the loop.
1203 /// \param LoopType Type of workshare loop.
1204 /// \param HasDistSchedule Defines if the clause being lowered is
1205 /// dist_schedule as this is handled slightly differently
1206 /// \param DistScheduleSchedType Defines the Schedule Type for the Distribute
1207 /// loop. Defaults to None if no Distribute loop is present.
1208 ///
1209 /// \returns Point where to insert code after the workshare construct.
1210 InsertPointOrErrorTy applyStaticWorkshareLoop(
1212 omp::WorksharingLoopType LoopType, bool NeedsBarrier,
1213 bool HasDistSchedule = false,
1214 omp::OMPScheduleType DistScheduleSchedType = omp::OMPScheduleType::None);
1215
1216 /// Modifies the canonical loop a statically-scheduled workshare loop with a
1217 /// user-specified chunk size.
1218 ///
1219 /// \param DL Debug location for instructions added for the
1220 /// workshare-loop construct itself.
1221 /// \param CLI A descriptor of the canonical loop to workshare.
1222 /// \param AllocaIP An insertion point for Alloca instructions usable in
1223 /// the preheader of the loop.
1224 /// \param NeedsBarrier Indicates whether a barrier must be inserted after the
1225 /// loop.
1226 /// \param ChunkSize The user-specified chunk size.
1227 /// \param SchedType Optional type of scheduling to be passed to the init
1228 /// function.
1229 /// \param DistScheduleChunkSize The size of dist_shcedule chunk considered
1230 /// as a unit when
1231 /// scheduling. If \p nullptr, defaults to 1.
1232 /// \param DistScheduleSchedType Defines the Schedule Type for the Distribute
1233 /// loop. Defaults to None if no Distribute loop is present.
1234 ///
1235 /// \returns Point where to insert code after the workshare construct.
1236 InsertPointOrErrorTy applyStaticChunkedWorkshareLoop(
1238 bool NeedsBarrier, Value *ChunkSize,
1239 omp::OMPScheduleType SchedType =
1241 Value *DistScheduleChunkSize = nullptr,
1242 omp::OMPScheduleType DistScheduleSchedType = omp::OMPScheduleType::None);
1243
1244 /// Modifies the canonical loop to be a dynamically-scheduled workshare loop.
1245 ///
1246 /// This takes a \p LoopInfo representing a canonical loop, such as the one
1247 /// created by \p createCanonicalLoop and emits additional instructions to
1248 /// turn it into a workshare loop. In particular, it calls to an OpenMP
1249 /// runtime function in the preheader to obtain, and then in each iteration
1250 /// to update the loop counter.
1251 ///
1252 /// \param DL Debug location for instructions added for the
1253 /// workshare-loop construct itself.
1254 /// \param CLI A descriptor of the canonical loop to workshare.
1255 /// \param AllocaIP An insertion point for Alloca instructions usable in the
1256 /// preheader of the loop.
1257 /// \param SchedType Type of scheduling to be passed to the init function.
1258 /// \param NeedsBarrier Indicates whether a barrier must be insterted after
1259 /// the loop.
1260 /// \param Chunk The size of loop chunk considered as a unit when
1261 /// scheduling. If \p nullptr, defaults to 1.
1262 ///
1263 /// \returns Point where to insert code after the workshare construct.
1264 InsertPointOrErrorTy applyDynamicWorkshareLoop(DebugLoc DL,
1265 CanonicalLoopInfo *CLI,
1266 InsertPointTy AllocaIP,
1267 omp::OMPScheduleType SchedType,
1268 bool NeedsBarrier,
1269 Value *Chunk = nullptr);
1270
1271 /// Create alternative version of the loop to support if clause
1272 ///
1273 /// OpenMP if clause can require to generate second loop. This loop
1274 /// will be executed when if clause condition is not met. createIfVersion
1275 /// adds branch instruction to the copied loop if \p ifCond is not met.
1276 ///
1277 /// \param Loop Original loop which should be versioned.
1278 /// \param IfCond Value which corresponds to if clause condition
1279 /// \param VMap Value to value map to define relation between
1280 /// original and copied loop values and loop blocks.
1281 /// \param NamePrefix Optional name prefix for if.then if.else blocks.
1282 void createIfVersion(CanonicalLoopInfo *Loop, Value *IfCond,
1284 LoopAnalysis &LIA, LoopInfo &LI, llvm::Loop *L,
1285 const Twine &NamePrefix = "");
1286
1287 /// Creates a task duplication function to be passed to kmpc_taskloop.
1288 ///
1289 /// The OpenMP runtime defines this function as taking the destination
1290 /// kmp_task_t, source kmp_task_t, and a lastprivate flag. This function is
1291 /// called on the source and destination tasks after the source task has been
1292 /// duplicated to create the destination task. At this point the destination
1293 /// task has been otherwise set up from the runtime's perspective, but this
1294 /// function is needed to fix up any data for the duplicated task e.g. private
1295 /// variables with non-trivial constructors.
1296 ///
1297 /// \param PrivatesTy The type of the privates structure for the task.
1298 /// \param PrivatesIndex The index inside the privates structure containing
1299 /// the data for the callback.
1300 /// \param DupCB The callback to generate the duplication code. See
1301 /// documentation for \ref TaskDupCallbackTy. This can be
1302 /// nullptr.
1303 Expected<Value *> createTaskDuplicationFunction(Type *PrivatesTy,
1304 int32_t PrivatesIndex,
1305 TaskDupCallbackTy DupCB);
1306
1307public:
1308 /// Modifies the canonical loop to be a workshare loop.
1309 ///
1310 /// This takes a \p LoopInfo representing a canonical loop, such as the one
1311 /// created by \p createCanonicalLoop and emits additional instructions to
1312 /// turn it into a workshare loop. In particular, it calls to an OpenMP
1313 /// runtime function in the preheader to obtain the loop bounds to be used in
1314 /// the current thread, updates the relevant instructions in the canonical
1315 /// loop and calls to an OpenMP runtime finalization function after the loop.
1316 ///
1317 /// The concrete transformation is done by applyStaticWorkshareLoop,
1318 /// applyStaticChunkedWorkshareLoop, or applyDynamicWorkshareLoop, depending
1319 /// on the value of \p SchedKind and \p ChunkSize.
1320 ///
1321 /// \param DL Debug location for instructions added for the
1322 /// workshare-loop construct itself.
1323 /// \param CLI A descriptor of the canonical loop to workshare.
1324 /// \param AllocaIP An insertion point for Alloca instructions usable in the
1325 /// preheader of the loop.
1326 /// \param NeedsBarrier Indicates whether a barrier must be insterted after
1327 /// the loop.
1328 /// \param SchedKind Scheduling algorithm to use.
1329 /// \param ChunkSize The chunk size for the inner loop.
1330 /// \param HasSimdModifier Whether the simd modifier is present in the
1331 /// schedule clause.
1332 /// \param HasMonotonicModifier Whether the monotonic modifier is present in
1333 /// the schedule clause.
1334 /// \param HasNonmonotonicModifier Whether the nonmonotonic modifier is
1335 /// present in the schedule clause.
1336 /// \param HasOrderedClause Whether the (parameterless) ordered clause is
1337 /// present.
1338 /// \param LoopType Information about type of loop worksharing.
1339 /// It corresponds to type of loop workshare OpenMP pragma.
1340 /// \param NoLoop If true, no-loop code is generated.
1341 /// \param HasDistSchedule Defines if the clause being lowered is
1342 /// dist_schedule as this is handled slightly differently
1343 ///
1344 /// \param DistScheduleChunkSize The chunk size for dist_schedule loop
1345 ///
1346 /// \returns Point where to insert code after the workshare construct.
1349 bool NeedsBarrier,
1350 llvm::omp::ScheduleKind SchedKind = llvm::omp::OMP_SCHEDULE_Default,
1351 Value *ChunkSize = nullptr, bool HasSimdModifier = false,
1352 bool HasMonotonicModifier = false, bool HasNonmonotonicModifier = false,
1353 bool HasOrderedClause = false,
1354 omp::WorksharingLoopType LoopType =
1356 bool NoLoop = false, bool HasDistSchedule = false,
1357 Value *DistScheduleChunkSize = nullptr);
1358
1359 /// Tile a loop nest.
1360 ///
1361 /// Tiles the loops of \p Loops by the tile sizes in \p TileSizes. Loops in
1362 /// \p/ Loops must be perfectly nested, from outermost to innermost loop
1363 /// (i.e. Loops.front() is the outermost loop). The trip count llvm::Value
1364 /// of every loop and every tile sizes must be usable in the outermost
1365 /// loop's preheader. This implies that the loop nest is rectangular.
1366 ///
1367 /// Example:
1368 /// \code
1369 /// for (int i = 0; i < 15; ++i) // Canonical loop "i"
1370 /// for (int j = 0; j < 14; ++j) // Canonical loop "j"
1371 /// body(i, j);
1372 /// \endcode
1373 ///
1374 /// After tiling with Loops={i,j} and TileSizes={5,7}, the loop is changed to
1375 /// \code
1376 /// for (int i1 = 0; i1 < 3; ++i1)
1377 /// for (int j1 = 0; j1 < 2; ++j1)
1378 /// for (int i2 = 0; i2 < 5; ++i2)
1379 /// for (int j2 = 0; j2 < 7; ++j2)
1380 /// body(i1*3+i2, j1*3+j2);
1381 /// \endcode
1382 ///
1383 /// The returned vector are the loops {i1,j1,i2,j2}. The loops i1 and j1 are
1384 /// referred to the floor, and the loops i2 and j2 are the tiles. Tiling also
1385 /// handles non-constant trip counts, non-constant tile sizes and trip counts
1386 /// that are not multiples of the tile size. In the latter case the tile loop
1387 /// of the last floor-loop iteration will have fewer iterations than specified
1388 /// as its tile size.
1389 ///
1390 ///
1391 /// @param DL Debug location for instructions added by tiling, for
1392 /// instance the floor- and tile trip count computation.
1393 /// @param Loops Loops to tile. The CanonicalLoopInfo objects are
1394 /// invalidated by this method, i.e. should not used after
1395 /// tiling.
1396 /// @param TileSizes For each loop in \p Loops, the tile size for that
1397 /// dimensions.
1398 ///
1399 /// \returns A list of generated loops. Contains twice as many loops as the
1400 /// input loop nest; the first half are the floor loops and the
1401 /// second half are the tile loops.
1402 LLVM_ABI std::vector<CanonicalLoopInfo *>
1404 ArrayRef<Value *> TileSizes);
1405
1406 /// Fuse a sequence of loops.
1407 ///
1408 /// Fuses the loops of \p Loops.
1409 /// The merging of the loops is done in the following structure:
1410 ///
1411 /// Example:
1412 /// \code
1413 /// for (int i = lb0; i < ub0; i += st0) // trip count is calculated as:
1414 /// body(i) // tc0 = (ub0 - lb0 + st0) / st0
1415 /// for (int j = lb1; j < ub1; j += st1)
1416 /// body(j);
1417 ///
1418 /// ...
1419 ///
1420 /// for (int k = lbk; j < ubk; j += stk)
1421 /// body(k);
1422 /// \endcode
1423 ///
1424 /// After fusing the loops a single loop is left:
1425 /// \code
1426 /// for (fuse.index = 0; fuse.index < max(tc0, tc1, ... tck); ++fuse.index) {
1427 /// if (fuse.index < tc0){
1428 /// iv0 = lb0 + st0 * fuse.index;
1429 /// original.index0 = iv0
1430 /// body(0);
1431 /// }
1432 /// if (fuse.index < tc1){
1433 /// iv1 = lb1 + st1 * fuse.index;
1434 /// original.index1 = iv1
1435 /// body(1);
1436 /// }
1437 ///
1438 /// ...
1439 ///
1440 /// if (fuse.index < tck){
1441 /// ivk = lbk + stk * fuse.index;
1442 /// original.indexk = ivk
1443 /// body(k);
1444 /// }
1445 /// }
1446 /// \endcode
1447 ///
1448 ///
1449 /// @param DL Debug location for instructions added by fusion.
1450 ///
1451 /// @param Loops Loops to fuse. The CanonicalLoopInfo objects are
1452 /// invalidated by this method, i.e. should not used after
1453 /// fusion.
1454 ///
1455 /// \returns A single loop generated by the loop fusion
1458
1459 /// Fully unroll a loop.
1460 ///
1461 /// Instead of unrolling the loop immediately (and duplicating its body
1462 /// instructions), it is deferred to LLVM's LoopUnrollPass by adding loop
1463 /// metadata.
1464 ///
1465 /// \param DL Debug location for instructions added by unrolling.
1466 /// \param Loop The loop to unroll. The loop will be invalidated.
1468
1469 /// Fully or partially unroll a loop. How the loop is unrolled is determined
1470 /// using LLVM's LoopUnrollPass.
1471 ///
1472 /// \param DL Debug location for instructions added by unrolling.
1473 /// \param Loop The loop to unroll. The loop will be invalidated.
1475
1476 /// Partially unroll a loop.
1477 ///
1478 /// The CanonicalLoopInfo of the unrolled loop for use with chained
1479 /// loop-associated directive can be requested using \p UnrolledCLI. Not
1480 /// needing the CanonicalLoopInfo allows more efficient code generation by
1481 /// deferring the actual unrolling to the LoopUnrollPass using loop metadata.
1482 /// A loop-associated directive applied to the unrolled loop needs to know the
1483 /// new trip count which means that if using a heuristically determined unroll
1484 /// factor (\p Factor == 0), that factor must be computed immediately. We are
1485 /// using the same logic as the LoopUnrollPass to derived the unroll factor,
1486 /// but which assumes that some canonicalization has taken place (e.g.
1487 /// Mem2Reg, LICM, GVN, Inlining, etc.). That is, the heuristic will perform
1488 /// better when the unrolled loop's CanonicalLoopInfo is not needed.
1489 ///
1490 /// \param DL Debug location for instructions added by unrolling.
1491 /// \param Loop The loop to unroll. The loop will be invalidated.
1492 /// \param Factor The factor to unroll the loop by. A factor of 0
1493 /// indicates that a heuristic should be used to determine
1494 /// the unroll-factor.
1495 /// \param UnrolledCLI If non-null, receives the CanonicalLoopInfo of the
1496 /// partially unrolled loop. Otherwise, uses loop metadata
1497 /// to defer unrolling to the LoopUnrollPass.
1499 int32_t Factor,
1500 CanonicalLoopInfo **UnrolledCLI);
1501
1502 /// Add metadata to simd-ize a loop. If IfCond is not nullptr, the loop
1503 /// is cloned. The metadata which prevents vectorization is added to
1504 /// to the cloned loop. The cloned loop is executed when ifCond is evaluated
1505 /// to false.
1506 ///
1507 /// \param Loop The loop to simd-ize.
1508 /// \param AlignedVars The map which containts pairs of the pointer
1509 /// and its corresponding alignment.
1510 /// \param IfCond The value which corresponds to the if clause
1511 /// condition.
1512 /// \param Order The enum to map order clause.
1513 /// \param Simdlen The Simdlen length to apply to the simd loop.
1514 /// \param Safelen The Safelen length to apply to the simd loop.
1516 MapVector<Value *, Value *> AlignedVars,
1517 Value *IfCond, omp::OrderKind Order,
1518 ConstantInt *Simdlen, ConstantInt *Safelen);
1519
1520 /// Generator for '#omp flush'
1521 ///
1522 /// \param Loc The location where the flush directive was encountered
1523 LLVM_ABI void createFlush(const LocationDescription &Loc);
1524
1525 /// Generate a call to the runtime to emit the diagnostic of an OpenMP
1526 /// `error` directive with `at(execution)`.
1527 ///
1528 /// \param Loc The location where the error directive was encountered; it is
1529 /// used to build the `ident_t` passed to the runtime.
1530 /// \param IsFatal Selects `severity(fatal)` (true) or `severity(warning)`
1531 /// (false).
1532 /// \param Message The message string (an `i8*`) to display, or null when no
1533 /// `message` clause is present.
1534 LLVM_ABI void createError(const LocationDescription &Loc, bool IsFatal,
1535 Value *Message);
1536
1537 /// Generator for '#omp taskyield'
1538 ///
1539 /// \param Loc The location where the taskyield directive was encountered.
1540 LLVM_ABI void createTaskyield(const LocationDescription &Loc);
1541
1542 /// A struct to pack the relevant information for an OpenMP depend clause.
1552
1553 /// A struct to pack static and dynamic dependency information for a task.
1554 ///
1555 /// For fixed-count (non-iterator) dependencies, callers populate \p Deps
1556 /// and the builder allocates and fills the kmp_depend_info array internally.
1557 /// For iterator-based dependencies, the caller pre-builds the array and
1558 /// sets \p NumDeps and \p DepArray directly.
1560 SmallVector<DependData> Deps; // vector of dependencies
1561 Value *NumDeps; // number of kmp_depend_info entries (used by iterator path)
1562 Value *DepArray; // kmp_depend_info array (used by iterator path)
1563
1564 DependenciesInfo() : Deps(), NumDeps(nullptr), DepArray(nullptr) {}
1567
1568 bool empty() const { return Deps.empty() && DepArray == nullptr; }
1569 };
1570
1571 /// Store one kmp_depend_info entry at the given \p Entry pointer.
1572 LLVM_ABI void emitTaskDependency(IRBuilderBase &Builder, Value *Entry,
1573 const DependData &Dep);
1574
1575 /// Generator for '#omp taskwait'
1576 ///
1577 /// \param Loc The location where the taskwait directive was encountered.
1578 /// \param Dependencies dependencies as specified by the 'depend' clause.
1579 LLVM_ABI void createTaskwait(const LocationDescription &Loc,
1580 DependenciesInfo Dependencies = {});
1581
1582 /// Return the LLVM struct type matching runtime `kmp_task_affinity_info_t`.
1583 /// `{ kmp_intptr_t base_addr; size_t len; flags (bitfield storage as i32) }`
1585
1586 /// A struct to pack the relevant information for an OpenMP affinity clause.
1588 Value *Count; // number of kmp_task_affinity_info_t entries
1589 Value *Info; // kmp_task_affinity_info_t
1590 };
1591
1592 /// Generator for `#omp taskloop`
1593 ///
1594 /// \param Loc The location where the taskloop construct was encountered.
1595 /// \param AllocaIP The insertion point to be used for alloca instructions.
1596 /// \param DeallocBlocks The list of insertion blocks where explicit
1597 /// deallocations, if needed, should be placed.
1598 /// \param BodyGenCB Callback that will generate the region code.
1599 /// \param LoopInfo Callback that return the CLI
1600 /// \param LBVal Lowerbound value of loop
1601 /// \param UBVal Upperbound value of loop
1602 /// \param StepVal Step value of loop
1603 /// \param Untied True if the task is untied, false if the task is tied.
1604 /// \param IfCond i1 value. If it evaluates to `false`, an undeferred
1605 /// task is generated, and the encountering thread must
1606 /// suspend the current task region, for which execution
1607 /// cannot be resumed until execution of the structured
1608 /// block that is associated with the generated task is
1609 /// completed.
1610 /// \param GrainSize Value of the GrainSize/Num of Tasks if present
1611 /// \param NoGroup False if NoGroup is defined, true if not
1612 /// \param Sched If Grainsize is defined, Sched is 1. Num Tasks, Shed is 2.
1613 /// Otherwise Sched is 0
1614 /// \param Final i1 value which is `true` if the task is final, `false` if the
1615 /// task is not final.
1616 /// \param Mergeable If the given task is `mergeable`
1617 /// \param Priority `priority-value' specifies the execution order of the
1618 /// tasks that is generated by the construct
1619 /// \param NumOfCollapseLoops Defines the number of loops that are being
1620 /// collapsed. The default value is 1, as thats the value when collapse is not
1621 /// used.
1622 /// \param DupCB The callback to generate the duplication code. See
1623 /// documentation for \ref TaskDupCallbackTy. This can be nullptr.
1624 /// \param TaskContextStructPtrVal If non-null, a pointer to to be placed
1625 /// immediately after the {lower bound, upper
1626 /// bound, step} values in the task data.
1627 /// \param FreeAgent If `true`, the generated tasks are eligible to be
1628 /// executed by a free-agent thread (threadset(omp_pool)).
1629 LLVM_ABI InsertPointOrErrorTy createTaskloop(
1630 const LocationDescription &Loc, InsertPointTy AllocaIP,
1631 ArrayRef<BasicBlock *> DeallocBlocks, BodyGenCallbackTy BodyGenCB,
1633 Value *LBVal, Value *UBVal, Value *StepVal, bool Untied = false,
1634 Value *IfCond = nullptr, Value *GrainSize = nullptr, bool NoGroup = false,
1635 int Sched = 0, Value *Final = nullptr, bool Mergeable = false,
1636 Value *Priority = nullptr, uint64_t NumOfCollapseLoops = 1,
1637 TaskDupCallbackTy DupCB = nullptr,
1638 Value *TaskContextStructPtrVal = nullptr, bool FreeAgent = false);
1639
1640 /// Generator for `#omp task`
1641 ///
1642 /// \param Loc The location where the task construct was encountered.
1643 /// \param AllocaIP The insertion point to be used for allocations.
1644 /// \param DeallocBlocks The insertion blocks to be used for explicit
1645 /// deallocations, if needed.
1646 /// \param BodyGenCB Callback that will generate the region code.
1647 /// \param Tied True if the task is tied, false if the task is untied.
1648 /// \param Final i1 value which is `true` if the task is final, `false` if the
1649 /// task is not final.
1650 /// \param IfCondition i1 value. If it evaluates to `false`, an undeferred
1651 /// task is generated, and the encountering thread must
1652 /// suspend the current task region, for which execution
1653 /// cannot be resumed until execution of the structured
1654 /// block that is associated with the generated task is
1655 /// completed.
1656 /// \param Dependencies Dependencies info holding either a vector of
1657 /// DependData objects or a pre-built dependency array.
1658 /// \param Affinities AffinityData object holding information of accumulated
1659 /// affinities as specified by the 'affinity' clause.
1660 /// \param EventHandle If present, signifies the event handle as part of
1661 /// the detach clause
1662 /// \param Mergeable If the given task is `mergeable`
1663 /// \param priority `priority-value' specifies the execution order of the
1664 /// tasks that is generated by the construct
1665 /// \param FreeAgent If `true`, the task is eligible to be executed by a
1666 /// free-agent thread (threadset(omp_pool)).
1668 const LocationDescription &Loc, InsertPointTy AllocaIP,
1669 ArrayRef<BasicBlock *> DeallocBlocks, BodyGenCallbackTy BodyGenCB,
1670 bool Tied = true, Value *Final = nullptr, Value *IfCondition = nullptr,
1671 const DependenciesInfo &Dependencies = {},
1672 const AffinityData &Affinities = {}, bool Mergeable = false,
1673 Value *EventHandle = nullptr, Value *Priority = nullptr,
1674 bool FreeAgent = false);
1675
1676 /// Generator for the taskgroup construct
1677 ///
1678 /// \param Loc The location where the taskgroup construct was encountered.
1679 /// \param AllocaIP The insertion point to be used for allocations.
1680 /// \param DeallocBlocks The insertion blocks to be used for explicit
1681 /// deallocation instructions, if needed.
1682 /// \param BodyGenCB Callback that will generate the region code.
1684 const LocationDescription &Loc, InsertPointTy AllocaIP,
1685 ArrayRef<BasicBlock *> DeallocBlocks, BodyGenCallbackTy BodyGenCB);
1686
1688 std::function<std::tuple<std::string, uint64_t>()>;
1689
1690 /// Creates a unique info for a target entry when provided a filename and
1691 /// line number from.
1692 ///
1693 /// \param CallBack A callback function which should return filename the entry
1694 /// resides in as well as the line number for the target entry
1695 /// \param ParentName The name of the parent the target entry resides in, if
1696 /// any.
1699 vfs::FileSystem &VFS, StringRef ParentName = "");
1700
1701 /// Enum class for the RedctionGen CallBack type to be used.
1703
1704 /// ReductionGen CallBack for Clang
1705 ///
1706 /// \param CodeGenIP InsertPoint for CodeGen.
1707 /// \param Index Index of the ReductionInfo to generate code for.
1708 /// \param LHSPtr Optionally used by Clang to return the LHSPtr it used for
1709 /// codegen, used for fixup later.
1710 /// \param RHSPtr Optionally used by Clang to
1711 /// return the RHSPtr it used for codegen, used for fixup later.
1712 /// \param CurFn Optionally used by Clang to pass in the Current Function as
1713 /// Clang context may be old.
1715 std::function<InsertPointTy(InsertPointTy CodeGenIP, unsigned Index,
1716 Value **LHS, Value **RHS, Function *CurFn)>;
1717
1718 /// ReductionGen CallBack for MLIR
1719 ///
1720 /// \param CodeGenIP InsertPoint for CodeGen.
1721 /// \param LHS Pass in the LHS Value to be used for CodeGen.
1722 /// \param RHS Pass in the RHS Value to be used for CodeGen.
1724 InsertPointTy CodeGenIP, Value *LHS, Value *RHS, Value *&Res)>;
1725
1726 /// Functions used to generate atomic reductions. Such functions take two
1727 /// Values representing pointers to LHS and RHS of the reduction, as well as
1728 /// the element type of these pointers. They are expected to atomically
1729 /// update the LHS to the reduced value.
1731 InsertPointTy, Type *, Value *, Value *)>;
1732
1734 InsertPointTy, Value *ByRefVal, Value *&Res)>;
1735
1736 /// Enum class for reduction evaluation types scalar, complex and aggregate.
1738
1739 /// Information about an OpenMP reduction.
1754
1760
1761 /// Reduction element type, must match pointee type of variable. For by-ref
1762 /// reductions, this would be just an opaque `ptr`.
1764
1765 /// Reduction variable of pointer type.
1767
1768 /// Thread-private partial reduction variable.
1770
1771 /// Reduction evaluation kind - scalar, complex or aggregate.
1773
1774 /// Callback for generating the reduction body. The IR produced by this will
1775 /// be used to combine two values in a thread-safe context, e.g., under
1776 /// lock or within the same thread, and therefore need not be atomic.
1778
1779 /// Clang callback for generating the reduction body. The IR produced by
1780 /// this will be used to combine two values in a thread-safe context, e.g.,
1781 /// under lock or within the same thread, and therefore need not be atomic.
1783
1784 /// Callback for generating the atomic reduction body, may be null. The IR
1785 /// produced by this will be used to atomically combine two values during
1786 /// reduction. If null, the implementation will use the non-atomic version
1787 /// along with the appropriate synchronization mechanisms.
1789
1791
1792 /// For by-ref reductions, we need to keep track of 2 extra types that are
1793 /// potentially different:
1794 /// * The allocated type is the type of the storage allocated by the
1795 /// reduction op's `alloc` region. For example, for allocatables and arrays,
1796 /// this type would be the descriptor/box struct.
1798
1799 /// * The by-ref element type is the type of the actual storage needed for
1800 /// the data of the allocatable or array. For example, an float allocatable
1801 /// of would need some float storage to store intermediate reduction
1802 /// results.
1804 };
1805
1806 enum class CopyAction : unsigned {
1807 // RemoteLaneToThread: Copy over a Reduce list from a remote lane in
1808 // the warp using shuffle instructions.
1810 // ThreadCopy: Make a copy of a Reduce list on the thread's stack.
1812 };
1813
1819
1820 /// Supporting functions for Reductions CodeGen.
1821private:
1822 /// Get the id of the current thread on the GPU.
1823 Value *getGPUThreadID();
1824
1825 /// Get the GPU warp size.
1826 Value *getGPUWarpSize();
1827
1828 /// Get the id of the warp in the block.
1829 /// We assume that the warp size is 32, which is always the case
1830 /// on the NVPTX device, to generate more efficient code.
1831 Value *getNVPTXWarpID();
1832
1833 /// Get the id of the current lane in the Warp.
1834 /// We assume that the warp size is 32, which is always the case
1835 /// on the NVPTX device, to generate more efficient code.
1836 Value *getNVPTXLaneID();
1837
1838 /// Cast value to the specified type.
1839 Value *castValueToType(InsertPointTy AllocaIP, Value *From, Type *ToType);
1840
1841 /// This function creates calls to one of two shuffle functions to copy
1842 /// variables between lanes in a warp. The returned value has \p ElementType,
1843 /// even though the shuffle runtime functions operate on 32- or 64-bit values.
1844 Value *createRuntimeShuffleFunction(InsertPointTy AllocaIP, Value *Element,
1845 Type *ElementType, Value *Offset);
1846
1847 /// Function to shuffle over the value from the remote lane.
1848 void shuffleAndStore(InsertPointTy AllocaIP, Value *SrcAddr, Value *DstAddr,
1849 Type *ElementType, Value *Offset, Type *ReductionArrayTy,
1850 bool IsByRefElem);
1851
1852 /// Emit instructions to copy a Reduce list, which contains partially
1853 /// aggregated values, in the specified direction.
1854 Error emitReductionListCopy(
1855 InsertPointTy AllocaIP, CopyAction Action, Type *ReductionArrayTy,
1856 ArrayRef<ReductionInfo> ReductionInfos, Value *SrcBase, Value *DestBase,
1857 ArrayRef<bool> IsByRef,
1858 CopyOptionsTy CopyOptions = {nullptr, nullptr, nullptr});
1859
1860 /// Emit a helper that reduces data across two OpenMP threads (lanes)
1861 /// in the same warp. It uses shuffle instructions to copy over data from
1862 /// a remote lane's stack. The reduction algorithm performed is specified
1863 /// by the fourth parameter.
1864 ///
1865 /// Algorithm Versions.
1866 /// Full Warp Reduce (argument value 0):
1867 /// This algorithm assumes that all 32 lanes are active and gathers
1868 /// data from these 32 lanes, producing a single resultant value.
1869 /// Contiguous Partial Warp Reduce (argument value 1):
1870 /// This algorithm assumes that only a *contiguous* subset of lanes
1871 /// are active. This happens for the last warp in a parallel region
1872 /// when the user specified num_threads is not an integer multiple of
1873 /// 32. This contiguous subset always starts with the zeroth lane.
1874 /// Partial Warp Reduce (argument value 2):
1875 /// This algorithm gathers data from any number of lanes at any position.
1876 /// All reduced values are stored in the lowest possible lane. The set
1877 /// of problems every algorithm addresses is a super set of those
1878 /// addressable by algorithms with a lower version number. Overhead
1879 /// increases as algorithm version increases.
1880 ///
1881 /// Terminology
1882 /// Reduce element:
1883 /// Reduce element refers to the individual data field with primitive
1884 /// data types to be combined and reduced across threads.
1885 /// Reduce list:
1886 /// Reduce list refers to a collection of local, thread-private
1887 /// reduce elements.
1888 /// Remote Reduce list:
1889 /// Remote Reduce list refers to a collection of remote (relative to
1890 /// the current thread) reduce elements.
1891 ///
1892 /// We distinguish between three states of threads that are important to
1893 /// the implementation of this function.
1894 /// Alive threads:
1895 /// Threads in a warp executing the SIMT instruction, as distinguished from
1896 /// threads that are inactive due to divergent control flow.
1897 /// Active threads:
1898 /// The minimal set of threads that has to be alive upon entry to this
1899 /// function. The computation is correct iff active threads are alive.
1900 /// Some threads are alive but they are not active because they do not
1901 /// contribute to the computation in any useful manner. Turning them off
1902 /// may introduce control flow overheads without any tangible benefits.
1903 /// Effective threads:
1904 /// In order to comply with the argument requirements of the shuffle
1905 /// function, we must keep all lanes holding data alive. But at most
1906 /// half of them perform value aggregation; we refer to this half of
1907 /// threads as effective. The other half is simply handing off their
1908 /// data.
1909 ///
1910 /// Procedure
1911 /// Value shuffle:
1912 /// In this step active threads transfer data from higher lane positions
1913 /// in the warp to lower lane positions, creating Remote Reduce list.
1914 /// Value aggregation:
1915 /// In this step, effective threads combine their thread local Reduce list
1916 /// with Remote Reduce list and store the result in the thread local
1917 /// Reduce list.
1918 /// Value copy:
1919 /// In this step, we deal with the assumption made by algorithm 2
1920 /// (i.e. contiguity assumption). When we have an odd number of lanes
1921 /// active, say 2k+1, only k threads will be effective and therefore k
1922 /// new values will be produced. However, the Reduce list owned by the
1923 /// (2k+1)th thread is ignored in the value aggregation. Therefore
1924 /// we copy the Reduce list from the (2k+1)th lane to (k+1)th lane so
1925 /// that the contiguity assumption still holds.
1926 ///
1927 /// \param ReductionInfos Array type containing the ReductionOps.
1928 /// \param ReduceFn The reduction function.
1929 /// \param FuncAttrs Optional param to specify any function attributes that
1930 /// need to be copied to the new function.
1931 /// \param IsByRef For each reduction clause, whether the reduction is by-ref
1932 /// or not.
1933 ///
1934 /// \return The ShuffleAndReduce function.
1935 Expected<Function *> emitShuffleAndReduceFunction(
1937 Function *ReduceFn, AttributeList FuncAttrs, ArrayRef<bool> IsByRef);
1938
1939 /// Helper function for CreateCanonicalScanLoops to create InputLoop
1940 /// in the firstGen and Scan Loop in the SecondGen
1941 /// \param InputLoopGen Callback for generating the loop for input phase
1942 /// \param ScanLoopGen Callback for generating the loop for scan phase
1943 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
1944 /// `ScanInfoInitialize`.
1945 ///
1946 /// \return error if any produced, else return success.
1947 Error emitScanBasedDirectiveIR(
1948 llvm::function_ref<Error()> InputLoopGen,
1949 llvm::function_ref<Error(LocationDescription Loc)> ScanLoopGen,
1950 ScanInfo *ScanRedInfo);
1951
1952 /// Creates the basic blocks required for scan reduction.
1953 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
1954 /// `ScanInfoInitialize`.
1955 void createScanBBs(ScanInfo *ScanRedInfo);
1956
1957 /// Dynamically allocates the buffer needed for scan reduction.
1958 /// \param AllocaIP The IP where possibly-shared pointer of buffer needs to
1959 /// be declared.
1960 /// \param ScanVars Scan Variables.
1961 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
1962 /// `ScanInfoInitialize`.
1963 ///
1964 /// \return error if any produced, else return success.
1965 Error emitScanBasedDirectiveDeclsIR(InsertPointTy AllocaIP,
1966 ArrayRef<llvm::Value *> ScanVars,
1967 ArrayRef<llvm::Type *> ScanVarsType,
1968 ScanInfo *ScanRedInfo);
1969
1970 /// Copies the result back to the reduction variable.
1971 /// \param ReductionInfos Array type containing the ReductionOps.
1972 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
1973 /// `ScanInfoInitialize`.
1974 ///
1975 /// \return error if any produced, else return success.
1976 Error emitScanBasedDirectiveFinalsIR(
1979
1980 /// This function emits a helper that gathers Reduce lists from the first
1981 /// lane of every active warp to lanes in the first warp.
1982 ///
1983 /// void inter_warp_copy_func(void* reduce_data, num_warps)
1984 /// shared smem[warp_size];
1985 /// For all data entries D in reduce_data:
1986 /// sync
1987 /// If (I am the first lane in each warp)
1988 /// Copy my local D to smem[warp_id]
1989 /// sync
1990 /// if (I am the first warp)
1991 /// Copy smem[thread_id] to my local D
1992 ///
1993 /// \param Loc The insert and source location description.
1994 /// \param ReductionInfos Array type containing the ReductionOps.
1995 /// \param FuncAttrs Optional param to specify any function attributes that
1996 /// need to be copied to the new function.
1997 /// \param IsByRef For each reduction clause, whether the reduction is by-ref
1998 /// or not.
1999 ///
2000 /// \return The InterWarpCopy function.
2002 emitInterWarpCopyFunction(const LocationDescription &Loc,
2003 ArrayRef<ReductionInfo> ReductionInfos,
2004 AttributeList FuncAttrs, ArrayRef<bool> IsByRef);
2005
2006 /// This function emits a helper that copies all the reduction variables from
2007 /// the team into the provided global buffer for the reduction variables.
2008 ///
2009 /// void list_to_global_copy_func(void *buffer, int Idx, void *reduce_data)
2010 /// For all data entries D in reduce_data:
2011 /// Copy local D to buffer.D[Idx]
2012 ///
2013 /// \param ReductionInfos Array type containing the ReductionOps.
2014 /// \param ReductionsBufferTy The StructTy for the reductions buffer.
2015 /// \param FuncAttrs Optional param to specify any function attributes that
2016 /// need to be copied to the new function.
2017 ///
2018 /// \return The ListToGlobalCopy function.
2020 emitListToGlobalCopyFunction(ArrayRef<ReductionInfo> ReductionInfos,
2021 Type *ReductionsBufferTy,
2022 AttributeList FuncAttrs, ArrayRef<bool> IsByRef);
2023
2024 /// This function emits a helper that copies all the reduction variables from
2025 /// the team into the provided global buffer for the reduction variables.
2026 ///
2027 /// void list_to_global_copy_func(void *buffer, int Idx, void *reduce_data)
2028 /// For all data entries D in reduce_data:
2029 /// Copy buffer.D[Idx] to local D;
2030 ///
2031 /// \param ReductionInfos Array type containing the ReductionOps.
2032 /// \param ReductionsBufferTy The StructTy for the reductions buffer.
2033 /// \param FuncAttrs Optional param to specify any function attributes that
2034 /// need to be copied to the new function.
2035 ///
2036 /// \return The GlobalToList function.
2038 emitGlobalToListCopyFunction(ArrayRef<ReductionInfo> ReductionInfos,
2039 Type *ReductionsBufferTy,
2040 AttributeList FuncAttrs, ArrayRef<bool> IsByRef);
2041
2042 /// This function emits a helper that reduces all the reduction variables from
2043 /// the team into the provided global buffer for the reduction variables.
2044 ///
2045 /// void list_to_global_reduce_func(void *buffer, int Idx, void *reduce_data)
2046 /// void *GlobPtrs[];
2047 /// GlobPtrs[0] = (void*)&buffer.D0[Idx];
2048 /// ...
2049 /// GlobPtrs[N] = (void*)&buffer.DN[Idx];
2050 /// reduce_function(GlobPtrs, reduce_data);
2051 ///
2052 /// \param ReductionInfos Array type containing the ReductionOps.
2053 /// \param ReduceFn The reduction function.
2054 /// \param ReductionsBufferTy The StructTy for the reductions buffer.
2055 /// \param FuncAttrs Optional param to specify any function attributes that
2056 /// need to be copied to the new function.
2057 ///
2058 /// \return The ListToGlobalReduce function.
2060 emitListToGlobalReduceFunction(ArrayRef<ReductionInfo> ReductionInfos,
2061 Function *ReduceFn, Type *ReductionsBufferTy,
2062 AttributeList FuncAttrs,
2063 ArrayRef<bool> IsByRef);
2064
2065 /// This function emits a helper that reduces all the reduction variables from
2066 /// the team into the provided global buffer for the reduction variables.
2067 ///
2068 /// void global_to_list_reduce_func(void *buffer, int Idx, void *reduce_data)
2069 /// void *GlobPtrs[];
2070 /// GlobPtrs[0] = (void*)&buffer.D0[Idx];
2071 /// ...
2072 /// GlobPtrs[N] = (void*)&buffer.DN[Idx];
2073 /// reduce_function(reduce_data, GlobPtrs);
2074 ///
2075 /// \param ReductionInfos Array type containing the ReductionOps.
2076 /// \param ReduceFn The reduction function.
2077 /// \param ReductionsBufferTy The StructTy for the reductions buffer.
2078 /// \param FuncAttrs Optional param to specify any function attributes that
2079 /// need to be copied to the new function.
2080 ///
2081 /// \return The GlobalToListReduce function.
2083 emitGlobalToListReduceFunction(ArrayRef<ReductionInfo> ReductionInfos,
2084 Function *ReduceFn, Type *ReductionsBufferTy,
2085 AttributeList FuncAttrs,
2086 ArrayRef<bool> IsByRef);
2087
2088 /// Get the function name of a reduction function.
2089 std::string getReductionFuncName(StringRef Name) const;
2090
2091 /// Generate a Fortran descriptor for array reductions
2092 ///
2093 /// \param DescriptorAddr Address of the descriptor to initialize
2094 /// \param DataPtr Pointer to the actual data the descriptor should reference
2095 /// \param SrcDescriptorAddr Address of the descriptor to copy metadata from
2096 /// \param DescriptorType Type of the descriptor structure
2097 /// \param DataPtrPtrGen Callback to get the base_ptr field in the descriptor
2098 ///
2099 /// \return Error if DataPtrPtrGen fails, otherwise success.
2100 InsertPointOrErrorTy generateReductionDescriptor(
2101 Value *DescriptorAddr, Value *DataPtr, Value *SrcDescriptorAddr,
2102 Type *DescriptorType,
2104 DataPtrPtrGen);
2105
2106 /// Allocate a by-ref reduction descriptor, copy \p SrcDescriptorAddr into it,
2107 /// and update its data pointer to reference \p DataPtr.
2108 ///
2109 /// \param AllocaIP Insertion point for the descriptor allocation.
2110 /// \param RI Reduction info containing descriptor type and access callback.
2111 /// \param DataPtr Pointer to the actual data the descriptor should reference.
2112 /// \param SrcDescriptorAddr Address of the descriptor to copy metadata from.
2113 /// \param DescriptorPtrTy Pointer type expected by the descriptor consumer.
2114 ///
2115 /// \return The new descriptor address, or an Error if descriptor generation
2116 /// fails.
2117 Expected<Value *> createReductionDescriptorCopy(
2118 InsertPointTy AllocaIP, const ReductionInfo &RI, Value *DataPtr,
2119 Value *SrcDescriptorAddr, Type *DescriptorPtrTy,
2120 const Twine &Name = ".omp.reduction.byref_descriptor");
2121
2122 /// Emits reduction function.
2123 /// \param ReducerName Name of the function calling the reduction.
2124 /// \param ReductionInfos Array type containing the ReductionOps.
2125 /// \param ReductionGenCBKind Optional param to specify Clang or MLIR
2126 /// CodeGenCB kind.
2127 /// \param FuncAttrs Optional param to specify any function attributes that
2128 /// need to be copied to the new function.
2129 ///
2130 /// \return The reduction function.
2131 Expected<Function *> createReductionFunction(
2132 StringRef ReducerName, ArrayRef<ReductionInfo> ReductionInfos,
2133 ArrayRef<bool> IsByRef,
2135 AttributeList FuncAttrs = {});
2136
2137 /// Rewrites uses of registered declare target globals to their
2138 /// replacements if we are processing a device module.
2139 void applyDeclareTargetGlobalReplacements();
2140
2141public:
2142 ///
2143 /// Design of OpenMP reductions on the GPU
2144 ///
2145 /// Consider a typical OpenMP program with one or more reduction
2146 /// clauses:
2147 ///
2148 /// float foo;
2149 /// double bar;
2150 /// #pragma omp target teams distribute parallel for \
2151 /// reduction(+:foo) reduction(*:bar)
2152 /// for (int i = 0; i < N; i++) {
2153 /// foo += A[i]; bar *= B[i];
2154 /// }
2155 ///
2156 /// where 'foo' and 'bar' are reduced across all OpenMP threads in
2157 /// all teams. In our OpenMP implementation on the NVPTX device an
2158 /// OpenMP team is mapped to a CUDA threadblock and OpenMP threads
2159 /// within a team are mapped to CUDA threads within a threadblock.
2160 /// Our goal is to efficiently aggregate values across all OpenMP
2161 /// threads such that:
2162 ///
2163 /// - the compiler and runtime are logically concise, and
2164 /// - the reduction is performed efficiently in a hierarchical
2165 /// manner as follows: within OpenMP threads in the same warp,
2166 /// across warps in a threadblock, and finally across teams on
2167 /// the NVPTX device.
2168 ///
2169 /// Introduction to Decoupling
2170 ///
2171 /// We would like to decouple the compiler and the runtime so that the
2172 /// latter is ignorant of the reduction variables (number, data types)
2173 /// and the reduction operators. This allows a simpler interface
2174 /// and implementation while still attaining good performance.
2175 ///
2176 /// Pseudocode for the aforementioned OpenMP program generated by the
2177 /// compiler is as follows:
2178 ///
2179 /// 1. Create private copies of reduction variables on each OpenMP
2180 /// thread: 'foo_private', 'bar_private'
2181 /// 2. Each OpenMP thread reduces the chunk of 'A' and 'B' assigned
2182 /// to it and writes the result in 'foo_private' and 'bar_private'
2183 /// respectively.
2184 /// 3. Call the OpenMP runtime on the GPU to reduce within a team
2185 /// and store the result on the team master:
2186 ///
2187 /// __kmpc_nvptx_parallel_reduce_nowait_v2(...,
2188 /// reduceData, shuffleReduceFn, interWarpCpyFn)
2189 ///
2190 /// where:
2191 /// struct ReduceData {
2192 /// double *foo;
2193 /// double *bar;
2194 /// } reduceData
2195 /// reduceData.foo = &foo_private
2196 /// reduceData.bar = &bar_private
2197 ///
2198 /// 'shuffleReduceFn' and 'interWarpCpyFn' are pointers to two
2199 /// auxiliary functions generated by the compiler that operate on
2200 /// variables of type 'ReduceData'. They aid the runtime perform
2201 /// algorithmic steps in a data agnostic manner.
2202 ///
2203 /// 'shuffleReduceFn' is a pointer to a function that reduces data
2204 /// of type 'ReduceData' across two OpenMP threads (lanes) in the
2205 /// same warp. It takes the following arguments as input:
2206 ///
2207 /// a. variable of type 'ReduceData' on the calling lane,
2208 /// b. its lane_id,
2209 /// c. an offset relative to the current lane_id to generate a
2210 /// remote_lane_id. The remote lane contains the second
2211 /// variable of type 'ReduceData' that is to be reduced.
2212 /// d. an algorithm version parameter determining which reduction
2213 /// algorithm to use.
2214 ///
2215 /// 'shuffleReduceFn' retrieves data from the remote lane using
2216 /// efficient GPU shuffle intrinsics and reduces, using the
2217 /// algorithm specified by the 4th parameter, the two operands
2218 /// element-wise. The result is written to the first operand.
2219 ///
2220 /// Different reduction algorithms are implemented in different
2221 /// runtime functions, all calling 'shuffleReduceFn' to perform
2222 /// the essential reduction step. Therefore, based on the 4th
2223 /// parameter, this function behaves slightly differently to
2224 /// cooperate with the runtime to ensure correctness under
2225 /// different circumstances.
2226 ///
2227 /// 'InterWarpCpyFn' is a pointer to a function that transfers
2228 /// reduced variables across warps. It tunnels, through CUDA
2229 /// shared memory, the thread-private data of type 'ReduceData'
2230 /// from lane 0 of each warp to a lane in the first warp.
2231 /// 4. Call the OpenMP runtime on the GPU to reduce across teams.
2232 /// The last team writes the global reduced value to memory.
2233 ///
2234 /// ret = __kmpc_gpu_teams_reduce_nowait(...,
2235 /// reduceData, shuffleReduceFn, interWarpCpyFn,
2236 /// scratchpadCopyFn, loadAndReduceFn)
2237 ///
2238 /// 'scratchpadCopyFn' is a helper that stores reduced
2239 /// data from the team master to a scratchpad array in
2240 /// global memory.
2241 ///
2242 /// 'loadAndReduceFn' is a helper that loads data from
2243 /// the scratchpad array and reduces it with the input
2244 /// operand.
2245 ///
2246 /// These compiler generated functions hide address
2247 /// calculation and alignment information from the runtime.
2248 /// 5. if ret == 1:
2249 /// The team master of the last team stores the reduced
2250 /// result to the globals in memory.
2251 /// foo += reduceData.foo; bar *= reduceData.bar
2252 ///
2253 ///
2254 /// Warp Reduction Algorithms
2255 ///
2256 /// On the warp level, we have three algorithms implemented in the
2257 /// OpenMP runtime depending on the number of active lanes:
2258 ///
2259 /// Full Warp Reduction
2260 ///
2261 /// The reduce algorithm within a warp where all lanes are active
2262 /// is implemented in the runtime as follows:
2263 ///
2264 /// full_warp_reduce(void *reduce_data,
2265 /// kmp_ShuffleReductFctPtr ShuffleReduceFn) {
2266 /// for (int offset = WARPSIZE/2; offset > 0; offset /= 2)
2267 /// ShuffleReduceFn(reduce_data, 0, offset, 0);
2268 /// }
2269 ///
2270 /// The algorithm completes in log(2, WARPSIZE) steps.
2271 ///
2272 /// 'ShuffleReduceFn' is used here with lane_id set to 0 because it is
2273 /// not used therefore we save instructions by not retrieving lane_id
2274 /// from the corresponding special registers. The 4th parameter, which
2275 /// represents the version of the algorithm being used, is set to 0 to
2276 /// signify full warp reduction.
2277 ///
2278 /// In this version, 'ShuffleReduceFn' behaves, per element, as follows:
2279 ///
2280 /// #reduce_elem refers to an element in the local lane's data structure
2281 /// #remote_elem is retrieved from a remote lane
2282 /// remote_elem = shuffle_down(reduce_elem, offset, WARPSIZE);
2283 /// reduce_elem = reduce_elem REDUCE_OP remote_elem;
2284 ///
2285 /// Contiguous Partial Warp Reduction
2286 ///
2287 /// This reduce algorithm is used within a warp where only the first
2288 /// 'n' (n <= WARPSIZE) lanes are active. It is typically used when the
2289 /// number of OpenMP threads in a parallel region is not a multiple of
2290 /// WARPSIZE. The algorithm is implemented in the runtime as follows:
2291 ///
2292 /// void
2293 /// contiguous_partial_reduce(void *reduce_data,
2294 /// kmp_ShuffleReductFctPtr ShuffleReduceFn,
2295 /// int size, int lane_id) {
2296 /// int curr_size;
2297 /// int offset;
2298 /// curr_size = size;
2299 /// mask = curr_size/2;
2300 /// while (offset>0) {
2301 /// ShuffleReduceFn(reduce_data, lane_id, offset, 1);
2302 /// curr_size = (curr_size+1)/2;
2303 /// offset = curr_size/2;
2304 /// }
2305 /// }
2306 ///
2307 /// In this version, 'ShuffleReduceFn' behaves, per element, as follows:
2308 ///
2309 /// remote_elem = shuffle_down(reduce_elem, offset, WARPSIZE);
2310 /// if (lane_id < offset)
2311 /// reduce_elem = reduce_elem REDUCE_OP remote_elem
2312 /// else
2313 /// reduce_elem = remote_elem
2314 ///
2315 /// This algorithm assumes that the data to be reduced are located in a
2316 /// contiguous subset of lanes starting from the first. When there is
2317 /// an odd number of active lanes, the data in the last lane is not
2318 /// aggregated with any other lane's dat but is instead copied over.
2319 ///
2320 /// Dispersed Partial Warp Reduction
2321 ///
2322 /// This algorithm is used within a warp when any discontiguous subset of
2323 /// lanes are active. It is used to implement the reduction operation
2324 /// across lanes in an OpenMP simd region or in a nested parallel region.
2325 ///
2326 /// void
2327 /// dispersed_partial_reduce(void *reduce_data,
2328 /// kmp_ShuffleReductFctPtr ShuffleReduceFn) {
2329 /// int size, remote_id;
2330 /// int logical_lane_id = number_of_active_lanes_before_me() * 2;
2331 /// do {
2332 /// remote_id = next_active_lane_id_right_after_me();
2333 /// # the above function returns 0 of no active lane
2334 /// # is present right after the current lane.
2335 /// size = number_of_active_lanes_in_this_warp();
2336 /// logical_lane_id /= 2;
2337 /// ShuffleReduceFn(reduce_data, logical_lane_id,
2338 /// remote_id-1-threadIdx.x, 2);
2339 /// } while (logical_lane_id % 2 == 0 && size > 1);
2340 /// }
2341 ///
2342 /// There is no assumption made about the initial state of the reduction.
2343 /// Any number of lanes (>=1) could be active at any position. The reduction
2344 /// result is returned in the first active lane.
2345 ///
2346 /// In this version, 'ShuffleReduceFn' behaves, per element, as follows:
2347 ///
2348 /// remote_elem = shuffle_down(reduce_elem, offset, WARPSIZE);
2349 /// if (lane_id % 2 == 0 && offset > 0)
2350 /// reduce_elem = reduce_elem REDUCE_OP remote_elem
2351 /// else
2352 /// reduce_elem = remote_elem
2353 ///
2354 ///
2355 /// Intra-Team Reduction
2356 ///
2357 /// This function, as implemented in the runtime call
2358 /// '__kmpc_nvptx_parallel_reduce_nowait_v2', aggregates data across OpenMP
2359 /// threads in a team. It first reduces within a warp using the
2360 /// aforementioned algorithms. We then proceed to gather all such
2361 /// reduced values at the first warp.
2362 ///
2363 /// The runtime makes use of the function 'InterWarpCpyFn', which copies
2364 /// data from each of the "warp master" (zeroth lane of each warp, where
2365 /// warp-reduced data is held) to the zeroth warp. This step reduces (in
2366 /// a mathematical sense) the problem of reduction across warp masters in
2367 /// a block to the problem of warp reduction.
2368 ///
2369 ///
2370 /// Inter-Team Reduction
2371 ///
2372 /// Once a team has reduced its data to a single value, it is stored in
2373 /// a global scratchpad array. Since each team has a distinct slot, this
2374 /// can be done without locking.
2375 ///
2376 /// The last team to write to the scratchpad array proceeds to reduce the
2377 /// scratchpad array. One or more workers in the last team use the helper
2378 /// 'loadAndReduceDataFn' to load and reduce values from the array, i.e.,
2379 /// the k'th worker reduces every k'th element.
2380 ///
2381 /// Finally, a call is made to '__kmpc_nvptx_parallel_reduce_nowait_v2' to
2382 /// reduce across workers and compute a globally reduced value.
2383 ///
2384 /// \param Loc The location where the reduction was
2385 /// encountered. Must be within the associate
2386 /// directive and after the last local access to the
2387 /// reduction variables.
2388 /// \param AllocaIP An insertion point suitable for allocas usable
2389 /// in reductions.
2390 /// \param CodeGenIP An insertion point suitable for code
2391 /// generation.
2392 /// \param ReductionInfos A list of info on each reduction
2393 /// variable.
2394 /// \param IsNoWait Optional flag set if the reduction is
2395 /// marked as nowait.
2396 /// \param IsByRef For each reduction clause, whether the reduction is by-ref.
2397 /// \param IsTeamsReduction Optional flag set if it is a teams
2398 /// reduction.
2399 /// \param IsSPMD Optional flag set when the surrounding kernel
2400 /// is compiled in SPMD execution mode (every
2401 /// reduction private is then known to be a
2402 /// per-thread scratch alloca). When false, the
2403 /// teams-reduction call site emits per-thread
2404 /// scratch and copies the team-local value in so
2405 /// the runtime's cross-team work cannot race on
2406 /// team-shared LDS storage produced by Generic
2407 /// globalization (Generic-SPMD case after
2408 /// OpenMPOpt SPMD-ization).
2409 /// \param GridValue Optional GPU grid value.
2410 /// used for teams reduction.
2411 /// \param SrcLocInfo Source location information global.
2413 const LocationDescription &Loc, InsertPointTy AllocaIP,
2414 InsertPointTy CodeGenIP, ArrayRef<ReductionInfo> ReductionInfos,
2415 ArrayRef<bool> IsByRef, bool IsNoWait = false,
2416 bool IsTeamsReduction = false, bool IsSPMD = false,
2418 std::optional<omp::GV> GridValue = {}, Value *SrcLocInfo = nullptr);
2419
2420 // TODO: provide atomic and non-atomic reduction generators for reduction
2421 // operators defined by the OpenMP specification.
2422
2423 /// Generator for '#omp reduction'.
2424 ///
2425 /// Emits the IR instructing the runtime to perform the specific kind of
2426 /// reductions. Expects reduction variables to have been privatized and
2427 /// initialized to reduction-neutral values separately. Emits the calls to
2428 /// runtime functions as well as the reduction function and the basic blocks
2429 /// performing the reduction atomically and non-atomically.
2430 ///
2431 /// The code emitted for the following:
2432 ///
2433 /// \code
2434 /// type var_1;
2435 /// type var_2;
2436 /// #pragma omp <directive> reduction(reduction-op:var_1,var_2)
2437 /// /* body */;
2438 /// \endcode
2439 ///
2440 /// corresponds to the following sketch.
2441 ///
2442 /// \code
2443 /// void _outlined_par() {
2444 /// // N is the number of different reductions.
2445 /// void *red_array[] = {privatized_var_1, privatized_var_2, ...};
2446 /// switch(__kmpc_reduce(..., N, /*size of data in red array*/, red_array,
2447 /// _omp_reduction_func,
2448 /// _gomp_critical_user.reduction.var)) {
2449 /// case 1: {
2450 /// var_1 = var_1 <reduction-op> privatized_var_1;
2451 /// var_2 = var_2 <reduction-op> privatized_var_2;
2452 /// // ...
2453 /// __kmpc_end_reduce(...);
2454 /// break;
2455 /// }
2456 /// case 2: {
2457 /// _Atomic<ReductionOp>(var_1, privatized_var_1);
2458 /// _Atomic<ReductionOp>(var_2, privatized_var_2);
2459 /// // ...
2460 /// break;
2461 /// }
2462 /// default: break;
2463 /// }
2464 /// }
2465 ///
2466 /// void _omp_reduction_func(void **lhs, void **rhs) {
2467 /// *(type *)lhs[0] = *(type *)lhs[0] <reduction-op> *(type *)rhs[0];
2468 /// *(type *)lhs[1] = *(type *)lhs[1] <reduction-op> *(type *)rhs[1];
2469 /// // ...
2470 /// }
2471 /// \endcode
2472 ///
2473 /// \param Loc The location where the reduction was
2474 /// encountered. Must be within the associate
2475 /// directive and after the last local access to the
2476 /// reduction variables.
2477 /// \param AllocaIP An insertion point suitable for allocas usable
2478 /// in reductions.
2479 /// \param ReductionInfos A list of info on each reduction variable.
2480 /// \param IsNoWait A flag set if the reduction is marked as nowait.
2481 /// \param IsByRef A flag set if the reduction is using reference
2482 /// or direct value.
2483 /// \param IsTeamsReduction Optional flag set if it is a teams
2484 /// reduction.
2486 const LocationDescription &Loc, InsertPointTy AllocaIP,
2487 ArrayRef<ReductionInfo> ReductionInfos, ArrayRef<bool> IsByRef,
2488 bool IsNoWait = false, bool IsTeamsReduction = false);
2489
2490 ///}
2491
2492 /// Emit the host runtime lookups that redirect the `in_reduction`
2493 /// list items of an `omp.target` region to their per-task reduction-private
2494 /// storage.
2495 ///
2496 /// This owns the complete runtime-generation path for target `in_reduction`.
2497 /// It computes the executing thread's gtid once for the whole target body (so
2498 /// a target with several in_reduction items does not emit a redundant
2499 /// `__kmpc_global_thread_num` per item), then for each item emits
2500 /// `__kmpc_task_reduction_get_th_data(gtid, /*descriptor=*/null, origPtr)`
2501 /// with the address-space normalization required by the runtime entry point.
2502 /// The NULL descriptor makes the runtime walk the enclosing taskgroups to
2503 /// find the matching `task_reduction` registration for the item. The lookups
2504 /// are emitted at \p Loc, which must be inside the target task body.
2505 ///
2506 /// The front-end-specific work (matching each `in_reduction` item to its
2507 /// mapped storage and binding the generated private pointer back to the
2508 /// right value) stays with the caller: each generated private pointer is
2509 /// handed back through \p MapPrivateCB.
2510 ///
2511 /// \param Loc Insertion point for the target body.
2512 /// \param OrigPtrs Per item, the mapped original pointer used as the
2513 /// runtime `orig` argument.
2514 /// \param ResultPtrTys Per item, the type the returned private pointer must
2515 /// have (for address-space normalization). Must have the
2516 /// same length as \p OrigPtrs.
2517 /// \param MapPrivateCB Called once per item, in list order, with the item
2518 /// index and the generated per-task private pointer.
2519 /// \returns The insertion point after the emitted lookups.
2520 LLVM_ABI InsertPointTy createTargetInReduction(
2521 const LocationDescription &Loc, ArrayRef<Value *> OrigPtrs,
2522 ArrayRef<Type *> ResultPtrTys,
2523 function_ref<void(unsigned, Value *)> MapPrivateCB);
2524
2525 /// Return the insertion point used by the underlying IRBuilder.
2527
2528 /// Update the internal location to \p Loc.
2530 Builder.restoreIP(Loc.IP);
2531 Builder.SetCurrentDebugLocation(Loc.DL);
2532 return Loc.IP.getBlock() != nullptr;
2533 }
2534
2535 /// Return the function declaration for the runtime function with \p FnID.
2538
2540
2542 ArrayRef<Value *> Args,
2543 StringRef Name = "");
2544
2545 /// Return the (LLVM-IR) string describing the source location \p LocStr.
2547 uint32_t &SrcLocStrSize);
2548
2549 /// Return the (LLVM-IR) string describing the default source location.
2551
2552 /// Return the (LLVM-IR) string describing the source location identified by
2553 /// the arguments.
2555 StringRef FileName, unsigned Line,
2556 unsigned Column,
2557 uint32_t &SrcLocStrSize);
2558
2559 /// Return the (LLVM-IR) string describing the DebugLoc \p DL. Use \p F as
2560 /// fallback if \p DL does not specify the function name.
2562 Function *F = nullptr);
2563
2564 /// Return the (LLVM-IR) string describing the source location \p Loc.
2565 LLVM_ABI Constant *getOrCreateSrcLocStr(const LocationDescription &Loc,
2566 uint32_t &SrcLocStrSize);
2567
2568 /// Return an ident_t* encoding the source location \p SrcLocStr and \p Flags.
2569 /// TODO: Create a enum class for the Reserve2Flags
2571 uint32_t SrcLocStrSize,
2572 omp::IdentFlag Flags = omp::IdentFlag(0),
2573 unsigned Reserve2Flags = 0);
2574
2575 /// Create a hidden global flag \p Name in the module with initial value \p
2576 /// Value.
2578
2579 /// Emit the llvm.used metadata.
2581
2582 /// Emit the kernel execution mode.
2585
2586 /// Generate control flow and cleanup for cancellation.
2587 ///
2588 /// \param CancelFlag Flag indicating if the cancellation is performed.
2589 /// \param CanceledDirective The kind of directive that is cancled.
2590 /// \param ExitCB Extra code to be generated in the exit block.
2591 ///
2592 /// \return an error, if any were triggered during execution.
2594 omp::Directive CanceledDirective);
2595
2596 /// Generate a target region entry call.
2597 ///
2598 /// \param Loc The location at which the request originated and is fulfilled.
2599 /// \param AllocaIP The insertion point to be used for alloca instructions.
2600 /// \param Return Return value of the created function returned by reference.
2601 /// \param DeviceID Identifier for the device via the 'device' clause.
2602 /// \param NumTeams Numer of teams for the region via the 'num_teams' clause
2603 /// or 0 if unspecified and -1 if there is no 'teams' clause.
2604 /// \param NumThreads Number of threads via the 'thread_limit' clause.
2605 /// \param HostPtr Pointer to the host-side pointer of the target kernel.
2606 /// \param KernelArgs Array of arguments to the kernel.
2607 LLVM_ABI InsertPointTy emitTargetKernel(const LocationDescription &Loc,
2608 InsertPointTy AllocaIP,
2609 Value *&Return, Value *Ident,
2610 Value *DeviceID, Value *NumTeams,
2611 Value *NumThreads, Value *HostPtr,
2612 ArrayRef<Value *> KernelArgs);
2613
2614 /// Generate a flush runtime call.
2615 ///
2616 /// \param Loc The location at which the request originated and is fulfilled.
2617 LLVM_ABI void emitFlush(const LocationDescription &Loc);
2618
2619 /// The finalization stack made up of finalize callbacks currently in-flight,
2620 /// wrapped into FinalizationInfo objects that reference also the finalization
2621 /// target block and the kind of cancellable directive.
2623
2624 /// Return true if the last entry in the finalization stack is of kind \p DK
2625 /// and cancellable.
2626 bool isLastFinalizationInfoCancellable(omp::Directive DK) {
2627 return !FinalizationStack.empty() &&
2628 FinalizationStack.back().IsCancellable &&
2629 FinalizationStack.back().DK == DK;
2630 }
2631
2632 /// Generate a taskwait runtime call.
2633 ///
2634 /// \param Loc The location at which the request originated and is fulfilled.
2635 LLVM_ABI void emitTaskwaitImpl(const LocationDescription &Loc);
2636
2637 /// Generate a taskyield runtime call.
2638 ///
2639 /// \param Loc The location at which the request originated and is fulfilled.
2640 LLVM_ABI void emitTaskyieldImpl(const LocationDescription &Loc);
2641
2642 /// Return the current thread ID.
2643 ///
2644 /// \param Ident The ident (ident_t*) describing the query origin.
2646
2647 /// The OpenMPIRBuilder Configuration
2649
2650 /// The underlying LLVM-IR module
2652
2653 /// The LLVM-IR Builder used to create IR.
2655
2656 /// Map to remember source location strings
2658
2659 /// Map to remember existing ident_t*.
2661
2662 /// Info manager to keep track of target regions.
2664
2665 /// The target triple of the underlying module.
2666 const Triple T;
2667
2668 /// Helper that contains information about regions we need to outline
2669 /// during finalization.
2671 using PostOutlineCBTy = std::function<void(Function &)>;
2677 // TODO: this should be safe to enable by default
2679
2680 virtual ~OutlineInfo() = default;
2681
2682 /// Collect all blocks in between EntryBB and ExitBB in both the given
2683 /// vector and set.
2685 SmallVectorImpl<BasicBlock *> &BlockVector);
2686
2687 /// Create a CodeExtractor instance based on the information stored in this
2688 /// structure, the list of collected blocks from a previous call to
2689 /// \c collectBlocks and a flag stating whether arguments must be passed in
2690 /// address space 0.
2691 virtual std::unique_ptr<CodeExtractor>
2693 bool ArgsInZeroAddressSpace, Twine Suffix = Twine(""));
2694
2695 /// Return the function that contains the region to be outlined.
2696 Function *getFunction() const { return EntryBB->getParent(); }
2697 };
2698
2699 /// Collection of regions that need to be outlined during finalization.
2701
2702 /// A collection of candidate target functions that's constant allocas will
2703 /// attempt to be raised on a call of finalize after all currently enqueued
2704 /// outline info's have been processed.
2706
2707 /// Describes a declare target global variable replacement to be applied
2708 /// during finalization.
2713
2714 /// Collection of declare target globals to rewrite uses of during
2715 /// device module finalizaiton.
2718
2719 /// Collection of owned canonical loop objects that eventually need to be
2720 /// free'd.
2721 std::forward_list<CanonicalLoopInfo> LoopInfos;
2722
2723 /// Collection of owned ScanInfo objects that eventually need to be free'd.
2724 std::forward_list<ScanInfo> ScanInfos;
2725
2726 /// Add a new region that will be outlined later.
2727 void addOutlineInfo(std::unique_ptr<OutlineInfo> &&OI) {
2728 OutlineInfos.emplace_back(std::move(OI));
2729 }
2730
2731 /// An ordered map of auto-generated variables to their unique names.
2732 /// It stores variables with the following names: 1) ".gomp_critical_user_" +
2733 /// <critical_section_name> + ".var" for "omp critical" directives; 2)
2734 /// <mangled_name_for_global_var> + ".cache." for cache for threadprivate
2735 /// variables.
2737
2738 /// Computes the size of type in bytes.
2740
2741 // Emit a branch from the current block to the Target block only if
2742 // the current block has a terminator.
2744
2745 // If BB has no use then delete it and return. Else place BB after the current
2746 // block, if possible, or else at the end of the function. Also add a branch
2747 // from current block to BB if current block does not have a terminator.
2748 LLVM_ABI void emitBlock(BasicBlock *BB, Function *CurFn,
2749 bool IsFinished = false);
2750
2751 /// Emits code for OpenMP 'if' clause using specified \a BodyGenCallbackTy
2752 /// Here is the logic:
2753 /// if (Cond) {
2754 /// ThenGen();
2755 /// } else {
2756 /// ElseGen();
2757 /// }
2758 ///
2759 /// \return an error, if any were triggered during execution.
2761 BodyGenCallbackTy ElseGen,
2762 InsertPointTy AllocaIP = {},
2763 ArrayRef<BasicBlock *> DeallocBlocks = {});
2764
2765 /// Create the global variable holding the offload mappings information.
2766 LLVM_ABI GlobalVariable *
2767 createOffloadMaptypes(SmallVectorImpl<uint64_t> &Mappings,
2768 std::string VarName);
2769
2770 /// Create the global variable holding the offload names information.
2771 LLVM_ABI GlobalVariable *
2772 createOffloadMapnames(SmallVectorImpl<llvm::Constant *> &Names,
2773 std::string VarName);
2774
2777 AllocaInst *Args = nullptr;
2779 };
2780
2781 /// Create the allocas instruction used in call to mapper functions.
2783 InsertPointTy AllocaIP,
2784 unsigned NumOperands,
2786
2787 /// Create the call for the target mapper function.
2788 /// \param Loc The source location description.
2789 /// \param MapperFunc Function to be called.
2790 /// \param SrcLocInfo Source location information global.
2791 /// \param MaptypesArg The argument types.
2792 /// \param MapnamesArg The argument names.
2793 /// \param MapperAllocas The AllocaInst used for the call.
2794 /// \param DeviceID Device ID for the call.
2795 /// \param NumOperands Number of operands in the call.
2797 Function *MapperFunc, Value *SrcLocInfo,
2798 Value *MaptypesArg, Value *MapnamesArg,
2800 int64_t DeviceID, unsigned NumOperands);
2801
2802 /// Container for the arguments used to pass data to the runtime library.
2804 /// The array of base pointer passed to the runtime library.
2806 /// The array of section pointers passed to the runtime library.
2808 /// The array of sizes passed to the runtime library.
2809 Value *SizesArray = nullptr;
2810 /// The array of map types passed to the runtime library for the beginning
2811 /// of the region or for the entire region if there are no separate map
2812 /// types for the region end.
2814 /// The array of map types passed to the runtime library for the end of the
2815 /// region, or nullptr if there are no separate map types for the region
2816 /// end.
2818 /// The array of user-defined mappers passed to the runtime library.
2820 /// The array of original declaration names of mapped pointers sent to the
2821 /// runtime library for debugging
2823
2824 explicit TargetDataRTArgs() = default;
2833 };
2834
2835 /// Container to pass the default attributes with which a kernel must be
2836 /// launched, used to set kernel attributes and populate associated static
2837 /// structures.
2838 ///
2839 /// For max values, < 0 means unset, == 0 means set but unknown at compile
2840 /// time. The number of max values will be 1 except for the case where
2841 /// ompx_bare is set.
2851
2852 /// Container to pass LLVM IR runtime values or constants related to the
2853 /// number of teams and threads with which the kernel must be launched, as
2854 /// well as the trip count of the loop, if it is an SPMD or Generic-SPMD
2855 /// kernel. These must be defined in the host prior to the call to the kernel
2856 /// launch OpenMP RTL function.
2862
2863 /// 'parallel' construct 'num_threads' clause value, if present and it is an
2864 /// SPMD kernel.
2866
2867 /// Total number of iterations of the SPMD or Generic-SPMD kernel or null if
2868 /// it is a generic kernel.
2870
2871 /// Device ID value used in the kernel launch.
2872 Value *DeviceID = nullptr;
2873 };
2874
2875 /// Data structure that contains the needed information to construct the
2876 /// kernel args vector.
2878 /// Number of arguments passed to the runtime library.
2879 unsigned NumTargetItems = 0;
2880 /// Arguments passed to the runtime library
2882 /// The number of iterations
2884 /// The number of teams.
2886 /// The number of threads.
2888 /// The size of the dynamic shared memory.
2890 /// True if the kernel has 'no wait' clause.
2891 bool HasNoWait = false;
2892 /// True if the kernel strictly requires the number of blocks and threads
2893 /// above to run.
2894 bool StrictBlocks = false;
2895 bool StrictThreads = false;
2896 /// The fallback mechanism for the shared memory.
2899
2900 // Constructors for TargetKernelArgs.
2901 TargetKernelArgs() = default;
2913 };
2914
2915 /// Create the kernel args vector used by emitTargetKernel. This function
2916 /// creates various constant values that are used in the resulting args
2917 /// vector.
2918 LLVM_ABI static void getKernelArgsVector(TargetKernelArgs &KernelArgs,
2919 IRBuilderBase &Builder,
2920 SmallVector<Value *> &ArgsVector);
2921
2922 /// Struct that keeps the information that should be kept throughout
2923 /// a 'target data' region.
2925 /// Set to true if device pointer information have to be obtained.
2926 bool RequiresDevicePointerInfo = false;
2927 /// Set to true if Clang emits separate runtime calls for the beginning and
2928 /// end of the region. These calls might have separate map type arrays.
2929 bool SeparateBeginEndCalls = false;
2930
2931 public:
2933
2936
2937 /// Indicate whether any user-defined mapper exists.
2938 bool HasMapper = false;
2939 /// The total number of pointers passed to the runtime library.
2940 unsigned NumberOfPtrs = 0u;
2941
2942 bool EmitDebug = false;
2943
2944 /// Whether the `target ... data` directive has a `nowait` clause.
2945 bool HasNoWait = false;
2946
2947 explicit TargetDataInfo() = default;
2948 explicit TargetDataInfo(bool RequiresDevicePointerInfo,
2949 bool SeparateBeginEndCalls)
2950 : RequiresDevicePointerInfo(RequiresDevicePointerInfo),
2951 SeparateBeginEndCalls(SeparateBeginEndCalls) {}
2952 /// Clear information about the data arrays.
2955 HasMapper = false;
2956 NumberOfPtrs = 0u;
2957 }
2958 /// Return true if the current target data information has valid arrays.
2959 bool isValid() {
2960 return RTArgs.BasePointersArray && RTArgs.PointersArray &&
2961 RTArgs.SizesArray && RTArgs.MapTypesArray &&
2962 (!HasMapper || RTArgs.MappersArray) && NumberOfPtrs;
2963 }
2964 bool requiresDevicePointerInfo() { return RequiresDevicePointerInfo; }
2965 bool separateBeginEndCalls() { return SeparateBeginEndCalls; }
2966 };
2967
2976
2977 /// This structure contains combined information generated for mappable
2978 /// clauses, including base pointers, pointers, sizes, map types, user-defined
2979 /// mappers, and non-contiguous information.
2980 struct MapInfosTy {
2994 /// True for entries that have an attach ptr, and thus an accompanying
2995 /// ATTACH entry linking that ptr to its ptee.
2996 ///
2997 /// This is a property of the storage block an entry describes, not of the
2998 /// map clause list item: it is true iff the entry's storage is the pointee
2999 /// storage reached through some attach ptr. So, for `int *p; map(p[1:10])`,
3000 /// which produces
3001 /// &p[1], &p[1], 10*sizeof(int), TO|FROM <- HasAttachPtr = true
3002 /// &p, &p[1], sizeof(void*), ATTACH <- HasAttachPtr = false
3003 /// it is set on the pointee entry only. It is never set on the ATTACH entry
3004 /// itself, nor on an entry that maps the pointer as an object in its own
3005 /// right (e.g. the `map(p)` entry for the pointer's own storage).
3006 ///
3007 /// It is set on every entry whose storage lies in a pointee block,
3008 /// including a combined struct entry for such a block and the individual
3009 /// member entries that are MEMBER_OF it. e.g. for
3010 /// `map(s2.s1p->x, s2.s1p->y)`:
3011 /// &s2.s1p[0], &s2.s1p->x, sizeof(x..y), ALLOC <- true
3012 /// &s2.s1p[0], &s2.s1p->x, 4, MEMBER_OF(1)|TO|FROM <- true
3013 /// &s2.s1p[0], &s2.s1p->y, 4, MEMBER_OF(1)|TO|FROM <- true
3014 /// &s2.s1p, &s2.s1p->x, sizeof(void*), ATTACH <- false
3015 /// with s2.s1p as the attach ptr for all three.
3018
3019 /// Append arrays in \a CurInfo.
3020 void append(MapInfosTy &CurInfo) {
3021 BasePointers.append(CurInfo.BasePointers.begin(),
3022 CurInfo.BasePointers.end());
3023 Pointers.append(CurInfo.Pointers.begin(), CurInfo.Pointers.end());
3024 DevicePointers.append(CurInfo.DevicePointers.begin(),
3025 CurInfo.DevicePointers.end());
3026 Sizes.append(CurInfo.Sizes.begin(), CurInfo.Sizes.end());
3027 Types.append(CurInfo.Types.begin(), CurInfo.Types.end());
3028 Names.append(CurInfo.Names.begin(), CurInfo.Names.end());
3029 HasAttachPtr.append(CurInfo.HasAttachPtr.begin(),
3030 CurInfo.HasAttachPtr.end());
3031 NonContigInfo.Dims.append(CurInfo.NonContigInfo.Dims.begin(),
3032 CurInfo.NonContigInfo.Dims.end());
3033 NonContigInfo.Offsets.append(CurInfo.NonContigInfo.Offsets.begin(),
3034 CurInfo.NonContigInfo.Offsets.end());
3035 NonContigInfo.Counts.append(CurInfo.NonContigInfo.Counts.begin(),
3036 CurInfo.NonContigInfo.Counts.end());
3037 NonContigInfo.Strides.append(CurInfo.NonContigInfo.Strides.begin(),
3038 CurInfo.NonContigInfo.Strides.end());
3039 }
3040 };
3042
3043 /// Callback function type for functions emitting the host fallback code that
3044 /// is executed when the kernel launch fails. It takes an insertion point as
3045 /// parameter where the code should be emitted. It returns an insertion point
3046 /// that points right after after the emitted code.
3049
3050 // Callback function type for emitting and fetching user defined custom
3051 // mappers.
3053 function_ref<Expected<Function *>(unsigned int)>;
3054
3055 /// Generate a target region entry call and host fallback call.
3056 ///
3057 /// \param Loc The location at which the request originated and is fulfilled.
3058 /// \param OutlinedFnID The ooulined function ID.
3059 /// \param EmitTargetCallFallbackCB Call back function to generate host
3060 /// fallback code.
3061 /// \param Args Data structure holding information about the kernel arguments.
3062 /// \param DeviceID Identifier for the device via the 'device' clause.
3063 /// \param RTLoc Source location identifier
3064 /// \param AllocaIP The insertion point to be used for alloca instructions.
3066 const LocationDescription &Loc, Value *OutlinedFnID,
3067 EmitFallbackCallbackTy EmitTargetCallFallbackCB, TargetKernelArgs &Args,
3068 Value *DeviceID, Value *RTLoc, InsertPointTy AllocaIP);
3069
3070 /// Callback type for generating the bodies of device directives that require
3071 /// outer target tasks (e.g. in case of having `nowait` or `depend` clauses).
3072 ///
3073 /// \param DeviceID The ID of the device on which the target region will
3074 /// execute.
3075 /// \param RTLoc Source location identifier
3076 /// \Param TargetTaskAllocaIP Insertion point for the alloca block of the
3077 /// generated task.
3078 ///
3079 /// \return an error, if any were triggered during execution.
3081 function_ref<Error(Value *DeviceID, Value *RTLoc,
3082 IRBuilderBase::InsertPoint TargetTaskAllocaIP)>;
3083
3084 /// Generate a target-task for the target construct
3085 ///
3086 /// \param TaskBodyCB Callback to generate the actual body of the target task.
3087 /// \param DeviceID Identifier for the device via the 'device' clause.
3088 /// \param RTLoc Source location identifier
3089 /// \param AllocaIP The insertion point to be used for alloca instructions.
3090 /// \param Dependencies Dependencies info as specified by the 'depend' clause.
3091 /// \param HasNoWait True if the target construct had 'nowait' on it, false
3092 /// otherwise
3094 emitTargetTask(TargetTaskBodyCallbackTy TaskBodyCB, Value *DeviceID,
3095 Value *RTLoc, OpenMPIRBuilder::InsertPointTy AllocaIP,
3096 const DependenciesInfo &Dependencies,
3097 const TargetDataRTArgs &RTArgs, bool HasNoWait);
3098
3099 /// Emit the arguments to be passed to the runtime library based on the
3100 /// arrays of base pointers, pointers, sizes, map types, and mappers. If
3101 /// ForEndCall, emit map types to be passed for the end of the region instead
3102 /// of the beginning.
3105 OpenMPIRBuilder::TargetDataInfo &Info, bool ForEndCall = false);
3106
3107 /// Emit an array of struct descriptors to be assigned to the offload args.
3109 InsertPointTy CodeGenIP,
3110 MapInfosTy &CombinedInfo,
3111 TargetDataInfo &Info);
3112
3113 /// Emit the arrays used to pass the captures and map information to the
3114 /// offloading runtime library. If there is no map or capture information,
3115 /// return nullptr by reference. Accepts a reference to a MapInfosTy object
3116 /// that contains information generated for mappable clauses,
3117 /// including base pointers, pointers, sizes, map types, user-defined mappers.
3119 InsertPointTy AllocaIP, InsertPointTy CodeGenIP, MapInfosTy &CombinedInfo,
3120 TargetDataInfo &Info, CustomMapperCallbackTy CustomMapperCB,
3121 bool IsNonContiguous = false,
3122 function_ref<void(unsigned int, Value *)> DeviceAddrCB = nullptr);
3123
3124 /// Allocates memory for and populates the arrays required for offloading
3125 /// (offload_{baseptrs|ptrs|mappers|sizes|maptypes|mapnames}). Then, it
3126 /// emits their base addresses as arguments to be passed to the runtime
3127 /// library. In essence, this function is a combination of
3128 /// emitOffloadingArrays and emitOffloadingArraysArgument and should arguably
3129 /// be preferred by clients of OpenMPIRBuilder.
3131 InsertPointTy AllocaIP, InsertPointTy CodeGenIP, TargetDataInfo &Info,
3132 TargetDataRTArgs &RTArgs, MapInfosTy &CombinedInfo,
3133 CustomMapperCallbackTy CustomMapperCB, bool IsNonContiguous = false,
3134 bool ForEndCall = false,
3135 function_ref<void(unsigned int, Value *)> DeviceAddrCB = nullptr);
3136
3137 /// Creates offloading entry for the provided entry ID \a ID, address \a
3138 /// Addr, size \a Size, and flags \a Flags.
3140 int32_t Flags, GlobalValue::LinkageTypes,
3141 StringRef Name = "");
3142
3143 /// The kind of errors that can occur when emitting the offload entries and
3144 /// metadata.
3151
3152 /// Callback function type
3154 std::function<void(EmitMetadataErrorKind, TargetRegionEntryInfo)>;
3155
3156 // Emit the offloading entries and metadata so that the device codegen side
3157 // can easily figure out what to emit. The produced metadata looks like
3158 // this:
3159 //
3160 // !omp_offload.info = !{!1, ...}
3161 //
3162 // We only generate metadata for function that contain target regions.
3164 EmitMetadataErrorReportFunctionTy &ErrorReportFunction);
3165
3166public:
3167 /// Generator for __kmpc_copyprivate
3168 ///
3169 /// \param Loc The source location description.
3170 /// \param BufSize Number of elements in the buffer.
3171 /// \param CpyBuf List of pointers to data to be copied.
3172 /// \param CpyFn function to call for copying data.
3173 /// \param DidIt flag variable; 1 for 'single' thread, 0 otherwise.
3174 ///
3175 /// \return The insertion position *after* the CopyPrivate call.
3176
3178 llvm::Value *BufSize,
3179 llvm::Value *CpyBuf,
3180 llvm::Value *CpyFn,
3181 llvm::Value *DidIt);
3182
3183 /// Generator for '#omp single'
3184 ///
3185 /// \param Loc The source location description.
3186 /// \param BodyGenCB Callback that will generate the region code.
3187 /// \param FiniCB Callback to finalize variable copies.
3188 /// \param IsNowait If false, a barrier is emitted.
3189 /// \param CPVars copyprivate variables.
3190 /// \param CPFuncs copy functions to use for each copyprivate variable.
3191 ///
3192 /// \returns The insertion position *after* the single call.
3195 FinalizeCallbackTy FiniCB, bool IsNowait,
3196 ArrayRef<llvm::Value *> CPVars = {},
3197 ArrayRef<llvm::Function *> CPFuncs = {});
3198
3199 /// Generator for '#omp scope'
3200 ///
3201 /// \param Loc The source location description.
3202 /// \param BodyGenCB Callback that will generate the region code.
3203 /// \param FiniCB Callback to finalize variable copies.
3204 /// \param IsNowait If false, a barrier is emitted.
3205 ///
3206 /// \returns The insertion position *after* the scope.
3207 LLVM_ABI InsertPointOrErrorTy createScope(const LocationDescription &Loc,
3208 BodyGenCallbackTy BodyGenCB,
3209 FinalizeCallbackTy FiniCB,
3210 bool IsNowait);
3211
3212 /// Generator for '#omp master'
3213 ///
3214 /// \param Loc The insert and source location description.
3215 /// \param BodyGenCB Callback that will generate the region code.
3216 /// \param FiniCB Callback to finalize variable copies.
3217 ///
3218 /// \returns The insertion position *after* the master.
3219 LLVM_ABI InsertPointOrErrorTy createMaster(const LocationDescription &Loc,
3220 BodyGenCallbackTy BodyGenCB,
3221 FinalizeCallbackTy FiniCB);
3222
3223 /// Generator for '#omp masked'
3224 ///
3225 /// \param Loc The insert and source location description.
3226 /// \param BodyGenCB Callback that will generate the region code.
3227 /// \param FiniCB Callback to finialize variable copies.
3228 ///
3229 /// \returns The insertion position *after* the masked.
3230 LLVM_ABI InsertPointOrErrorTy createMasked(const LocationDescription &Loc,
3231 BodyGenCallbackTy BodyGenCB,
3232 FinalizeCallbackTy FiniCB,
3233 Value *Filter);
3234
3235 /// This function performs the scan reduction of the values updated in
3236 /// the input phase. The reduction logic needs to be emitted between input
3237 /// and scan loop returned by `CreateCanonicalScanLoops`. The following
3238 /// is the code that is generated, `buffer` and `span` are expected to be
3239 /// populated before executing the generated code.
3240 /// \code{c}
3241 /// for (int k = 0; k != ceil(log2(span)); ++k) {
3242 /// i=pow(2,k)
3243 /// for (size cnt = last_iter; cnt >= i; --cnt)
3244 /// buffer[cnt] op= buffer[cnt-i];
3245 /// }
3246 /// \endcode
3247 /// \param Loc The insert and source location description.
3248 /// \param ReductionInfos Array type containing the ReductionOps.
3249 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
3250 /// `ScanInfoInitialize`.
3251 ///
3252 /// \returns The insertion position *after* the masked.
3254 const LocationDescription &Loc,
3256 ScanInfo *ScanRedInfo);
3257
3258 /// This directive split and directs the control flow to input phase
3259 /// blocks or scan phase blocks based on 1. whether input loop or scan loop
3260 /// is executed, 2. whether exclusive or inclusive scan is used.
3261 ///
3262 /// \param Loc The insert and source location description.
3263 /// \param AllocaIP The IP where the temporary buffer for scan reduction
3264 // needs to be allocated.
3265 /// \param ScanVars Scan Variables.
3266 /// \param IsInclusive Whether it is an inclusive or exclusive scan.
3267 /// \param ScanRedInfo Pointer to the ScanInfo objected created using
3268 /// `ScanInfoInitialize`.
3269 ///
3270 /// \returns The insertion position *after* the scan.
3271 LLVM_ABI InsertPointOrErrorTy createScan(const LocationDescription &Loc,
3272 InsertPointTy AllocaIP,
3273 ArrayRef<llvm::Value *> ScanVars,
3274 ArrayRef<llvm::Type *> ScanVarsType,
3275 bool IsInclusive,
3276 ScanInfo *ScanRedInfo);
3277
3278 /// Generator for '#omp critical'
3279 ///
3280 /// \param Loc The insert and source location description.
3281 /// \param BodyGenCB Callback that will generate the region body code.
3282 /// \param FiniCB Callback to finalize variable copies.
3283 /// \param CriticalName name of the lock used by the critical directive
3284 /// \param HintInst Hint Instruction for hint clause associated with critical
3285 ///
3286 /// \returns The insertion position *after* the critical.
3287 LLVM_ABI InsertPointOrErrorTy createCritical(const LocationDescription &Loc,
3288 BodyGenCallbackTy BodyGenCB,
3289 FinalizeCallbackTy FiniCB,
3290 StringRef CriticalName,
3291 Value *HintInst);
3292
3293 /// Generator for '#omp ordered depend (source | sink)'
3294 ///
3295 /// \param Loc The insert and source location description.
3296 /// \param AllocaIP The insertion point to be used for alloca instructions.
3297 /// \param NumLoops The number of loops in depend clause.
3298 /// \param StoreValues The value will be stored in vector address.
3299 /// \param Name The name of alloca instruction.
3300 /// \param IsDependSource If true, depend source; otherwise, depend sink.
3301 ///
3302 /// \return The insertion position *after* the ordered.
3304 createOrderedDepend(const LocationDescription &Loc, InsertPointTy AllocaIP,
3305 unsigned NumLoops, ArrayRef<llvm::Value *> StoreValues,
3306 const Twine &Name, bool IsDependSource);
3307
3308 /// Generator for '#omp ordered [threads | simd]'
3309 ///
3310 /// \param Loc The insert and source location description.
3311 /// \param BodyGenCB Callback that will generate the region code.
3312 /// \param FiniCB Callback to finalize variable copies.
3313 /// \param IsThreads If true, with threads clause or without clause;
3314 /// otherwise, with simd clause;
3315 ///
3316 /// \returns The insertion position *after* the ordered.
3318 const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB,
3319 FinalizeCallbackTy FiniCB, bool IsThreads);
3320
3321 /// Generator for '#omp sections'
3322 ///
3323 /// \param Loc The insert and source location description.
3324 /// \param AllocaIP The insertion points to be used for alloca instructions.
3325 /// \param SectionCBs Callbacks that will generate body of each section.
3326 /// \param PrivCB Callback to copy a given variable (think copy constructor).
3327 /// \param FiniCB Callback to finalize variable copies.
3328 /// \param IsCancellable Flag to indicate a cancellable parallel region.
3329 /// \param IsNowait If true, barrier - to ensure all sections are executed
3330 /// before moving forward will not be generated.
3331 /// \returns The insertion position *after* the sections.
3333 createSections(const LocationDescription &Loc, InsertPointTy AllocaIP,
3336 bool IsCancellable, bool IsNowait);
3337
3338 /// Generator for '#omp section'
3339 ///
3340 /// \param Loc The insert and source location description.
3341 /// \param BodyGenCB Callback that will generate the region body code.
3342 /// \param FiniCB Callback to finalize variable copies.
3343 /// \returns The insertion position *after* the section.
3344 LLVM_ABI InsertPointOrErrorTy createSection(const LocationDescription &Loc,
3345 BodyGenCallbackTy BodyGenCB,
3346 FinalizeCallbackTy FiniCB);
3347
3348 /// Generator for `#omp teams`
3349 ///
3350 /// \param Loc The location where the teams construct was encountered.
3351 /// \param BodyGenCB Callback that will generate the region code.
3352 /// \param NumTeamsLower Lower bound on number of teams. If this is nullptr,
3353 /// it is as if lower bound is specified as equal to upperbound. If
3354 /// this is non-null, then upperbound must also be non-null.
3355 /// \param NumTeamsUpper Upper bound on the number of teams.
3356 /// \param ThreadLimit on the number of threads that may participate in a
3357 /// contention group created by each team.
3358 /// \param IfExpr is the integer argument value of the if condition on the
3359 /// teams clause.
3360 LLVM_ABI InsertPointOrErrorTy createTeams(const LocationDescription &Loc,
3361 BodyGenCallbackTy BodyGenCB,
3362 Value *NumTeamsLower = nullptr,
3363 Value *NumTeamsUpper = nullptr,
3364 Value *ThreadLimit = nullptr,
3365 Value *IfExpr = nullptr);
3366
3367 /// Generator for `#omp distribute`
3368 ///
3369 /// \param Loc The location where the distribute construct was encountered.
3370 /// \param AllocaIP The insertion point to be used for allocations.
3371 /// \param DeallocBlocks The insertion blocks to be used for explicit
3372 /// deallocations, if needed.
3373 /// \param BodyGenCB Callback that will generate the region code.
3375 const LocationDescription &Loc, InsertPointTy AllocaIP,
3376 ArrayRef<BasicBlock *> DeallocBlocks, BodyGenCallbackTy BodyGenCB);
3377
3378 /// Generate conditional branch and relevant BasicBlocks through which private
3379 /// threads copy the 'copyin' variables from Master copy to threadprivate
3380 /// copies.
3381 ///
3382 /// \param IP insertion block for copyin conditional
3383 /// \param MasterVarPtr a pointer to the master variable
3384 /// \param PrivateVarPtr a pointer to the threadprivate variable
3385 /// \param IntPtrTy Pointer size type
3386 /// \param BranchtoEnd Create a branch between the copyin.not.master blocks
3387 // and copy.in.end block
3388 ///
3389 /// \returns The insertion point where copying operation to be emitted.
3391 Value *MasterAddr,
3392 Value *PrivateAddr,
3394 bool BranchtoEnd = true);
3395
3396 /// Create a runtime call for kmpc_alloc
3397 ///
3398 /// \param Loc The insert and source location description.
3399 /// \param Size Size of allocated memory space
3400 /// \param Allocator Allocator information instruction
3401 /// \param Name Name of call Instruction for OMP_alloc
3402 ///
3403 /// \returns CallInst to the OMP_Alloc call
3404 LLVM_ABI CallInst *createOMPAlloc(const LocationDescription &Loc, Value *Size,
3405 Value *Allocator, std::string Name = "");
3406
3407 /// Create a runtime call for kmpc_align_alloc
3408 ///
3409 /// \param Loc The insert and source location description.
3410 /// \param Align Align value
3411 /// \param Size Size of allocated memory space
3412 /// \param Allocator Allocator information instruction
3413 /// \param Name Name of call Instruction for OMP_Align_Alloc
3414 ///
3415 /// \returns CallInst to the OMP_Align_Alloc call
3416 LLVM_ABI CallInst *createOMPAlignedAlloc(const LocationDescription &Loc,
3417 Value *Align, Value *Size,
3418 Value *Allocator,
3419 std::string Name = "");
3420
3421 /// Create a runtime call for kmpc_free
3422 ///
3423 /// \param Loc The insert and source location description.
3424 /// \param Addr Address of memory space to be freed
3425 /// \param Allocator Allocator information instruction
3426 /// \param Name Name of call Instruction for OMP_Free
3427 ///
3428 /// \returns CallInst to the OMP_Free call
3429 LLVM_ABI CallInst *createOMPFree(const LocationDescription &Loc, Value *Addr,
3430 Value *Allocator, std::string Name = "");
3431
3432 /// Create a runtime call for kmpc_alloc_shared.
3433 ///
3434 /// \param Loc The insert and source location description.
3435 /// \param Size Size of allocated memory space.
3436 /// \param Name Name of call Instruction.
3437 ///
3438 /// \returns CallInst to the kmpc_alloc_shared call.
3439 LLVM_ABI CallInst *createOMPAllocShared(const LocationDescription &Loc,
3440 Value *Size,
3441 const Twine &Name = Twine(""));
3442
3443 /// Create a runtime call for kmpc_alloc_shared.
3444 ///
3445 /// \param Loc The insert and source location description.
3446 /// \param VarType Type of variable to be allocated.
3447 /// \param Name Name of call Instruction.
3448 ///
3449 /// \returns CallInst to the kmpc_alloc_shared call.
3450 LLVM_ABI CallInst *createOMPAllocShared(const LocationDescription &Loc,
3451 Type *VarType,
3452 const Twine &Name = Twine(""));
3453
3454 /// Create a runtime call for kmpc_free_shared.
3455 ///
3456 /// \param Loc The insert and source location description.
3457 /// \param Addr Value obtained from the corresponding kmpc_alloc_shared call.
3458 /// \param Size Size of allocated memory space.
3459 /// \param Name Name of call Instruction.
3460 ///
3461 /// \returns CallInst to the kmpc_free_shared call.
3462 LLVM_ABI CallInst *createOMPFreeShared(const LocationDescription &Loc,
3463 Value *Addr, Value *Size,
3464 const Twine &Name = Twine(""));
3465
3466 /// Create a runtime call for kmpc_free_shared.
3467 ///
3468 /// \param Loc The insert and source location description.
3469 /// \param Addr Value obtained from the corresponding kmpc_alloc_shared call.
3470 /// \param VarType Type of variable to be freed.
3471 /// \param Name Name of call Instruction.
3472 ///
3473 /// \returns CallInst to the kmpc_free_shared call.
3474 LLVM_ABI CallInst *createOMPFreeShared(const LocationDescription &Loc,
3475 Value *Addr, Type *VarType,
3476 const Twine &Name = Twine(""));
3477
3478 /// Create a runtime call for kmpc_threadprivate_cached
3479 ///
3480 /// \param Loc The insert and source location description.
3481 /// \param Pointer pointer to data to be cached
3482 /// \param Size size of data to be cached
3483 /// \param Name Name of call Instruction for callinst
3484 ///
3485 /// \returns CallInst to the thread private cache call.
3486 LLVM_ABI CallInst *
3487 createCachedThreadPrivate(const LocationDescription &Loc,
3489 const llvm::Twine &Name = Twine(""));
3490
3491 /// Create a runtime call for __tgt_interop_init
3492 ///
3493 /// \param Loc The insert and source location description.
3494 /// \param InteropVar variable to be allocated
3495 /// \param InteropType type of interop operation
3496 /// \param Device devide to which offloading will occur
3497 /// \param NumDependences number of dependence variables
3498 /// \param DependenceAddress pointer to dependence variables
3499 /// \param HaveNowaitClause does nowait clause exist
3500 ///
3501 /// \returns CallInst to the __tgt_interop_init call
3502 LLVM_ABI CallInst *createOMPInteropInit(const LocationDescription &Loc,
3503 Value *InteropVar,
3504 omp::OMPInteropType InteropType,
3505 Value *Device, Value *NumDependences,
3506 Value *DependenceAddress,
3507 bool HaveNowaitClause);
3508
3509 /// Create a runtime call for __tgt_interop_destroy
3510 ///
3511 /// \param Loc The insert and source location description.
3512 /// \param InteropVar variable to be allocated
3513 /// \param Device devide to which offloading will occur
3514 /// \param NumDependences number of dependence variables
3515 /// \param DependenceAddress pointer to dependence variables
3516 /// \param HaveNowaitClause does nowait clause exist
3517 ///
3518 /// \returns CallInst to the __tgt_interop_destroy call
3519 LLVM_ABI CallInst *createOMPInteropDestroy(const LocationDescription &Loc,
3520 Value *InteropVar, Value *Device,
3521 Value *NumDependences,
3522 Value *DependenceAddress,
3523 bool HaveNowaitClause);
3524
3525 /// Create a runtime call for __tgt_interop_use
3526 ///
3527 /// \param Loc The insert and source location description.
3528 /// \param InteropVar variable to be allocated
3529 /// \param Device devide to which offloading will occur
3530 /// \param NumDependences number of dependence variables
3531 /// \param DependenceAddress pointer to dependence variables
3532 /// \param HaveNowaitClause does nowait clause exist
3533 ///
3534 /// \returns CallInst to the __tgt_interop_use call
3535 LLVM_ABI CallInst *createOMPInteropUse(const LocationDescription &Loc,
3536 Value *InteropVar, Value *Device,
3537 Value *NumDependences,
3538 Value *DependenceAddress,
3539 bool HaveNowaitClause);
3540
3541 /// The `omp target` interface
3542 ///
3543 /// For more information about the usage of this interface,
3544 /// \see openmp/device/include/Interface.h
3545 ///
3546 ///{
3547
3548 /// Create a runtime call for kmpc_target_init
3549 ///
3550 /// \param Loc The insert and source location description.
3551 /// \param Attrs Structure containing the default attributes, including
3552 /// numbers of threads and teams to launch the kernel with.
3554 const LocationDescription &Loc,
3556
3557 /// Create a runtime call for kmpc_target_deinit
3558 ///
3559 /// \param Loc The insert and source location description.
3560 /// \param TeamsReductionDataSize The maximal size of all the reduction data
3561 /// for teams reduction.
3562 LLVM_ABI void createTargetDeinit(const LocationDescription &Loc,
3563 int32_t TeamsReductionDataSize = 0);
3564
3565 ///}
3566
3567 /// Helpers to read/write kernel annotations from the IR.
3568 ///
3569 ///{
3570
3571 /// Read/write a bounds on threads for \p Kernel. Read will return 0 if none
3572 /// is set.
3573 LLVM_ABI static std::pair<int32_t, int32_t>
3574 readThreadBoundsForKernel(const Triple &T, Function &Kernel);
3575 LLVM_ABI static void writeThreadBoundsForKernel(const Triple &T,
3576 Function &Kernel, int32_t LB,
3577 int32_t UB);
3578
3579 /// Read/write a bounds on teams for \p Kernel. Read will return 0 if none
3580 /// is set.
3581 LLVM_ABI static std::pair<int32_t, int32_t>
3582 readTeamBoundsForKernel(const Triple &T, Function &Kernel);
3583 LLVM_ABI static void writeTeamsForKernel(const Triple &T, Function &Kernel,
3584 int32_t LB, int32_t UB);
3585 ///}
3586
3587private:
3588 // Sets the function attributes expected for the outlined function
3589 void setOutlinedTargetRegionFunctionAttributes(Function *OutlinedFn);
3590
3591 // Creates the function ID/Address for the given outlined function.
3592 // In the case of an embedded device function the address of the function is
3593 // used, in the case of a non-offload function a constant is created.
3594 Constant *createOutlinedFunctionID(Function *OutlinedFn,
3595 StringRef EntryFnIDName);
3596
3597 // Creates the region entry address for the outlined function
3598 Constant *createTargetRegionEntryAddr(Function *OutlinedFunction,
3599 StringRef EntryFnName);
3600
3601public:
3602 /// Functions used to generate a function with the given name.
3604 std::function<Expected<Function *>(StringRef FunctionName)>;
3605
3606 /// Create a unique name for the entry function using the source location
3607 /// information of the current target region. The name will be something like:
3608 ///
3609 /// __omp_offloading_DD_FFFF_PP_lBB[_CC]
3610 ///
3611 /// where DD_FFFF is an ID unique to the file (device and file IDs), PP is the
3612 /// mangled name of the function that encloses the target region and BB is the
3613 /// line number of the target region. CC is a count added when more than one
3614 /// region is located at the same location.
3615 ///
3616 /// If this target outline function is not an offload entry, we don't need to
3617 /// register it. This may happen if it is guarded by an if clause that is
3618 /// false at compile time, or no target archs have been specified.
3619 ///
3620 /// The created target region ID is used by the runtime library to identify
3621 /// the current target region, so it only has to be unique and not
3622 /// necessarily point to anything. It could be the pointer to the outlined
3623 /// function that implements the target region, but we aren't using that so
3624 /// that the compiler doesn't need to keep that, and could therefore inline
3625 /// the host function if proven worthwhile during optimization. In the other
3626 /// hand, if emitting code for the device, the ID has to be the function
3627 /// address so that it can retrieved from the offloading entry and launched
3628 /// by the runtime library. We also mark the outlined function to have
3629 /// external linkage in case we are emitting code for the device, because
3630 /// these functions will be entry points to the device.
3631 ///
3632 /// \param InfoManager The info manager keeping track of the offload entries
3633 /// \param EntryInfo The entry information about the function
3634 /// \param GenerateFunctionCallback The callback function to generate the code
3635 /// \param OutlinedFunction Pointer to the outlined function
3636 /// \param EntryFnIDName Name of the ID o be created
3638 TargetRegionEntryInfo &EntryInfo,
3639 FunctionGenCallback &GenerateFunctionCallback, bool IsOffloadEntry,
3640 Function *&OutlinedFn, Constant *&OutlinedFnID);
3641
3642 /// Registers the given function and sets up the attribtues of the function
3643 /// Returns the FunctionID.
3644 ///
3645 /// \param InfoManager The info manager keeping track of the offload entries
3646 /// \param EntryInfo The entry information about the function
3647 /// \param OutlinedFunction Pointer to the outlined function
3648 /// \param EntryFnName Name of the outlined function
3649 /// \param EntryFnIDName Name of the ID o be created
3652 Function *OutlinedFunction,
3653 StringRef EntryFnName, StringRef EntryFnIDName);
3654
3655 /// Type of BodyGen to use for region codegen
3656 ///
3657 /// Priv: If device pointer privatization is required, emit the body of the
3658 /// region here. It will have to be duplicated: with and without
3659 /// privatization.
3660 /// DupNoPriv: If we need device pointer privatization, we need
3661 /// to emit the body of the region with no privatization in the 'else' branch
3662 /// of the conditional.
3663 /// NoPriv: If we don't require privatization of device
3664 /// pointers, we emit the body in between the runtime calls. This avoids
3665 /// duplicating the body code.
3667
3668 /// Callback type for creating the map infos for the kernel parameters.
3669 /// \param CodeGenIP is the insertion point where code should be generated,
3670 /// if any.
3673
3674private:
3675 /// Emit the array initialization or deletion portion for user-defined mapper
3676 /// code generation. First, it evaluates whether an array section is mapped
3677 /// and whether the \a MapType instructs to delete this section. If \a IsInit
3678 /// is true, and \a MapType indicates to not delete this array, array
3679 /// initialization code is generated. If \a IsInit is false, and \a MapType
3680 /// indicates to delete this array, array deletion code is generated.
3681 void emitUDMapperArrayInitOrDel(Function *MapperFn, llvm::Value *MapperHandle,
3682 llvm::Value *Base, llvm::Value *Begin,
3683 llvm::Value *Size, llvm::Value *MapType,
3684 llvm::Value *MapName, TypeSize ElementSize,
3685 llvm::BasicBlock *ExitBB, bool IsInit);
3686
3687public:
3688 /// Emit the user-defined mapper function. The code generation follows the
3689 /// pattern in the example below.
3690 /// \code
3691 /// void .omp_mapper.<type_name>.<mapper_id>.(void *rt_mapper_handle,
3692 /// void *base, void *begin,
3693 /// int64_t size, int64_t type,
3694 /// void *name = nullptr) {
3695 /// // Allocate space for an array section first or add a base/begin for
3696 /// // pointer dereference.
3697 /// if ((size > 1 || (base != begin && maptype.IsPtrAndObj)) &&
3698 /// !maptype.IsDelete)
3699 /// __tgt_push_mapper_component(rt_mapper_handle, base, begin,
3700 /// size*sizeof(Ty), clearToFromMember(type));
3701 /// // Map members.
3702 /// for (unsigned i = 0; i < size; i++) {
3703 /// // For each component specified by this mapper:
3704 /// for (auto c : begin[i]->all_components) {
3705 /// // Map-type-modifying bits (ALWAYS, DELETE, CLOSE) from the outer
3706 /// // map clause are propagated to each component, except ATTACH
3707 /// // entries (ATTACH|ALWAYS is reserved for attach(always), and other
3708 /// // modifier bits have no meaning for ATTACH). PRESENT is
3709 /// // additionally propagated to components with HasAttachPtr (the
3710 /// // pointee data) when PropagatePresentToPointee is set
3711 /// // (OpenMP >= 6.0).
3712 /// present_bit = (PropagatePresentToPointee && c.hasAttachPtr())
3713 /// ? PRESENT : 0;
3714 /// imported_modifier_bits = type & (ALWAYS | DELETE | CLOSE |
3715 /// present_bit);
3716 /// effective_type = c.isAttach() ? c.arg_type
3717 /// : c.arg_type | imported_modifier_bits;
3718 /// if (c.hasMapper())
3719 /// (*c.Mapper())(rt_mapper_handle, c.arg_base, c.arg_begin,
3720 /// c.arg_size, effective_type, c.arg_name);
3721 /// else
3722 /// __tgt_push_mapper_component(rt_mapper_handle, c.arg_base,
3723 /// c.arg_begin, c.arg_size,
3724 /// effective_type, c.arg_name);
3725 /// }
3726 /// }
3727 /// // Delete the array section.
3728 /// if (size > 1 && maptype.IsDelete)
3729 /// __tgt_push_mapper_component(rt_mapper_handle, base, begin,
3730 /// size*sizeof(Ty), clearToFromMember(type));
3731 /// }
3732 /// \endcode
3733 ///
3734 /// \param PrivAndGenMapInfoCB Callback that privatizes code and populates the
3735 /// MapInfos and returns.
3736 /// \param ElemTy DeclareMapper element type.
3737 /// \param FuncName Optional param to specify mapper function name.
3738 /// \param CustomMapperCB Optional callback to generate code related to
3739 /// custom mappers.
3740 /// \param PropagatePresentToPointee If true, the PRESENT map-type modifier
3741 /// from the outer clause is propagated to the pointee entries the mapper
3742 /// inserts, i.e. those with HasAttachPtr. Callers set this only for
3743 /// OpenMP >= 6.0; at earlier versions the present modifier is treated as not
3744 /// applying to the pointee.
3747 InsertPointTy CodeGenIP, llvm::Value *PtrPHI, llvm::Value *BeginArg)>
3748 PrivAndGenMapInfoCB,
3749 llvm::Type *ElemTy, StringRef FuncName,
3750 CustomMapperCallbackTy CustomMapperCB, bool PreserveMemberOfFlags = false,
3751 bool PropagatePresentToPointee = false);
3752
3753 /// Generator for '#omp target data'
3754 ///
3755 /// \param Loc The location where the target data construct was encountered.
3756 /// \param AllocaIP The insertion points to be used for allocations.
3757 /// \param CodeGenIP The insertion point at which the target directive code
3758 /// should be placed.
3759 /// \param DeallocBlocks The insertion blocks at which explicit deallocations
3760 /// should be placed, if needed.
3761 /// \param IsBegin If true then emits begin mapper call otherwise emits
3762 /// end mapper call.
3763 /// \param DeviceID Stores the DeviceID from the device clause.
3764 /// \param IfCond Value which corresponds to the if clause condition.
3765 /// \param Info Stores all information realted to the Target Data directive.
3766 /// \param GenMapInfoCB Callback that populates the MapInfos and returns.
3767 /// \param CustomMapperCB Callback to generate code related to
3768 /// custom mappers.
3769 /// \param BodyGenCB Optional Callback to generate the region code.
3770 /// \param DeviceAddrCB Optional callback to generate code related to
3771 /// use_device_ptr and use_device_addr.
3773 const LocationDescription &Loc, InsertPointTy AllocaIP,
3774 InsertPointTy CodeGenIP, ArrayRef<BasicBlock *> DeallocBlocks,
3775 Value *DeviceID, Value *IfCond, TargetDataInfo &Info,
3776 GenMapInfoCallbackTy GenMapInfoCB, CustomMapperCallbackTy CustomMapperCB,
3777 omp::RuntimeFunction *MapperFunc = nullptr,
3779 BodyGenTy BodyGenType)>
3780 BodyGenCB = nullptr,
3781 function_ref<void(unsigned int, Value *)> DeviceAddrCB = nullptr,
3782 Value *SrcLocInfo = nullptr);
3783
3785 InsertPointTy AllocaIP, InsertPointTy CodeGenIP,
3786 ArrayRef<BasicBlock *> DeallocBlocks)>;
3787
3789 Argument &Arg, Value *Input, Value *&RetVal, InsertPointTy AllocaIP,
3790 InsertPointTy CodeGenIP, ArrayRef<InsertPointTy> DeallocIPs)>;
3791
3792 /// Generator for '#omp target'
3793 ///
3794 /// \param Loc where the target data construct was encountered.
3795 /// \param IsOffloadEntry whether it is an offload entry.
3796 /// \param CodeGenIP The insertion point where the call to the outlined
3797 /// function should be emitted.
3798 /// \param DeallocBlocks The insertion points at which explicit deallocations
3799 /// should be placed, if needed.
3800 /// \param Info Stores all information realted to the Target directive.
3801 /// \param EntryInfo The entry information about the function.
3802 /// \param DefaultAttrs Structure containing the default attributes, including
3803 /// numbers of threads and teams to launch the kernel with.
3804 /// \param RuntimeAttrs Structure containing the runtime numbers of threads
3805 /// and teams to launch the kernel with.
3806 /// \param IfCond value of the `if` clause.
3807 /// \param Inputs The input values to the region that will be passed.
3808 /// as arguments to the outlined function.
3809 /// \param BodyGenCB Callback that will generate the region code.
3810 /// \param ArgAccessorFuncCB Callback that will generate accessors
3811 /// instructions for passed in target arguments where neccessary
3812 /// \param CustomMapperCB Callback to generate code related to
3813 /// custom mappers.
3814 /// \param Dependencies A vector of DependData objects that carry
3815 /// dependency information as passed in the depend clause
3816 /// \param HasNowait Whether the target construct has a `nowait` clause or
3817 /// not.
3818 /// \param DynCGroupMem The size of the dynamic groupprivate memory for each
3819 /// cgroup.
3820 /// \param DynCGroupMem The fallback mechanism to execute if the requested
3821 /// cgroup memory cannot be provided.
3822 /// \param OutlinedFnLoc Location scoped to the DISubprogram that the caller
3823 /// will attach to the outlined function. \p Loc is scoped to the
3824 /// parent function, so it cannot be used for code emitted inside the
3825 /// outlined function. If this is empty, such code is emitted without a
3826 /// debug location.
3828 const LocationDescription &Loc, bool IsOffloadEntry,
3831 ArrayRef<BasicBlock *> DeallocBlocks, TargetDataInfo &Info,
3832 TargetRegionEntryInfo &EntryInfo,
3833 const TargetKernelDefaultAttrs &DefaultAttrs,
3834 const TargetKernelRuntimeAttrs &RuntimeAttrs, Value *IfCond,
3835 SmallVectorImpl<Value *> &Inputs, GenMapInfoCallbackTy GenMapInfoCB,
3836 TargetBodyGenCallbackTy BodyGenCB,
3837 TargetGenArgAccessorsCallbackTy ArgAccessorFuncCB,
3838 CustomMapperCallbackTy CustomMapperCB,
3839 const DependenciesInfo &Dependencies = {}, bool HasNowait = false,
3840 Value *DynCGroupMem = nullptr,
3841 omp::OMPDynGroupprivateFallbackType DynCGroupMemFallback =
3843 DebugLoc OutlinedFnLoc = {});
3844
3845 /// Returns __kmpc_for_static_init_* runtime function for the specified
3846 /// size \a IVSize and sign \a IVSigned. Will create a distribute call
3847 /// __kmpc_distribute_static_init* if \a IsGPUDistribute is set.
3848 LLVM_ABI FunctionCallee createForStaticInitFunction(unsigned IVSize,
3849 bool IVSigned,
3850 bool IsGPUDistribute);
3851
3852 /// Returns __kmpc_dispatch_init_* runtime function for the specified
3853 /// size \a IVSize and sign \a IVSigned.
3854 LLVM_ABI FunctionCallee createDispatchInitFunction(unsigned IVSize,
3855 bool IVSigned);
3856
3857 /// Returns __kmpc_dispatch_next_* runtime function for the specified
3858 /// size \a IVSize and sign \a IVSigned.
3859 LLVM_ABI FunctionCallee createDispatchNextFunction(unsigned IVSize,
3860 bool IVSigned);
3861
3862 /// Returns __kmpc_dispatch_fini_* runtime function for the specified
3863 /// size \a IVSize and sign \a IVSigned.
3864 LLVM_ABI FunctionCallee createDispatchFiniFunction(unsigned IVSize,
3865 bool IVSigned);
3866
3867 /// Returns __kmpc_dispatch_deinit runtime function.
3868 LLVM_ABI FunctionCallee createDispatchDeinitFunction();
3869
3870 /// Declarations for LLVM-IR types (simple, array, function and structure) are
3871 /// generated below. Their names are defined and used in OpenMPKinds.def. Here
3872 /// we provide the declarations, the initializeTypes function will provide the
3873 /// values.
3874 ///
3875 ///{
3876#define OMP_TYPE(VarName, InitValue) Type *VarName = nullptr;
3877#define OMP_ARRAY_TYPE(VarName, ElemTy, ArraySize) \
3878 ArrayType *VarName##Ty = nullptr; \
3879 PointerType *VarName##PtrTy = nullptr;
3880#define OMP_FUNCTION_TYPE(VarName, IsVarArg, ReturnType, ...) \
3881 FunctionType *VarName = nullptr; \
3882 PointerType *VarName##Ptr = nullptr;
3883#define OMP_STRUCT_TYPE(VarName, StrName, ...) \
3884 StructType *VarName = nullptr; \
3885 PointerType *VarName##Ptr = nullptr;
3886#include "llvm/Frontend/OpenMP/OMPKinds.def"
3887
3888 ///}
3889
3890private:
3891 /// Create all simple and struct types exposed by the runtime and remember
3892 /// the llvm::PointerTypes of them for easy access later.
3893 void initializeTypes(Module &M);
3894
3895 /// Common interface for generating entry calls for OMP Directives.
3896 /// if the directive has a region/body, It will set the insertion
3897 /// point to the body
3898 ///
3899 /// \param OMPD Directive to generate entry blocks for
3900 /// \param EntryCall Call to the entry OMP Runtime Function
3901 /// \param ExitBB block where the region ends.
3902 /// \param Conditional indicate if the entry call result will be used
3903 /// to evaluate a conditional of whether a thread will execute
3904 /// body code or not.
3905 ///
3906 /// \return The insertion position in exit block
3907 InsertPointTy emitCommonDirectiveEntry(omp::Directive OMPD, Value *EntryCall,
3908 BasicBlock *ExitBB,
3909 bool Conditional = false);
3910
3911 /// Common interface to finalize the region
3912 ///
3913 /// \param OMPD Directive to generate exiting code for
3914 /// \param FinIP Insertion point for emitting Finalization code and exit call.
3915 /// This block must not contain any non-finalization code.
3916 /// \param ExitCall Call to the ending OMP Runtime Function
3917 /// \param HasFinalize indicate if the directive will require finalization
3918 /// and has a finalization callback in the stack that
3919 /// should be called.
3920 ///
3921 /// \return The insertion position in exit block
3922 InsertPointOrErrorTy emitCommonDirectiveExit(omp::Directive OMPD,
3923 InsertPointTy FinIP,
3924 Instruction *ExitCall,
3925 bool HasFinalize = true);
3926
3927 /// Common Interface to generate OMP inlined regions
3928 ///
3929 /// \param OMPD Directive to generate inlined region for
3930 /// \param EntryCall Call to the entry OMP Runtime Function
3931 /// \param ExitCall Call to the ending OMP Runtime Function
3932 /// \param BodyGenCB Body code generation callback.
3933 /// \param FiniCB Finalization Callback. Will be called when finalizing region
3934 /// \param Conditional indicate if the entry call result will be used
3935 /// to evaluate a conditional of whether a thread will execute
3936 /// body code or not.
3937 /// \param HasFinalize indicate if the directive will require finalization
3938 /// and has a finalization callback in the stack that
3939 /// should be called.
3940 /// \param IsCancellable if HasFinalize is set to true, indicate if the
3941 /// the directive should be cancellable.
3942 /// \return The insertion point after the region
3944 EmitOMPInlinedRegion(omp::Directive OMPD, Instruction *EntryCall,
3945 Instruction *ExitCall, BodyGenCallbackTy BodyGenCB,
3946 FinalizeCallbackTy FiniCB, bool Conditional = false,
3947 bool HasFinalize = true, bool IsCancellable = false);
3948
3949 /// Get the platform-specific name separator.
3950 /// \param Parts different parts of the final name that needs separation
3951 /// \param FirstSeparator First separator used between the initial two
3952 /// parts of the name.
3953 /// \param Separator separator used between all of the rest consecutive
3954 /// parts of the name
3955 static std::string getNameWithSeparators(ArrayRef<StringRef> Parts,
3956 StringRef FirstSeparator,
3957 StringRef Separator);
3958
3959 /// Returns corresponding lock object for the specified critical region
3960 /// name. If the lock object does not exist it is created, otherwise the
3961 /// reference to the existing copy is returned.
3962 /// \param CriticalName Name of the critical region.
3963 ///
3964 Value *getOMPCriticalRegionLock(StringRef CriticalName);
3965
3966 /// Callback type for Atomic Expression update
3967 /// ex:
3968 /// \code{.cpp}
3969 /// unsigned x = 0;
3970 /// #pragma omp atomic update
3971 /// x = Expr(x_old); //Expr() is any legal operation
3972 /// \endcode
3973 ///
3974 /// \param XOld the value of the atomic memory address to use for update
3975 /// \param IRB reference to the IRBuilder to use
3976 ///
3977 /// \returns Value to update X to.
3978 using AtomicUpdateCallbackTy =
3979 const function_ref<Expected<Value *>(Value *XOld, IRBuilder<> &IRB)>;
3980
3981private:
3982 enum AtomicKind { Read, Write, Update, Capture, Compare };
3983
3984 /// Determine whether to emit flush or not
3985 ///
3986 /// \param Loc The insert and source location description.
3987 /// \param AO The required atomic ordering
3988 /// \param AK The OpenMP atomic operation kind used.
3989 ///
3990 /// \returns wether a flush was emitted or not
3991 bool checkAndEmitFlushAfterAtomic(const LocationDescription &Loc,
3992 AtomicOrdering AO, AtomicKind AK);
3993
3994 /// Emit atomic update for constructs: X = X BinOp Expr ,or X = Expr BinOp X
3995 /// For complex Operations: X = UpdateOp(X) => CmpExch X, old_X, UpdateOp(X)
3996 /// Only Scalar data types.
3997 ///
3998 /// \param AllocaIP The insertion point to be used for alloca
3999 /// instructions.
4000 /// \param X The target atomic pointer to be updated
4001 /// \param XElemTy The element type of the atomic pointer.
4002 /// \param Expr The value to update X with.
4003 /// \param AO Atomic ordering of the generated atomic
4004 /// instructions.
4005 /// \param RMWOp The binary operation used for update. If
4006 /// operation is not supported by atomicRMW,
4007 /// or belong to {FADD, FSUB, BAD_BINOP}.
4008 /// Then a `cmpExch` based atomic will be generated.
4009 /// \param UpdateOp Code generator for complex expressions that cannot be
4010 /// expressed through atomicrmw instruction.
4011 /// \param VolatileX true if \a X volatile?
4012 /// \param IsXBinopExpr true if \a X is Left H.S. in Right H.S. part of the
4013 /// update expression, false otherwise.
4014 /// (e.g. true for X = X BinOp Expr)
4015 ///
4016 /// \returns A pair of the old value of X before the update, and the value
4017 /// used for the update.
4018 Expected<std::pair<Value *, Value *>>
4019 emitAtomicUpdate(InsertPointTy AllocaIP, Value *X, Type *XElemTy, Value *Expr,
4021 AtomicUpdateCallbackTy &UpdateOp, bool VolatileX,
4022 bool IsXBinopExpr, bool IsIgnoreDenormalMode,
4023 bool IsFineGrainedMemory, bool IsRemoteMemory);
4024
4025 /// Emit the binary op. described by \p RMWOp, using \p Src1 and \p Src2 .
4026 ///
4027 /// \Return The instruction
4028 Value *emitRMWOpAsInstruction(Value *Src1, Value *Src2,
4029 AtomicRMWInst::BinOp RMWOp);
4030
4031 bool IsFinalized;
4032
4033public:
4034 /// a struct to pack relevant information while generating atomic Ops
4036 Value *Var = nullptr;
4037 Type *ElemTy = nullptr;
4038 bool IsSigned = false;
4039 bool IsVolatile = false;
4040 };
4041
4042 /// Emit atomic Read for : V = X --- Only Scalar data types.
4043 ///
4044 /// \param Loc The insert and source location description.
4045 /// \param X The target pointer to be atomically read
4046 /// \param V Memory address where to store atomically read
4047 /// value
4048 /// \param AO Atomic ordering of the generated atomic
4049 /// instructions.
4050 /// \param AllocaIP Insert point for allocas
4051 //
4052 /// \return Insertion point after generated atomic read IR.
4055 AtomicOrdering AO,
4056 InsertPointTy AllocaIP);
4057
4058 /// Emit atomic write for : X = Expr --- Only Scalar data types.
4059 ///
4060 /// \param Loc The insert and source location description.
4061 /// \param X The target pointer to be atomically written to
4062 /// \param Expr The value to store.
4063 /// \param AO Atomic ordering of the generated atomic
4064 /// instructions.
4065 /// \param AllocaIP Insert point for allocas
4066 ///
4067 /// \return Insertion point after generated atomic Write IR.
4069 AtomicOpValue &X, Value *Expr,
4070 AtomicOrdering AO,
4071 InsertPointTy AllocaIP);
4072
4073 /// Emit atomic update for constructs: X = X BinOp Expr ,or X = Expr BinOp X
4074 /// For complex Operations: X = UpdateOp(X) => CmpExch X, old_X, UpdateOp(X)
4075 /// Only Scalar data types.
4076 ///
4077 /// \param Loc The insert and source location description.
4078 /// \param AllocaIP The insertion point to be used for alloca instructions.
4079 /// \param X The target atomic pointer to be updated
4080 /// \param Expr The value to update X with.
4081 /// \param AO Atomic ordering of the generated atomic instructions.
4082 /// \param RMWOp The binary operation used for update. If operation
4083 /// is not supported by atomicRMW, or belong to
4084 /// {FADD, FSUB, BAD_BINOP}. Then a `cmpExch` based
4085 /// atomic will be generated.
4086 /// \param UpdateOp Code generator for complex expressions that cannot be
4087 /// expressed through atomicrmw instruction.
4088 /// \param IsXBinopExpr true if \a X is Left H.S. in Right H.S. part of the
4089 /// update expression, false otherwise.
4090 /// (e.g. true for X = X BinOp Expr)
4091 ///
4092 /// \return Insertion point after generated atomic update IR.
4095 Value *Expr, AtomicOrdering AO, AtomicRMWInst::BinOp RMWOp,
4096 AtomicUpdateCallbackTy &UpdateOp, bool IsXBinopExpr,
4097 bool IsIgnoreDenormalMode = false, bool IsFineGrainedMemory = false,
4098 bool IsRemoteMemory = false);
4099
4100 /// Emit atomic update for constructs: --- Only Scalar data types
4101 /// V = X; X = X BinOp Expr ,
4102 /// X = X BinOp Expr; V = X,
4103 /// V = X; X = Expr BinOp X,
4104 /// X = Expr BinOp X; V = X,
4105 /// V = X; X = UpdateOp(X),
4106 /// X = UpdateOp(X); V = X,
4107 ///
4108 /// \param Loc The insert and source location description.
4109 /// \param AllocaIP The insertion point to be used for alloca instructions.
4110 /// \param X The target atomic pointer to be updated
4111 /// \param V Memory address where to store captured value
4112 /// \param Expr The value to update X with.
4113 /// \param AO Atomic ordering of the generated atomic instructions
4114 /// \param RMWOp The binary operation used for update. If
4115 /// operation is not supported by atomicRMW, or belong to
4116 /// {FADD, FSUB, BAD_BINOP}. Then a cmpExch based
4117 /// atomic will be generated.
4118 /// \param UpdateOp Code generator for complex expressions that cannot be
4119 /// expressed through atomicrmw instruction.
4120 /// \param UpdateExpr true if X is an in place update of the form
4121 /// X = X BinOp Expr or X = Expr BinOp X
4122 /// \param IsXBinopExpr true if X is Left H.S. in Right H.S. part of the
4123 /// update expression, false otherwise.
4124 /// (e.g. true for X = X BinOp Expr)
4125 /// \param IsPostfixUpdate true if original value of 'x' must be stored in
4126 /// 'v', not an updated one.
4127 ///
4128 /// \return Insertion point after generated atomic capture IR.
4131 AtomicOpValue &V, Value *Expr, AtomicOrdering AO,
4132 AtomicRMWInst::BinOp RMWOp, AtomicUpdateCallbackTy &UpdateOp,
4133 bool UpdateExpr, bool IsPostfixUpdate, bool IsXBinopExpr,
4134 bool IsIgnoreDenormalMode = false, bool IsFineGrainedMemory = false,
4135 bool IsRemoteMemory = false);
4136
4137 /// Emit atomic compare for constructs: --- Only scalar data types
4138 /// cond-expr-stmt:
4139 /// x = x ordop expr ? expr : x;
4140 /// x = expr ordop x ? expr : x;
4141 /// x = x == e ? d : x;
4142 /// x = e == x ? d : x; (this one is not in the spec)
4143 /// cond-update-stmt:
4144 /// if (x ordop expr) { x = expr; }
4145 /// if (expr ordop x) { x = expr; }
4146 /// if (x == e) { x = d; }
4147 /// if (e == x) { x = d; } (this one is not in the spec)
4148 /// conditional-update-capture-atomic:
4149 /// v = x; cond-update-stmt; (IsPostfixUpdate=true, IsFailOnly=false)
4150 /// cond-update-stmt; v = x; (IsPostfixUpdate=false, IsFailOnly=false)
4151 /// if (x == e) { x = d; } else { v = x; } (IsPostfixUpdate=false,
4152 /// IsFailOnly=true)
4153 /// r = x == e; if (r) { x = d; } (IsPostfixUpdate=false, IsFailOnly=false)
4154 /// r = x == e; if (r) { x = d; } else { v = x; } (IsPostfixUpdate=false,
4155 /// IsFailOnly=true)
4156 ///
4157 /// \param Loc The insert and source location description.
4158 /// \param X The target atomic pointer to be updated.
4159 /// \param V Memory address where to store captured value (for
4160 /// compare capture only).
4161 /// \param R Memory address where to store comparison result
4162 /// (for compare capture with '==' only).
4163 /// \param E The expected value ('e') for forms that use an
4164 /// equality comparison or an expression ('expr') for
4165 /// forms that use 'ordop' (logically an atomic maximum or
4166 /// minimum).
4167 /// \param D The desired value for forms that use an equality
4168 /// comparison. If forms that use 'ordop', it should be
4169 /// \p nullptr.
4170 /// \param AO Atomic ordering of the generated atomic instructions.
4171 /// \param Op Atomic compare operation. It can only be ==, <, or >.
4172 /// \param IsXBinopExpr True if the conditional statement is in the form where
4173 /// x is on LHS. It only matters for < or >.
4174 /// \param IsPostfixUpdate True if original value of 'x' must be stored in
4175 /// 'v', not an updated one (for compare capture
4176 /// only).
4177 /// \param IsFailOnly True if the original value of 'x' is stored to 'v'
4178 /// only when the comparison fails. This is only valid for
4179 /// the case the comparison is '=='.
4180 ///
4181 /// \return Insertion point after generated atomic capture IR.
4182 /// Whether to emit special handling for IEEE 754 -0.0 == +0.0 in
4183 /// atomic compare operations on floating-point types.
4184 bool HandleFPNegZero = false;
4185
4186 /// Set whether atomic compare should handle -0.0/+0.0 equivalence.
4187 /// Returns the previous value so callers can save and restore it.
4188 bool setHandleFPNegZero(bool FPNegZero) {
4189 bool Old = HandleFPNegZero;
4190 HandleFPNegZero = FPNegZero;
4191 return Old;
4192 }
4193
4195 const LocationDescription &Loc, AtomicOpValue &X, AtomicOpValue &V,
4196 AtomicOpValue &R, Value *E, Value *D, AtomicOrdering AO,
4197 omp::OMPAtomicCompareOp Op, bool IsXBinopExpr, bool IsPostfixUpdate,
4198 bool IsFailOnly, bool IsWeak = false);
4200 const LocationDescription &Loc, AtomicOpValue &X, AtomicOpValue &V,
4201 AtomicOpValue &R, Value *E, Value *D, AtomicOrdering AO,
4202 omp::OMPAtomicCompareOp Op, bool IsXBinopExpr, bool IsPostfixUpdate,
4203 bool IsFailOnly, AtomicOrdering Failure, bool IsWeak = false);
4204
4205 /// Create the control flow structure of a canonical OpenMP loop.
4206 ///
4207 /// The emitted loop will be disconnected, i.e. no edge to the loop's
4208 /// preheader and no terminator in the AfterBB. The OpenMPIRBuilder's
4209 /// IRBuilder location is not preserved.
4210 ///
4211 /// \param DL DebugLoc used for the instructions in the skeleton.
4212 /// \param TripCount Value to be used for the trip count.
4213 /// \param F Function in which to insert the BasicBlocks.
4214 /// \param PreInsertBefore Where to insert BBs that execute before the body,
4215 /// typically the body itself.
4216 /// \param PostInsertBefore Where to insert BBs that execute after the body.
4217 /// \param Name Base name used to derive BB
4218 /// and instruction names.
4219 /// \param IsCollapsed Whether this is a collapsed loop.
4220 ///
4221 /// \returns The CanonicalLoopInfo that represents the emitted loop.
4224 BasicBlock *PreInsertBefore, BasicBlock *PostInsertBefore,
4225 const Twine &Name = {}, bool IsCollapsed = false);
4226 /// OMP Offload Info Metadata name string
4227 const std::string ompOffloadInfoName = "omp_offload.info";
4228
4229 /// Loads all the offload entries information from the host IR
4230 /// metadata. This function is only meant to be used with device code
4231 /// generation.
4232 ///
4233 /// \param M Module to load Metadata info from. Module passed maybe
4234 /// loaded from bitcode file, i.e, different from OpenMPIRBuilder::M module.
4236
4237 /// Loads all the offload entries information from the host IR
4238 /// metadata read from the file passed in as the HostFilePath argument. This
4239 /// function is only meant to be used with device code generation.
4240 ///
4241 /// \param HostFilePath The path to the host IR file,
4242 /// used to load in offload metadata for the device, allowing host and device
4243 /// to maintain the same metadata mapping.
4245 StringRef HostFilePath);
4246
4247 /// Gets (if variable with the given name already exist) or creates
4248 /// internal global variable with the specified Name. The created variable has
4249 /// linkage CommonLinkage by default and is initialized by null value.
4250 /// \param Ty Type of the global variable. If it is exist already the type
4251 /// must be the same.
4252 /// \param Name Name of the variable.
4255 std::optional<unsigned> AddressSpace = {});
4256
4258 InsertPointTy BodyIP, llvm::Value *LinearIV)>;
4259
4260 /// Create a canonical iterator loop at the current insertion point.
4261 ///
4262 /// This helper splits the current block and builds a canonical loop
4263 /// using createLoopSkeleton(). The resulting control flow looks like:
4264 ///
4265 /// CurBB -> Preheader -> Header -> Body -> Latch -> After -> ContBB
4266 ///
4267 /// The body of the loop is produced by calling \p BodyGen with the insertion
4268 /// point for the loop body and the induction variable.
4269 /// Unlike createCanonicalLoop(), this function is intended for \p BodyGen
4270 /// that may perform region lowering (e.g., translating MLIR regions) and are
4271 /// not guaranteed to preserve the canonical skeleton's body terminator. In
4272 /// particular:
4273 ///
4274 /// - The skeleton’s unconditional branch from the loop body is removed
4275 /// before invoking \p BodyGen.
4276 /// - \p BodyGen may freely emit instructions and temporarily introduce
4277 /// control flow.
4278 /// - If the loop body does not end with a terminator after \p BodyGen
4279 /// returns, a branch to the latch is inserted to restore canonical form.
4280 ///
4281 /// \param Loc The location where the iterator modifier was encountered.
4282 /// \param TripCount Number of loop iterations.
4283 /// \param BodyGen Callback to generate the loop body.
4284 /// \param Name Base name used for creating the loop
4285 /// \returns The insertion position *after* the iterator loop
4288 IteratorBodyGenTy BodyGen, llvm::StringRef Name = "iterator");
4289
4290 /// Kind of parameter in a function with 'declare simd' directive.
4299
4300 /// Attribute set of the `declare simd` parameter.
4307
4313
4314 /// Emit x86 vector-function ABI attributes for a `declare simd` function.
4315 ///
4316 /// Generates and attaches `_ZGV*` vector function ABI attributes to \p Fn
4317 /// following the x86 vector ABI used by OpenMP `declare simd`. For each
4318 /// supported ISA (SSE, AVX, AVX2, AVX512) and masking variant, this
4319 /// constructs the appropriate mangled vector-function name and adds it as a
4320 /// function attribute.
4321 ///
4322 /// \param Fn The scalar function to which vector-function attributes
4323 /// are attached.
4324 /// \param NumElements Number of elements used to derive the vector length
4325 /// when
4326 /// \p VLENVal is not specified.
4327 /// \param VLENVal User provided vector length.
4328 /// \param ParamAttrs Array of attribute set of the `declare simd` parameter.
4329 /// \param Branch `undefined`, `inbranch` or `notinbranch` clause.
4331 llvm::Function *Fn, unsigned NumElements, const llvm::APSInt &VLENVal,
4333
4334 /// Emit AArch64 vector-function ABI attributes for a `declare simd` function.
4335 ///
4336 /// Generates and attaches `_ZGV*` vector function ABI attributes to \p Fn
4337 /// following the AArch64 vector-function ABI. The emitted names depend on the
4338 /// selected ISA, user-specified vector length, parameter attribute mangling,
4339 /// and the declare simd branch clause.
4340 ///
4341 /// \param Fn The scalar function to which vector-function
4342 /// attributes are attached.
4343 /// \param VLENVal User provided vector length.
4344 /// \param ParamAttrs Array of attribute set of the `declare simd`
4345 /// parameter.
4346 /// \param Branch `undefined`, `inbranch` or `notinbranch`
4347 /// clause.
4348 /// \param ISA `'n'` for Advanced SIMD or `'s'` for SVE.
4349 /// \param NarrowestDataSize Narrowest data size in bits used to infer the
4350 /// default vector length when \p VLENVal is
4351 /// absent.
4352 /// \param OutputBecomesInput Whether result values are represented as input
4353 /// parameters in the emitted vector-function ABI
4354 /// name.
4356 llvm::Function *Fn, unsigned VLENVal,
4358 char ISA, unsigned NarrowestDataSize, bool OutputBecomesInput);
4359};
4360
4361/// Class to represented the control flow structure of an OpenMP canonical loop.
4362///
4363/// The control-flow structure is standardized for easy consumption by
4364/// directives associated with loops. For instance, the worksharing-loop
4365/// construct may change this control flow such that each loop iteration is
4366/// executed on only one thread. The constraints of a canonical loop in brief
4367/// are:
4368///
4369/// * The number of loop iterations must have been computed before entering the
4370/// loop.
4371///
4372/// * Has an (unsigned) logical induction variable that starts at zero and
4373/// increments by one.
4374///
4375/// * The loop's CFG itself has no side-effects. The OpenMP specification
4376/// itself allows side-effects, but the order in which they happen, including
4377/// how often or whether at all, is unspecified. We expect that the frontend
4378/// will emit those side-effect instructions somewhere (e.g. before the loop)
4379/// such that the CanonicalLoopInfo itself can be side-effect free.
4380///
4381/// Keep in mind that CanonicalLoopInfo is meant to only describe a repeated
4382/// execution of a loop body that satifies these constraints. It does NOT
4383/// represent arbitrary SESE regions that happen to contain a loop. Do not use
4384/// CanonicalLoopInfo for such purposes.
4385///
4386/// The control flow can be described as follows:
4387///
4388/// Preheader
4389/// |
4390/// /-> Header
4391/// | |
4392/// | Cond---\
4393/// | | |
4394/// | Body |
4395/// | | | |
4396/// | <...> |
4397/// | | | |
4398/// \--Latch |
4399/// |
4400/// Exit
4401/// |
4402/// After
4403///
4404/// The loop is thought to start at PreheaderIP (at the Preheader's terminator,
4405/// including) and end at AfterIP (at the After's first instruction, excluding).
4406/// That is, instructions in the Preheader and After blocks (except the
4407/// Preheader's terminator) are out of CanonicalLoopInfo's control and may have
4408/// side-effects. Typically, the Preheader is used to compute the loop's trip
4409/// count. The instructions from BodyIP (at the Body block's first instruction,
4410/// excluding) until the Latch are also considered outside CanonicalLoopInfo's
4411/// control and thus can have side-effects. The body block is the single entry
4412/// point into the loop body, which may contain arbitrary control flow as long
4413/// as all control paths eventually branch to the Latch block.
4414///
4415/// TODO: Consider adding another standardized BasicBlock between Body CFG and
4416/// Latch to guarantee that there is only a single edge to the latch. It would
4417/// make loop transformations easier to not needing to consider multiple
4418/// predecessors of the latch (See redirectAllPredecessorsTo) and would give us
4419/// an equivalant to PreheaderIP, AfterIP and BodyIP for inserting code that
4420/// executes after each body iteration.
4421///
4422/// There must be no loop-carried dependencies through llvm::Values. This is
4423/// equivalant to that the Latch has no PHINode and the Header's only PHINode is
4424/// for the induction variable.
4425///
4426/// All code in Header, Cond, Latch and Exit (plus the terminator of the
4427/// Preheader) are CanonicalLoopInfo's responsibility and their build-up checked
4428/// by assertOK(). They are expected to not be modified unless explicitly
4429/// modifying the CanonicalLoopInfo through a methods that applies a OpenMP
4430/// loop-associated construct such as applyWorkshareLoop, tileLoops, unrollLoop,
4431/// etc. These methods usually invalidate the CanonicalLoopInfo and re-use its
4432/// basic blocks. After invalidation, the CanonicalLoopInfo must not be used
4433/// anymore as its underlying control flow may not exist anymore.
4434/// Loop-transformation methods such as tileLoops, collapseLoops and unrollLoop
4435/// may also return a new CanonicalLoopInfo that can be passed to other
4436/// loop-associated construct implementing methods. These loop-transforming
4437/// methods may either create a new CanonicalLoopInfo usually using
4438/// createLoopSkeleton and invalidate the input CanonicalLoopInfo, or reuse and
4439/// modify one of the input CanonicalLoopInfo and return it as representing the
4440/// modified loop. What is done is an implementation detail of
4441/// transformation-implementing method and callers should always assume that the
4442/// CanonicalLoopInfo passed to it is invalidated and a new object is returned.
4443/// Returned CanonicalLoopInfo have the same structure and guarantees as the one
4444/// created by createCanonicalLoop, such that transforming methods do not have
4445/// to special case where the CanonicalLoopInfo originated from.
4446///
4447/// Generally, methods consuming CanonicalLoopInfo do not need an
4448/// OpenMPIRBuilder::InsertPointTy as argument, but use the locations of the
4449/// CanonicalLoopInfo to insert new or modify existing instructions. Unless
4450/// documented otherwise, methods consuming CanonicalLoopInfo do not invalidate
4451/// any InsertPoint that is outside CanonicalLoopInfo's control. Specifically,
4452/// any InsertPoint in the Preheader, After or Block can still be used after
4453/// calling such a method.
4454///
4455/// TODO: Provide mechanisms for exception handling and cancellation points.
4456///
4457/// Defined outside OpenMPIRBuilder because nested classes cannot be
4458/// forward-declared, e.g. to avoid having to include the entire OMPIRBuilder.h.
4460 friend class OpenMPIRBuilder;
4461
4462private:
4463 BasicBlock *Header = nullptr;
4464 BasicBlock *Cond = nullptr;
4465 BasicBlock *Latch = nullptr;
4466 BasicBlock *Exit = nullptr;
4467
4468 // Hold the MLIR value for the `lastiter` of the canonical loop.
4469 Value *LastIter = nullptr;
4470
4471 /// Add the control blocks of this loop to \p BBs.
4472 ///
4473 /// This does not include any block from the body, including the one returned
4474 /// by getBody().
4475 ///
4476 /// FIXME: This currently includes the Preheader and After blocks even though
4477 /// their content is (mostly) not under CanonicalLoopInfo's control.
4478 /// Re-evaluated whether this makes sense.
4479 void collectControlBlocks(SmallVectorImpl<BasicBlock *> &BBs);
4480
4481 /// Sets the number of loop iterations to the given value. This value must be
4482 /// valid in the condition block (i.e., defined in the preheader) and is
4483 /// interpreted as an unsigned integer.
4484 void setTripCount(Value *TripCount);
4485
4486 /// Replace all uses of the canonical induction variable in the loop body with
4487 /// a new one.
4488 ///
4489 /// The intended use case is to update the induction variable for an updated
4490 /// iteration space such that it can stay normalized in the 0...tripcount-1
4491 /// range.
4492 ///
4493 /// The \p Updater is called with the (presumable updated) current normalized
4494 /// induction variable and is expected to return the value that uses of the
4495 /// pre-updated induction values should use instead, typically dependent on
4496 /// the new induction variable. This is a lambda (instead of e.g. just passing
4497 /// the new value) to be able to distinguish the uses of the pre-updated
4498 /// induction variable and uses of the induction varible to compute the
4499 /// updated induction variable value.
4500 void mapIndVar(llvm::function_ref<Value *(Instruction *)> Updater);
4501
4502public:
4503 /// Sets the last iteration variable for this loop.
4504 void setLastIter(Value *IterVar) { LastIter = std::move(IterVar); }
4505
4506 /// Returns the last iteration variable for this loop.
4507 /// Certain use-cases (like translation of linear clause) may access
4508 /// this variable even after a loop transformation. Hence, do not guard
4509 /// this getter function by `isValid`. It is the responsibility of the
4510 /// callee to ensure this functionality is not invoked by a non-outlined
4511 /// CanonicalLoopInfo object (in which case, `setLastIter` will never be
4512 /// invoked and `LastIter` will be by default `nullptr`).
4513 Value *getLastIter() { return LastIter; }
4514
4515 /// Returns whether this object currently represents the IR of a loop. If
4516 /// returning false, it may have been consumed by a loop transformation or not
4517 /// been initialized. Do not use in this case;
4518 bool isValid() const { return Header; }
4519
4520 /// The preheader ensures that there is only a single edge entering the loop.
4521 /// Code that must be execute before any loop iteration can be emitted here,
4522 /// such as computing the loop trip count and begin lifetime markers. Code in
4523 /// the preheader is not considered part of the canonical loop.
4525
4526 /// The header is the entry for each iteration. In the canonical control flow,
4527 /// it only contains the PHINode for the induction variable.
4529 assert(isValid() && "Requires a valid canonical loop");
4530 return Header;
4531 }
4532
4533 /// The condition block computes whether there is another loop iteration. If
4534 /// yes, branches to the body; otherwise to the exit block.
4536 assert(isValid() && "Requires a valid canonical loop");
4537 return Cond;
4538 }
4539
4540 /// The body block is the single entry for a loop iteration and not controlled
4541 /// by CanonicalLoopInfo. It can contain arbitrary control flow but must
4542 /// eventually branch to the \p Latch block.
4544 assert(isValid() && "Requires a valid canonical loop");
4545 return cast<CondBrInst>(Cond->getTerminator())->getSuccessor(0);
4546 }
4547
4548 /// Reaching the latch indicates the end of the loop body code. In the
4549 /// canonical control flow, it only contains the increment of the induction
4550 /// variable.
4552 assert(isValid() && "Requires a valid canonical loop");
4553 return Latch;
4554 }
4555
4556 /// Reaching the exit indicates no more iterations are being executed.
4558 assert(isValid() && "Requires a valid canonical loop");
4559 return Exit;
4560 }
4561
4562 /// The after block is intended for clean-up code such as lifetime end
4563 /// markers. It is separate from the exit block to ensure, analogous to the
4564 /// preheader, it having just a single entry edge and being free from PHI
4565 /// nodes should there be multiple loop exits (such as from break
4566 /// statements/cancellations).
4568 assert(isValid() && "Requires a valid canonical loop");
4569 return Exit->getSingleSuccessor();
4570 }
4571
4572 /// Returns the llvm::Value containing the number of loop iterations. It must
4573 /// be valid in the preheader and always interpreted as an unsigned integer of
4574 /// any bit-width.
4576 assert(isValid() && "Requires a valid canonical loop");
4577 Instruction *CmpI = &Cond->front();
4578 assert(isa<CmpInst>(CmpI) && "First inst must compare IV with TripCount");
4579 return CmpI->getOperand(1);
4580 }
4581
4582 /// Returns the instruction representing the current logical induction
4583 /// variable. Always unsigned, always starting at 0 with an increment of one.
4585 assert(isValid() && "Requires a valid canonical loop");
4586 Instruction *IndVarPHI = &Header->front();
4587 assert(isa<PHINode>(IndVarPHI) && "First inst must be the IV PHI");
4588 return IndVarPHI;
4589 }
4590
4591 /// Return the type of the induction variable (and the trip count).
4593 assert(isValid() && "Requires a valid canonical loop");
4594 return getIndVar()->getType();
4595 }
4596
4597 /// Return the insertion point for user code before the loop.
4599 assert(isValid() && "Requires a valid canonical loop");
4600 BasicBlock *Preheader = getPreheader();
4601 return {Preheader, std::prev(Preheader->end())};
4602 };
4603
4604 /// Return the insertion point for user code in the body.
4606 assert(isValid() && "Requires a valid canonical loop");
4607 BasicBlock *Body = getBody();
4608 return {Body, Body->begin()};
4609 };
4610
4611 /// Return the insertion point for user code after the loop.
4613 assert(isValid() && "Requires a valid canonical loop");
4614 BasicBlock *After = getAfter();
4615 return {After, After->begin()};
4616 };
4617
4619 assert(isValid() && "Requires a valid canonical loop");
4620 return Header->getParent();
4621 }
4622
4623 /// Consistency self-check.
4624 LLVM_ABI void assertOK() const;
4625
4626 /// Invalidate this loop. That is, the underlying IR does not fulfill the
4627 /// requirements of an OpenMP canonical loop anymore.
4628 LLVM_ABI void invalidate();
4629};
4630
4631/// ScanInfo holds the information to assist in lowering of Scan reduction.
4632/// Before lowering, the body of the for loop specifying scan reduction is
4633/// expected to have the following structure
4634///
4635/// Loop Body Entry
4636/// |
4637/// Code before the scan directive
4638/// |
4639/// Scan Directive
4640/// |
4641/// Code after the scan directive
4642/// |
4643/// Loop Body Exit
4644/// When `createCanonicalScanLoops` is executed, the bodyGen callback of it
4645/// transforms the body to:
4646///
4647/// Loop Body Entry
4648/// |
4649/// OMPScanDispatch
4650///
4651/// OMPBeforeScanBlock
4652/// |
4653/// OMPScanLoopExit
4654/// |
4655/// Loop Body Exit
4656///
4657/// The insert point is updated to the first insert point of OMPBeforeScanBlock.
4658/// It dominates the control flow of code generated until
4659/// scan directive is encountered and OMPAfterScanBlock dominates the
4660/// control flow of code generated after scan is encountered. The successor
4661/// of OMPScanDispatch can be OMPBeforeScanBlock or OMPAfterScanBlock based
4662/// on 1.whether it is in Input phase or Scan Phase , 2. whether it is an
4663/// exclusive or inclusive scan. This jump is added when `createScan` is
4664/// executed. If input loop is being generated, if it is inclusive scan,
4665/// `OMPAfterScanBlock` succeeds `OMPScanDispatch` , if exclusive,
4666/// `OMPBeforeScanBlock` succeeds `OMPDispatch` and vice versa for scan loop. At
4667/// the end of the input loop, temporary buffer is populated and at the
4668/// beginning of the scan loop, temporary buffer is read. After scan directive
4669/// is encountered, insertion point is updated to `OMPAfterScanBlock` as it is
4670/// expected to dominate the code after the scan directive. Both Before and
4671/// After scan blocks are succeeded by `OMPScanLoopExit`.
4672/// Temporary buffer allocations are done in `ScanLoopInit` block before the
4673/// lowering of for-loop. The results are copied back to reduction variable in
4674/// `ScanLoopFinish` block.
4676public:
4677 /// Dominates the body of the loop before scan directive
4679
4680 /// Dominates the body of the loop before scan directive
4682
4683 /// Controls the flow to before or after scan blocks
4685
4686 /// Exit block of loop body
4688
4689 /// Block before loop body where scan initializations are done
4691
4692 /// Block after loop body where scan finalizations are done
4694
4695 /// If true, it indicates Input phase is lowered; else it indicates
4696 /// ScanPhase is lowered
4697 bool OMPFirstScanLoop = false;
4698
4699 /// Maps the private reduction variable to the pointer of the temporary
4700 /// buffer
4702
4703 /// Keeps track of value of iteration variable for input/scan loop to be
4704 /// used for Scan directive lowering
4705 llvm::Value *IV = nullptr;
4706
4707 /// Stores the span of canonical loop being lowered to be used for temporary
4708 /// buffer allocation or Finalization.
4709 llvm::Value *Span = nullptr;
4710
4714 ScanInfo(ScanInfo &) = delete;
4715 ScanInfo &operator=(const ScanInfo &) = delete;
4716
4717 ~ScanInfo() { delete (ScanBuffPtrs); }
4718};
4719
4720} // end namespace llvm
4721
4722#endif // LLVM_FRONTEND_OPENMP_OMPIRBUILDER_H
assert(UImm &&(UImm !=~static_cast< T >(0)) &&"Invalid immediate!")
unsigned uint64_t
This file implements the APSInt class, which is a simple class that represents an arbitrary sized int...
arc branch finalize
MachineBasicBlock MachineBasicBlock::iterator DebugLoc DL
This file defines the BumpPtrAllocator interface.
#define X(NUM, ENUM, NAME)
Definition ELF.h:857
static GCRegistry::Add< ShadowStackGC > C("shadow-stack", "Very portable GC for uncooperative code generators")
static GCRegistry::Add< StatepointGC > D("statepoint-example", "an example strategy for statepoint")
static GCRegistry::Add< CoreCLRGC > E("coreclr", "CoreCLR-compatible GC")
#define LLVM_ABI
Definition Compiler.h:215
DXIL Finalize Linkage
Hexagon Hardware Loops
Module.h This file contains the declarations for the Module class.
#define F(x, y, z)
Definition MD5.cpp:54
#define I(x, y, z)
Definition MD5.cpp:57
#define G(x, y, z)
Definition MD5.cpp:55
Machine Check Debug Module
#define T
This file defines constans and helpers used when dealing with OpenMP.
Provides definitions for Target specific Grid Values.
const SmallVectorImpl< MachineOperand > & Cond
Basic Register Allocator
static cl::opt< RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode > Mode("regalloc-enable-advisor", cl::Hidden, cl::init(RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Default), cl::desc("Enable regalloc advisor mode"), cl::values(clEnumValN(RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Default, "default", "Default"), clEnumValN(RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Release, "release", "precompiled"), clEnumValN(RegAllocEvictionAdvisorAnalysisLegacy::AdvisorMode::Development, "development", "for training")))
Func getContext().diagnose(DiagnosticInfoUnsupported(Func
SmallPtrSet< BasicBlock *, 0 > BlockSet
This file implements a set that has insertion order iteration characteristics.
Contains the forward declaration for vfs::FileSystem, as well as the IntrusiveRefCntPtrInfo specializ...
Value * RHS
Value * LHS
The Input class is used to parse a yaml document into in-memory structs and vectors.
An arbitrary precision integer that knows its signedness.
Definition APSInt.h:24
an instruction to allocate memory on the stack
This class represents an incoming formal argument to a Function.
Definition Argument.h:32
Represent a constant reference to an array (0 or more elements consecutively in memory),...
Definition ArrayRef.h:40
Align AtomicAlign
Definition Atomic.h:23
bool UseLibcall
Definition Atomic.h:25
IRBuilderBase * Builder
Definition Atomic.h:19
uint64_t AtomicSizeInBits
Definition Atomic.h:21
uint64_t ValueSizeInBits
Definition Atomic.h:22
IRBuilderBase::InsertPoint AllocaIP
Definition Atomic.h:26
Align ValueAlign
Definition Atomic.h:24
BinOp
This enumeration lists the possible modifications atomicrmw can make.
LLVM Basic Block Representation.
Definition BasicBlock.h:62
iterator end()
Definition BasicBlock.h:459
iterator begin()
Instruction iterator methods.
Definition BasicBlock.h:446
This class represents a function call, abstracting a target machine's calling convention.
Class to represented the control flow structure of an OpenMP canonical loop.
Value * getTripCount() const
Returns the llvm::Value containing the number of loop iterations.
BasicBlock * getHeader() const
The header is the entry for each iteration.
LLVM_ABI void assertOK() const
Consistency self-check.
Type * getIndVarType() const
Return the type of the induction variable (and the trip count).
BasicBlock * getBody() const
The body block is the single entry for a loop iteration and not controlled by CanonicalLoopInfo.
bool isValid() const
Returns whether this object currently represents the IR of a loop.
void setLastIter(Value *IterVar)
Sets the last iteration variable for this loop.
OpenMPIRBuilder::InsertPointTy getAfterIP() const
Return the insertion point for user code after the loop.
Value * getLastIter()
Returns the last iteration variable for this loop.
OpenMPIRBuilder::InsertPointTy getBodyIP() const
Return the insertion point for user code in the body.
BasicBlock * getAfter() const
The after block is intended for clean-up code such as lifetime end markers.
Function * getFunction() const
LLVM_ABI void invalidate()
Invalidate this loop.
BasicBlock * getLatch() const
Reaching the latch indicates the end of the loop body code.
OpenMPIRBuilder::InsertPointTy getPreheaderIP() const
Return the insertion point for user code before the loop.
BasicBlock * getCond() const
The condition block computes whether there is another loop iteration.
BasicBlock * getExit() const
Reaching the exit indicates no more iterations are being executed.
LLVM_ABI BasicBlock * getPreheader() const
The preheader ensures that there is only a single edge entering the loop.
Instruction * getIndVar() const
Returns the instruction representing the current logical induction variable.
Utility class for extracting code into a new function.
This is the shared class of boolean and integer constants.
Definition Constants.h:87
This is an important base class in LLVM.
Definition Constant.h:43
A debug info location.
Definition DebugLoc.h:126
Lightweight error class with error context and mandatory checking.
Definition Error.h:159
Tagged union holding either a T or a Error.
Definition Error.h:485
A handy container for a FunctionType+Callee-pointer pair, which can be passed around as a single enti...
LinkageTypes
An enumeration for the kinds of linkage for global values.
Definition GlobalValue.h:52
InsertPoint - A saved insertion point.
Definition IRBuilder.h:246
Common base class shared among various IRBuilders.
Definition IRBuilder.h:114
This provides a uniform API for creating instructions and inserting them into a basic block: either a...
Definition IRBuilder.h:2908
Class to represent integer types.
Analysis pass that exposes the LoopInfo for a function.
Definition LoopInfo.h:594
Represents a single loop in the control flow graph.
Definition LoopInfo.h:40
This class implements a map that also provides access to all stored values in a deterministic order.
Definition MapVector.h:38
A Module instance is used to store all the information related to an LLVM module.
Definition Module.h:68
OffloadEntryInfoDeviceGlobalVar(unsigned Order, OMPTargetGlobalVarEntryKind Flags)
OffloadEntryInfoDeviceGlobalVar(unsigned Order, Constant *Addr, int64_t VarSize, OMPTargetGlobalVarEntryKind Flags, GlobalValue::LinkageTypes Linkage, const std::string &VarName)
static bool classof(const OffloadEntryInfo *Info)
OffloadEntryInfoTargetRegion(unsigned Order, Constant *Addr, Constant *ID, OMPTargetRegionEntryKind Flags)
@ OffloadingEntryInfoTargetRegion
Entry is a target region.
@ OffloadingEntryInfoDeviceGlobalVar
Entry is a declare target variable.
OffloadingEntryInfoKinds getKind() const
OffloadEntryInfo(OffloadingEntryInfoKinds Kind)
static bool classof(const OffloadEntryInfo *Info)
OffloadEntryInfo(OffloadingEntryInfoKinds Kind, unsigned Order, uint32_t Flags)
Class that manages information about offload code regions and data.
function_ref< void(StringRef, const OffloadEntryInfoDeviceGlobalVar &)> OffloadDeviceGlobalVarEntryInfoActTy
Applies action Action on all registered entries.
OMPTargetDeviceClauseKind
Kind of device clause for declare target variables and functions NOTE: Currently not used as a part o...
@ OMPTargetDeviceClauseNoHost
The target is marked for non-host devices.
@ OMPTargetDeviceClauseAny
The target is marked for all devices.
@ OMPTargetDeviceClauseNone
The target is marked as having no clause.
@ OMPTargetDeviceClauseHost
The target is marked for host devices.
LLVM_ABI void registerDeviceGlobalVarEntryInfo(StringRef VarName, Constant *Addr, int64_t VarSize, OMPTargetGlobalVarEntryKind Flags, GlobalValue::LinkageTypes Linkage)
Register device global variable entry.
LLVM_ABI void initializeDeviceGlobalVarEntryInfo(StringRef Name, OMPTargetGlobalVarEntryKind Flags, unsigned Order)
Initialize device global variable entry.
LLVM_ABI void actOnDeviceGlobalVarEntriesInfo(const OffloadDeviceGlobalVarEntryInfoActTy &Action)
OMPTargetRegionEntryKind
Kind of the target registry entry.
@ OMPTargetRegionEntryTargetRegion
Mark the entry as target region.
OffloadEntriesInfoManager(OpenMPIRBuilder *builder)
LLVM_ABI void getTargetRegionEntryFnName(SmallVectorImpl< char > &Name, const TargetRegionEntryInfo &EntryInfo)
LLVM_ABI bool hasTargetRegionEntryInfo(TargetRegionEntryInfo EntryInfo, bool IgnoreAddressId=false) const
Return true if a target region entry with the provided information exists.
LLVM_ABI void registerTargetRegionEntryInfo(TargetRegionEntryInfo EntryInfo, Constant *Addr, Constant *ID, OMPTargetRegionEntryKind Flags)
Register target region entry.
LLVM_ABI void actOnTargetRegionEntriesInfo(const OffloadTargetRegionEntryInfoActTy &Action)
unsigned size() const
Return number of entries defined so far.
LLVM_ABI void initializeTargetRegionEntryInfo(const TargetRegionEntryInfo &EntryInfo, unsigned Order)
Initialize target region entry.
OMPTargetGlobalVarEntryKind
Kind of the global variable entry..
@ OMPTargetGlobalVarEntryEnter
Mark the entry as a declare target enter.
@ OMPTargetGlobalVarEntryNone
Mark the entry as having no declare target entry kind.
@ OMPTargetGlobalRegisterRequires
Mark the entry as a register requires global.
@ OMPTargetGlobalVarEntryIndirect
Mark the entry as a declare target indirect global.
@ OMPTargetGlobalVarEntryLink
Mark the entry as a to declare target link.
@ OMPTargetGlobalVarEntryTo
Mark the entry as a to declare target.
@ OMPTargetGlobalVarEntryIndirectVTable
Mark the entry as a declare target indirect vtable.
function_ref< void(const TargetRegionEntryInfo &EntryInfo, const OffloadEntryInfoTargetRegion &)> OffloadTargetRegionEntryInfoActTy
brief Applies action Action on all registered entries.
bool hasDeviceGlobalVarEntryInfo(StringRef VarName) const
Checks if the variable with the given name has been registered already.
LLVM_ABI bool empty() const
Return true if a there are no entries defined.
Captures attributes that affect generating LLVM-IR using the OpenMPIRBuilder and related classes.
std::optional< bool > NoSignedWrap
Flag for specifying whether the no-signed-wrap (nsw) flag should be added to loop induction variable ...
std::optional< bool > IsTargetDevice
Flag to define whether to generate code for the role of the OpenMP host (if set to false) or device (...
std::optional< bool > IsGPU
Flag for specifying if the compilation is done for an accelerator.
std::optional< StringRef > FirstSeparator
First separator used between the initial two parts of a name.
StringRef separator() const
LLVM_ABI int64_t getRequiresFlags() const
Returns requires directive clauses as flags compatible with those expected by libomptarget.
void setFirstSeparator(StringRef FS)
void setDefaultTargetAS(unsigned AS)
StringRef firstSeparator() const
std::optional< bool > OpenMPOffloadMandatory
Flag for specifying if offloading is mandatory.
std::optional< bool > EmitLLVMUsedMetaInfo
Flag for specifying if LLVMUsed information should be emitted.
SmallVector< Triple > TargetTriples
When compilation is being done for the OpenMP host (i.e.
LLVM_ABI void setHasRequiresReverseOffload(bool Value)
void setNoSignedWrap(bool Value)
LLVM_ABI bool hasRequiresUnifiedSharedMemory() const
LLVM_ABI void setHasRequiresUnifiedSharedMemory(bool Value)
unsigned getDefaultTargetAS() const
std::optional< StringRef > Separator
Separator used between all of the rest consecutive parts of s name.
LLVM_ABI bool hasRequiresDynamicAllocators() const
bool openMPOffloadMandatory() const
CallingConv::ID getRuntimeCC() const
LLVM_ABI void setHasRequiresUnifiedAddress(bool Value)
void setOpenMPOffloadMandatory(bool Value)
void setIsTargetDevice(bool Value)
void setSeparator(StringRef S)
void setRuntimeCC(CallingConv::ID CC)
LLVM_ABI void setHasRequiresDynamicAllocators(bool Value)
void setEmitLLVMUsed(bool Value=true)
std::optional< omp::GV > GridValue
LLVM_ABI bool hasRequiresReverseOffload() const
LLVM_ABI bool hasRequiresUnifiedAddress() const
llvm::AllocaInst * CreateAlloca(llvm::Type *Ty, const llvm::Twine &Name) const override
void decorateWithTBAA(llvm::Instruction *I) override
AtomicInfo(IRBuilder<> *Builder, llvm::Type *Ty, uint64_t AtomicSizeInBits, uint64_t ValueSizeInBits, llvm::Align AtomicAlign, llvm::Align ValueAlign, bool UseLibcall, IRBuilderBase::InsertPoint AllocaIP, llvm::Value *AtomicVar)
llvm::Value * getAtomicPointer() const override
Struct that keeps the information that should be kept throughout a 'target data' region.
TargetDataInfo(bool RequiresDevicePointerInfo, bool SeparateBeginEndCalls)
SmallMapVector< const Value *, std::pair< Value *, Value * >, 4 > DevicePtrInfoMap
void clearArrayInfo()
Clear information about the data arrays.
unsigned NumberOfPtrs
The total number of pointers passed to the runtime library.
bool HasNoWait
Whether the target ... data directive has a nowait clause.
bool isValid()
Return true if the current target data information has valid arrays.
bool HasMapper
Indicate whether any user-defined mapper exists.
An interface to create LLVM-IR for OpenMP directives.
LLVM_ABI InsertPointOrErrorTy createOrderedThreadsSimd(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, FinalizeCallbackTy FiniCB, bool IsThreads)
Generator for 'omp ordered [threads | simd]'.
LLVM_ABI void emitAArch64DeclareSimdFunction(llvm::Function *Fn, unsigned VLENVal, llvm::ArrayRef< DeclareSimdAttrTy > ParamAttrs, DeclareSimdBranch Branch, char ISA, unsigned NarrowestDataSize, bool OutputBecomesInput)
Emit AArch64 vector-function ABI attributes for a declare simd function.
LLVM_ABI Constant * getOrCreateIdent(Constant *SrcLocStr, uint32_t SrcLocStrSize, omp::IdentFlag Flags=omp::IdentFlag(0), unsigned Reserve2Flags=0)
Return an ident_t* encoding the source location SrcLocStr and Flags.
LLVM_ABI void registerDeclareTargetGlobalReplacement(GlobalValue *Original, GlobalValue *Replacement)
Register a module-scope replacement of a declare target global variable.
LLVM_ABI FunctionCallee getOrCreateRuntimeFunction(Module &M, omp::RuntimeFunction FnID)
Return the function declaration for the runtime function with FnID.
LLVM_ABI InsertPointOrErrorTy createCancel(const LocationDescription &Loc, Value *IfCondition, omp::Directive CanceledDirective)
Generator for 'omp cancel'.
std::function< Expected< Function * >(StringRef FunctionName)> FunctionGenCallback
Functions used to generate a function with the given name.
LLVM_ABI CallInst * createOMPAllocShared(const LocationDescription &Loc, Value *Size, const Twine &Name=Twine(""))
Create a runtime call for kmpc_alloc_shared.
ReductionGenCBKind
Enum class for the RedctionGen CallBack type to be used.
LLVM_ABI CanonicalLoopInfo * collapseLoops(DebugLoc DL, ArrayRef< CanonicalLoopInfo * > Loops, InsertPointTy ComputeIP)
Collapse a loop nest into a single loop.
LLVM_ABI void createTaskyield(const LocationDescription &Loc)
Generator for 'omp taskyield'.
std::function< Error(InsertPointTy CodeGenIP)> FinalizeCallbackTy
Callback type for variable finalization (think destructors).
LLVM_ABI void emitBranch(BasicBlock *Target)
LLVM_ABI Error emitCancelationCheckImpl(Value *CancelFlag, omp::Directive CanceledDirective)
Generate control flow and cleanup for cancellation.
static LLVM_ABI void writeThreadBoundsForKernel(const Triple &T, Function &Kernel, int32_t LB, int32_t UB)
EvalKind
Enum class for reduction evaluation types scalar, complex and aggregate.
LLVM_ABI void emitTaskwaitImpl(const LocationDescription &Loc)
Generate a taskwait runtime call.
LLVM_ABI Constant * registerTargetRegionFunction(TargetRegionEntryInfo &EntryInfo, Function *OutlinedFunction, StringRef EntryFnName, StringRef EntryFnIDName)
Registers the given function and sets up the attribtues of the function Returns the FunctionID.
LLVM_ABI GlobalVariable * emitKernelExecutionMode(StringRef KernelName, omp::OMPTgtExecModeFlags Mode)
Emit the kernel execution mode.
LLVM_ABI void initialize()
Initialize the internal state, this will put structures types and potentially other helpers into the ...
LLVM_ABI InsertPointTy createAtomicCompare(const LocationDescription &Loc, AtomicOpValue &X, AtomicOpValue &V, AtomicOpValue &R, Value *E, Value *D, AtomicOrdering AO, omp::OMPAtomicCompareOp Op, bool IsXBinopExpr, bool IsPostfixUpdate, bool IsFailOnly, bool IsWeak=false)
std::function< InsertPointTy(InsertPointTy CodeGenIP, unsigned Index, Value **LHS, Value **RHS, Function *CurFn)> ReductionGenClangCBTy
ReductionGen CallBack for Clang.
LLVM_ABI InsertPointTy createAtomicWrite(const LocationDescription &Loc, AtomicOpValue &X, Value *Expr, AtomicOrdering AO, InsertPointTy AllocaIP)
Emit atomic write for : X = Expr — Only Scalar data types.
LLVM_ABI void loadOffloadInfoMetadata(Module &M)
Loads all the offload entries information from the host IR metadata.
function_ref< MapInfosTy &(InsertPointTy CodeGenIP)> GenMapInfoCallbackTy
Callback type for creating the map infos for the kernel parameters.
LLVM_ABI Error emitOffloadingArrays(InsertPointTy AllocaIP, InsertPointTy CodeGenIP, MapInfosTy &CombinedInfo, TargetDataInfo &Info, CustomMapperCallbackTy CustomMapperCB, bool IsNonContiguous=false, function_ref< void(unsigned int, Value *)> DeviceAddrCB=nullptr)
Emit the arrays used to pass the captures and map information to the offloading runtime library.
LLVM_ABI void unrollLoopFull(DebugLoc DL, CanonicalLoopInfo *Loop)
Fully unroll a loop.
function_ref< Error(InsertPointTy CodeGenIP, Value *IndVar)> LoopBodyGenCallbackTy
Callback type for loop body code generation.
LLVM_ABI InsertPointOrErrorTy emitScanReduction(const LocationDescription &Loc, ArrayRef< llvm::OpenMPIRBuilder::ReductionInfo > ReductionInfos, ScanInfo *ScanRedInfo)
This function performs the scan reduction of the values updated in the input phase.
LLVM_ABI void emitFlush(const LocationDescription &Loc)
Generate a flush runtime call.
LLVM_ABI InsertPointOrErrorTy createScope(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, FinalizeCallbackTy FiniCB, bool IsNowait)
Generator for 'omp scope'.
static LLVM_ABI std::pair< int32_t, int32_t > readThreadBoundsForKernel(const Triple &T, Function &Kernel)
}
origPtr *with the address space normalization required by the runtime entry point *The NULL descriptor makes the runtime walk the enclosing taskgroups to *find the matching task_reduction registration for the item The lookups *are emitted at p which must be inside the target task body **The front end specific work(matching each `in_reduction` item to its *mapped storage and binding the generated private pointer back to the *right value) stays with the caller InsertPointT getInsertionPoint)()
Return the insertion point used by the underlying IRBuilder.
SmallVector< bool, 4 > MapHasAttachPtrArrayTy
OpenMPIRBuilderConfig Config
The OpenMPIRBuilder Configuration.
LLVM_ABI CallInst * createOMPInteropDestroy(const LocationDescription &Loc, Value *InteropVar, Value *Device, Value *NumDependences, Value *DependenceAddress, bool HaveNowaitClause)
Create a runtime call for __tgt_interop_destroy.
std::function< InsertPointOrErrorTy( InsertPointTy CodeGenIP, Value *LHS, Value *RHS, Value *&Res)> ReductionGenCBTy
ReductionGen CallBack for MLIR.
LLVM_ABI void emitUsed(StringRef Name, ArrayRef< llvm::WeakTrackingVH > List)
Emit the llvm.used metadata.
void setConfig(OpenMPIRBuilderConfig C)
LLVM_ABI InsertPointOrErrorTy createSingle(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, FinalizeCallbackTy FiniCB, bool IsNowait, ArrayRef< llvm::Value * > CPVars={}, ArrayRef< llvm::Function * > CPFuncs={})
Generator for 'omp single'.
LLVM_ABI InsertPointOrErrorTy createTeams(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, Value *NumTeamsLower=nullptr, Value *NumTeamsUpper=nullptr, Value *ThreadLimit=nullptr, Value *IfExpr=nullptr)
Generator for #omp teams
std::forward_list< CanonicalLoopInfo > LoopInfos
Collection of owned canonical loop objects that eventually need to be free'd.
bool setHandleFPNegZero(bool FPNegZero)
Set whether atomic compare should handle -0.0/+0.0 equivalence.
LLVM_ABI llvm::StructType * getKmpTaskAffinityInfoTy()
Return the LLVM struct type matching runtime kmp_task_affinity_info_t.
SmallVector< uint64_t, 4 > MapDimArrayTy
std::function< Error(InsertPointTy AllocaIP, InsertPointTy CodeGenIP, ArrayRef< BasicBlock * > DeallocBlocks)> StorableBodyGenCallbackTy
LLVM_ABI std::string createPlatformSpecificName(ArrayRef< StringRef > Parts) const
Get the create a name using the platform specific separators.
LLVM_ABI FunctionCallee createDispatchNextFunction(unsigned IVSize, bool IVSigned)
Returns __kmpc_dispatch_next_* runtime function for the specified size IVSize and sign IVSigned.
static LLVM_ABI void getKernelArgsVector(TargetKernelArgs &KernelArgs, IRBuilderBase &Builder, SmallVector< Value * > &ArgsVector)
Create the kernel args vector used by emitTargetKernel.
SmallVector< Constant *, 4 > MapNamesArrayTy
LLVM_ABI void unrollLoopHeuristic(DebugLoc DL, CanonicalLoopInfo *Loop)
Fully or partially unroll a loop.
LLVM_ABI omp::OpenMPOffloadMappingFlags getMemberOfFlag(unsigned Position)
Get OMP_MAP_MEMBER_OF flag with extra bits reserved based on the position given.
LLVM_ABI void addAttributes(omp::RuntimeFunction FnID, Function &Fn)
Add attributes known for FnID to Fn.
Module & M
The underlying LLVM-IR module.
StringMap< Constant * > SrcLocStrMap
Map to remember source location strings.
LLVM_ABI void createMapperAllocas(const LocationDescription &Loc, InsertPointTy AllocaIP, unsigned NumOperands, struct MapperAllocas &MapperAllocas)
Create the allocas instruction used in call to mapper functions.
SmallVector< DeclareTargetGlobalReplacement, 8 > DeclareTargetGlobalReplacements
Collection of declare target globals to rewrite uses of during device module finalizaiton.
LLVM_ABI Constant * getOrCreateSrcLocStr(StringRef LocStr, uint32_t &SrcLocStrSize)
Return the (LLVM-IR) string describing the source location LocStr.
LLVM_ABI Error emitTargetRegionFunction(TargetRegionEntryInfo &EntryInfo, FunctionGenCallback &GenerateFunctionCallback, bool IsOffloadEntry, Function *&OutlinedFn, Constant *&OutlinedFnID)
Create a unique name for the entry function using the source location information of the current targ...
LLVM_ABI InsertPointOrErrorTy createIteratorLoop(LocationDescription Loc, llvm::Value *TripCount, IteratorBodyGenTy BodyGen, llvm::StringRef Name="iterator")
Create a canonical iterator loop at the current insertion point.
LLVM_ABI Expected< SmallVector< llvm::CanonicalLoopInfo * > > createCanonicalScanLoops(const LocationDescription &Loc, LoopBodyGenCallbackTy BodyGenCB, Value *Start, Value *Stop, Value *Step, bool IsSigned, bool InclusiveStop, InsertPointTy ComputeIP, const Twine &Name, ScanInfo *ScanRedInfo)
Generator for the control flow structure of an OpenMP canonical loops if the parent directive has an ...
LLVM_ABI FunctionCallee createDispatchFiniFunction(unsigned IVSize, bool IVSigned)
Returns __kmpc_dispatch_fini_* runtime function for the specified size IVSize and sign IVSigned.
function_ref< InsertPointOrErrorTy( InsertPointTy AllocaIP, InsertPointTy CodeGenIP, ArrayRef< BasicBlock * > DeallocBlocks)> TargetBodyGenCallbackTy
LLVM_ABI void unrollLoopPartial(DebugLoc DL, CanonicalLoopInfo *Loop, int32_t Factor, CanonicalLoopInfo **UnrolledCLI)
Partially unroll a loop.
function_ref< Error(Value *DeviceID, Value *RTLoc, IRBuilderBase::InsertPoint TargetTaskAllocaIP)> TargetTaskBodyCallbackTy
Callback type for generating the bodies of device directives that require outer target tasks (e....
Expected< MapInfosTy & > MapInfosOrErrorTy
bool HandleFPNegZero
Emit atomic compare for constructs: — Only scalar data types cond-expr-stmt: x = x ordop expr ?
SmallVector< omp::OpenMPOffloadMappingFlags, 4 > MapFlagsArrayTy
LLVM_ABI void emitTaskyieldImpl(const LocationDescription &Loc)
Generate a taskyield runtime call.
LLVM_ABI void emitMapperCall(const LocationDescription &Loc, Function *MapperFunc, Value *SrcLocInfo, Value *MaptypesArg, Value *MapnamesArg, struct MapperAllocas &MapperAllocas, int64_t DeviceID, unsigned NumOperands)
Create the call for the target mapper function.
LLVM_ABI InsertPointOrErrorTy createDistribute(const LocationDescription &Loc, InsertPointTy AllocaIP, ArrayRef< BasicBlock * > DeallocBlocks, BodyGenCallbackTy BodyGenCB)
Generator for #omp distribute
function_ref< Expected< Function * >(unsigned int)> CustomMapperCallbackTy
LLVM_ABI InsertPointTy createOrderedDepend(const LocationDescription &Loc, InsertPointTy AllocaIP, unsigned NumLoops, ArrayRef< llvm::Value * > StoreValues, const Twine &Name, bool IsDependSource)
Generator for 'omp ordered depend (source | sink)'.
LLVM_ABI InsertPointTy createCopyinClauseBlocks(InsertPointTy IP, Value *MasterAddr, Value *PrivateAddr, llvm::IntegerType *IntPtrTy, bool BranchtoEnd=true)
Generate conditional branch and relevant BasicBlocks through which private threads copy the 'copyin' ...
SmallVector< MapValuesArrayTy, 4 > MapNonContiguousArrayTy
function_ref< InsertPointOrErrorTy( InsertPointTy AllocaIP, InsertPointTy CodeGenIP, Value &Original, Value &Inner, Value *&ReplVal)> PrivatizeCallbackTy
Callback type for variable privatization (think copy & default constructor).
LLVM_ABI bool isFinalized()
Check whether the finalize function has already run.
SmallVector< DeviceInfoTy, 4 > MapDeviceInfoArrayTy
SmallVector< FinalizationInfo, 8 > FinalizationStack
The finalization stack made up of finalize callbacks currently in-flight, wrapped into FinalizationIn...
LLVM_ABI std::vector< CanonicalLoopInfo * > tileLoops(DebugLoc DL, ArrayRef< CanonicalLoopInfo * > Loops, ArrayRef< Value * > TileSizes)
Tile a loop nest.
LLVM_ABI CallInst * createOMPInteropInit(const LocationDescription &Loc, Value *InteropVar, omp::OMPInteropType InteropType, Value *Device, Value *NumDependences, Value *DependenceAddress, bool HaveNowaitClause)
Create a runtime call for __tgt_interop_init.
LLVM_ABI Error emitIfClause(Value *Cond, BodyGenCallbackTy ThenGen, BodyGenCallbackTy ElseGen, InsertPointTy AllocaIP={}, ArrayRef< BasicBlock * > DeallocBlocks={})
Emits code for OpenMP 'if' clause using specified BodyGenCallbackTy Here is the logic: if (Cond) { Th...
LLVM_ABI Function * getOrCreateRuntimeFunctionPtr(omp::RuntimeFunction FnID)
std::function< InsertPointOrErrorTy( InsertPointTy, Value *ByRefVal, Value *&Res)> ReductionGenDataPtrPtrCBTy
void addOutlineInfo(std::unique_ptr< OutlineInfo > &&OI)
Add a new region that will be outlined later.
LLVM_ABI InsertPointTy createTargetInit(const LocationDescription &Loc, const llvm::OpenMPIRBuilder::TargetKernelDefaultAttrs &Attrs)
The omp target interface.
LLVM_ABI InsertPointOrErrorTy createReductions(const LocationDescription &Loc, InsertPointTy AllocaIP, ArrayRef< ReductionInfo > ReductionInfos, ArrayRef< bool > IsByRef, bool IsNoWait=false, bool IsTeamsReduction=false)
Generator for 'omp reduction'.
const Triple T
The target triple of the underlying module.
DenseMap< std::pair< Constant *, uint64_t >, Constant * > IdentMap
Map to remember existing ident_t*.
LLVM_ABI CallInst * createOMPFree(const LocationDescription &Loc, Value *Addr, Value *Allocator, std::string Name="")
Create a runtime call for kmpc_free.
LLVM_ABI InsertPointOrErrorTy createReductionsGPU(const LocationDescription &Loc, InsertPointTy AllocaIP, InsertPointTy CodeGenIP, ArrayRef< ReductionInfo > ReductionInfos, ArrayRef< bool > IsByRef, bool IsNoWait=false, bool IsTeamsReduction=false, bool IsSPMD=false, ReductionGenCBKind ReductionGenCBKind=ReductionGenCBKind::MLIR, std::optional< omp::GV > GridValue={}, Value *SrcLocInfo=nullptr)
Design of OpenMP reductions on the GPU.
LLVM_ABI FunctionCallee createForStaticInitFunction(unsigned IVSize, bool IVSigned, bool IsGPUDistribute)
Returns __kmpc_for_static_init_* runtime function for the specified size IVSize and sign IVSigned.
LLVM_ABI CallInst * createOMPAlloc(const LocationDescription &Loc, Value *Size, Value *Allocator, std::string Name="")
Create a runtime call for kmpc_alloc.
LLVM_ABI void emitNonContiguousDescriptor(InsertPointTy AllocaIP, InsertPointTy CodeGenIP, MapInfosTy &CombinedInfo, TargetDataInfo &Info)
Emit an array of struct descriptors to be assigned to the offload args.
SmallVector< Value *, 4 > MapValuesArrayTy
LLVM_ABI InsertPointOrErrorTy createSection(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, FinalizeCallbackTy FiniCB)
Generator for 'omp section'.
LLVM_ABI InsertPointOrErrorTy createTaskgroup(const LocationDescription &Loc, InsertPointTy AllocaIP, ArrayRef< BasicBlock * > DeallocBlocks, BodyGenCallbackTy BodyGenCB)
Generator for the taskgroup construct.
LLVM_ABI InsertPointOrErrorTy createParallel(const LocationDescription &Loc, InsertPointTy AllocaIP, ArrayRef< BasicBlock * > DeallocBlocks, BodyGenCallbackTy BodyGenCB, PrivatizeCallbackTy PrivCB, FinalizeCallbackTy FiniCB, Value *IfCondition, Value *NumThreads, omp::ProcBindKind ProcBind, bool IsCancellable)
Generator for 'omp parallel'.
function_ref< InsertPointOrErrorTy(InsertPointTy)> EmitFallbackCallbackTy
Callback function type for functions emitting the host fallback code that is executed when the kernel...
static LLVM_ABI TargetRegionEntryInfo getTargetEntryUniqueInfo(FileIdentifierInfoCallbackTy CallBack, vfs::FileSystem &VFS, StringRef ParentName="")
Creates a unique info for a target entry when provided a filename and line number from.
LLVM_ABI void emitTaskDependency(IRBuilderBase &Builder, Value *Entry, const DependData &Dep)
Store one kmp_depend_info entry at the given Entry pointer.
LLVM_ABI void emitBlock(BasicBlock *BB, Function *CurFn, bool IsFinished=false)
LLVM_ABI Value * getOrCreateThreadID(Value *Ident)
Return the current thread ID.
LLVM_ABI InsertPointOrErrorTy createMaster(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, FinalizeCallbackTy FiniCB)
Generator for 'omp master'.
void pushFinalizationCB(const FinalizationInfo &FI)
Push a finalization callback on the finalization stack.
LLVM_ABI InsertPointOrErrorTy createTarget(const LocationDescription &Loc, bool IsOffloadEntry, OpenMPIRBuilder::InsertPointTy AllocaIP, OpenMPIRBuilder::InsertPointTy CodeGenIP, ArrayRef< BasicBlock * > DeallocBlocks, TargetDataInfo &Info, TargetRegionEntryInfo &EntryInfo, const TargetKernelDefaultAttrs &DefaultAttrs, const TargetKernelRuntimeAttrs &RuntimeAttrs, Value *IfCond, SmallVectorImpl< Value * > &Inputs, GenMapInfoCallbackTy GenMapInfoCB, TargetBodyGenCallbackTy BodyGenCB, TargetGenArgAccessorsCallbackTy ArgAccessorFuncCB, CustomMapperCallbackTy CustomMapperCB, const DependenciesInfo &Dependencies={}, bool HasNowait=false, Value *DynCGroupMem=nullptr, omp::OMPDynGroupprivateFallbackType DynCGroupMemFallback=omp::OMPDynGroupprivateFallbackType::Abort, DebugLoc OutlinedFnLoc={})
Generator for 'omp target'.
LLVM_ABI InsertPointOrErrorTy createTargetData(const LocationDescription &Loc, InsertPointTy AllocaIP, InsertPointTy CodeGenIP, ArrayRef< BasicBlock * > DeallocBlocks, Value *DeviceID, Value *IfCond, TargetDataInfo &Info, GenMapInfoCallbackTy GenMapInfoCB, CustomMapperCallbackTy CustomMapperCB, omp::RuntimeFunction *MapperFunc=nullptr, function_ref< InsertPointOrErrorTy(InsertPointTy CodeGenIP, BodyGenTy BodyGenType)> BodyGenCB=nullptr, function_ref< void(unsigned int, Value *)> DeviceAddrCB=nullptr, Value *SrcLocInfo=nullptr)
Generator for 'omp target data'.
LLVM_ABI CallInst * createRuntimeFunctionCall(FunctionCallee Callee, ArrayRef< Value * > Args, StringRef Name="")
LLVM_ABI InsertPointOrErrorTy emitKernelLaunch(const LocationDescription &Loc, Value *OutlinedFnID, EmitFallbackCallbackTy EmitTargetCallFallbackCB, TargetKernelArgs &Args, Value *DeviceID, Value *RTLoc, InsertPointTy AllocaIP)
Generate a target region entry call and host fallback call.
StringMap< GlobalVariable *, BumpPtrAllocator > InternalVars
An ordered map of auto-generated variables to their unique names.
LLVM_ABI InsertPointOrErrorTy createCancellationPoint(const LocationDescription &Loc, omp::Directive CanceledDirective)
Generator for 'omp cancellation point'.
LLVM_ABI CallInst * createOMPAlignedAlloc(const LocationDescription &Loc, Value *Align, Value *Size, Value *Allocator, std::string Name="")
Create a runtime call for kmpc_align_alloc.
LLVM_ABI FunctionCallee createDispatchInitFunction(unsigned IVSize, bool IVSigned)
Returns __kmpc_dispatch_init_* runtime function for the specified size IVSize and sign IVSigned.
LLVM_ABI InsertPointOrErrorTy createScan(const LocationDescription &Loc, InsertPointTy AllocaIP, ArrayRef< llvm::Value * > ScanVars, ArrayRef< llvm::Type * > ScanVarsType, bool IsInclusive, ScanInfo *ScanRedInfo)
This directive split and directs the control flow to input phase blocks or scan phase blocks based on...
LLVM_ABI CallInst * createOMPFreeShared(const LocationDescription &Loc, Value *Addr, Value *Size, const Twine &Name=Twine(""))
Create a runtime call for kmpc_free_shared.
LLVM_ABI CallInst * createOMPInteropUse(const LocationDescription &Loc, Value *InteropVar, Value *Device, Value *NumDependences, Value *DependenceAddress, bool HaveNowaitClause)
Create a runtime call for __tgt_interop_use.
IRBuilder<>::InsertPoint InsertPointTy
Type used throughout for insertion points.
LLVM_ABI GlobalVariable * getOrCreateInternalVariable(Type *Ty, const StringRef &Name, std::optional< unsigned > AddressSpace={})
Gets (if variable with the given name already exist) or creates internal global variable with the spe...
LLVM_ABI GlobalVariable * createOffloadMapnames(SmallVectorImpl< llvm::Constant * > &Names, std::string VarName)
Create the global variable holding the offload names information.
LLVM_ABI InsertPointOrErrorTy createTask(const LocationDescription &Loc, InsertPointTy AllocaIP, ArrayRef< BasicBlock * > DeallocBlocks, BodyGenCallbackTy BodyGenCB, bool Tied=true, Value *Final=nullptr, Value *IfCondition=nullptr, const DependenciesInfo &Dependencies={}, const AffinityData &Affinities={}, bool Mergeable=false, Value *EventHandle=nullptr, Value *Priority=nullptr, bool FreeAgent=false)
Generator for #omp taskloop
std::forward_list< ScanInfo > ScanInfos
Collection of owned ScanInfo objects that eventually need to be free'd.
static LLVM_ABI void writeTeamsForKernel(const Triple &T, Function &Kernel, int32_t LB, int32_t UB)
std::function< InsertPointOrErrorTy( InsertPointTy, Type *, Value *, Value *)> ReductionGenAtomicCBTy
Functions used to generate atomic reductions.
LLVM_ABI Value * calculateCanonicalLoopTripCount(const LocationDescription &Loc, Value *Start, Value *Stop, Value *Step, bool IsSigned, bool InclusiveStop, const Twine &Name="loop")
Calculate the trip count of a canonical loop.
DeclareSimdKindTy
Kind of parameter in a function with 'declare simd' directive.
LLVM_ABI InsertPointOrErrorTy createBarrier(const LocationDescription &Loc, omp::Directive Kind, bool ForceSimpleCall=false, bool CheckCancelFlag=true)
Emitter methods for OpenMP directives.
LLVM_ABI void setCorrectMemberOfFlag(omp::OpenMPOffloadMappingFlags &Flags, omp::OpenMPOffloadMappingFlags MemberOfFlag)
Given an initial flag set, this function modifies it to contain the passed in MemberOfFlag generated ...
LLVM_ABI Error emitOffloadingArraysAndArgs(InsertPointTy AllocaIP, InsertPointTy CodeGenIP, TargetDataInfo &Info, TargetDataRTArgs &RTArgs, MapInfosTy &CombinedInfo, CustomMapperCallbackTy CustomMapperCB, bool IsNonContiguous=false, bool ForEndCall=false, function_ref< void(unsigned int, Value *)> DeviceAddrCB=nullptr)
Allocates memory for and populates the arrays required for offloading (offload_{baseptrs|ptrs|mappers...
LLVM_ABI Constant * getOrCreateDefaultSrcLocStr(uint32_t &SrcLocStrSize)
Return the (LLVM-IR) string describing the default source location.
LLVM_ABI InsertPointOrErrorTy createCritical(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, FinalizeCallbackTy FiniCB, StringRef CriticalName, Value *HintInst)
Generator for 'omp critical'.
LLVM_ABI void createError(const LocationDescription &Loc, bool IsFatal, Value *Message)
Generate a call to the runtime to emit the diagnostic of an OpenMP error directive with at(execution)...
LLVM_ABI void createOffloadEntry(Constant *ID, Constant *Addr, uint64_t Size, int32_t Flags, GlobalValue::LinkageTypes, StringRef Name="")
Creates offloading entry for the provided entry ID ID, address Addr, size Size, and flags Flags.
static LLVM_ABI unsigned getOpenMPDefaultSimdAlign(const Triple &TargetTriple, const StringMap< bool > &Features)
Get the default alignment value for given target.
LLVM_ABI unsigned getFlagMemberOffset()
Get the offset of the OMP_MAP_MEMBER_OF field.
LLVM_ABI InsertPointOrErrorTy applyWorkshareLoop(DebugLoc DL, CanonicalLoopInfo *CLI, InsertPointTy AllocaIP, bool NeedsBarrier, llvm::omp::ScheduleKind SchedKind=llvm::omp::OMP_SCHEDULE_Default, Value *ChunkSize=nullptr, bool HasSimdModifier=false, bool HasMonotonicModifier=false, bool HasNonmonotonicModifier=false, bool HasOrderedClause=false, omp::WorksharingLoopType LoopType=omp::WorksharingLoopType::ForStaticLoop, bool NoLoop=false, bool HasDistSchedule=false, Value *DistScheduleChunkSize=nullptr)
Modifies the canonical loop to be a workshare loop.
LLVM_ABI InsertPointOrErrorTy createAtomicCapture(const LocationDescription &Loc, InsertPointTy AllocaIP, AtomicOpValue &X, AtomicOpValue &V, Value *Expr, AtomicOrdering AO, AtomicRMWInst::BinOp RMWOp, AtomicUpdateCallbackTy &UpdateOp, bool UpdateExpr, bool IsPostfixUpdate, bool IsXBinopExpr, bool IsIgnoreDenormalMode=false, bool IsFineGrainedMemory=false, bool IsRemoteMemory=false)
Emit atomic update for constructs: — Only Scalar data types V = X; X = X BinOp Expr ,...
LLVM_ABI CanonicalLoopInfo * createLoopSkeleton(DebugLoc DL, Value *TripCount, Function *F, BasicBlock *PreInsertBefore, BasicBlock *PostInsertBefore, const Twine &Name={}, bool IsCollapsed=false)
Create the control flow structure of a canonical OpenMP loop.
LLVM_ABI void createOffloadEntriesAndInfoMetadata(EmitMetadataErrorReportFunctionTy &ErrorReportFunction)
LLVM_ABI void applySimd(CanonicalLoopInfo *Loop, MapVector< Value *, Value * > AlignedVars, Value *IfCond, omp::OrderKind Order, ConstantInt *Simdlen, ConstantInt *Safelen)
Add metadata to simd-ize a loop.
SmallVector< std::unique_ptr< OutlineInfo >, 16 > OutlineInfos
Collection of regions that need to be outlined during finalization.
LLVM_ABI InsertPointOrErrorTy createAtomicUpdate(const LocationDescription &Loc, InsertPointTy AllocaIP, AtomicOpValue &X, Value *Expr, AtomicOrdering AO, AtomicRMWInst::BinOp RMWOp, AtomicUpdateCallbackTy &UpdateOp, bool IsXBinopExpr, bool IsIgnoreDenormalMode=false, bool IsFineGrainedMemory=false, bool IsRemoteMemory=false)
Emit atomic update for constructs: X = X BinOp Expr ,or X = Expr BinOp X For complex Operations: X = ...
std::function< std::tuple< std::string, uint64_t >()> FileIdentifierInfoCallbackTy
bool isLastFinalizationInfoCancellable(omp::Directive DK)
Return true if the last entry in the finalization stack is of kind DK and cancellable.
LLVM_ABI InsertPointTy emitTargetKernel(const LocationDescription &Loc, InsertPointTy AllocaIP, Value *&Return, Value *Ident, Value *DeviceID, Value *NumTeams, Value *NumThreads, Value *HostPtr, ArrayRef< Value * > KernelArgs)
Generate a target region entry call.
LLVM_ABI GlobalVariable * createOffloadMaptypes(SmallVectorImpl< uint64_t > &Mappings, std::string VarName)
Create the global variable holding the offload mappings information.
LLVM_ABI Expected< Function * > emitUserDefinedMapper(function_ref< MapInfosOrErrorTy(InsertPointTy CodeGenIP, llvm::Value *PtrPHI, llvm::Value *BeginArg)> PrivAndGenMapInfoCB, llvm::Type *ElemTy, StringRef FuncName, CustomMapperCallbackTy CustomMapperCB, bool PreserveMemberOfFlags=false, bool PropagatePresentToPointee=false)
Emit the user-defined mapper function.
LLVM_ABI CallInst * createCachedThreadPrivate(const LocationDescription &Loc, llvm::Value *Pointer, llvm::ConstantInt *Size, const llvm::Twine &Name=Twine(""))
Create a runtime call for kmpc_threadprivate_cached.
IRBuilder Builder
The LLVM-IR Builder used to create IR.
LLVM_ABI GlobalValue * createGlobalFlag(unsigned Value, StringRef Name)
Create a hidden global flag Name in the module with initial value Value.
LLVM_ABI void emitOffloadingArraysArgument(IRBuilderBase &Builder, OpenMPIRBuilder::TargetDataRTArgs &RTArgs, OpenMPIRBuilder::TargetDataInfo &Info, bool ForEndCall=false)
Emit the arguments to be passed to the runtime library based on the arrays of base pointers,...
LLVM_ABI InsertPointOrErrorTy createMasked(const LocationDescription &Loc, BodyGenCallbackTy BodyGenCB, FinalizeCallbackTy FiniCB, Value *Filter)
Generator for 'omp masked'.
LLVM_ABI Expected< CanonicalLoopInfo * > createCanonicalLoop(const LocationDescription &Loc, LoopBodyGenCallbackTy BodyGenCB, Value *TripCount, const Twine &Name="loop")
Generator for the control flow structure of an OpenMP canonical loop.
function_ref< Expected< InsertPointTy >( InsertPointTy AllocaIP, InsertPointTy CodeGenIP, Value *DestPtr, Value *SrcPtr)> TaskDupCallbackTy
Callback type for task duplication function code generation.
LLVM_ABI Value * getSizeInBytes(Value *BasePtr)
Computes the size of type in bytes.
llvm::function_ref< llvm::Error( InsertPointTy BodyIP, llvm::Value *LinearIV)> IteratorBodyGenTy
OpenMPIRBuilder(Module &M)
Create a new OpenMPIRBuilder operating on the given module M.
LLVM_ABI FunctionCallee createDispatchDeinitFunction()
Returns __kmpc_dispatch_deinit runtime function.
LLVM_ABI void registerTargetGlobalVariable(OffloadEntriesInfoManager::OMPTargetGlobalVarEntryKind CaptureClause, OffloadEntriesInfoManager::OMPTargetDeviceClauseKind DeviceClause, bool IsDeclaration, bool IsExternallyVisible, TargetRegionEntryInfo EntryInfo, StringRef MangledName, std::vector< GlobalVariable * > &GeneratedRefs, bool OpenMPSIMD, std::vector< Triple > TargetTriple, std::function< Constant *()> GlobalInitializer, std::function< GlobalValue::LinkageTypes()> VariableLinkage, Type *LlvmPtrTy, Constant *Addr)
Registers a target variable for device or host.
LLVM_ABI void createTargetDeinit(const LocationDescription &Loc, int32_t TeamsReductionDataSize=0)
Create a runtime call for kmpc_target_deinit.
BodyGenTy
Type of BodyGen to use for region codegen.
LLVM_ABI CanonicalLoopInfo * fuseLoops(DebugLoc DL, ArrayRef< CanonicalLoopInfo * > Loops)
Fuse a sequence of loops.
LLVM_ABI void emitX86DeclareSimdFunction(llvm::Function *Fn, unsigned NumElements, const llvm::APSInt &VLENVal, llvm::ArrayRef< DeclareSimdAttrTy > ParamAttrs, DeclareSimdBranch Branch)
Emit x86 vector-function ABI attributes for a declare simd function.
SmallVector< llvm::Function *, 16 > ConstantAllocaRaiseCandidates
A collection of candidate target functions that's constant allocas will attempt to be raised on a cal...
OffloadEntriesInfoManager OffloadInfoManager
Info manager to keep track of target regions.
static LLVM_ABI std::pair< int32_t, int32_t > readTeamBoundsForKernel(const Triple &T, Function &Kernel)
Read/write a bounds on teams for Kernel.
const std::string ompOffloadInfoName
OMP Offload Info Metadata name string.
Expected< InsertPointTy > InsertPointOrErrorTy
Type used to represent an insertion point or an error value.
LLVM_ABI InsertPointTy createCopyPrivate(const LocationDescription &Loc, llvm::Value *BufSize, llvm::Value *CpyBuf, llvm::Value *CpyFn, llvm::Value *DidIt)
Generator for __kmpc_copyprivate.
void popFinalizationCB()
Pop the last finalization callback from the finalization stack.
LLVM_ABI InsertPointOrErrorTy createSections(const LocationDescription &Loc, InsertPointTy AllocaIP, ArrayRef< StorableBodyGenCallbackTy > SectionCBs, PrivatizeCallbackTy PrivCB, FinalizeCallbackTy FiniCB, bool IsCancellable, bool IsNowait)
Generator for 'omp sections'.
std::function< void(EmitMetadataErrorKind, TargetRegionEntryInfo)> EmitMetadataErrorReportFunctionTy
Callback function type.
function_ref< InsertPointOrErrorTy( Argument &Arg, Value *Input, Value *&RetVal, InsertPointTy AllocaIP, InsertPointTy CodeGenIP, ArrayRef< InsertPointTy > DeallocIPs)> TargetGenArgAccessorsCallbackTy
LLVM_ABI Expected< ScanInfo * > scanInfoInitialize()
Creates a ScanInfo object, allocates and returns the pointer.
LLVM_ABI InsertPointOrErrorTy emitTargetTask(TargetTaskBodyCallbackTy TaskBodyCB, Value *DeviceID, Value *RTLoc, OpenMPIRBuilder::InsertPointTy AllocaIP, const DependenciesInfo &Dependencies, const TargetDataRTArgs &RTArgs, bool HasNoWait)
Generate a target-task for the target construct.
LLVM_ABI InsertPointTy createAtomicRead(const LocationDescription &Loc, AtomicOpValue &X, AtomicOpValue &V, AtomicOrdering AO, InsertPointTy AllocaIP)
Emit atomic Read for : V = X — Only Scalar data types.
function_ref< Error(InsertPointTy AllocaIP, InsertPointTy CodeGenIP, ArrayRef< BasicBlock * > DeallocBlocks)> BodyGenCallbackTy
Callback type for body (=inner region) code generation.
bool updateToLocation(const LocationDescription &Loc)
Update the internal location to Loc.
LLVM_ABI void createFlush(const LocationDescription &Loc)
Generator for 'omp flush'.
LLVM_ABI void createTaskwait(const LocationDescription &Loc, DependenciesInfo Dependencies={})
Generator for 'omp taskwait'.
LLVM_ABI Constant * getAddrOfDeclareTargetVar(OffloadEntriesInfoManager::OMPTargetGlobalVarEntryKind CaptureClause, OffloadEntriesInfoManager::OMPTargetDeviceClauseKind DeviceClause, bool IsDeclaration, bool IsExternallyVisible, TargetRegionEntryInfo EntryInfo, StringRef MangledName, std::vector< GlobalVariable * > &GeneratedRefs, bool OpenMPSIMD, std::vector< Triple > TargetTriple, Type *LlvmPtrTy, std::function< Constant *()> GlobalInitializer, std::function< GlobalValue::LinkageTypes()> VariableLinkage)
Retrieve (or create if non-existent) the address of a declare target variable, used in conjunction wi...
origPtr *with the address space normalization required by the runtime entry point *The NULL descriptor makes the runtime walk the enclosing taskgroups to *find the matching task_reduction registration for the item The lookups *are emitted at p Loc
EmitMetadataErrorKind
The kind of errors that can occur when emitting the offload entries and metadata.
ScanInfo holds the information to assist in lowering of Scan reduction.
llvm::SmallDenseMap< llvm::Value *, llvm::Value * > * ScanBuffPtrs
Maps the private reduction variable to the pointer of the temporary buffer.
llvm::BasicBlock * OMPScanLoopExit
Exit block of loop body.
llvm::Value * IV
Keeps track of value of iteration variable for input/scan loop to be used for Scan directive lowering...
llvm::BasicBlock * OMPAfterScanBlock
Dominates the body of the loop before scan directive.
llvm::BasicBlock * OMPScanInit
Block before loop body where scan initializations are done.
llvm::BasicBlock * OMPBeforeScanBlock
Dominates the body of the loop before scan directive.
llvm::BasicBlock * OMPScanFinish
Block after loop body where scan finalizations are done.
ScanInfo & operator=(const ScanInfo &)=delete
llvm::Value * Span
Stores the span of canonical loop being lowered to be used for temporary buffer allocation or Finaliz...
bool OMPFirstScanLoop
If true, it indicates Input phase is lowered; else it indicates ScanPhase is lowered.
ScanInfo(ScanInfo &)=delete
llvm::BasicBlock * OMPScanDispatch
Controls the flow to before or after scan blocks.
A vector that has set insertion semantics.
Definition SetVector.h:57
A templated base class for SmallPtrSet which provides the typesafe interface that is common across al...
This class consists of common code factored out of the SmallVector class to reduce code duplication b...
This is a 'vector' (really, a variable-sized array), optimized for the case when the array is small.
StringMap - This is an unconventional map that is specialized for handling keys that are "strings",...
Definition StringMap.h:129
Represent a constant reference to a string, i.e.
Definition StringRef.h:56
Class to represent struct types.
Target - Wrapper for Target specific information.
Triple - Helper class for working with autoconf configuration names.
Definition Triple.h:48
Twine - A lightweight data structure for efficiently representing the concatenation of temporary valu...
Definition Twine.h:82
The instances of the Type class are immutable: once they are created, they are never changed.
Definition Type.h:46
Value * getOperand(unsigned i) const
Definition User.h:207
See the file comment.
Definition ValueMap.h:84
LLVM Value Representation.
Definition Value.h:75
Type * getType() const
All values are typed, get the type of this value.
Definition Value.h:257
LLVM_ABI void setName(const Twine &Name)
Change the name of the value.
Definition Value.cpp:394
Value handle that is nullable, but tries to track the Value.
An efficient, type-erasing, non-owning reference to a callable.
The virtual file system interface.
unsigned ID
LLVM IR allows to use arbitrary numbers as calling convention identifiers.
Definition CallingConv.h:24
@ C
The default llvm calling convention, compatible with C.
Definition CallingConv.h:34
OpenMPOffloadMappingFlags
Values for bit flags used to specify the mapping type for offloading.
IdentFlag
IDs for all omp runtime library ident_t flag encodings (see their defintion in openmp/runtime/src/kmp...
RTLDependenceKindTy
Dependence kind for RTL.
RuntimeFunction
IDs for all omp runtime library (RTL) functions.
OMPDynGroupprivateFallbackType
The fallback types for the dyn_groupprivate clause.
WorksharingLoopType
A type of worksharing loop construct.
OMPAtomicCompareOp
Atomic compare operations. Currently OpenMP only supports ==, >, and <.
This is an optimization pass for GlobalISel generic memory operations.
LLVM_ABI BasicBlock * splitBBWithSuffix(IRBuilderBase &Builder, bool CreateBranch, llvm::Twine Suffix=".split")
Like splitBB, but reuses the current block's name for the new name.
@ Offset
Definition DWP.cpp:577
LLVM_ABI BasicBlock * splitBB(IRBuilderBase::InsertPoint IP, bool CreateBranch, DebugLoc DL, llvm::Twine Name={})
Split a BasicBlock at an InsertPoint, even if the block is degenerate (missing the terminator).
auto cast_or_null(const Y &Val)
Definition Casting.h:714
RelativeUniformCounterPtr ValuesPtrExpr VTableAddr Value
Definition InstrProf.h:143
bool isa(const From &Val)
isa<X> - Return true if the parameter to the template is an instance of one of the template type argu...
Definition Casting.h:547
AtomicOrdering
Atomic ordering for LLVM's memory model.
IRBuilder(LLVMContext &, FolderTy, InserterTy, MDNode *, ArrayRef< OperandBundleDef >) -> IRBuilder< FolderTy, InserterTy >
RelativeUniformCounterPtr ValuesPtrExpr VTableAddr Count
Definition InstrProf.h:145
IntPtrTy
Definition InstrProf.h:82
DWARFExpression::Operation Op
ArrayRef(const T &OneElt) -> ArrayRef< T >
OutputIt move(R &&Range, OutputIt Out)
Provide wrappers to std::move which take ranges instead of having to pass begin/end explicitly.
Definition STLExtras.h:1933
LLVM_ABI void spliceBB(IRBuilderBase::InsertPoint IP, BasicBlock *New, bool CreateBranch, DebugLoc DL)
Move the instruction after an InsertPoint to the beginning of another BasicBlock.
decltype(auto) cast(const From &Val)
cast<X> - Return the argument parameter cast to the specified type.
Definition Casting.h:559
Implement std::hash so that hash_code can be used in STL containers.
Definition BitVector.h:878
This struct is a compact representation of a valid (non-zero power of two) alignment.
Definition Alignment.h:39
A struct to pack the relevant information for an OpenMP affinity clause.
a struct to pack relevant information while generating atomic Ops
Attribute set of the declare simd parameter.
Describes a declare target global variable replacement to be applied during finalization.
DependData(omp::RTLDependenceKindTy DepKind, Type *DepValueType, Value *DepVal)
omp::RTLDependenceKindTy DepKind
A struct to pack static and dynamic dependency information for a task.
DependenciesInfo(SmallVector< DependData > D)
const omp::Directive DK
The directive kind of the innermost directive that has an associated region which might require final...
const bool IsCancellable
Flag to indicate if the directive is cancellable.
LLVM_ABI Error mergeFiniBB(IRBuilderBase &Builder, BasicBlock *ExistingFiniBB)
For cases where there is an unavoidable existing finalization block (e.g.
FinalizationInfo(FinalizeCallbackTy FiniCB, omp::Directive DK, bool IsCancellable)
LLVM_ABI Expected< BasicBlock * > getFiniBB(IRBuilderBase &Builder)
The basic block to which control should be transferred to implement the FiniCB.
Description of a LLVM-IR insertion point (IP) and a debug/source location (filename,...
LocationDescription(const InsertPointTy &IP, const DebugLoc &DL)
LocationDescription(const IRBuilderBase &IRB)
This structure contains combined information generated for mappable clauses, including base pointers,...
void append(MapInfosTy &CurInfo)
Append arrays in CurInfo.
MapDeviceInfoArrayTy DevicePointers
MapHasAttachPtrArrayTy HasAttachPtr
True for entries that have an attach ptr, and thus an accompanying ATTACH entry linking that ptr to i...
StructNonContiguousInfo NonContigInfo
Helper that contains information about regions we need to outline during finalization.
void collectBlocks(SmallPtrSetImpl< BasicBlock * > &BlockSet, SmallVectorImpl< BasicBlock * > &BlockVector)
Collect all blocks in between EntryBB and ExitBB in both the given vector and set.
Function * getFunction() const
Return the function that contains the region to be outlined.
SmallVector< Value *, 2 > ExcludeArgsFromAggregate
virtual std::unique_ptr< CodeExtractor > createCodeExtractor(ArrayRef< BasicBlock * > Blocks, bool ArgsInZeroAddressSpace, Twine Suffix=Twine(""))
Create a CodeExtractor instance based on the information stored in this structure,...
std::function< void(Function &)> PostOutlineCBTy
SmallVector< BasicBlock * > OuterDeallocBBs
EvalKind EvaluationKind
Reduction evaluation kind - scalar, complex or aggregate.
ReductionInfo(Type *ElementType, Value *Variable, Value *PrivateVariable, EvalKind EvaluationKind, ReductionGenCBTy ReductionGen, ReductionGenClangCBTy ReductionGenClang, ReductionGenAtomicCBTy AtomicReductionGen, ReductionGenDataPtrPtrCBTy DataPtrPtrGen, Type *ByRefAllocatedType=nullptr, Type *ByRefElementType=nullptr)
ReductionGenAtomicCBTy AtomicReductionGen
Callback for generating the atomic reduction body, may be null.
ReductionGenCBTy ReductionGen
Callback for generating the reduction body.
ReductionInfo(Value *PrivateVariable)
Type * ByRefAllocatedType
For by-ref reductions, we need to keep track of 2 extra types that are potentially different:
Value * Variable
Reduction variable of pointer type.
Value * PrivateVariable
Thread-private partial reduction variable.
ReductionGenClangCBTy ReductionGenClang
Clang callback for generating the reduction body.
Type * ElementType
Reduction element type, must match pointee type of variable.
ReductionGenDataPtrPtrCBTy DataPtrPtrGen
Container for the arguments used to pass data to the runtime library.
Value * SizesArray
The array of sizes passed to the runtime library.
TargetDataRTArgs(Value *BasePointersArray, Value *PointersArray, Value *SizesArray, Value *MapTypesArray, Value *MapTypesArrayEnd, Value *MappersArray, Value *MapNamesArray)
Value * PointersArray
The array of section pointers passed to the runtime library.
Value * MappersArray
The array of user-defined mappers passed to the runtime library.
Value * MapTypesArrayEnd
The array of map types passed to the runtime library for the end of the region, or nullptr if there a...
Value * BasePointersArray
The array of base pointer passed to the runtime library.
Value * MapTypesArray
The array of map types passed to the runtime library for the beginning of the region or for the entir...
Value * MapNamesArray
The array of original declaration names of mapped pointers sent to the runtime library for debugging.
Data structure that contains the needed information to construct the kernel args vector.
bool StrictBlocks
True if the kernel strictly requires the number of blocks and threads above to run.
ArrayRef< Value * > NumThreads
The number of threads.
TargetDataRTArgs RTArgs
Arguments passed to the runtime library.
TargetKernelArgs(unsigned NumTargetItems, TargetDataRTArgs RTArgs, Value *NumIterations, ArrayRef< Value * > NumTeams, ArrayRef< Value * > NumThreads, Value *DynCGroupMem, bool HasNoWait, bool StrictBlocks, bool StrictThreads, omp::OMPDynGroupprivateFallbackType DynCGroupMemFallback)
Value * NumIterations
The number of iterations.
Value * DynCGroupMem
The size of the dynamic shared memory.
unsigned NumTargetItems
Number of arguments passed to the runtime library.
bool HasNoWait
True if the kernel has 'no wait' clause.
ArrayRef< Value * > NumTeams
The number of teams.
omp::OMPDynGroupprivateFallbackType DynCGroupMemFallback
The fallback mechanism for the shared memory.
Container to pass the default attributes with which a kernel must be launched, used to set kernel att...
Container to pass LLVM IR runtime values or constants related to the number of teams and threads with...
Value * DeviceID
Device ID value used in the kernel launch.
Value * LoopTripCount
Total number of iterations of the SPMD or Generic-SPMD kernel or null if it is a generic kernel.
SmallVector< Value * > MaxThreads
'parallel' construct 'num_threads' clause value, if present and it is an SPMD kernel.
A MapVector that performs no allocations if smaller than a certain size.
Definition MapVector.h:342
Data structure to contain the information needed to uniquely identify a target entry.
static LLVM_ABI void getTargetRegionEntryFnName(SmallVectorImpl< char > &Name, StringRef ParentName, unsigned DeviceID, unsigned FileID, unsigned Line, unsigned Count)
static constexpr const char * KernelNamePrefix
The prefix used for kernel names.
bool operator<(const TargetRegionEntryInfo &RHS) const
TargetRegionEntryInfo(StringRef ParentName, unsigned DeviceID, unsigned FileID, unsigned Line, unsigned Count=0)
Defines various target-specific GPU grid values that must be consistent between host RTL (plugin),...