|
LLVM 23.0.0git
|
This file implements the OpenMPIRBuilder class, which is used as a convenient way to create LLVM instructions for OpenMP directives. More...
#include "llvm/Frontend/OpenMP/OMPIRBuilder.h"#include "llvm/ADT/SmallBitVector.h"#include "llvm/ADT/SmallSet.h"#include "llvm/ADT/StringExtras.h"#include "llvm/ADT/StringRef.h"#include "llvm/Analysis/AssumptionCache.h"#include "llvm/Analysis/CodeMetrics.h"#include "llvm/Analysis/LoopInfo.h"#include "llvm/Analysis/OptimizationRemarkEmitter.h"#include "llvm/Analysis/PostDominators.h"#include "llvm/Analysis/ScalarEvolution.h"#include "llvm/Analysis/TargetLibraryInfo.h"#include "llvm/Bitcode/BitcodeReader.h"#include "llvm/Frontend/Offloading/Utility.h"#include "llvm/Frontend/OpenMP/OMPGridValues.h"#include "llvm/IR/Attributes.h"#include "llvm/IR/BasicBlock.h"#include "llvm/IR/CFG.h"#include "llvm/IR/CallingConv.h"#include "llvm/IR/Constant.h"#include "llvm/IR/Constants.h"#include "llvm/IR/DIBuilder.h"#include "llvm/IR/DebugInfoMetadata.h"#include "llvm/IR/DerivedTypes.h"#include "llvm/IR/Function.h"#include "llvm/IR/GlobalVariable.h"#include "llvm/IR/IRBuilder.h"#include "llvm/IR/InstIterator.h"#include "llvm/IR/IntrinsicInst.h"#include "llvm/IR/LLVMContext.h"#include "llvm/IR/MDBuilder.h"#include "llvm/IR/Metadata.h"#include "llvm/IR/PassInstrumentation.h"#include "llvm/IR/PassManager.h"#include "llvm/IR/ReplaceConstant.h"#include "llvm/IR/Value.h"#include "llvm/MC/TargetRegistry.h"#include "llvm/Support/CommandLine.h"#include "llvm/Support/Error.h"#include "llvm/Support/ErrorHandling.h"#include "llvm/Support/FileSystem.h"#include "llvm/Support/NVVMAttributes.h"#include "llvm/Support/VirtualFileSystem.h"#include "llvm/Target/TargetMachine.h"#include "llvm/Target/TargetOptions.h"#include "llvm/Transforms/Utils/BasicBlockUtils.h"#include "llvm/Transforms/Utils/Cloning.h"#include "llvm/Transforms/Utils/CodeExtractor.h"#include "llvm/Transforms/Utils/LoopPeel.h"#include "llvm/Transforms/Utils/UnrollLoop.h"#include <cstdint>#include <optional>#include "llvm/Frontend/OpenMP/OMPKinds.def"Go to the source code of this file.
Macros | |
| #define | DEBUG_TYPE "openmp-ir-builder" |
| #define | OMP_ATTRS_SET(VarName, AttrSet) |
| #define | OMP_RTL_ATTRS(Enum, FnAttrSet, RetAttrSet, ArgAttrSets) |
| #define | OMP_RTL(Enum, Str, IsVarArg, ReturnType, ...) |
| #define | OMP_RTL(Enum, Str, ...) |
| #define | OMP_CANCEL_KIND(Enum, Str, DirectiveEnum, Value) |
| #define | OMP_CANCEL_KIND(Enum, Str, DirectiveEnum, Value) |
| #define | OMP_TYPE(VarName, InitValue) |
| #define | OMP_ARRAY_TYPE(VarName, ElemTy, ArraySize) |
| #define | OMP_FUNCTION_TYPE(VarName, IsVarArg, ReturnType, ...) |
| #define | OMP_STRUCT_TYPE(VarName, StructName, Packed, ...) |
Functions | |
| static bool | isConflictIP (IRBuilder<>::InsertPoint IP1, IRBuilder<>::InsertPoint IP2) |
| Return whether IP1 and IP2 are ambiguous, i.e. | |
| static bool | isValidWorkshareLoopScheduleType (OMPScheduleType SchedType) |
| static void | restoreIPandDebugLoc (llvm::IRBuilderBase &Builder, llvm::IRBuilderBase::InsertPoint IP) |
| This is wrapper over IRBuilderBase::restoreIP that also restores the current debug location to the last instruction in the specified basic block if the insert point points to the end of the block. | |
| static bool | hasGridValue (const Triple &T) |
| static const omp::GV & | getGridValue (const Triple &T, Function *Kernel) |
| static OMPScheduleType | getOpenMPBaseScheduleType (llvm::omp::ScheduleKind ClauseKind, bool HasChunks, bool HasSimdModifier, bool HasDistScheduleChunks) |
| Determine which scheduling algorithm to use, determined from schedule clause arguments. | |
| static OMPScheduleType | getOpenMPOrderingScheduleType (OMPScheduleType BaseScheduleType, bool HasOrderedClause) |
| Adds ordering modifier flags to schedule type. | |
| static OMPScheduleType | getOpenMPMonotonicityScheduleType (OMPScheduleType ScheduleType, bool HasSimdModifier, bool HasMonotonic, bool HasNonmonotonic, bool HasOrderedClause) |
| Adds monotonicity modifier flags to schedule type. | |
| static OMPScheduleType | computeOpenMPScheduleType (ScheduleKind ClauseKind, bool HasChunks, bool HasSimdModifier, bool HasMonotonicModifier, bool HasNonmonotonicModifier, bool HasOrderedClause, bool HasDistScheduleChunks) |
| Determine the schedule type using schedule and ordering clause arguments. | |
| static std::optional< omp::OMPTgtExecModeFlags > | getTargetKernelExecMode (Function &Kernel) |
| Given a function, if it represents the entry point of a target kernel, this returns the execution mode flags associated with that kernel. | |
| static bool | isGenericKernel (Function &Fn) |
| static void | redirectTo (BasicBlock *Source, BasicBlock *Target, DebugLoc DL) |
Make Source branch to Target. | |
| Value * | createFakeIntVal (IRBuilderBase &Builder, OpenMPIRBuilder::InsertPointTy OuterAllocaIP, llvm::SmallVectorImpl< Instruction * > &ToBeDeleted, OpenMPIRBuilder::InsertPointTy InnerAllocaIP, const Twine &Name="", bool AsPtr=true, bool Is64Bit=false) |
| static void | raiseUserConstantDataAllocasToEntryBlock (IRBuilderBase &Builder, Function *Function) |
| static void | hoistNonEntryAllocasToEntryBlock (llvm::BasicBlock &Block) |
| static void | hoistNonEntryAllocasToEntryBlock (llvm::Function *Func) |
| static Function * | createTargetParallelWrapper (OpenMPIRBuilder *OMPIRBuilder, Function &OutlinedFn) |
| Create wrapper function used to gather the outlined function's argument structure from a shared buffer and to forward them to it when running in Generic mode. | |
| static void | targetParallelCallback (OpenMPIRBuilder *OMPIRBuilder, Function &OutlinedFn, Function *OuterFn, BasicBlock *OuterAllocaBB, Value *Ident, Value *IfCondition, Value *NumThreads, Instruction *PrivTID, AllocaInst *PrivTIDAddr, Value *ThreadID, const SmallVector< Instruction *, 4 > &ToBeDeleted) |
| static void | hostParallelCallback (OpenMPIRBuilder *OMPIRBuilder, Function &OutlinedFn, Function *OuterFn, Value *Ident, Value *IfCondition, Instruction *PrivTID, AllocaInst *PrivTIDAddr, const SmallVector< Instruction *, 4 > &ToBeDeleted) |
| static Value * | emitTaskDependencies (OpenMPIRBuilder &OMPBuilder, const SmallVectorImpl< OpenMPIRBuilder::DependData > &Dependencies) |
| static OpenMPIRBuilder::InsertPointTy | getInsertPointAfterInstr (Instruction *I) |
| static void | checkReductionInfos (ArrayRef< OpenMPIRBuilder::ReductionInfo > ReductionInfos, bool IsGPU) |
| static Function * | getFreshReductionFunc (Module &M) |
| static Error | populateReductionFunction (Function *ReductionFunc, ArrayRef< OpenMPIRBuilder::ReductionInfo > ReductionInfos, IRBuilder<> &Builder, ArrayRef< bool > IsByRef, bool IsGPU) |
| static llvm::CallInst * | emitNoUnwindRuntimeCall (IRBuilder<> &Builder, llvm::FunctionCallee Callee, ArrayRef< llvm::Value * > Args, const llvm::Twine &Name) |
| static FunctionCallee | getKmpcDistForStaticInitForType (Type *Ty, Module &M, OpenMPIRBuilder &OMPBuilder) |
| static FunctionCallee | getKmpcForStaticInitForType (Type *Ty, Module &M, OpenMPIRBuilder &OMPBuilder) |
| static void | addAccessGroupMetadata (BasicBlock *Block, MDNode *AccessGroup, LoopInfo &LI) |
Attach llvm.access.group metadata to the memref instructions of Block. | |
| static void | addLoopMetadata (CanonicalLoopInfo *Loop, ArrayRef< Metadata * > Properties) |
Attach loop metadata Properties to the loop described by Loop. | |
| static void | applyParallelAccessesMetadata (CanonicalLoopInfo *CLI, LLVMContext &Ctx, Loop *Loop, LoopInfo &LoopInfo, SmallVector< Metadata * > &LoopMDList) |
| static FunctionCallee | getKmpcForStaticLoopForType (Type *Ty, OpenMPIRBuilder *OMPBuilder, WorksharingLoopType LoopType) |
| static void | createTargetLoopWorkshareCall (OpenMPIRBuilder *OMPBuilder, WorksharingLoopType LoopType, BasicBlock *InsertBlock, Value *Ident, Value *LoopBodyArg, Value *TripCount, Function &LoopBodyFn, bool NoLoop) |
| static void | workshareLoopTargetCallback (OpenMPIRBuilder *OMPIRBuilder, CanonicalLoopInfo *CLI, Value *Ident, Function &OutlinedFn, const SmallVector< Instruction *, 4 > &ToBeDeleted, WorksharingLoopType LoopType, bool NoLoop) |
| static FunctionCallee | getKmpcForDynamicInitForType (Type *Ty, Module &M, OpenMPIRBuilder &OMPBuilder) |
| Returns an LLVM function to call for initializing loop bounds using OpenMP dynamic scheduling depending on type. | |
| static FunctionCallee | getKmpcForDynamicNextForType (Type *Ty, Module &M, OpenMPIRBuilder &OMPBuilder) |
| Returns an LLVM function to call for updating the next loop using OpenMP dynamic scheduling depending on type. | |
| static FunctionCallee | getKmpcForDynamicFiniForType (Type *Ty, Module &M, OpenMPIRBuilder &OMPBuilder) |
| Returns an LLVM function to call for finalizing the dynamic loop using depending on type. | |
| static void | redirectAllPredecessorsTo (BasicBlock *OldTarget, BasicBlock *NewTarget, DebugLoc DL) |
Redirect all edges that branch to OldTarget to NewTarget. | |
| static void | removeUnusedBlocksFromParent (ArrayRef< BasicBlock * > BBs) |
Determine which blocks in BBs are reachable from outside and remove the ones that are not reachable from the function. | |
| static void | addBasicBlockMetadata (BasicBlock *BB, ArrayRef< Metadata * > Properties) |
Attach metadata Properties to the basic block described by BB. | |
| static std::unique_ptr< TargetMachine > | createTargetMachine (Function *F, CodeGenOptLevel OptLevel) |
| Create the TargetMachine object to query the backend for optimization preferences. | |
| static int32_t | computeHeuristicUnrollFactor (CanonicalLoopInfo *CLI) |
Heuristically determine the best-performant unroll factor for CLI. | |
| static void | updateNVPTXAttr (Function &Kernel, StringRef Name, int32_t Value, bool Min) |
| static void | FixupDebugInfoForOutlinedFunction (OpenMPIRBuilder &OMPBuilder, IRBuilderBase &Builder, Function *Func, DenseMap< Value *, std::tuple< Value *, unsigned > > &ValueReplacementMap) |
| static Value * | removeASCastIfPresent (Value *V) |
| static Expected< Function * > | createOutlinedFunction (OpenMPIRBuilder &OMPBuilder, IRBuilderBase &Builder, const OpenMPIRBuilder::TargetKernelDefaultAttrs &DefaultAttrs, StringRef FuncName, SmallVectorImpl< Value * > &Inputs, OpenMPIRBuilder::TargetBodyGenCallbackTy &CBFunc, OpenMPIRBuilder::TargetGenArgAccessorsCallbackTy &ArgAccessorFuncCB) |
| static LoadInst * | loadSharedDataFromTaskDescriptor (OpenMPIRBuilder &OMPIRBuilder, IRBuilderBase &Builder, Value *TaskWithPrivates, Type *TaskWithPrivatesTy) |
| Given a task descriptor, TaskWithPrivates, return the pointer to the block of pointers containing shared data between the parent task and the created task. | |
| static Function * | emitTargetTaskProxyFunction (OpenMPIRBuilder &OMPBuilder, IRBuilderBase &Builder, CallInst *StaleCI, StructType *PrivatesTy, StructType *TaskWithPrivatesTy, const size_t NumOffloadingArrays, const int SharedArgsOperandNo) |
| Create an entry point for a target task with the following. | |
| static Type * | getOffloadingArrayType (Value *V) |
| static StructType * | createTaskWithPrivatesTy (OpenMPIRBuilder &OMPIRBuilder, ArrayRef< Value * > OffloadingArraysToPrivatize) |
| static Error | emitTargetOutlinedFunction (OpenMPIRBuilder &OMPBuilder, IRBuilderBase &Builder, bool IsOffloadEntry, TargetRegionEntryInfo &EntryInfo, const OpenMPIRBuilder::TargetKernelDefaultAttrs &DefaultAttrs, Function *&OutlinedFn, Constant *&OutlinedFnID, SmallVectorImpl< Value * > &Inputs, OpenMPIRBuilder::TargetBodyGenCallbackTy &CBFunc, OpenMPIRBuilder::TargetGenArgAccessorsCallbackTy &ArgAccessorFuncCB) |
| static void | emitTargetCall (OpenMPIRBuilder &OMPBuilder, IRBuilderBase &Builder, OpenMPIRBuilder::InsertPointTy AllocaIP, ArrayRef< BasicBlock * > DeallocBlocks, OpenMPIRBuilder::TargetDataInfo &Info, const OpenMPIRBuilder::TargetKernelDefaultAttrs &DefaultAttrs, const OpenMPIRBuilder::TargetKernelRuntimeAttrs &RuntimeAttrs, Value *IfCond, Function *OutlinedFn, Constant *OutlinedFnID, SmallVectorImpl< Value * > &Args, OpenMPIRBuilder::GenMapInfoCallbackTy GenMapInfoCB, OpenMPIRBuilder::CustomMapperCallbackTy CustomMapperCB, const OpenMPIRBuilder::DependenciesInfo &Dependencies, bool HasNoWait, Value *DynCGroupMem, OMPDynGroupprivateFallbackType DynCGroupMemFallback) |
| static std::string | mangleVectorParameters (ArrayRef< llvm::OpenMPIRBuilder::DeclareSimdAttrTy > ParamAttrs) |
| Mangle the parameter part of the vector function name according to their OpenMP classification. | |
| template<typename T> | |
| static void | addAArch64VectorName (T VLEN, StringRef LMask, StringRef Prefix, char ISA, StringRef ParSeq, StringRef MangledName, bool OutputBecomesInput, llvm::Function *Fn) |
| static void | addAArch64AdvSIMDNDSNames (unsigned NDS, StringRef Mask, StringRef Prefix, char ISA, StringRef ParSeq, StringRef MangledName, bool OutputBecomesInput, llvm::Function *Fn) |
Variables | |
| static cl::opt< bool > | OptimisticAttributes ("openmp-ir-builder-optimistic-attributes", cl::Hidden, cl::desc("Use optimistic attributes describing " "'as-if' properties of runtime calls."), cl::init(false)) |
| static cl::opt< double > | UnrollThresholdFactor ("openmp-ir-builder-unroll-threshold-factor", cl::Hidden, cl::desc("Factor for the unroll threshold to account for code " "simplifications still taking place"), cl::init(1.5)) |
This file implements the OpenMPIRBuilder class, which is used as a convenient way to create LLVM instructions for OpenMP directives.
Definition in file OMPIRBuilder.cpp.
| #define DEBUG_TYPE "openmp-ir-builder" |
Definition at line 69 of file OMPIRBuilder.cpp.
| #define OMP_ARRAY_TYPE | ( | VarName, | |
| ElemTy, | |||
| ArraySize ) |
| #define OMP_ATTRS_SET | ( | VarName, | |
| AttrSet ) |
| #define OMP_CANCEL_KIND | ( | Enum, | |
| Str, | |||
| DirectiveEnum, | |||
| Value ) |
| #define OMP_CANCEL_KIND | ( | Enum, | |
| Str, | |||
| DirectiveEnum, | |||
| Value ) |
| #define OMP_FUNCTION_TYPE | ( | VarName, | |
| IsVarArg, | |||
| ReturnType, | |||
| ... ) |
| #define OMP_RTL | ( | Enum, | |
| Str, | |||
| IsVarArg, | |||
| ReturnType, | |||
| ... ) |
| #define OMP_RTL | ( | Enum, | |
| Str, | |||
| ... ) |
| #define OMP_RTL_ATTRS | ( | Enum, | |
| FnAttrSet, | |||
| RetAttrSet, | |||
| ArgAttrSets ) |
| #define OMP_STRUCT_TYPE | ( | VarName, | |
| StructName, | |||
| Packed, | |||
| ... ) |
| #define OMP_TYPE | ( | VarName, | |
| InitValue ) |
|
static |
Definition at line 12215 of file OMPIRBuilder.cpp.
References addAArch64VectorName(), and llvm_unreachable.
Referenced by llvm::OpenMPIRBuilder::emitAArch64DeclareSimdFunction().
|
static |
Definition at line 12200 of file OMPIRBuilder.cpp.
References llvm::Function::addFnAttr(), llvm::raw_svector_ostream::str(), and T.
Referenced by addAArch64AdvSIMDNDSNames(), and llvm::OpenMPIRBuilder::emitAArch64DeclareSimdFunction().
|
static |
Attach llvm.access.group metadata to the memref instructions of Block.
Definition at line 6926 of file OMPIRBuilder.cpp.
References llvm::Block, and I.
Referenced by applyParallelAccessesMetadata().
|
static |
Attach metadata Properties to the basic block described by BB.
If the basic block already has metadata, the basic block properties are appended.
Definition at line 6891 of file OMPIRBuilder.cpp.
References llvm::append_range(), llvm::drop_begin(), llvm::ArrayRef< T >::empty(), llvm::BasicBlock::getContext(), llvm::MDNode::getDistinct(), llvm::Instruction::getMetadata(), llvm::BasicBlock::getTerminator(), llvm::MDNode::operands(), llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::MDNode::replaceOperandWith(), and llvm::Instruction::setMetadata().
Referenced by addLoopMetadata().
|
static |
Attach loop metadata Properties to the loop described by Loop.
If the loop already has metadata, the loop properties are appended.
Definition at line 6915 of file OMPIRBuilder.cpp.
References addBasicBlockMetadata(), and assert().
Referenced by llvm::OpenMPIRBuilder::applySimd(), llvm::OpenMPIRBuilder::unrollLoopFull(), llvm::OpenMPIRBuilder::unrollLoopHeuristic(), and llvm::OpenMPIRBuilder::unrollLoopPartial().
|
static |
Definition at line 5812 of file OMPIRBuilder.cpp.
References addAccessGroupMetadata(), llvm::Block, llvm::MDNode::get(), llvm::MDString::get(), llvm::LoopBase< BlockT, LoopT >::getBlocks(), llvm::CanonicalLoopInfo::getCond(), llvm::MDNode::getDistinct(), llvm::CanonicalLoopInfo::getHeader(), and llvm::SmallVectorTemplateBase< T, bool >::push_back().
Referenced by llvm::OpenMPIRBuilder::applySimd().
|
static |
Definition at line 4496 of file OMPIRBuilder.cpp.
References assert().
Referenced by llvm::OpenMPIRBuilder::createReductions(), and llvm::OpenMPIRBuilder::createReductionsGPU().
|
static |
Heuristically determine the best-performant unroll factor for CLI.
This depends on the target processor. We are re-using the same heuristics as the LoopUnrollPass.
Definition at line 7307 of file OMPIRBuilder.cpp.
References llvm::Aggressive, assert(), llvm::TargetTransformInfo::UnrollingPreferences::BEInsns, llvm::UnrollCostEstimator::canUnroll(), llvm::CodeMetrics::collectEphemeralValues(), llvm::computeUnrollCount(), llvm::TargetTransformInfo::UnrollingPreferences::Count, createTargetMachine(), llvm::dbgs(), llvm::dyn_cast(), F, FAM, llvm::TargetTransformInfo::UnrollingPreferences::Force, llvm::gatherPeelingPreferences(), llvm::gatherUnrollingPreferences(), llvm::CanonicalLoopInfo::getFunction(), llvm::CanonicalLoopInfo::getHeader(), llvm::LoopInfoBase< BlockT, LoopT >::getLoopFor(), llvm::UnrollCostEstimator::getRolledLoopSize(), I, llvm::SmallPtrSetImpl< PtrType >::insert(), LLVM_DEBUG, llvm::TargetTransformInfo::UnrollingPreferences::OptSizeThreshold, llvm::TargetTransformInfo::UnrollingPreferences::PartialOptSizeThreshold, llvm::TargetTransformInfo::UnrollingPreferences::PartialThreshold, llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::AssumptionAnalysis::run(), llvm::DominatorTreeAnalysis::run(), llvm::LoopAnalysis::run(), llvm::ScalarEvolutionAnalysis::run(), llvm::TargetIRAnalysis::run(), llvm::Value::stripPointerCasts(), llvm::TargetTransformInfo::UnrollingPreferences::Threshold, and UnrollThresholdFactor.
Referenced by llvm::OpenMPIRBuilder::unrollLoopPartial().
|
static |
Determine the schedule type using schedule and ordering clause arguments.
Definition at line 281 of file OMPIRBuilder.cpp.
References assert(), getOpenMPBaseScheduleType(), getOpenMPMonotonicityScheduleType(), getOpenMPOrderingScheduleType(), and isValidWorkshareLoopScheduleType().
Referenced by llvm::OpenMPIRBuilder::applyWorkshareLoop().
| Value * createFakeIntVal | ( | IRBuilderBase & | Builder, |
| OpenMPIRBuilder::InsertPointTy | OuterAllocaIP, | ||
| llvm::SmallVectorImpl< Instruction * > & | ToBeDeleted, | ||
| OpenMPIRBuilder::InsertPointTy | InnerAllocaIP, | ||
| const Twine & | Name = "", | ||
| bool | AsPtr = true, | ||
| bool | Is64Bit = false ) |
Definition at line 444 of file OMPIRBuilder.cpp.
References llvm::cast(), llvm::Type::getInt32Ty(), llvm::Type::getInt64Ty(), and llvm::SmallVectorTemplateBase< T, bool >::push_back().
Referenced by llvm::OpenMPIRBuilder::createTask(), llvm::OpenMPIRBuilder::createTeams(), and llvm::OpenMPIRBuilder::emitTargetTask().
|
static |
Definition at line 8797 of file OMPIRBuilder.cpp.
References llvm::BasicBlock::begin(), llvm::OpenMPIRBuilder::Config, llvm::OpenMPIRBuilder::ConstantAllocaRaiseCandidates, llvm::convertUsersOfConstantsToInstructions(), llvm::BasicBlock::Create(), llvm::Function::Create(), llvm::OpenMPIRBuilder::createTargetDeinit(), llvm::OpenMPIRBuilder::createTargetInit(), llvm::dyn_cast(), llvm::OpenMPIRBuilder::emitKernelExecutionMode(), llvm::OpenMPIRBuilder::emitUsed(), llvm::SmallVectorImpl< T >::emplace_back(), llvm::OpenMPIRBuilder::TargetKernelDefaultAttrs::ExecFlags, FixupDebugInfoForOutlinedFunction(), llvm::FunctionType::get(), llvm::Argument::getArgNo(), llvm::BasicBlock::getFirstNonPHIIt(), llvm::BasicBlock::getFirstNonPHIOrDbg(), llvm::Function::getFnAttribute(), llvm::Type::getInt64Ty(), llvm::GlobalValue::getParent(), llvm::PointerType::getUnqual(), llvm::GlobalValue::InternalLinkage, llvm::isa(), llvm::OpenMPIRBuilderConfig::isTargetDevice(), llvm::make_early_inc_range(), llvm::make_range(), llvm::SmallVectorTemplateBase< T, bool >::push_back(), removeASCastIfPresent(), llvm::splitBB(), llvm::Expected< T >::takeError(), Users, and llvm::zip().
Referenced by emitTargetOutlinedFunction().
|
static |
Definition at line 6084 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::Builder, llvm::OpenMPIRBuilder::createRuntimeFunctionCall(), llvm::BasicBlock::end(), getKmpcForStaticLoopForType(), llvm::OpenMPIRBuilder::getOrCreateRuntimeFunction(), llvm::Value::getType(), llvm::OpenMPIRBuilder::M, and llvm::SmallVectorTemplateBase< T, bool >::push_back().
Referenced by workshareLoopTargetCallback().
|
static |
Create the TargetMachine object to query the backend for optimization preferences.
Ideally, this would be passed from the front-end to the OpenMPBuilder, but e.g. Clang does not pass it to its CodeGen layer and creates it only when needed for the LLVM pass pipline. We use some default options to avoid having to pass too many settings from the frontend that probably do not matter.
Currently, TargetMachine is only used sometimes by the unrollLoopPartial method. If we are going to use TargetMachine for more purposes, especially those that are sensitive to TargetOptions, RelocModel and CodeModel, it might become be worth requiring front-ends to pass on their TargetMachine, or at least cache it between methods. Note that while fontends such as Clang have just a single main TargetMachine per translation unit, "target-cpu" and "target-features" that determine the TargetMachine are per-function and can be overrided using attribute((target("OPTIONS"))).
Definition at line 7286 of file OMPIRBuilder.cpp.
References llvm::Target::createTargetMachine(), F, llvm::TargetRegistry::lookupTarget(), and Options.
Referenced by llvm::lto::backend(), computeHeuristicUnrollFactor(), LLVMCreateTargetMachineWithOptions(), splitCodeGen(), and llvm::lto::thinBackend().
|
static |
Create wrapper function used to gather the outlined function's argument structure from a shared buffer and to forward them to it when running in Generic mode.
The outlined function is expected to receive 2 integer arguments followed by an optional pointer argument to an argument structure holding the rest.
Definition at line 1500 of file OMPIRBuilder.cpp.
References llvm::Function::arg_size(), assert(), llvm::OpenMPIRBuilder::Builder, llvm::BasicBlock::Create(), llvm::Function::Create(), llvm::FunctionType::get(), llvm::Module::getContext(), llvm::Value::getName(), llvm::OpenMPIRBuilder::getOrCreateRuntimeFunctionPtr(), llvm::GlobalValue::InternalLinkage, and llvm::OpenMPIRBuilder::M.
Referenced by targetParallelCallback().
|
static |
Definition at line 9151 of file OMPIRBuilder.cpp.
References assert(), llvm::StructType::create(), llvm::ArrayRef< T >::empty(), getOffloadingArrayType(), and llvm::SmallVectorTemplateBase< T, bool >::push_back().
Referenced by llvm::OpenMPIRBuilder::emitTargetTask().
|
static |
Definition at line 5053 of file OMPIRBuilder.cpp.
References Call.
Referenced by llvm::OpenMPIRBuilder::emitScanReduction().
|
static |
Definition at line 9597 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::Builder, llvm::cantFail(), llvm::OpenMPIRBuilder::createRuntimeFunctionCall(), llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs::DeviceID, llvm::OpenMPIRBuilder::emitIfClause(), llvm::OpenMPIRBuilder::emitKernelLaunch(), llvm::OpenMPIRBuilder::emitOffloadingArraysAndArgs(), llvm::OpenMPIRBuilder::emitTargetTask(), llvm::OpenMPIRBuilder::DependenciesInfo::empty(), llvm::Constant::getNullValue(), llvm::OpenMPIRBuilder::getOrCreateDefaultSrcLocStr(), llvm::OpenMPIRBuilder::getOrCreateIdent(), llvm::PointerType::getUnqual(), llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs::LoopTripCount, llvm::OpenMPIRBuilder::TargetKernelDefaultAttrs::MaxTeams, llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs::MaxTeams, llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs::MaxThreads, llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::IRBuilderBase::restoreIP(), llvm::OpenMPIRBuilder::TargetKernelArgs::RTArgs, llvm::IRBuilderBase::saveIP(), llvm::Error::success(), llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs::TargetThreadLimit, llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs::TeamsThreadLimit, and llvm::zip_equal().
Referenced by llvm::OpenMPIRBuilder::createTarget().
|
static |
Definition at line 9171 of file OMPIRBuilder.cpp.
References createOutlinedFunction(), and llvm::OpenMPIRBuilder::emitTargetRegionFunction().
Referenced by llvm::OpenMPIRBuilder::createTarget().
|
static |
Create an entry point for a target task with the following.
It'll have the following signature void .omp_target_task_proxy_func(i32 thread.id, ptr task) This function is called from emitTargetTask once the code to launch the target kernel has been outlined already. NumOffloadingArrays is the number of offloading arrays that we need to copy into the task structure so that the deferred target task can access this data even after the stack frame of the generating task has been rolled back. Offloading arrays contain base pointers, pointers, sizes etc of the data that the target kernel will access. These in effect are the non-empty arrays of pointers held by OpenMPIRBuilder::TargetDataRTArgs.
Definition at line 9018 of file OMPIRBuilder.cpp.
References llvm::CallBase::arg_size(), assert(), llvm::cast(), llvm::BasicBlock::Create(), llvm::Function::Create(), llvm::OpenMPIRBuilder::createRuntimeFunctionCall(), llvm::dyn_cast(), llvm::FunctionType::get(), llvm::AllocaInst::getAlign(), llvm::AllocaInst::getAllocatedType(), llvm::AllocaInst::getAllocationSize(), llvm::CallBase::getArgOperand(), llvm::CallBase::getCalledFunction(), llvm::Type::getInt32Ty(), llvm::ilist_node_impl< OptionsT >::getIterator(), llvm::ilist_detail::node_parent_access< NodeTy, ParentTy >::getParent(), llvm::Value::getPointerAlignment(), llvm::GlobalValue::InternalLinkage, loadSharedDataFromTaskDescriptor(), llvm::OpenMPIRBuilder::M, llvm::SmallVectorTemplateBase< T, bool >::push_back(), llvm::SmallVectorImpl< T >::reserve(), and llvm::Value::setName().
Referenced by llvm::OpenMPIRBuilder::emitTargetTask().
|
static |
Definition at line 2194 of file OMPIRBuilder.cpp.
References llvm::sampleprof::Base, llvm::OpenMPIRBuilder::Builder, llvm::OpenMPIRBuilder::emitTaskDependency(), llvm::SmallVectorTemplateCommon< T, typename >::empty(), llvm::enumerate(), llvm::ArrayType::get(), llvm::IRBuilderBase::InsertPoint::getBlock(), llvm::Function::getEntryBlock(), llvm::BasicBlock::getParent(), llvm::BasicBlock::getTerminator(), and llvm::SmallVectorTemplateCommon< T, typename >::size().
Referenced by llvm::OpenMPIRBuilder::createTask(), and llvm::OpenMPIRBuilder::emitTargetTask().
|
static |
Definition at line 8700 of file OMPIRBuilder.cpp.
References assert(), llvm::BasicBlock::back(), llvm::OpenMPIRBuilder::Config, llvm::dyn_cast(), llvm::BasicBlock::empty(), llvm::filterDbgVars(), llvm::MDNode::get(), llvm::DIVariable::getAlignInBits(), llvm::DILocalVariable::getAnnotations(), llvm::DILocalVariable::getArg(), llvm::DIScope::getFile(), llvm::DIVariable::getFile(), llvm::DILocalVariable::getFlags(), llvm::ilist_node_impl< OptionsT >::getIterator(), llvm::DIVariable::getLine(), llvm::DIVariable::getName(), llvm::BasicBlock::getParent(), llvm::DILocalVariable::getScope(), llvm::DIVariable::getType(), I, llvm::BasicBlock::insertDbgRecordBefore(), instructions, llvm::isa(), llvm::OpenMPIRBuilderConfig::isTargetDevice(), and llvm::SmallVectorTemplateBase< T, bool >::push_back().
Referenced by createOutlinedFunction().
Definition at line 4771 of file OMPIRBuilder.cpp.
References llvm::Function::Create(), llvm::FunctionType::get(), llvm::PointerType::getUnqual(), llvm::Type::getVoidTy(), and llvm::GlobalValue::InternalLinkage.
Referenced by llvm::OpenMPIRBuilder::createReductions().
Definition at line 176 of file OMPIRBuilder.cpp.
References llvm::StringRef::count(), llvm::omp::getAMDGPUGridValues(), llvm_unreachable, llvm::omp::NVPTXGridValues, llvm::omp::SPIRVGridValues, and T.
Referenced by llvm::OpenMPIRBuilder::createReductionsGPU(), and llvm::OpenMPIRBuilder::createTargetInit().
|
static |
Definition at line 3082 of file OMPIRBuilder.cpp.
|
static |
Definition at line 5646 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::getOrCreateRuntimeFunction(), and llvm_unreachable.
|
static |
Returns an LLVM function to call for finalizing the dynamic loop using depending on type.
Only i32 and i64 are supported by the runtime. Always interpret integers as unsigned similarly to CanonicalLoopInfo.
Definition at line 6391 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::getOrCreateRuntimeFunction(), and llvm_unreachable.
|
static |
Returns an LLVM function to call for initializing loop bounds using OpenMP dynamic scheduling depending on type.
Only i32 and i64 are supported by the runtime. Always interpret integers as unsigned similarly to CanonicalLoopInfo.
Definition at line 6360 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::getOrCreateRuntimeFunction(), and llvm_unreachable.
|
static |
Returns an LLVM function to call for updating the next loop using OpenMP dynamic scheduling depending on type.
Only i32 and i64 are supported by the runtime. Always interpret integers as unsigned similarly to CanonicalLoopInfo.
Definition at line 6376 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::getOrCreateRuntimeFunction(), and llvm_unreachable.
|
static |
Definition at line 5662 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::getOrCreateRuntimeFunction(), and llvm_unreachable.
|
static |
Definition at line 6046 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::getOrCreateRuntimeFunction(), llvm_unreachable, and llvm::OpenMPIRBuilder::M.
Referenced by createTargetLoopWorkshareCall().
Definition at line 9126 of file OMPIRBuilder.cpp.
References llvm::dyn_cast(), GEP, and llvm_unreachable.
Referenced by createTaskWithPrivatesTy(), and llvm::OpenMPIRBuilder::emitTargetTask().
|
static |
Determine which scheduling algorithm to use, determined from schedule clause arguments.
Definition at line 194 of file OMPIRBuilder.cpp.
References llvm::omp::BaseAuto, and llvm_unreachable.
Referenced by computeOpenMPScheduleType().
|
static |
Adds monotonicity modifier flags to schedule type.
Definition at line 245 of file OMPIRBuilder.cpp.
References assert(), and llvm::omp::ModifierMask.
Referenced by computeOpenMPScheduleType().
|
static |
Adds ordering modifier flags to schedule type.
Definition at line 221 of file OMPIRBuilder.cpp.
References assert().
Referenced by computeOpenMPScheduleType().
|
static |
Given a function, if it represents the entry point of a target kernel, this returns the execution mode flags associated with that kernel.
Definition at line 300 of file OMPIRBuilder.cpp.
References Call, llvm::cast(), llvm::dyn_cast(), llvm::CallBase::getArgOperand(), and llvm::GlobalVariable::getInitializer().
Referenced by isGenericKernel().
Definition at line 172 of file OMPIRBuilder.cpp.
References T.
Referenced by llvm::OpenMPIRBuilder::createTargetInit().
|
static |
Definition at line 865 of file OMPIRBuilder.cpp.
References llvm::Block, llvm::dyn_cast(), llvm::AllocaInst::isArrayAllocation(), llvm::Instruction::moveBefore(), and llvm::SmallVectorTemplateBase< T, bool >::push_back().
Referenced by llvm::OpenMPIRBuilder::finalize(), and hoistNonEntryAllocasToEntryBlock().
|
static |
Definition at line 887 of file OMPIRBuilder.cpp.
References hoistNonEntryAllocasToEntryBlock(), and llvm::DominatorTreeBase< NodeT, IsPostDom >::properlyDominates().
|
static |
Definition at line 1676 of file OMPIRBuilder.cpp.
References llvm::Function::addFnAttr(), llvm::Function::addParamAttr(), llvm::SmallVectorImpl< T >::append(), llvm::CallBase::arg_begin(), llvm::Function::arg_begin(), llvm::CallBase::arg_end(), llvm::Function::arg_size(), assert(), llvm::OpenMPIRBuilder::Builder, llvm::cast(), Cond, llvm::MDBuilder::createCallbackEncoding(), llvm::OpenMPIRBuilder::createRuntimeFunctionCall(), llvm::dbgs(), llvm::dyn_cast(), llvm::Instruction::eraseFromParent(), F, llvm::MDNode::get(), llvm::FunctionCallee::getCallee(), llvm::Constant::getNullValue(), llvm::OpenMPIRBuilder::getOrCreateRuntimeFunctionPtr(), llvm::ilist_detail::node_parent_access< NodeTy, ParentTy >::getParent(), I, LLVM_DEBUG, llvm::SmallVectorTemplateBase< T, bool >::push_back(), and llvm::Value::user_back().
Referenced by llvm::OpenMPIRBuilder::createParallel().
|
static |
Return whether IP1 and IP2 are ambiguous, i.e.
that inserting instructions at position IP1 may change the meaning of IP2 or vice-versa. This is because an InsertPoint stores the instruction before something is inserted. For instance, if both point to the same instruction, two IRBuilders alternating creating instruction will cause the instructions to be interleaved.
Definition at line 92 of file OMPIRBuilder.cpp.
Referenced by llvm::OpenMPIRBuilder::createAtomicUpdate(), llvm::OpenMPIRBuilder::createParallel(), and llvm::OpenMPIRBuilder::createSections().
Definition at line 329 of file OMPIRBuilder.cpp.
References getTargetKernelExecMode(), and llvm::omp::OMP_TGT_EXEC_MODE_GENERIC.
Referenced by llvm::OpenMPIRBuilder::createParallel(), and targetParallelCallback().
|
static |
Definition at line 99 of file OMPIRBuilder.cpp.
Referenced by computeOpenMPScheduleType().
|
static |
Given a task descriptor, TaskWithPrivates, return the pointer to the block of pointers containing shared data between the parent task and the created task.
Definition at line 8984 of file OMPIRBuilder.cpp.
References llvm::PointerType::getUnqual().
Referenced by llvm::OpenMPIRBuilder::emitTargetTask(), and emitTargetTaskProxyFunction().
|
static |
Mangle the parameter part of the vector function name according to their OpenMP classification.
The mangling function is defined in section 4.5 of the AAVFABI(2021Q1).
Definition at line 12103 of file OMPIRBuilder.cpp.
References llvm::OpenMPIRBuilder::Linear, llvm::OpenMPIRBuilder::LinearRef, llvm::OpenMPIRBuilder::LinearUVal, llvm::OpenMPIRBuilder::LinearVal, ParamAttr, llvm::raw_svector_ostream::str(), llvm::OpenMPIRBuilder::Uniform, and llvm::OpenMPIRBuilder::Vector.
Referenced by llvm::OpenMPIRBuilder::emitAArch64DeclareSimdFunction(), and llvm::OpenMPIRBuilder::emitX86DeclareSimdFunction().
|
static |
Definition at line 4780 of file OMPIRBuilder.cpp.
References llvm::BasicBlock::Create(), llvm::OpenMPIRBuilder::ReductionInfo::ElementType, llvm::enumerate(), llvm::ArrayType::get(), llvm::Function::getArg(), llvm::Module::getContext(), llvm::Value::getName(), llvm::GlobalValue::getParent(), llvm::Value::getType(), LHS, llvm::OpenMPIRBuilder::ReductionInfo::PrivateVariable, llvm::OpenMPIRBuilder::ReductionInfo::ReductionGen, RHS, llvm::ArrayRef< T >::size(), llvm::Error::success(), llvm::Expected< T >::takeError(), and llvm::OpenMPIRBuilder::ReductionInfo::Variable.
Referenced by llvm::OpenMPIRBuilder::createReductions().
|
static |
Definition at line 842 of file OMPIRBuilder.cpp.
References llvm::Function::begin(), llvm::Block, llvm::dyn_cast_if_present(), llvm::Function::end(), llvm::AllocaInst::getArraySize(), llvm::Function::getEntryBlock(), llvm::BasicBlock::getFirstNonPHIIt(), llvm::isa(), and llvm::Instruction::moveBeforePreserving().
Referenced by llvm::OpenMPIRBuilder::finalize().
|
static |
Redirect all edges that branch to OldTarget to NewTarget.
That is, after this OldTarget will be orphaned.
Definition at line 6539 of file OMPIRBuilder.cpp.
References DL, llvm::make_early_inc_range(), llvm::predecessors(), and redirectTo().
Referenced by llvm::OpenMPIRBuilder::collapseLoops(), llvm::OpenMPIRBuilder::fuseLoops(), and llvm::OpenMPIRBuilder::tileLoops().
|
static |
Make Source branch to Target.
Handles two situations:
Source already has an unconditional branch.Source is a degenerate block (no terminator because the BB is the current head of the IR construction). Definition at line 341 of file OMPIRBuilder.cpp.
References llvm::cast(), llvm::UncondBrInst::Create(), DL, and llvm::BasicBlock::removePredecessor().
Referenced by llvm::OpenMPIRBuilder::collapseLoops(), llvm::OpenMPIRBuilder::createIteratorLoop(), llvm::OpenMPIRBuilder::fuseLoops(), redirectAllPredecessorsTo(), and llvm::OpenMPIRBuilder::tileLoops().
Definition at line 8791 of file OMPIRBuilder.cpp.
References llvm::cast(), and llvm::Operator::getOpcode().
Referenced by createOutlinedFunction().
|
static |
Determine which blocks in BBs are reachable from outside and remove the ones that are not reachable from the function.
Definition at line 6547 of file OMPIRBuilder.cpp.
References llvm::SmallPtrSetImpl< PtrType >::begin(), llvm::SmallPtrSetImpl< PtrType >::count(), llvm::DeleteDeadBlocks(), llvm::dyn_cast(), llvm::SmallPtrSetImpl< PtrType >::end(), llvm::from_range, llvm::SmallPtrSetImpl< PtrType >::remove_if(), and llvm::Value::uses().
Referenced by llvm::OpenMPIRBuilder::collapseLoops(), llvm::OpenMPIRBuilder::fuseLoops(), and llvm::OpenMPIRBuilder::tileLoops().
|
static |
This is wrapper over IRBuilderBase::restoreIP that also restores the current debug location to the last instruction in the specified basic block if the insert point points to the end of the block.
Definition at line 163 of file OMPIRBuilder.cpp.
References llvm::BasicBlock::back(), llvm::BasicBlock::empty(), llvm::BasicBlock::end(), llvm::Instruction::getStableDebugLoc(), and I.
Referenced by llvm::OpenMPIRBuilder::createTargetData(), and llvm::OpenMPIRBuilder::emitOffloadingArrays().
|
static |
Definition at line 1578 of file OMPIRBuilder.cpp.
References llvm::Function::addFnAttr(), llvm::Function::addParamAttr(), llvm::CallBase::arg_begin(), llvm::Function::arg_begin(), llvm::Function::arg_size(), assert(), llvm::OpenMPIRBuilder::Builder, llvm::cast(), Cond, llvm::OpenMPIRBuilder::createRuntimeFunctionCall(), createTargetParallelWrapper(), llvm::dbgs(), llvm::Instruction::eraseFromParent(), llvm::ArrayType::get(), llvm::AllocaInst::getAddressSpace(), llvm::BasicBlock::getFirstInsertionPt(), llvm::Constant::getNullValue(), llvm::OpenMPIRBuilder::getOrCreateRuntimeFunctionPtr(), llvm::ilist_detail::node_parent_access< NodeTy, ParentTy >::getParent(), I, isGenericKernel(), LLVM_DEBUG, and llvm::Value::user_back().
Referenced by llvm::OpenMPIRBuilder::createParallel().
Definition at line 8295 of file OMPIRBuilder.cpp.
References llvm::utostr().
Referenced by llvm::OpenMPIRBuilder::writeThreadBoundsForKernel().
|
static |
Definition at line 6124 of file OMPIRBuilder.cpp.
References llvm::CallBase::arg_size(), assert(), llvm::BasicBlock::begin(), llvm::OpenMPIRBuilder::Builder, llvm::OpenMPIRBuilder::OutlineInfo::collectBlocks(), createTargetLoopWorkshareCall(), llvm::DeleteDeadBlocks(), llvm::dyn_cast(), llvm::BasicBlock::end(), llvm::OpenMPIRBuilder::OutlineInfo::EntryBB, llvm::Instruction::eraseFromParent(), llvm::OpenMPIRBuilder::OutlineInfo::ExitBB, llvm::CallBase::getArgOperand(), llvm::CanonicalLoopInfo::getBody(), llvm::Instruction::getDebugLoc(), llvm::CanonicalLoopInfo::getExit(), llvm::CanonicalLoopInfo::getHeader(), llvm::Constant::getNullValue(), llvm::ilist_detail::node_parent_access< NodeTy, ParentTy >::getParent(), llvm::CanonicalLoopInfo::getPreheader(), llvm::BasicBlock::getTerminator(), llvm::CanonicalLoopInfo::getTripCount(), llvm::Value::getUniqueUndroppableUser(), llvm::CanonicalLoopInfo::invalidate(), and llvm::BasicBlock::splice().
|
static |
|
static |
Referenced by computeHeuristicUnrollFactor().