MLIR 22.0.0git
OpenMPToLLVMIRTranslation.cpp File Reference
#include "mlir/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.h"
#include "mlir/Analysis/TopologicalSortUtils.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/Dialect/LLVMIR/LLVMTypes.h"
#include "mlir/Dialect/OpenMP/OpenMPDialect.h"
#include "mlir/Dialect/OpenMP/OpenMPInterfaces.h"
#include "mlir/IR/Operation.h"
#include "mlir/Support/LLVM.h"
#include "mlir/Target/LLVMIR/Dialect/OpenMPCommon.h"
#include "mlir/Target/LLVMIR/ModuleTranslation.h"
#include "llvm/ADT/ArrayRef.h"
#include "llvm/ADT/SmallVector.h"
#include "llvm/ADT/TypeSwitch.h"
#include "llvm/Frontend/OpenMP/OMPConstants.h"
#include "llvm/Frontend/OpenMP/OMPIRBuilder.h"
#include "llvm/IR/Constants.h"
#include "llvm/IR/DebugInfoMetadata.h"
#include "llvm/IR/DerivedTypes.h"
#include "llvm/IR/IRBuilder.h"
#include "llvm/IR/MDBuilder.h"
#include "llvm/IR/ReplaceConstant.h"
#include "llvm/Support/FileSystem.h"
#include "llvm/Support/VirtualFileSystem.h"
#include "llvm/TargetParser/Triple.h"
#include "llvm/Transforms/Utils/ModuleUtils.h"
#include <cstdint>
#include <iterator>
#include <numeric>
#include <optional>
#include <utility>

Go to the source code of this file.

Classes

struct  PrivateVarsInfo
 A util to collect info needed to convert delayed privatizers from MLIR to LLVM. More...

Functions

static omp::PrivateClauseOp findPrivatizer (Operation *from, SymbolRefAttr symbolName)
 Looks up from the operation from and returns the PrivateClauseOp with name symbolName.
static LogicalResult checkImplementationStatus (Operation &op)
 Check whether translation to LLVM IR for the given operation is currently supported.
static LogicalResult handleError (llvm::Error error, Operation &op)
template<typename T>
static LogicalResult handleError (llvm::Expected< T > &result, Operation &op)
static llvm::OpenMPIRBuilder::InsertPointTy findAllocaInsertPoint (llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Find the insertion point for allocas given the current insertion point for normal operations in the builder.
static llvm::CanonicalLoopInfo * findCurrentLoopInfo (LLVM::ModuleTranslation &moduleTranslation)
 Find the loop information structure for the loop nest being translated.
static llvm::Expected< llvm::BasicBlock * > convertOmpOpRegions (Region &region, StringRef blockName, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, SmallVectorImpl< llvm::PHINode * > *continuationBlockPHIs=nullptr)
 Converts the given region that appears within an OpenMP dialect operation to LLVM IR, creating a branch from the sourceBlock to the entry block of the region, and a branch from any block with an successor-less OpenMP terminator to continuationBlock.
static llvm::omp::ProcBindKind getProcBindKind (omp::ClauseProcBindKind kind)
 Convert ProcBindKind from MLIR-generated enum to LLVM enum.
static void forwardArgs (LLVM::ModuleTranslation &moduleTranslation, omp::BlockArgOpenMPOpInterface blockArgIface)
 Maps block arguments from blockArgIface (which are MLIR values) to the corresponding LLVM values of the interface's operands.
static LogicalResult convertOmpMasked (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP 'masked' operation into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpMaster (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP 'master' operation into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpCritical (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP 'critical' operation into LLVM IR using OpenMPIRBuilder.
template<typename T>
static void collectReductionDecls (T op, SmallVectorImpl< omp::DeclareReductionOp > &reductions)
 Populates reductions with reduction declarations used in the given op.
static LogicalResult inlineConvertOmpRegions (Region &region, StringRef blockName, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, SmallVectorImpl< llvm::Value * > *continuationBlockArgs=nullptr)
 Translates the blocks contained in the given region and appends them to at the current insertion point of builder.
static OwningReductionGen makeReductionGen (omp::DeclareReductionOp decl, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Create an OpenMPIRBuilder-compatible reduction generator for the given reduction declaration.
static OwningAtomicReductionGen makeAtomicReductionGen (omp::DeclareReductionOp decl, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Create an OpenMPIRBuilder-compatible atomic reduction generator for the given reduction declaration.
static LogicalResult convertOmpOrdered (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP 'ordered' operation into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpOrderedRegion (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP 'ordered_region' operation into LLVM IR using OpenMPIRBuilder.
template<typename T>
static LogicalResult allocReductionVars (T loop, ArrayRef< BlockArgument > reductionArgs, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, const llvm::OpenMPIRBuilder::InsertPointTy &allocaIP, SmallVectorImpl< omp::DeclareReductionOp > &reductionDecls, SmallVectorImpl< llvm::Value * > &privateReductionVariables, DenseMap< Value, llvm::Value * > &reductionVariableMap, SmallVectorImpl< DeferredStore > &deferredStores, llvm::ArrayRef< bool > isByRefs)
 Allocate space for privatized reduction variables.
template<typename T>
static void mapInitializationArgs (T loop, LLVM::ModuleTranslation &moduleTranslation, SmallVectorImpl< omp::DeclareReductionOp > &reductionDecls, DenseMap< Value, llvm::Value * > &reductionVariableMap, unsigned i)
 Map input arguments to reduction initialization region.
static void setInsertPointForPossiblyEmptyBlock (llvm::IRBuilderBase &builder, llvm::BasicBlock *block=nullptr)
static LogicalResult inlineOmpRegionCleanup (llvm::SmallVectorImpl< Region * > &cleanupRegions, llvm::ArrayRef< llvm::Value * > privateVariables, LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase &builder, StringRef regionName, bool shouldLoadCleanupRegionArg=true)
 handling of DeclareReductionOp's cleanup region
template<class OP>
static LogicalResult createReductionsAndCleanup (OP op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, llvm::OpenMPIRBuilder::InsertPointTy &allocaIP, SmallVectorImpl< omp::DeclareReductionOp > &reductionDecls, ArrayRef< llvm::Value * > privateReductionVariables, ArrayRef< bool > isByRef, bool isNowait=false, bool isTeamsReduction=false)
static ArrayRef< boolgetIsByRef (std::optional< ArrayRef< bool > > attr)
template<typename OP>
static LogicalResult allocAndInitializeReductionVars (OP op, ArrayRef< BlockArgument > reductionArgs, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, llvm::OpenMPIRBuilder::InsertPointTy &allocaIP, SmallVectorImpl< omp::DeclareReductionOp > &reductionDecls, SmallVectorImpl< llvm::Value * > &privateReductionVariables, DenseMap< Value, llvm::Value * > &reductionVariableMap, llvm::ArrayRef< bool > isByRef)
static llvm::Value * findAssociatedValue (Value privateVar, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, llvm::DenseMap< Value, Value > *mappedPrivateVars=nullptr)
 Return the llvm::Value * corresponding to the privateVar that is being privatized. It isn't always as simple as looking up moduleTranslation with privateVar. For instance, in case of an allocatable, the descriptor for the allocatable is privatized. This descriptor is mapped using an MapInfoOp. So, this function will return a pointer to the llvm::Value corresponding to the block argument for the mapped descriptor.
static llvm::Expected< llvm::Value * > initPrivateVar (llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, omp::PrivateClauseOp &privDecl, Value mlirPrivVar, BlockArgument &blockArg, llvm::Value *llvmPrivateVar, llvm::BasicBlock *privInitBlock, llvm::DenseMap< Value, Value > *mappedPrivateVars=nullptr)
 Initialize a single (first)private variable. You probably want to use allocateAndInitPrivateVars instead of this. This returns the private variable which has been initialized. This variable should be mapped before constructing the body of the Op.
static llvm::Error initPrivateVars (llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, PrivateVarsInfo &privateVarsInfo, llvm::DenseMap< Value, Value > *mappedPrivateVars=nullptr)
static llvm::Expected< llvm::BasicBlock * > allocatePrivateVars (llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, PrivateVarsInfo &privateVarsInfo, const llvm::OpenMPIRBuilder::InsertPointTy &allocaIP, llvm::DenseMap< Value, Value > *mappedPrivateVars=nullptr)
 Allocate and initialize delayed private variables. Returns the basic block which comes after all of these allocations. llvm::Value * for each of these private variables are populated in llvmPrivateVars.
static LogicalResult copyFirstPrivateVars (mlir::Operation *op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, SmallVectorImpl< mlir::Value > &mlirPrivateVars, ArrayRef< llvm::Value * > llvmPrivateVars, SmallVectorImpl< omp::PrivateClauseOp > &privateDecls, bool insertBarrier, llvm::DenseMap< Value, Value > *mappedPrivateVars=nullptr)
static LogicalResult cleanupPrivateVars (llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, Location loc, SmallVectorImpl< llvm::Value * > &llvmPrivateVars, SmallVectorImpl< omp::PrivateClauseOp > &privateDecls)
static bool constructIsCancellable (Operation *op)
 Returns true if the construct contains omp.cancel or omp.cancellation_point.
static LogicalResult convertOmpSections (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertOmpSingle (omp::SingleOp &singleOp, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP single construct into LLVM IR using OpenMPIRBuilder.
static bool teamsReductionContainedInDistribute (omp::TeamsOp teamsOp)
static LogicalResult convertOmpTeams (omp::TeamsOp op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static void buildDependData (std::optional< ArrayAttr > dependKinds, OperandRange dependVars, LLVM::ModuleTranslation &moduleTranslation, SmallVectorImpl< llvm::OpenMPIRBuilder::DependData > &dds)
static void pushCancelFinalizationCB (SmallVectorImpl< llvm::BranchInst * > &cancelTerminators, llvm::IRBuilderBase &llvmBuilder, llvm::OpenMPIRBuilder &ompBuilder, mlir::Operation *op, llvm::omp::Directive cancelDirective)
 Shared implementation of a callback which adds a termiator for the new block created for the branch taken when an openmp construct is cancelled. The terminator is saved in cancelTerminators. This callback is invoked only if there is cancellation inside of the taskgroup body. The terminator will need to be fixed to branch to the correct block to cleanup the construct.
static void popCancelFinalizationCB (const ArrayRef< llvm::BranchInst * > cancelTerminators, llvm::OpenMPIRBuilder &ompBuilder, const llvm::OpenMPIRBuilder::InsertPointTy &afterIP)
 If we cancelled the construct, we should branch to the finalization block of that construct. OMPIRBuilder structures the CFG such that the cleanup block is immediately before the continuation block. Now this finalization has been created we can fix the branch.
static LogicalResult convertOmpTaskOp (omp::TaskOp taskOp, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP task construct into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpTaskgroupOp (omp::TaskgroupOp tgOp, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP taskgroup construct into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpTaskwaitOp (omp::TaskwaitOp twOp, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertOmpWsloop (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP workshare loop into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpParallel (omp::ParallelOp opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts the OpenMP parallel operation to LLVM IR.
static llvm::omp::OrderKind convertOrderKind (std::optional< omp::ClauseOrderKind > o)
 Convert Order attribute to llvm::omp::OrderKind.
static LogicalResult convertOmpSimd (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP simd loop into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpLoopNest (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP loop nest into LLVM IR using OpenMPIRBuilder.
static LogicalResult convertOmpCanonicalLoopOp (omp::CanonicalLoopOp op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Convert an omp.canonical_loop to LLVM-IR.
static LogicalResult applyUnrollHeuristic (omp::UnrollHeuristicOp op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Apply a #pragma omp unroll / "!$omp unroll" transformation using the OpenMPIRBuilder.
static LogicalResult applyTile (omp::TileOp op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Apply a #pragma omp tile / !$omp tile transformation using the OpenMPIRBuilder.
static llvm::AtomicOrdering convertAtomicOrdering (std::optional< omp::ClauseMemoryOrderKind > ao)
 Convert an Atomic Ordering attribute to llvm::AtomicOrdering.
static LogicalResult convertOmpAtomicRead (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Convert omp.atomic.read operation to LLVM IR.
static LogicalResult convertOmpAtomicWrite (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an omp.atomic.write operation to LLVM IR.
static llvm::AtomicRMWInst::BinOp convertBinOpToAtomic (Operation &op)
 Converts an LLVM dialect binary operation to the corresponding enum value for atomicrmw supported binary operation.
static void extractAtomicControlFlags (omp::AtomicUpdateOp atomicUpdateOp, bool &isIgnoreDenormalMode, bool &isFineGrainedMemory, bool &isRemoteMemory)
static LogicalResult convertOmpAtomicUpdate (omp::AtomicUpdateOp &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP atomic update operation using OpenMPIRBuilder.
static LogicalResult convertOmpAtomicCapture (omp::AtomicCaptureOp atomicCaptureOp, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static llvm::omp::Directive convertCancellationConstructType (omp::ClauseCancellationConstructType directive)
static LogicalResult convertOmpCancel (omp::CancelOp op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertOmpCancellationPoint (omp::CancellationPointOp op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertOmpThreadprivate (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Converts an OpenMP Threadprivate operation into LLVM IR using OpenMPIRBuilder.
static llvm::OffloadEntriesInfoManager::OMPTargetDeviceClauseKind convertToDeviceClauseKind (mlir::omp::DeclareTargetDeviceType deviceClause)
static llvm::OffloadEntriesInfoManager::OMPTargetGlobalVarEntryKind convertToCaptureClauseKind (mlir::omp::DeclareTargetCaptureClause captureClause)
static llvm::SmallString< 64 > getDeclareTargetRefPtrSuffix (LLVM::GlobalOp globalOp, llvm::OpenMPIRBuilder &ompBuilder)
static bool isDeclareTargetLink (mlir::Value value)
static llvm::Value * getRefPtrIfDeclareTarget (mlir::Value value, LLVM::ModuleTranslation &moduleTranslation)
static uint64_t getArrayElementSizeInBits (LLVM::LLVMArrayType arrTy, DataLayout &dl)
static llvm::Value * getSizeInBytes (DataLayout &dl, const mlir::Type &type, Operation *clauseOp, llvm::Value *basePointer, llvm::Type *baseType, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static llvm::omp::OpenMPOffloadMappingFlags convertClauseMapFlags (omp::ClauseMapFlags mlirFlags)
static void collectMapDataFromMapOperands (MapInfoData &mapData, SmallVectorImpl< Value > &mapVars, LLVM::ModuleTranslation &moduleTranslation, DataLayout &dl, llvm::IRBuilderBase &builder, ArrayRef< Value > useDevPtrOperands={}, ArrayRef< Value > useDevAddrOperands={}, ArrayRef< Value > hasDevAddrOperands={})
static int getMapDataMemberIdx (MapInfoData &mapData, omp::MapInfoOp memberOp)
static omp::MapInfoOp getFirstOrLastMappedMemberPtr (omp::MapInfoOp mapInfo, bool first)
static std::vector< llvm::Value * > calculateBoundsOffset (LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase &builder, bool isArrayTy, OperandRange bounds)
 This function calculates the array/pointer offset for map data provided with bounds operations, e.g. when provided something like the following:
static llvm::omp::OpenMPOffloadMappingFlags mapParentWithMembers (LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase &builder, llvm::OpenMPIRBuilder &ompBuilder, DataLayout &dl, MapInfosTy &combinedInfo, MapInfoData &mapData, uint64_t mapDataIndex, bool isTargetParams)
static bool checkIfPointerMap (omp::MapInfoOp mapOp)
static void processMapMembersWithParent (LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase &builder, llvm::OpenMPIRBuilder &ompBuilder, DataLayout &dl, MapInfosTy &combinedInfo, MapInfoData &mapData, uint64_t mapDataIndex, llvm::omp::OpenMPOffloadMappingFlags memberOfFlag)
static void processIndividualMap (MapInfoData &mapData, size_t mapDataIdx, MapInfosTy &combinedInfo, bool isTargetParams, int mapDataParentIdx=-1)
static void processMapWithMembersOf (LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase &builder, llvm::OpenMPIRBuilder &ompBuilder, DataLayout &dl, MapInfosTy &combinedInfo, MapInfoData &mapData, uint64_t mapDataIndex, bool isTargetParams)
static void createAlteredByCaptureMap (MapInfoData &mapData, LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase &builder)
static void genMapInfos (llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, DataLayout &dl, MapInfosTy &combinedInfo, MapInfoData &mapData, bool isTargetParams=false)
static llvm::Expected< llvm::Function * > emitUserDefinedMapper (Operation *declMapperOp, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, llvm::StringRef mapperFuncName)
static llvm::Expected< llvm::Function * > getOrCreateUserDefinedMapperFunc (Operation *op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertOmpTargetData (Operation *op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertOmpDistribute (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertFlagsAttr (Operation *op, mlir::omp::FlagsAttr attribute, LLVM::ModuleTranslation &moduleTranslation)
 Lowers the FlagsAttr which is applied to the module on the device pass when offloading, this attribute contains OpenMP RTL globals that can be passed as flags to the frontend, otherwise they are set to default.
static void getTargetEntryUniqueInfo (llvm::TargetRegionEntryInfo &targetInfo, omp::TargetOp targetOp, llvm::StringRef parentName="")
static void handleDeclareTargetMapVar (MapInfoData &mapData, LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase &builder, llvm::Function *func)
static llvm::IRBuilderBase::InsertPoint createDeviceArgumentAccessor (MapInfoData &mapData, llvm::Argument &arg, llvm::Value *input, llvm::Value *&retVal, llvm::IRBuilderBase &builder, llvm::OpenMPIRBuilder &ompBuilder, LLVM::ModuleTranslation &moduleTranslation, llvm::IRBuilderBase::InsertPoint allocaIP, llvm::IRBuilderBase::InsertPoint codeGenIP)
static void extractHostEvalClauses (omp::TargetOp targetOp, Value &numThreads, Value &numTeamsLower, Value &numTeamsUpper, Value &threadLimit, llvm::SmallVectorImpl< Value > *lowerBounds=nullptr, llvm::SmallVectorImpl< Value > *upperBounds=nullptr, llvm::SmallVectorImpl< Value > *steps=nullptr)
 Follow uses of host_eval-defined block arguments of the given omp.target operation and populate output variables with their corresponding host value (i.e. operand evaluated outside of the target region), based on their uses inside of the target region.
template<typename OpTy>
static OpTy castOrGetParentOfType (Operation *op, bool immediateParent=false)
 If op is of the given type parameter, return it casted to that type. Otherwise, if its immediate parent operation (or some other higher-level parent, if immediateParent is false) is of that type, return that parent casted to the given type.
static std::optional< int64_textractConstInteger (Value value)
 If the given value is defined by an llvm.mlir.constant operation and it is of an integer type, return its value.
static uint64_t getTypeByteSize (mlir::Type type, const DataLayout &dl)
template<typename OpTy>
static uint64_t getReductionDataSize (OpTy &op)
static void initTargetDefaultAttrs (omp::TargetOp targetOp, Operation *capturedOp, llvm::OpenMPIRBuilder::TargetKernelDefaultAttrs &attrs, bool isTargetDevice, bool isGPU)
 Populate default MinTeams, MaxTeams and MaxThreads to their default values as stated by the corresponding clauses, if constant.
static void initTargetRuntimeAttrs (llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation, omp::TargetOp targetOp, Operation *capturedOp, llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs &attrs)
 Gather LLVM runtime values for all clauses evaluated in the host that are passed to the kernel invocation.
static LogicalResult convertOmpTarget (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertDeclareTargetAttr (Operation *op, mlir::omp::DeclareTargetAttr attribute, LLVM::ModuleTranslation &moduleTranslation)
static bool isTargetDeviceOp (Operation *op)
static llvm::Function * getOmpTargetAlloc (llvm::IRBuilderBase &builder, llvm::Module *llvmModule)
static LogicalResult convertTargetAllocMemOp (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static llvm::Function * getOmpTargetFree (llvm::IRBuilderBase &builder, llvm::Module *llvmModule)
static LogicalResult convertTargetFreeMemOp (Operation &opInst, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertHostOrTargetOperation (Operation *op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
 Given an OpenMP MLIR operation, create the corresponding LLVM IR (including OpenMP runtime calls).
static LogicalResult convertTargetDeviceOp (Operation *op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)
static LogicalResult convertTargetOpsInNest (Operation *op, llvm::IRBuilderBase &builder, LLVM::ModuleTranslation &moduleTranslation)

Variables

 owningReductionGens [i]
 nullptr
 atomicGen

Function Documentation

◆ allocAndInitializeReductionVars()

template<typename OP>
LogicalResult allocAndInitializeReductionVars ( OP op,
ArrayRef< BlockArgument > reductionArgs,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
llvm::OpenMPIRBuilder::InsertPointTy & allocaIP,
SmallVectorImpl< omp::DeclareReductionOp > & reductionDecls,
SmallVectorImpl< llvm::Value * > & privateReductionVariables,
DenseMap< Value, llvm::Value * > & reductionVariableMap,
llvm::ArrayRef< bool > isByRef )
static

◆ allocatePrivateVars()

llvm::Expected< llvm::BasicBlock * > allocatePrivateVars ( llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
PrivateVarsInfo & privateVarsInfo,
const llvm::OpenMPIRBuilder::InsertPointTy & allocaIP,
llvm::DenseMap< Value, Value > * mappedPrivateVars = nullptr )
static

Allocate and initialize delayed private variables. Returns the basic block which comes after all of these allocations. llvm::Value * for each of these private variables are populated in llvmPrivateVars.

Definition at line 1584 of file OpenMPToLLVMIRTranslation.cpp.

References PrivateVarsInfo::blockArgs, mlir::LLVM::ModuleTranslation::convertType(), mlir::LLVM::ModuleTranslation::getLLVMModule(), PrivateVarsInfo::llvmVars, PrivateVarsInfo::mlirVars, and PrivateVarsInfo::privatizers.

Referenced by convertOmpDistribute(), convertOmpParallel(), convertOmpSimd(), convertOmpTarget(), and convertOmpWsloop().

◆ allocReductionVars()

template<typename T>
LogicalResult allocReductionVars ( T loop,
ArrayRef< BlockArgument > reductionArgs,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
const llvm::OpenMPIRBuilder::InsertPointTy & allocaIP,
SmallVectorImpl< omp::DeclareReductionOp > & reductionDecls,
SmallVectorImpl< llvm::Value * > & privateReductionVariables,
DenseMap< Value, llvm::Value * > & reductionVariableMap,
SmallVectorImpl< DeferredStore > & deferredStores,
llvm::ArrayRef< bool > isByRefs )
static

Allocate space for privatized reduction variables.

deferredStores contains information to create store operations which needs to be inserted after all allocas

Definition at line 1104 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::LLVM::ModuleTranslation::convertType(), mlir::Region::empty(), inlineConvertOmpRegions(), mlir::LLVM::ModuleTranslation::mapValue(), and success().

Referenced by allocAndInitializeReductionVars(), convertOmpParallel(), convertOmpSimd(), and convertOmpWsloop().

◆ applyTile()

LogicalResult applyTile ( omp::TileOp op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ applyUnrollHeuristic()

LogicalResult applyUnrollHeuristic ( omp::UnrollHeuristicOp op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

Apply a #pragma omp unroll / "!$omp unroll" transformation using the OpenMPIRBuilder.

Definition at line 3155 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::LLVM::ModuleTranslation::getOpenMPBuilder(), mlir::LLVM::ModuleTranslation::invalidateOmpLoop(), mlir::LLVM::ModuleTranslation::lookupOMPLoop(), and success().

Referenced by convertHostOrTargetOperation().

◆ buildDependData()

void buildDependData ( std::optional< ArrayAttr > dependKinds,
OperandRange dependVars,
LLVM::ModuleTranslation & moduleTranslation,
SmallVectorImpl< llvm::OpenMPIRBuilder::DependData > & dds )
static

◆ calculateBoundsOffset()

std::vector< llvm::Value * > calculateBoundsOffset ( LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase & builder,
bool isArrayTy,
OperandRange bounds )
static

This function calculates the array/pointer offset for map data provided with bounds operations, e.g. when provided something like the following:

Fortran map(tofrom: array(2:5, 3:2))

We must calculate the initial pointer offset to pass across, this function performs this using bounds.

TODO/WARNING: This only supports Fortran's column major indexing currently as is noted in the note below and comments in the function, we must extend this function when we add a C++ frontend. NOTE: which while specified in row-major order it currently needs to be flipped for Fortran's column order array allocation and access (as opposed to C++'s row-major, hence the backwards processing where order is important). This is likely important to keep in mind for the future when we incorporate a C++ frontend, both frontends will need to agree on the ordering of generated bounds operations (one may have to flip them) to make the below lowering frontend agnostic. The offload size calcualtion may also have to be adjusted for C++.

Definition at line 4103 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::LLVM::ModuleTranslation::lookupValue().

Referenced by createAlteredByCaptureMap().

◆ castOrGetParentOfType()

template<typename OpTy>
OpTy castOrGetParentOfType ( Operation * op,
bool immediateParent = false )
static

If op is of the given type parameter, return it casted to that type. Otherwise, if its immediate parent operation (or some other higher-level parent, if immediateParent is false) is of that type, return that parent casted to the given type.

If op is null or neither it or its parent(s) are of the specified type, return a null operation.

Definition at line 5349 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::Operation::getParentOfType(), and mlir::Operation::getParentOp().

Referenced by initTargetDefaultAttrs(), and initTargetRuntimeAttrs().

◆ checkIfPointerMap()

bool checkIfPointerMap ( omp::MapInfoOp mapOp)
static

◆ checkImplementationStatus()

◆ cleanupPrivateVars()

LogicalResult cleanupPrivateVars ( llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
Location loc,
SmallVectorImpl< llvm::Value * > & llvmPrivateVars,
SmallVectorImpl< omp::PrivateClauseOp > & privateDecls )
static

◆ collectMapDataFromMapOperands()

void collectMapDataFromMapOperands ( MapInfoData & mapData,
SmallVectorImpl< Value > & mapVars,
LLVM::ModuleTranslation & moduleTranslation,
DataLayout & dl,
llvm::IRBuilderBase & builder,
ArrayRef< Value > useDevPtrOperands = {},
ArrayRef< Value > useDevAddrOperands = {},
ArrayRef< Value > hasDevAddrOperands = {} )
static

◆ collectReductionDecls()

template<typename T>
void collectReductionDecls ( T op,
SmallVectorImpl< omp::DeclareReductionOp > & reductions )
static

Populates reductions with reduction declarations used in the given op.

Definition at line 863 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::SymbolTable::lookupNearestSymbolFrom().

Referenced by convertOmpDistribute(), convertOmpParallel(), convertOmpSections(), convertOmpSimd(), convertOmpTeams(), convertOmpWsloop(), and getReductionDataSize().

◆ constructIsCancellable()

bool constructIsCancellable ( Operation * op)
static

Returns true if the construct contains omp.cancel or omp.cancellation_point.

Definition at line 1717 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::WalkResult::advance(), mlir::WalkResult::interrupt(), and mlir::Operation::walk().

Referenced by convertOmpParallel(), convertOmpSections(), and pushCancelFinalizationCB().

◆ convertAtomicOrdering()

llvm::AtomicOrdering convertAtomicOrdering ( std::optional< omp::ClauseMemoryOrderKind > ao)
static

Convert an Atomic Ordering attribute to llvm::AtomicOrdering.

Definition at line 3212 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertOmpAtomicCapture(), convertOmpAtomicRead(), convertOmpAtomicUpdate(), and convertOmpAtomicWrite().

◆ convertBinOpToAtomic()

llvm::AtomicRMWInst::BinOp convertBinOpToAtomic ( Operation & op)
static

Converts an LLVM dialect binary operation to the corresponding enum value for atomicrmw supported binary operation.

Definition at line 3284 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertOmpAtomicCapture(), and convertOmpAtomicUpdate().

◆ convertCancellationConstructType()

llvm::omp::Directive convertCancellationConstructType ( omp::ClauseCancellationConstructType directive)
static

◆ convertClauseMapFlags()

llvm::omp::OpenMPOffloadMappingFlags convertClauseMapFlags ( omp::ClauseMapFlags mlirFlags)
static

Definition at line 3834 of file OpenMPToLLVMIRTranslation.cpp.

References mapTypeToBool().

Referenced by processMapMembersWithParent().

◆ convertDeclareTargetAttr()

◆ convertFlagsAttr()

LogicalResult convertFlagsAttr ( Operation * op,
mlir::omp::FlagsAttr attribute,
LLVM::ModuleTranslation & moduleTranslation )
static

Lowers the FlagsAttr which is applied to the module on the device pass when offloading, this attribute contains OpenMP RTL globals that can be passed as flags to the frontend, otherwise they are set to default.

Definition at line 5039 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::LLVM::ModuleTranslation::getOpenMPBuilder(), and success().

◆ convertHostOrTargetOperation()

◆ convertOmpAtomicCapture()

◆ convertOmpAtomicRead()

LogicalResult convertOmpAtomicRead ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpAtomicUpdate()

◆ convertOmpAtomicWrite()

LogicalResult convertOmpAtomicWrite ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpCancel()

LogicalResult convertOmpCancel ( omp::CancelOp op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpCancellationPoint()

LogicalResult convertOmpCancellationPoint ( omp::CancellationPointOp op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpCanonicalLoopOp()

LogicalResult convertOmpCanonicalLoopOp ( omp::CanonicalLoopOp op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpCritical()

LogicalResult convertOmpCritical ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpDistribute()

◆ convertOmpLoopNest()

LogicalResult convertOmpLoopNest ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpMasked()

LogicalResult convertOmpMasked ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpMaster()

LogicalResult convertOmpMaster ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

Converts an OpenMP 'master' operation into LLVM IR using OpenMPIRBuilder.

Definition at line 741 of file OpenMPToLLVMIRTranslation.cpp.

References checkImplementationStatus(), convertOmpOpRegions(), mlir::LLVM::ModuleTranslation::getOpenMPBuilder(), handleError(), and success().

Referenced by convertHostOrTargetOperation().

◆ convertOmpOpRegions()

llvm::Expected< llvm::BasicBlock * > convertOmpOpRegions ( Region & region,
StringRef blockName,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
SmallVectorImpl< llvm::PHINode * > * continuationBlockPHIs = nullptr )
static

◆ convertOmpOrdered()

LogicalResult convertOmpOrdered ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpOrderedRegion()

LogicalResult convertOmpOrderedRegion ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

Converts an OpenMP 'ordered_region' operation into LLVM IR using OpenMPIRBuilder.

Definition at line 1055 of file OpenMPToLLVMIRTranslation.cpp.

References checkImplementationStatus(), convertOmpOpRegions(), mlir::LLVM::ModuleTranslation::getOpenMPBuilder(), handleError(), and success().

Referenced by convertHostOrTargetOperation().

◆ convertOmpParallel()

◆ convertOmpSections()

◆ convertOmpSimd()

◆ convertOmpSingle()

LogicalResult convertOmpSingle ( omp::SingleOp & singleOp,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpTarget()

◆ convertOmpTargetData()

◆ convertOmpTaskgroupOp()

LogicalResult convertOmpTaskgroupOp ( omp::TaskgroupOp tgOp,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

Converts an OpenMP taskgroup construct into LLVM IR using OpenMPIRBuilder.

Definition at line 2424 of file OpenMPToLLVMIRTranslation.cpp.

References checkImplementationStatus(), convertOmpOpRegions(), findAllocaInsertPoint(), mlir::LLVM::ModuleTranslation::getOpenMPBuilder(), handleError(), and success().

Referenced by convertHostOrTargetOperation().

◆ convertOmpTaskOp()

◆ convertOmpTaskwaitOp()

LogicalResult convertOmpTaskwaitOp ( omp::TaskwaitOp twOp,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertOmpTeams()

◆ convertOmpThreadprivate()

◆ convertOmpWsloop()

◆ convertOrderKind()

llvm::omp::OrderKind convertOrderKind ( std::optional< omp::ClauseOrderKind > o)
static

Convert Order attribute to llvm::omp::OrderKind.

Definition at line 2825 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertOmpSimd().

◆ convertTargetAllocMemOp()

◆ convertTargetDeviceOp()

LogicalResult convertTargetDeviceOp ( Operation * op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

Definition at line 6309 of file OpenMPToLLVMIRTranslation.cpp.

References convertHostOrTargetOperation().

◆ convertTargetFreeMemOp()

LogicalResult convertTargetFreeMemOp ( Operation & opInst,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ convertTargetOpsInNest()

◆ convertToCaptureClauseKind()

llvm::OffloadEntriesInfoManager::OMPTargetGlobalVarEntryKind convertToCaptureClauseKind ( mlir::omp::DeclareTargetCaptureClause captureClause)
static

Definition at line 3626 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertDeclareTargetAttr().

◆ convertToDeviceClauseKind()

llvm::OffloadEntriesInfoManager::OMPTargetDeviceClauseKind convertToDeviceClauseKind ( mlir::omp::DeclareTargetDeviceType deviceClause)
static

Definition at line 3610 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertDeclareTargetAttr().

◆ copyFirstPrivateVars()

LogicalResult copyFirstPrivateVars ( mlir::Operation * op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
SmallVectorImpl< mlir::Value > & mlirPrivateVars,
ArrayRef< llvm::Value * > llvmPrivateVars,
SmallVectorImpl< omp::PrivateClauseOp > & privateDecls,
bool insertBarrier,
llvm::DenseMap< Value, Value > * mappedPrivateVars = nullptr )
static

◆ createAlteredByCaptureMap()

void createAlteredByCaptureMap ( MapInfoData & mapData,
LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase & builder )
static

◆ createDeviceArgumentAccessor()

llvm::IRBuilderBase::InsertPoint createDeviceArgumentAccessor ( MapInfoData & mapData,
llvm::Argument & arg,
llvm::Value * input,
llvm::Value *& retVal,
llvm::IRBuilderBase & builder,
llvm::OpenMPIRBuilder & ompBuilder,
LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase::InsertPoint allocaIP,
llvm::IRBuilderBase::InsertPoint codeGenIP )
static

◆ createReductionsAndCleanup()

template<class OP>
LogicalResult createReductionsAndCleanup ( OP op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
llvm::OpenMPIRBuilder::InsertPointTy & allocaIP,
SmallVectorImpl< omp::DeclareReductionOp > & reductionDecls,
ArrayRef< llvm::Value * > privateReductionVariables,
ArrayRef< bool > isByRef,
bool isNowait = false,
bool isTeamsReduction = false )
static

◆ emitUserDefinedMapper()

◆ extractAtomicControlFlags()

void extractAtomicControlFlags ( omp::AtomicUpdateOp atomicUpdateOp,
bool & isIgnoreDenormalMode,
bool & isFineGrainedMemory,
bool & isRemoteMemory )
static

◆ extractConstInteger()

std::optional< int64_t > extractConstInteger ( Value value)
static

If the given value is defined by an llvm.mlir.constant operation and it is of an integer type, return its value.

Definition at line 5364 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::Value::getDefiningOp().

Referenced by initTargetDefaultAttrs().

◆ extractHostEvalClauses()

void extractHostEvalClauses ( omp::TargetOp targetOp,
Value & numThreads,
Value & numTeamsLower,
Value & numTeamsUpper,
Value & threadLimit,
llvm::SmallVectorImpl< Value > * lowerBounds = nullptr,
llvm::SmallVectorImpl< Value > * upperBounds = nullptr,
llvm::SmallVectorImpl< Value > * steps = nullptr )
static

Follow uses of host_eval-defined block arguments of the given omp.target operation and populate output variables with their corresponding host value (i.e. operand evaluated outside of the target region), based on their uses inside of the target region.

Loop bounds and steps are only optionally populated, if output vectors are provided.

Definition at line 5285 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::Operation::getUsers().

Referenced by initTargetDefaultAttrs(), and initTargetRuntimeAttrs().

◆ findAllocaInsertPoint()

llvm::OpenMPIRBuilder::InsertPointTy findAllocaInsertPoint ( llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ findAssociatedValue()

llvm::Value * findAssociatedValue ( Value privateVar,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
llvm::DenseMap< Value, Value > * mappedPrivateVars = nullptr )
static

Return the llvm::Value * corresponding to the privateVar that is being privatized. It isn't always as simple as looking up moduleTranslation with privateVar. For instance, in case of an allocatable, the descriptor for the allocatable is privatized. This descriptor is mapped using an MapInfoOp. So, this function will return a pointer to the llvm::Value corresponding to the block argument for the mapped descriptor.

Definition at line 1482 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::LLVM::ModuleTranslation::convertType(), mlir::Value::getType(), and mlir::LLVM::ModuleTranslation::lookupValue().

Referenced by copyFirstPrivateVars(), and initPrivateVar().

◆ findCurrentLoopInfo()

llvm::CanonicalLoopInfo * findCurrentLoopInfo ( LLVM::ModuleTranslation & moduleTranslation)
static

Find the loop information structure for the loop nest being translated.

It will return a null value unless called from the translation function for a loop wrapper operation after successfully translating its body.

Definition at line 528 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::WalkResult::interrupt(), and mlir::LLVM::ModuleTranslation::stackWalk().

Referenced by convertOmpDistribute(), convertOmpSimd(), and convertOmpWsloop().

◆ findPrivatizer()

omp::PrivateClauseOp findPrivatizer ( Operation * from,
SymbolRefAttr symbolName )
static

Looks up from the operation from and returns the PrivateClauseOp with name symbolName.

Definition at line 268 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::SymbolTable::lookupNearestSymbolFrom().

Referenced by convertOmpTarget().

◆ forwardArgs()

void forwardArgs ( LLVM::ModuleTranslation & moduleTranslation,
omp::BlockArgOpenMPOpInterface blockArgIface )
static

Maps block arguments from blockArgIface (which are MLIR values) to the corresponding LLVM values of the interface's operands.

This is useful when an OpenMP region with entry block arguments is converted to LLVM. In this case the block arguments are (part of) of the OpenMP region's entry arguments and the operands are (part of) of the operands to the OpenMP op containing the region.

Definition at line 687 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::LLVM::ModuleTranslation::lookupValue(), and mlir::LLVM::ModuleTranslation::mapValue().

Referenced by convertTargetOpsInNest().

◆ genMapInfos()

void genMapInfos ( llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
DataLayout & dl,
MapInfosTy & combinedInfo,
MapInfoData & mapData,
bool isTargetParams = false )
static

◆ getArrayElementSizeInBits()

uint64_t getArrayElementSizeInBits ( LLVM::LLVMArrayType arrTy,
DataLayout & dl )
static

◆ getDeclareTargetRefPtrSuffix()

llvm::SmallString< 64 > getDeclareTargetRefPtrSuffix ( LLVM::GlobalOp globalOp,
llvm::OpenMPIRBuilder & ompBuilder )
static

Definition at line 3640 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::SymbolTable::Private.

Referenced by getRefPtrIfDeclareTarget().

◆ getFirstOrLastMappedMemberPtr()

omp::MapInfoOp getFirstOrLastMappedMemberPtr ( omp::MapInfoOp mapInfo,
bool first )
static

Definition at line 4045 of file OpenMPToLLVMIRTranslation.cpp.

References ArrayAttr(), b, and indices.

Referenced by mapParentWithMembers().

◆ getIsByRef()

ArrayRef< bool > getIsByRef ( std::optional< ArrayRef< bool > > attr)
static

◆ getMapDataMemberIdx()

int getMapDataMemberIdx ( MapInfoData & mapData,
omp::MapInfoOp memberOp )
static

◆ getOmpTargetAlloc()

llvm::Function * getOmpTargetAlloc ( llvm::IRBuilderBase & builder,
llvm::Module * llvmModule )
static

Definition at line 6035 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertTargetAllocMemOp().

◆ getOmpTargetFree()

llvm::Function * getOmpTargetFree ( llvm::IRBuilderBase & builder,
llvm::Module * llvmModule )
static

Definition at line 6079 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertTargetFreeMemOp().

◆ getOrCreateUserDefinedMapperFunc()

llvm::Expected< llvm::Function * > getOrCreateUserDefinedMapperFunc ( Operation * op,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ getProcBindKind()

llvm::omp::ProcBindKind getProcBindKind ( omp::ClauseProcBindKind kind)
static

Convert ProcBindKind from MLIR-generated enum to LLVM enum.

Definition at line 667 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertOmpParallel().

◆ getReductionDataSize()

template<typename OpTy>
uint64_t getReductionDataSize ( OpTy & op)
static

◆ getRefPtrIfDeclareTarget()

◆ getSizeInBytes()

llvm::Value * getSizeInBytes ( DataLayout & dl,
const mlir::Type & type,
Operation * clauseOp,
llvm::Value * basePointer,
llvm::Type * baseType,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

◆ getTargetEntryUniqueInfo()

void getTargetEntryUniqueInfo ( llvm::TargetRegionEntryInfo & targetInfo,
omp::TargetOp targetOp,
llvm::StringRef parentName = "" )
static

Definition at line 5076 of file OpenMPToLLVMIRTranslation.cpp.

References getDevice().

Referenced by convertOmpTarget().

◆ getTypeByteSize()

uint64_t getTypeByteSize ( mlir::Type type,
const DataLayout & dl )
static

◆ handleDeclareTargetMapVar()

void handleDeclareTargetMapVar ( MapInfoData & mapData,
LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase & builder,
llvm::Function * func )
static

◆ handleError() [1/2]

◆ handleError() [2/2]

template<typename T>
LogicalResult handleError ( llvm::Expected< T > & result,
Operation & op )
static

Definition at line 470 of file OpenMPToLLVMIRTranslation.cpp.

References handleError(), result, and success().

◆ initPrivateVar()

llvm::Expected< llvm::Value * > initPrivateVar ( llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
omp::PrivateClauseOp & privDecl,
Value mlirPrivVar,
BlockArgument & blockArg,
llvm::Value * llvmPrivateVar,
llvm::BasicBlock * privInitBlock,
llvm::DenseMap< Value, Value > * mappedPrivateVars = nullptr )
static

Initialize a single (first)private variable. You probably want to use allocateAndInitPrivateVars instead of this. This returns the private variable which has been initialized. This variable should be mapped before constructing the body of the Op.

Definition at line 1513 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::Region::empty(), findAssociatedValue(), mlir::LLVM::ModuleTranslation::forgetMapping(), inlineConvertOmpRegions(), and mlir::LLVM::ModuleTranslation::mapValue().

Referenced by convertOmpTaskOp(), and initPrivateVars().

◆ initPrivateVars()

◆ initTargetDefaultAttrs()

void initTargetDefaultAttrs ( omp::TargetOp targetOp,
Operation * capturedOp,
llvm::OpenMPIRBuilder::TargetKernelDefaultAttrs & attrs,
bool isTargetDevice,
bool isGPU )
static

Populate default MinTeams, MaxTeams and MaxThreads to their default values as stated by the corresponding clauses, if constant.

These default values must be set before the creation of the outlined LLVM function for the target region, so that they can be used to initialize the corresponding global ConfigurationEnvironmentTy structure.

Definition at line 5407 of file OpenMPToLLVMIRTranslation.cpp.

References castOrGetParentOfType(), extractConstInteger(), extractHostEvalClauses(), getReductionDataSize(), and result.

Referenced by convertOmpTarget().

◆ initTargetRuntimeAttrs()

void initTargetRuntimeAttrs ( llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
omp::TargetOp targetOp,
Operation * capturedOp,
llvm::OpenMPIRBuilder::TargetKernelRuntimeAttrs & attrs )
static

Gather LLVM runtime values for all clauses evaluated in the host that are passed to the kernel invocation.

This function must be called only when compiling for the host. Also, it will only provide correct results if it's called after the body of targetOp has been fully generated.

Definition at line 5533 of file OpenMPToLLVMIRTranslation.cpp.

References castOrGetParentOfType(), extractHostEvalClauses(), mlir::LLVM::ModuleTranslation::getOpenMPBuilder(), and mlir::LLVM::ModuleTranslation::lookupValue().

Referenced by convertOmpTarget().

◆ inlineConvertOmpRegions()

LogicalResult inlineConvertOmpRegions ( Region & region,
StringRef blockName,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation,
SmallVectorImpl< llvm::Value * > * continuationBlockArgs = nullptr )
static

Translates the blocks contained in the given region and appends them to at the current insertion point of builder.

The operations of the entry block are appended to the current insertion block. If set, continuationBlockArgs is populated with translated values that correspond to the values omp.yield'ed from the region.

Definition at line 882 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::Block::back(), mlir::LLVM::ModuleTranslation::convertBlock(), convertOmpOpRegions(), mlir::Region::empty(), mlir::LLVM::ModuleTranslation::forgetMapping(), mlir::Region::front(), mlir::Operation::getOperands(), mlir::Region::getParentOp(), handleError(), mlir::Region::hasOneBlock(), mlir::LLVM::ModuleTranslation::lookupValues(), mlir::LLVM::ModuleTranslation::mapBlock(), nullptr, and success().

Referenced by allocReductionVars(), convertOmpTargetData(), copyFirstPrivateVars(), initPrivateVar(), inlineOmpRegionCleanup(), makeAtomicReductionGen(), and makeReductionGen().

◆ inlineOmpRegionCleanup()

LogicalResult inlineOmpRegionCleanup ( llvm::SmallVectorImpl< Region * > & cleanupRegions,
llvm::ArrayRef< llvm::Value * > privateVariables,
LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase & builder,
StringRef regionName,
bool shouldLoadCleanupRegionArg = true )
static

◆ isDeclareTargetLink()

bool isDeclareTargetLink ( mlir::Value value)
static

Definition at line 3661 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::Value::getDefiningOp().

Referenced by checkIfPointerMap().

◆ isTargetDeviceOp()

bool isTargetDeviceOp ( Operation * op)
static

◆ makeAtomicReductionGen()

OwningAtomicReductionGen makeAtomicReductionGen ( omp::DeclareReductionOp decl,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

Create an OpenMPIRBuilder-compatible atomic reduction generator for the given reduction declaration.

The generator uses builder but ignores its insertion point. Returns null if there is no atomic region available in the reduction declaration.

Definition at line 992 of file OpenMPToLLVMIRTranslation.cpp.

References atomicGen, inlineConvertOmpRegions(), lhs, mlir::LLVM::ModuleTranslation::mapValue(), and rhs.

◆ makeReductionGen()

OwningReductionGen makeReductionGen ( omp::DeclareReductionOp decl,
llvm::IRBuilderBase & builder,
LLVM::ModuleTranslation & moduleTranslation )
static

Create an OpenMPIRBuilder-compatible reduction generator for the given reduction declaration.

The generator uses builder but ignores its insertion point.

Definition at line 962 of file OpenMPToLLVMIRTranslation.cpp.

References inlineConvertOmpRegions(), lhs, mlir::LLVM::ModuleTranslation::mapValue(), result, and rhs.

Referenced by convertOmpSimd().

◆ mapInitializationArgs()

template<typename T>
void mapInitializationArgs ( T loop,
LLVM::ModuleTranslation & moduleTranslation,
SmallVectorImpl< omp::DeclareReductionOp > & reductionDecls,
DenseMap< Value, llvm::Value * > & reductionVariableMap,
unsigned i )
static

◆ mapParentWithMembers()

llvm::omp::OpenMPOffloadMappingFlags mapParentWithMembers ( LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase & builder,
llvm::OpenMPIRBuilder & ompBuilder,
DataLayout & dl,
MapInfosTy & combinedInfo,
MapInfoData & mapData,
uint64_t mapDataIndex,
bool isTargetParams )
static

◆ popCancelFinalizationCB()

void popCancelFinalizationCB ( const ArrayRef< llvm::BranchInst * > cancelTerminators,
llvm::OpenMPIRBuilder & ompBuilder,
const llvm::OpenMPIRBuilder::InsertPointTy & afterIP )
static

If we cancelled the construct, we should branch to the finalization block of that construct. OMPIRBuilder structures the CFG such that the cleanup block is immediately before the continuation block. Now this finalization has been created we can fix the branch.

Definition at line 2065 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by convertOmpTaskOp(), and convertOmpWsloop().

◆ processIndividualMap()

void processIndividualMap ( MapInfoData & mapData,
size_t mapDataIdx,
MapInfosTy & combinedInfo,
bool isTargetParams,
int mapDataParentIdx = -1 )
static

Definition at line 4377 of file OpenMPToLLVMIRTranslation.cpp.

References checkIfPointerMap().

Referenced by genMapInfos(), and processMapWithMembersOf().

◆ processMapMembersWithParent()

void processMapMembersWithParent ( LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase & builder,
llvm::OpenMPIRBuilder & ompBuilder,
DataLayout & dl,
MapInfosTy & combinedInfo,
MapInfoData & mapData,
uint64_t mapDataIndex,
llvm::omp::OpenMPOffloadMappingFlags memberOfFlag )
static

◆ processMapWithMembersOf()

void processMapWithMembersOf ( LLVM::ModuleTranslation & moduleTranslation,
llvm::IRBuilderBase & builder,
llvm::OpenMPIRBuilder & ompBuilder,
DataLayout & dl,
MapInfosTy & combinedInfo,
MapInfoData & mapData,
uint64_t mapDataIndex,
bool isTargetParams )
static

◆ pushCancelFinalizationCB()

void pushCancelFinalizationCB ( SmallVectorImpl< llvm::BranchInst * > & cancelTerminators,
llvm::IRBuilderBase & llvmBuilder,
llvm::OpenMPIRBuilder & ompBuilder,
mlir::Operation * op,
llvm::omp::Directive cancelDirective )
static

Shared implementation of a callback which adds a termiator for the new block created for the branch taken when an openmp construct is cancelled. The terminator is saved in cancelTerminators. This callback is invoked only if there is cancellation inside of the taskgroup body. The terminator will need to be fixed to branch to the correct block to cleanup the construct.

Definition at line 2035 of file OpenMPToLLVMIRTranslation.cpp.

References constructIsCancellable().

Referenced by convertOmpTaskOp(), and convertOmpWsloop().

◆ setInsertPointForPossiblyEmptyBlock()

void setInsertPointForPossiblyEmptyBlock ( llvm::IRBuilderBase & builder,
llvm::BasicBlock * block = nullptr )
static

◆ teamsReductionContainedInDistribute()

bool teamsReductionContainedInDistribute ( omp::TeamsOp teamsOp)
static

Definition at line 1879 of file OpenMPToLLVMIRTranslation.cpp.

References mlir::Operation::erase().

Referenced by convertOmpDistribute(), and convertOmpTeams().

Variable Documentation

◆ atomicGen

atomicGen

Definition at line 1337 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by makeAtomicReductionGen().

◆ nullptr

nullptr

Definition at line 1337 of file OpenMPToLLVMIRTranslation.cpp.

Referenced by mlir::detail::AttrOpBinder< AttrT >::AttrOpBinder(), mlir::spirv::BlockMergeInfo::BlockMergeInfo(), mlir::BlockRange::BlockRange(), mlir::CallGraph::CallGraph(), mlir::sparse_tensor::CodegenEnv::CodegenEnv(), mlir::detail::constant_op_binder< AttrT >::constant_op_binder(), mlir::AsmParser::CyclicParseReset::CyclicParseReset(), mlir::AsmPrinter::CyclicPrintReset::CyclicPrintReset(), DynamicMemRefType< T >::DynamicMemRefType(), DynamicMemRefType< T >::DynamicMemRefType(), mlir::python::PyMlirContext::ErrorCapture::ErrorCapture(), executeGetOperandsResults(), mlir::ExecutionEngine::ExecutionEngine(), findPayloadRoot(), mlir::AsmParserState::getAttributeAliasDef(), mlir::AsmParserState::getBlockDef(), mlir::detail::DialectInterfaceCollectionBase::getInterfaceFor(), mlir::Liveness::getLiveness(), mlir::PassExecutionAction::getOp(), mlir::AsmParserState::getOpDef(), mlir::AsmParserState::getTypeAliasDef(), inlineConvertOmpRegions(), inlineOmpRegionCleanup(), mlir::detail::Interface< ConcreteType, Attribute, Traits, Attribute, AttributeTrait::TraitBase >::Interface(), mlir::detail::Interface< ConcreteType, Attribute, Traits, Attribute, AttributeTrait::TraitBase >::Interface(), mlir::detail::Interface< ConcreteType, Attribute, Traits, Attribute, AttributeTrait::TraitBase >::Interface(), mlir::detail::Interface< ConcreteType, Attribute, Traits, Attribute, AttributeTrait::TraitBase >::Interface(), mlir::Lexer::Lexer(), mlir::pdll::Lexer::Lexer(), mlir::sparse_tensor::LoopEmitter::locateLvlAtAffineAddress(), mlir::PassInfo::lookup(), mlir::PassPipelineInfo::lookup(), mlir::pdll::ods::Context::lookupDialect(), mlir::CallGraph::lookupNode(), makeArrayAttr(), makeDenseBoolArrayAttr(), makeDenseI64ArrayAttr(), mlir::affine::NestedPattern::NestedPattern(), mlir::Op< ConcreteType, Traits >::Op(), mlir::Op< ConcreteType, Traits >::Op(), mlir::OwningOpRef< OpTy >::OwningOpRef(), mlir::Pattern::Pattern(), mlir::ReductionNode::ReductionNode(), mlir::ShapedTypeComponents::ShapedTypeComponents(), mlir::ShapedTypeComponents::ShapedTypeComponents(), mlir::ShapedTypeComponents::ShapedTypeComponents(), mlir::ShapedTypeComponents::ShapedTypeComponents(), mlir::sparse_tensor::SparseIterator::SparseIterator(), mlir::detail::StringAttrStorage::StringAttrStorage(), mlir::spirv::detail::StructTypeStorage::StructTypeStorage(), mlir::pdl_to_pdl_interp::SuccessNode::SuccessNode(), and mlir::SuccessorRange::SuccessorRange().

◆ owningReductionGens

owningReductionGens[i]