|
MLIR 23.0.0git
|
#include "mlir/Dialect/OpenMP/OpenMPDialect.h"#include "mlir/Conversion/ConvertToLLVM/ToLLVMInterface.h"#include "mlir/Dialect/Func/IR/FuncOps.h"#include "mlir/Dialect/LLVMIR/LLVMTypes.h"#include "mlir/Dialect/OpenMP/OpenMPClauseOperands.h"#include "mlir/IR/Attributes.h"#include "mlir/IR/BuiltinAttributes.h"#include "mlir/IR/DialectImplementation.h"#include "mlir/IR/Matchers.h"#include "mlir/IR/OpImplementation.h"#include "mlir/IR/OperationSupport.h"#include "mlir/IR/SymbolTable.h"#include "mlir/Interfaces/FoldInterfaces.h"#include "llvm/ADT/ArrayRef.h"#include "llvm/ADT/PostOrderIterator.h"#include "llvm/ADT/STLExtras.h"#include "llvm/ADT/STLForwardCompat.h"#include "llvm/ADT/SmallString.h"#include "llvm/ADT/StringExtras.h"#include "llvm/ADT/StringRef.h"#include "llvm/ADT/TypeSwitch.h"#include "llvm/ADT/bit.h"#include "llvm/Support/InterleavedRange.h"#include <cstddef>#include <iterator>#include <optional>#include <variant>#include "mlir/Dialect/OpenMP/OpenMPOpsDialect.cpp.inc"#include "mlir/Dialect/OpenMP/OpenMPOpsEnums.cpp.inc"#include "mlir/Dialect/OpenMP/OpenMPOpsInterfaces.cpp.inc"#include "mlir/Dialect/OpenMP/OpenMPTypeInterfaces.cpp.inc"#include "mlir/Dialect/OpenMP/OpenMPOps.cpp.inc"#include "mlir/Dialect/OpenMP/OpenMPOpsAttributes.cpp.inc"#include "mlir/Dialect/OpenMP/OpenMPOpsTypes.cpp.inc"Go to the source code of this file.
Macros | |
| #define | GET_OP_LIST |
| #define | GET_ATTRDEF_LIST |
| #define | GET_TYPEDEF_LIST |
| #define | GET_ATTRDEF_CLASSES |
| #define | GET_OP_CLASSES |
| #define | GET_TYPEDEF_CLASSES |
Functions | |
| static ArrayAttr | makeArrayAttr (MLIRContext *context, llvm::ArrayRef< Attribute > attrs) |
| static DenseBoolArrayAttr | makeDenseBoolArrayAttr (MLIRContext *ctx, const ArrayRef< bool > boolArray) |
| static DenseI64ArrayAttr | makeDenseI64ArrayAttr (MLIRContext *ctx, const ArrayRef< int64_t > intArray) |
| static std::string | generateLoopNestingName (StringRef prefix, CanonicalLoopOp op) |
| Generate a name of a canonical loop nest of the format <prefix>(_r<idx>_s<idx>)*. | |
| static ParseResult | parseAllocateAndAllocator (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &allocateVars, SmallVectorImpl< Type > &allocateTypes, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &allocatorVars, SmallVectorImpl< Type > &allocatorTypes) |
| Parse an allocate clause with allocators and a list of operands with types. | |
| static void | printAllocateAndAllocator (OpAsmPrinter &p, Operation *op, OperandRange allocateVars, TypeRange allocateTypes, OperandRange allocatorVars, TypeRange allocatorTypes) |
| Print allocate clause. | |
| template<typename ClauseAttr> | |
| static ParseResult | parseClauseAttr (AsmParser &parser, ClauseAttr &attr) |
| template<typename ClauseAttr> | |
| static void | printClauseAttr (OpAsmPrinter &p, Operation *op, ClauseAttr attr) |
| static ParseResult | parseLinearClause (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &linearVars, SmallVectorImpl< Type > &linearTypes, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &linearStepVars, SmallVectorImpl< Type > &linearStepTypes, ArrayAttr &linearModifiers) |
| linear ::= linear ( linear-list ) linear-list := linear-val | linear-val linear-list linear-val := ssa-id-and-type = ssa-id-and-type | val ( ssa-id-and-type = ssa-id-and-type ) | ref ( ssa-id-and-type = ssa-id-and-type ) | uval ( ssa-id-and-type = ssa-id-and-type ) | |
| static void | printLinearClause (OpAsmPrinter &p, Operation *op, ValueRange linearVars, TypeRange linearTypes, ValueRange linearStepVars, TypeRange stepVarTypes, ArrayAttr linearModifiers) |
| Print Linear Clause. | |
| static LogicalResult | verifyLinearModifiers (Operation *op, std::optional< ArrayAttr > linearModifiers, OperandRange linearVars, bool isDeclareSimd=false) |
| OpenMP 5.2, Section 5.4.6: "A linear-modifier may be specified as ref or uval only on a declare simd directive. | |
| static LogicalResult | verifyNontemporalClause (Operation *op, OperandRange nontemporalVars) |
| static LogicalResult | verifyAlignedClause (Operation *op, std::optional< ArrayAttr > alignments, OperandRange alignedVars) |
| static ParseResult | parseAlignedClause (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &alignedVars, SmallVectorImpl< Type > &alignedTypes, ArrayAttr &alignmentsAttr) |
| aligned ::= aligned ( aligned-list ) aligned-list := aligned-val | aligned-val aligned-list aligned-val := ssa-id-and-type -> alignment | |
| static void | printAlignedClause (OpAsmPrinter &p, Operation *op, ValueRange alignedVars, TypeRange alignedTypes, std::optional< ArrayAttr > alignments) |
| Print Aligned Clause. | |
| static ParseResult | verifyScheduleModifiers (OpAsmParser &parser, SmallVectorImpl< SmallString< 12 > > &modifiers) |
| static ParseResult | parseScheduleClause (OpAsmParser &parser, ClauseScheduleKindAttr &scheduleAttr, ScheduleModifierAttr &scheduleMod, UnitAttr &scheduleSimd, std::optional< OpAsmParser::UnresolvedOperand > &chunkSize, Type &chunkType) |
| schedule ::= schedule ( sched-list ) sched-list ::= sched-val | sched-val sched-list | sched-val , sched-modifier sched-val ::= sched-with-chunk | sched-wo-chunk sched-with-chunk ::= sched-with-chunk-types (= ssa-id-and-type)? | |
| static void | printScheduleClause (OpAsmPrinter &p, Operation *op, ClauseScheduleKindAttr scheduleKind, ScheduleModifierAttr scheduleMod, UnitAttr scheduleSimd, Value scheduleChunk, Type scheduleChunkType) |
| Print schedule clause. | |
| static ParseResult | parseOrderClause (OpAsmParser &parser, ClauseOrderKindAttr &order, OrderModifierAttr &orderMod) |
| static void | printOrderClause (OpAsmPrinter &p, Operation *op, ClauseOrderKindAttr order, OrderModifierAttr orderMod) |
| template<typename ClauseTypeAttr, typename ClauseType> | |
| static ParseResult | parseGranularityClause (OpAsmParser &parser, ClauseTypeAttr &prescriptiveness, std::optional< OpAsmParser::UnresolvedOperand > &operand, Type &operandType, std::optional< ClauseType >(*symbolizeClause)(StringRef), StringRef clauseName) |
| template<typename ClauseTypeAttr, typename ClauseType> | |
| static void | printGranularityClause (OpAsmPrinter &p, Operation *op, ClauseTypeAttr prescriptiveness, Value operand, mlir::Type operandType, StringRef(*stringifyClauseType)(ClauseType)) |
| static ParseResult | parseGrainsizeClause (OpAsmParser &parser, ClauseGrainsizeTypeAttr &grainsizeMod, std::optional< OpAsmParser::UnresolvedOperand > &grainsize, Type &grainsizeType) |
| static void | printGrainsizeClause (OpAsmPrinter &p, Operation *op, ClauseGrainsizeTypeAttr grainsizeMod, Value grainsize, mlir::Type grainsizeType) |
| static ParseResult | parseNumTasksClause (OpAsmParser &parser, ClauseNumTasksTypeAttr &numTasksMod, std::optional< OpAsmParser::UnresolvedOperand > &numTasks, Type &numTasksType) |
| static void | printNumTasksClause (OpAsmPrinter &p, Operation *op, ClauseNumTasksTypeAttr numTasksMod, Value numTasks, mlir::Type numTasksType) |
| static constexpr StringRef | getPrivateNeedsBarrierSpelling () |
| static ParseResult | parseClauseWithRegionArgs (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &operands, SmallVectorImpl< Type > &types, SmallVectorImpl< OpAsmParser::Argument > ®ionPrivateArgs, ArrayAttr *symbols=nullptr, DenseI64ArrayAttr *mapIndices=nullptr, DenseBoolArrayAttr *byref=nullptr, ReductionModifierAttr *modifier=nullptr, UnitAttr *needsBarrier=nullptr) |
| static ParseResult | parseBlockArgClause (OpAsmParser &parser, llvm::SmallVectorImpl< OpAsmParser::Argument > &entryBlockArgs, StringRef keyword, std::optional< MapParseArgs > mapArgs) |
| static ParseResult | parseBlockArgClause (OpAsmParser &parser, llvm::SmallVectorImpl< OpAsmParser::Argument > &entryBlockArgs, StringRef keyword, std::optional< PrivateParseArgs > privateArgs) |
| static ParseResult | parseBlockArgClause (OpAsmParser &parser, llvm::SmallVectorImpl< OpAsmParser::Argument > &entryBlockArgs, StringRef keyword, std::optional< ReductionParseArgs > reductionArgs) |
| static ParseResult | parseBlockArgRegion (OpAsmParser &parser, Region ®ion, AllRegionParseArgs args) |
| static ParseResult | parseTargetOpRegion (OpAsmParser &parser, Region ®ion, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &hasDeviceAddrVars, SmallVectorImpl< Type > &hasDeviceAddrTypes, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &hostEvalVars, SmallVectorImpl< Type > &hostEvalTypes, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &inReductionVars, SmallVectorImpl< Type > &inReductionTypes, DenseBoolArrayAttr &inReductionByref, ArrayAttr &inReductionSyms, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &mapVars, SmallVectorImpl< Type > &mapTypes, llvm::SmallVectorImpl< OpAsmParser::UnresolvedOperand > &privateVars, llvm::SmallVectorImpl< Type > &privateTypes, ArrayAttr &privateSyms, UnitAttr &privateNeedsBarrier, DenseI64ArrayAttr &privateMaps) |
| static ParseResult | parseInReductionPrivateRegion (OpAsmParser &parser, Region ®ion, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &inReductionVars, SmallVectorImpl< Type > &inReductionTypes, DenseBoolArrayAttr &inReductionByref, ArrayAttr &inReductionSyms, llvm::SmallVectorImpl< OpAsmParser::UnresolvedOperand > &privateVars, llvm::SmallVectorImpl< Type > &privateTypes, ArrayAttr &privateSyms, UnitAttr &privateNeedsBarrier) |
| static ParseResult | parseInReductionPrivateReductionRegion (OpAsmParser &parser, Region ®ion, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &inReductionVars, SmallVectorImpl< Type > &inReductionTypes, DenseBoolArrayAttr &inReductionByref, ArrayAttr &inReductionSyms, llvm::SmallVectorImpl< OpAsmParser::UnresolvedOperand > &privateVars, llvm::SmallVectorImpl< Type > &privateTypes, ArrayAttr &privateSyms, UnitAttr &privateNeedsBarrier, ReductionModifierAttr &reductionMod, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &reductionVars, SmallVectorImpl< Type > &reductionTypes, DenseBoolArrayAttr &reductionByref, ArrayAttr &reductionSyms) |
| static ParseResult | parsePrivateRegion (OpAsmParser &parser, Region ®ion, llvm::SmallVectorImpl< OpAsmParser::UnresolvedOperand > &privateVars, llvm::SmallVectorImpl< Type > &privateTypes, ArrayAttr &privateSyms, UnitAttr &privateNeedsBarrier) |
| static ParseResult | parsePrivateReductionRegion (OpAsmParser &parser, Region ®ion, llvm::SmallVectorImpl< OpAsmParser::UnresolvedOperand > &privateVars, llvm::SmallVectorImpl< Type > &privateTypes, ArrayAttr &privateSyms, UnitAttr &privateNeedsBarrier, ReductionModifierAttr &reductionMod, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &reductionVars, SmallVectorImpl< Type > &reductionTypes, DenseBoolArrayAttr &reductionByref, ArrayAttr &reductionSyms) |
| static ParseResult | parseTaskReductionRegion (OpAsmParser &parser, Region ®ion, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &taskReductionVars, SmallVectorImpl< Type > &taskReductionTypes, DenseBoolArrayAttr &taskReductionByref, ArrayAttr &taskReductionSyms) |
| static ParseResult | parseUseDeviceAddrUseDevicePtrRegion (OpAsmParser &parser, Region ®ion, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &useDeviceAddrVars, SmallVectorImpl< Type > &useDeviceAddrTypes, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &useDevicePtrVars, SmallVectorImpl< Type > &useDevicePtrTypes) |
| static void | printClauseWithRegionArgs (OpAsmPrinter &p, MLIRContext *ctx, StringRef clauseName, ValueRange argsSubrange, ValueRange operands, TypeRange types, ArrayAttr symbols=nullptr, DenseI64ArrayAttr mapIndices=nullptr, DenseBoolArrayAttr byref=nullptr, ReductionModifierAttr modifier=nullptr, UnitAttr needsBarrier=nullptr) |
| static void | printBlockArgClause (OpAsmPrinter &p, MLIRContext *ctx, StringRef clauseName, ValueRange argsSubrange, std::optional< MapPrintArgs > mapArgs) |
| static void | printBlockArgClause (OpAsmPrinter &p, MLIRContext *ctx, StringRef clauseName, ValueRange argsSubrange, std::optional< PrivatePrintArgs > privateArgs) |
| static void | printBlockArgClause (OpAsmPrinter &p, MLIRContext *ctx, StringRef clauseName, ValueRange argsSubrange, std::optional< ReductionPrintArgs > reductionArgs) |
| static void | printBlockArgRegion (OpAsmPrinter &p, Operation *op, Region ®ion, const AllRegionPrintArgs &args) |
| static void | printTargetOpRegion (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange hasDeviceAddrVars, TypeRange hasDeviceAddrTypes, ValueRange hostEvalVars, TypeRange hostEvalTypes, ValueRange inReductionVars, TypeRange inReductionTypes, DenseBoolArrayAttr inReductionByref, ArrayAttr inReductionSyms, ValueRange mapVars, TypeRange mapTypes, ValueRange privateVars, TypeRange privateTypes, ArrayAttr privateSyms, UnitAttr privateNeedsBarrier, DenseI64ArrayAttr privateMaps) |
| static void | printInReductionPrivateRegion (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange inReductionVars, TypeRange inReductionTypes, DenseBoolArrayAttr inReductionByref, ArrayAttr inReductionSyms, ValueRange privateVars, TypeRange privateTypes, ArrayAttr privateSyms, UnitAttr privateNeedsBarrier) |
| static void | printInReductionPrivateReductionRegion (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange inReductionVars, TypeRange inReductionTypes, DenseBoolArrayAttr inReductionByref, ArrayAttr inReductionSyms, ValueRange privateVars, TypeRange privateTypes, ArrayAttr privateSyms, UnitAttr privateNeedsBarrier, ReductionModifierAttr reductionMod, ValueRange reductionVars, TypeRange reductionTypes, DenseBoolArrayAttr reductionByref, ArrayAttr reductionSyms) |
| static void | printPrivateRegion (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange privateVars, TypeRange privateTypes, ArrayAttr privateSyms, UnitAttr privateNeedsBarrier) |
| static void | printPrivateReductionRegion (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange privateVars, TypeRange privateTypes, ArrayAttr privateSyms, UnitAttr privateNeedsBarrier, ReductionModifierAttr reductionMod, ValueRange reductionVars, TypeRange reductionTypes, DenseBoolArrayAttr reductionByref, ArrayAttr reductionSyms) |
| static void | printTaskReductionRegion (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange taskReductionVars, TypeRange taskReductionTypes, DenseBoolArrayAttr taskReductionByref, ArrayAttr taskReductionSyms) |
| static void | printUseDeviceAddrUseDevicePtrRegion (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange useDeviceAddrVars, TypeRange useDeviceAddrTypes, ValueRange useDevicePtrVars, TypeRange useDevicePtrTypes) |
| template<typename ParsePrefixFn> | |
| static ParseResult | parseSplitIteratedList (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &iteratedVars, SmallVectorImpl< Type > &iteratedTypes, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &plainVars, SmallVectorImpl< Type > &plainTypes, ParsePrefixFn &&parsePrefix) |
| template<typename PrintPrefixFn> | |
| static void | printSplitIteratedList (OpAsmPrinter &p, ValueRange iteratedVars, TypeRange iteratedTypes, ValueRange plainVars, TypeRange plainTypes, PrintPrefixFn &&printPrefixForPlain, PrintPrefixFn &&printPrefixForIterated) |
| static LogicalResult | verifyReductionVarList (Operation *op, std::optional< ArrayAttr > reductionSyms, OperandRange reductionVars, std::optional< ArrayRef< bool > > reductionByref) |
| Verifies Reduction Clause. | |
| static ParseResult | parseCopyprivate (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > ©privateVars, SmallVectorImpl< Type > ©privateTypes, ArrayAttr ©privateSyms) |
| copyprivate-entry-list ::= copyprivate-entry | copyprivate-entry-list , copyprivate-entry copyprivate-entry ::= ssa-id -> symbol-ref : type | |
| static void | printCopyprivate (OpAsmPrinter &p, Operation *op, OperandRange copyprivateVars, TypeRange copyprivateTypes, std::optional< ArrayAttr > copyprivateSyms) |
| Print Copyprivate clause. | |
| static LogicalResult | verifyCopyprivateVarList (Operation *op, OperandRange copyprivateVars, std::optional< ArrayAttr > copyprivateSyms) |
| Verifies CopyPrivate Clause. | |
| static ParseResult | parseDependVarList (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &dependVars, SmallVectorImpl< Type > &dependTypes, ArrayAttr &dependKinds, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &iteratedVars, SmallVectorImpl< Type > &iteratedTypes, ArrayAttr &iteratedKinds) |
| depend-entry-list ::= depend-entry | depend-entry-list , depend-entry depend-entry ::= depend-kind -> ssa-id : type | depend-kind -> ssa-id : iterated-type | |
| static void | printDependVarList (OpAsmPrinter &p, Operation *op, OperandRange dependVars, TypeRange dependTypes, std::optional< ArrayAttr > dependKinds, OperandRange iteratedVars, TypeRange iteratedTypes, std::optional< ArrayAttr > iteratedKinds) |
| Print Depend clause. | |
| static LogicalResult | verifyDependVarList (Operation *op, std::optional< ArrayAttr > dependKinds, OperandRange dependVars, std::optional< ArrayAttr > iteratedKinds, OperandRange iteratedVars) |
| Verifies Depend clause. | |
| static ParseResult | parseSynchronizationHint (OpAsmParser &parser, IntegerAttr &hintAttr) |
| Parses a Synchronization Hint clause. | |
| static void | printSynchronizationHint (OpAsmPrinter &p, Operation *op, IntegerAttr hintAttr) |
| Prints a Synchronization Hint clause. | |
| static LogicalResult | verifySynchronizationHint (Operation *op, uint64_t hint) |
| Verifies a synchronization hint clause. | |
| static bool | mapTypeToBool (ClauseMapFlags value, ClauseMapFlags flag) |
| return | success () |
| if (parser.parseCommaSeparatedList(parseIndices)) return failure() | |
| if (failed(parser.parseRSquare())) return failure() | |
| memberIdxs | push_back (ArrayAttr::get(parser.getContext(), values)) |
| values | clear () |
| while (succeeded(parser.parseOptionalComma())) | |
| if (!memberIdxs.empty()) membersIdx | |
| static void | printMembersIndex (OpAsmPrinter &p, MapInfoOp op, ArrayAttr membersIdx) |
| static void | printCaptureType (OpAsmPrinter &p, Operation *op, VariableCaptureKindAttr mapCaptureType) |
| static ParseResult | parseCaptureType (OpAsmParser &parser, VariableCaptureKindAttr &mapCaptureType) |
| static LogicalResult | verifyMapClause (Operation *op, OperandRange mapVars) |
| static LogicalResult | verifyPrivateVarsMapping (TargetOp targetOp) |
| static LogicalResult | verifyMapInfoDefinedArgs (Operation *op, StringRef clauseName, OperandRange vars) |
| static Operation * | findCapturedOmpOp (Operation *rootOp, bool checkSingleMandatoryExec, llvm::function_ref< bool(Operation *)> siblingAllowedFn) |
| static bool | canPromoteToNoLoop (Operation *capturedOp, TeamsOp teamsOp, WsloopOp *wsLoopOp) |
| Check if we can promote SPMD kernel to No-Loop kernel. | |
| template<typename OpType> | |
| static LogicalResult | verifyPrivateVarList (OpType &op) |
| static bool | opInGlobalImplicitParallelRegion (Operation *op) |
| static LogicalResult | verifyNumTeamsClause (Operation *op, Value numTeamsLower, OperandRange numTeamsUpperVars) |
| static void | printLoopTransformClis (OpAsmPrinter &p, TileOp op, OperandRange generatees, OperandRange applyees) |
| static ParseResult | parseLoopTransformClis (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &generateesOperands, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &applyeesOperands) |
| static LogicalResult | checkApplyeesNesting (TileOp op) |
| Check properties of the loop nest consisting of the transformation's applyees: | |
| static void | printLoopTransformClis (OpAsmPrinter &p, FuseOp op, OperandRange generatees, OperandRange applyees) |
| static LogicalResult | verifyOrderedParent (Operation &op) |
| static Operation * | getParentInSameDialect (Operation *thisOp) |
| static mlir::ParseResult | parseTargetAllocMemOp (mlir::OpAsmParser &parser, mlir::OperationState &result) |
| operation ::= res = (omp.target_alloc_mem) $device : devicetype, $in_type ( ( $typeparams ) )? ( , $shape )? attr-dict-without-keyword | |
| static ParseResult | parseUniformClause (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &uniformVars, SmallVectorImpl< Type > &uniformTypes) |
| uniform ::= uniform ( uniform-list ) uniform-list := uniform-val (, uniform-val)* uniform-val := ssa-id : type */ | |
| static void | printUniformClause (OpAsmPrinter &p, Operation *op, ValueRange uniformVars, TypeRange uniformTypes) |
| Print Uniform Clauses. | |
| static ParseResult | parseAffinityClause (OpAsmParser &parser, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &iterated, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &affinityVars, SmallVectorImpl< Type > &iteratedTypes, SmallVectorImpl< Type > &affinityVarTypes) |
| static void | printAffinityClause (OpAsmPrinter &p, Operation *op, ValueRange iterated, ValueRange affinityVars, TypeRange iteratedTypes, TypeRange affinityVarTypes) |
| static ParseResult | parseIteratorHeader (OpAsmParser &parser, Region ®ion, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &lbs, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &ubs, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &steps, SmallVectorImpl< Type > &lbTypes, SmallVectorImpl< Type > &ubTypes, SmallVectorImpl< Type > &stepTypes) |
| static void | printIteratorHeader (OpAsmPrinter &p, Operation *op, Region ®ion, ValueRange lbs, ValueRange ubs, ValueRange steps, TypeRange, TypeRange, TypeRange) |
Variables | |
| false | |
| Parses a map_entries map type from a string format back into its numeric value. | |
| do | |
| #define GET_ATTRDEF_CLASSES |
Definition at line 4978 of file OpenMPDialect.cpp.
| #define GET_ATTRDEF_LIST |
| #define GET_OP_CLASSES |
Definition at line 4981 of file OpenMPDialect.cpp.
| #define GET_OP_LIST |
| #define GET_TYPEDEF_CLASSES |
Definition at line 4984 of file OpenMPDialect.cpp.
| #define GET_TYPEDEF_LIST |
Check if we can promote SPMD kernel to No-Loop kernel.
Definition at line 2541 of file OpenMPDialect.cpp.
References mlir::Operation::getParentOfType().
|
static |
Check properties of the loop nest consisting of the transformation's applyees:
TODO: Generalize for LoopTransformationInterface.
Definition at line 3942 of file OpenMPDialect.cpp.
References mlir::omp::decodeCli(), and success().
| values clear | ( | ) |
|
static |
Definition at line 2454 of file OpenMPDialect.cpp.
References mlir::WalkResult::advance(), mlir::DominanceInfo::dominates(), mlir::Operation::getBlock(), mlir::Operation::getDialect(), mlir::Operation::getNumRegions(), mlir::Region::getOps(), mlir::Operation::getParentRegion(), mlir::Block::getSuccessors(), mlir::WalkResult::interrupt(), mlir::detail::DominanceInfoBase< IsPostDom >::isReachableFromEntry(), mlir::PreOrder, mlir::WalkResult::skip(), and mlir::Operation::walk().
|
static |
Generate a name of a canonical loop nest of the format <prefix>(_r<idx>_s<idx>)*.
Hereby, _r<idx> identifies the region argument index of an operation that has multiple regions, if the operation has multiple regions. _s<idx> identifies the position of an operation within a region, where only operations that may potentially contain loops ("container operations" i.e. have region arguments) are counted. Again, it is omitted if there is only one such operation in a region. If there are canonical loops nested inside each other, also may also use the format _d<num> where <num> is the nesting depth of the loop.
The generated name is a best-effort to make canonical loop unique within an SSA namespace. This also means that regions with IsolatedFromAbove property do not consider any parents or siblings.
If true, this component describes a region operand of an operation (the operand's owner) If false, this component describes an operation located in a parent region
Definition at line 95 of file OpenMPDialect.cpp.
References b, mlir::Region::getBlocks(), mlir::Region::getParentOp(), mlir::Operation::getParentRegion(), mlir::Operation::getRegions(), mlir::Operation::hasTrait(), and isUnique().
Definition at line 4318 of file OpenMPDialect.cpp.
References mlir::Operation::getDialect(), and mlir::Operation::getParentOp().
|
inlinestaticconstexpr |
Definition at line 927 of file OpenMPDialect.cpp.
Referenced by printClauseWithRegionArgs().
| if | ( | !memberIdxs. | empty() | ) |
| if | ( | failed(parser.parseRSquare()) | ) |
| if | ( | parser. | parseCommaSeparatedListparseIndices | ) |
|
static |
Definition at line 50 of file OpenMPDialect.cpp.
References ArrayAttr(), and nullptr.
|
static |
Definition at line 56 of file OpenMPDialect.cpp.
References mlir::detail::DenseArrayAttrImpl< bool >::get(), and nullptr.
|
static |
Definition at line 61 of file OpenMPDialect.cpp.
References mlir::detail::DenseArrayAttrImpl< int64_t >::get(), and nullptr.
|
static |
Definition at line 1902 of file OpenMPDialect.cpp.
Referenced by convertClauseMapFlags(), and verifyMapClause().
Definition at line 2788 of file OpenMPDialect.cpp.
References mlir::Operation::getDialect(), and mlir::Operation::getParentOp().
|
static |
Definition at line 4810 of file OpenMPDialect.cpp.
References parseSplitIteratedList(), and success().
|
static |
aligned ::= aligned ( aligned-list ) aligned-list := aligned-val | aligned-val aligned-list aligned-val := ssa-id-and-type -> alignment
Definition at line 590 of file OpenMPDialect.cpp.
References ArrayAttr(), mlir::AsmParser::parseCommaSeparatedList(), and success().
|
static |
Parse an allocate clause with allocators and a list of operands with types.
allocate-operand-list :: = allocate-operand | allocator-operand , allocate-operand-list allocate-operand :: = ssa-id-and-type -> ssa-id-and-type ssa-id-and-type ::= ssa-id : type
Definition at line 355 of file OpenMPDialect.cpp.
References mlir::AsmParser::parseArrow(), mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), mlir::OpAsmParser::parseOperand(), and success().
|
static |
Definition at line 1035 of file OpenMPDialect.cpp.
References parseClauseWithRegionArgs(), mlir::AsmParser::parseOptionalKeyword(), and success().
Referenced by parseBlockArgRegion().
|
static |
Definition at line 1050 of file OpenMPDialect.cpp.
References parseClauseWithRegionArgs(), mlir::AsmParser::parseOptionalKeyword(), and success().
|
static |
Definition at line 1067 of file OpenMPDialect.cpp.
References parseClauseWithRegionArgs(), mlir::AsmParser::parseOptionalKeyword(), and success().
|
static |
Definition at line 1083 of file OpenMPDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getCurrentLocation(), parseBlockArgClause(), and mlir::OpAsmParser::parseRegion().
Referenced by parseInReductionPrivateReductionRegion(), parseInReductionPrivateRegion(), parsePrivateReductionRegion(), parsePrivateRegion(), parseTargetOpRegion(), parseTaskReductionRegion(), and parseUseDeviceAddrUseDevicePtrRegion().
|
static |
Definition at line 2129 of file OpenMPDialect.cpp.
References mlir::AsmParser::getContext(), mlir::AsmParser::parseKeyword(), and success().
|
static |
Definition at line 398 of file OpenMPDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getContext(), mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::parseKeyword(), and success().
|
static |
Definition at line 931 of file OpenMPDialect.cpp.
References ArrayAttr(), mlir::AsmParser::getContext(), mlir::AsmParser::parseColon(), mlir::AsmParser::parseComma(), mlir::AsmParser::parseCommaSeparatedList(), mlir::AsmParser::parseKeyword(), mlir::AsmParser::parseLParen(), mlir::AsmParser::parseOptionalKeyword(), and success().
Referenced by parseBlockArgClause(), parseBlockArgClause(), and parseBlockArgClause().
|
static |
copyprivate-entry-list ::= copyprivate-entry | copyprivate-entry-list , copyprivate-entry copyprivate-entry ::= ssa-id -> symbol-ref : type
Definition at line 1612 of file OpenMPDialect.cpp.
References ArrayAttr(), mlir::AsmParser::getContext(), mlir::AsmParser::parseCommaSeparatedList(), and success().
|
static |
depend-entry-list ::= depend-entry | depend-entry-list , depend-entry depend-entry ::= depend-kind -> ssa-id : type | depend-kind -> ssa-id : iterated-type
Definition at line 1716 of file OpenMPDialect.cpp.
References ArrayAttr(), mlir::AsmParser::parseCommaSeparatedList(), and success().
|
static |
Definition at line 839 of file OpenMPDialect.cpp.
References parseGranularityClause().
|
static |
Definition at line 785 of file OpenMPDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getContext(), mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::parseColonType(), mlir::AsmParser::parseComma(), mlir::OpAsmParser::parseOperand(), mlir::AsmParser::parseOptionalKeyword(), and success().
Referenced by parseGrainsizeClause(), and parseNumTasksClause().
|
static |
Definition at line 1178 of file OpenMPDialect.cpp.
References ArrayAttr(), and parseBlockArgRegion().
|
static |
Definition at line 1162 of file OpenMPDialect.cpp.
References ArrayAttr(), and parseBlockArgRegion().
|
static |
Definition at line 4839 of file OpenMPDialect.cpp.
References mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::parseCommaSeparatedList(), and success().
|
static |
linear ::= linear ( linear-list ) linear-list := linear-val | linear-val linear-list linear-val := ssa-id-and-type = ssa-id-and-type | val ( ssa-id-and-type = ssa-id-and-type ) | ref ( ssa-id-and-type = ssa-id-and-type ) | uval ( ssa-id-and-type = ssa-id-and-type )
Definition at line 426 of file OpenMPDialect.cpp.
References ArrayAttr(), mlir::AsmParser::getContext(), mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), mlir::AsmParser::parseEqual(), mlir::AsmParser::parseLParen(), mlir::OpAsmParser::parseOperand(), mlir::AsmParser::parseOptionalKeyword(), mlir::AsmParser::parseRParen(), result, and success().
|
static |
Definition at line 3905 of file OpenMPDialect.cpp.
References mlir::AsmParser::Paren, mlir::AsmParser::parseLess(), mlir::AsmParser::parseMinus(), mlir::OpAsmParser::parseOperandList(), mlir::AsmParser::parseOptionalLess(), and success().
|
static |
Definition at line 861 of file OpenMPDialect.cpp.
References parseGranularityClause().
|
static |
Definition at line 750 of file OpenMPDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getContext(), mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::parseKeyword(), mlir::AsmParser::parseOptionalColon(), and success().
|
static |
Definition at line 1210 of file OpenMPDialect.cpp.
References ArrayAttr(), and parseBlockArgRegion().
|
static |
Definition at line 1199 of file OpenMPDialect.cpp.
References ArrayAttr(), and parseBlockArgRegion().
|
static |
schedule ::= schedule ( sched-list ) sched-list ::= sched-val | sched-val sched-list | sched-val , sched-modifier sched-val ::= sched-with-chunk | sched-wo-chunk sched-with-chunk ::= sched-with-chunk-types (= ssa-id-and-type)?
sched-with-chunk-types ::= static | dynamic | guided sched-wo-chunk ::= auto | runtime sched-modifier ::= sched-mod-val | sched-mod-val , sched-mod-val sched-mod-val ::= monotonic | nonmonotonic | simd | none
Definition at line 668 of file OpenMPDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getBuilder(), mlir::AsmParser::getContext(), mlir::Builder::getContext(), mlir::AsmParser::getCurrentLocation(), mlir::AsmParser::getNameLoc(), mlir::AsmParser::parseColonType(), mlir::AsmParser::parseKeyword(), mlir::OpAsmParser::parseOperand(), mlir::AsmParser::parseOptionalComma(), mlir::AsmParser::parseOptionalEqual(), success(), and verifyScheduleModifiers().
|
static |
Definition at line 1510 of file OpenMPDialect.cpp.
References mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), mlir::OpAsmParser::parseOperand(), and success().
Referenced by parseAffinityClause().
|
static |
Parses a Synchronization Hint clause.
The value of hint is an integer which is a combination of different hints from omp_sync_hint_t.
hint-clause = hint ( hint-value )
Definition at line 1815 of file OpenMPDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getBuilder(), mlir::AsmParser::getCurrentLocation(), mlir::Builder::getI64Type(), mlir::AsmParser::parseCommaSeparatedList(), mlir::AsmParser::parseKeyword(), mlir::AsmParser::parseOptionalKeyword(), and success().
|
static |
operation ::= res = (omp.target_alloc_mem) $device : devicetype, $in_type ( ( $typeparams ) )? ( , $shape )? attr-dict-without-keyword
Definition at line 4603 of file OpenMPDialect.cpp.
References mlir::AsmParser::addTypeToList(), mlir::AsmParser::emitError(), mlir::AsmParser::getBuilder(), mlir::Builder::getDenseI32ArrayAttr(), mlir::Builder::getIndexType(), mlir::Builder::getIntegerType(), mlir::AsmParser::getNameLoc(), mlir::AsmParser::None, mlir::AsmParser::parseColonType(), mlir::AsmParser::parseColonTypeList(), mlir::AsmParser::parseComma(), mlir::OpAsmParser::parseOperand(), mlir::OpAsmParser::parseOperandList(), mlir::AsmParser::parseOptionalAttrDict(), mlir::AsmParser::parseOptionalComma(), mlir::AsmParser::parseOptionalLParen(), mlir::AsmParser::parseRParen(), mlir::AsmParser::parseType(), mlir::OpAsmParser::resolveOperand(), mlir::OpAsmParser::resolveOperands(), and result.
|
static |
Definition at line 1137 of file OpenMPDialect.cpp.
References ArrayAttr(), and parseBlockArgRegion().
|
static |
Definition at line 1226 of file OpenMPDialect.cpp.
References ArrayAttr(), and parseBlockArgRegion().
|
static |
uniform ::= uniform ( uniform-list ) uniform-list := uniform-val (, uniform-val)* uniform-val := ssa-id : type */
Definition at line 4785 of file OpenMPDialect.cpp.
References mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), and mlir::OpAsmParser::parseOperand().
|
static |
Definition at line 1237 of file OpenMPDialect.cpp.
References parseBlockArgRegion().
|
static |
Definition at line 4823 of file OpenMPDialect.cpp.
References printSplitIteratedList().
|
static |
Print Aligned Clause.
Definition at line 611 of file OpenMPDialect.cpp.
References mlir::ValueRange::getType().
|
static |
Print allocate clause.
Definition at line 381 of file OpenMPDialect.cpp.
|
static |
Definition at line 1345 of file OpenMPDialect.cpp.
References printClauseWithRegionArgs().
Referenced by printBlockArgRegion().
|
static |
Definition at line 1353 of file OpenMPDialect.cpp.
References printClauseWithRegionArgs().
|
static |
Definition at line 1364 of file OpenMPDialect.cpp.
References printClauseWithRegionArgs().
|
static |
Definition at line 1374 of file OpenMPDialect.cpp.
References mlir::Operation::getContext(), printBlockArgClause(), and mlir::OpAsmPrinter::printRegion().
Referenced by printInReductionPrivateReductionRegion(), printInReductionPrivateRegion(), printPrivateReductionRegion(), printPrivateRegion(), printTargetOpRegion(), printTaskReductionRegion(), and printUseDeviceAddrUseDevicePtrRegion().
|
static |
Definition at line 2114 of file OpenMPDialect.cpp.
|
static |
Definition at line 412 of file OpenMPDialect.cpp.
|
static |
Definition at line 1293 of file OpenMPDialect.cpp.
References ArrayAttr(), mlir::detail::DenseArrayAttrImpl< bool >::get(), mlir::detail::DenseArrayAttrImpl< int64_t >::get(), and getPrivateNeedsBarrierSpelling().
Referenced by printBlockArgClause(), printBlockArgClause(), and printBlockArgClause().
|
static |
Print Copyprivate clause.
Definition at line 1632 of file OpenMPDialect.cpp.
|
static |
Print Depend clause.
Definition at line 1757 of file OpenMPDialect.cpp.
|
static |
Definition at line 847 of file OpenMPDialect.cpp.
References printGranularityClause().
|
static |
Definition at line 821 of file OpenMPDialect.cpp.
Referenced by printGrainsizeClause(), and printNumTasksClause().
|
static |
Definition at line 1440 of file OpenMPDialect.cpp.
References ArrayAttr(), and printBlockArgRegion().
|
static |
Definition at line 1426 of file OpenMPDialect.cpp.
References ArrayAttr(), and printBlockArgRegion().
|
static |
Definition at line 4903 of file OpenMPDialect.cpp.
References mlir::Region::front(), mlir::Block::getArgument(), mlir::Block::getNumArguments(), mlir::OpAsmPrinter::printRegion(), and mlir::OpAsmPrinter::printRegionArgument().
|
static |
|
static |
Definition at line 4045 of file OpenMPDialect.cpp.
|
static |
Definition at line 3895 of file OpenMPDialect.cpp.
|
static |
Definition at line 2099 of file OpenMPDialect.cpp.
References ArrayAttr().
|
static |
Definition at line 869 of file OpenMPDialect.cpp.
References printGranularityClause().
|
static |
Definition at line 774 of file OpenMPDialect.cpp.
|
static |
Definition at line 1470 of file OpenMPDialect.cpp.
References ArrayAttr(), and printBlockArgRegion().
|
static |
Definition at line 1459 of file OpenMPDialect.cpp.
References ArrayAttr(), and printBlockArgRegion().
|
static |
Print schedule clause.
Definition at line 730 of file OpenMPDialect.cpp.
References mlir::Value::getType().
|
static |
Definition at line 1538 of file OpenMPDialect.cpp.
References emit().
Referenced by printAffinityClause().
|
static |
Prints a Synchronization Hint clause.
Definition at line 1846 of file OpenMPDialect.cpp.
|
static |
Definition at line 1406 of file OpenMPDialect.cpp.
References ArrayAttr(), and printBlockArgRegion().
|
static |
Definition at line 1485 of file OpenMPDialect.cpp.
References ArrayAttr(), and printBlockArgRegion().
|
static |
Print Uniform Clauses.
Definition at line 4797 of file OpenMPDialect.cpp.
|
static |
Definition at line 1497 of file OpenMPDialect.cpp.
References printBlockArgRegion().
| memberIdxs push_back | ( | ArrayAttr::get(parser.getContext(), values) | ) |
Referenced by mlir::NamedAttrList::append(), applyTilingToAll(), and getReassociationForFlattening().
| return success | ( | ) |
Referenced by checkApplyeesNesting(), parseAffinityClause(), parseAllocateAndAllocator(), parseBlockArgClause(), parseBlockArgClause(), parseBlockArgClause(), parseCaptureType(), parseClauseAttr(), parseCopyprivate(), parseGranularityClause(), parseLinearClause(), parseLoopTransformClis(), parseOrderClause(), parseScheduleClause(), parseSplitIteratedList(), parseSynchronizationHint(), verifyAlignedClause(), verifyCopyprivateVarList(), verifyDependVarList(), verifyLinearModifiers(), verifyMapClause(), verifyMapInfoDefinedArgs(), verifyNontemporalClause(), verifyNumTeamsClause(), verifyOrderedParent(), verifyPrivateVarList(), verifyPrivateVarsMapping(), verifyReductionVarList(), verifyScheduleModifiers(), and verifySynchronizationHint().
|
static |
Definition at line 550 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), and success().
|
static |
Verifies CopyPrivate Clause.
Definition at line 1648 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), mlir::SymbolTable::lookupNearestSymbolFrom(), and success().
|
static |
Verifies Depend clause.
Definition at line 1781 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), and success().
|
static |
OpenMP 5.2, Section 5.4.6: "A linear-modifier may be specified as ref or uval only on a declare simd directive.
" Also verifies that modifier count matches variable count.
Definition at line 507 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), and success().
|
static |
Definition at line 2151 of file OpenMPDialect.cpp.
References mlir::emitError(), mlir::Operation::getLoc(), mapTypeToBool(), and success().
|
static |
Definition at line 2242 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), and success().
|
static |
Definition at line 535 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), and success().
|
static |
Definition at line 2810 of file OpenMPDialect.cpp.
References mlir::Operation::emitError(), mlir::OperandRange::getType(), mlir::Value::getType(), and success().
|
static |
Definition at line 4126 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), mlir::Operation::getNumRegions(), mlir::Operation::getParentOfType(), mlir::Operation::getParentOp(), and success().
|
static |
Definition at line 2700 of file OpenMPDialect.cpp.
References mlir::SymbolTable::lookupNearestSymbolFrom(), and success().
|
static |
Definition at line 2220 of file OpenMPDialect.cpp.
References mlir::emitError(), and success().
|
static |
Verifies Reduction Clause.
Definition at line 1561 of file OpenMPDialect.cpp.
References mlir::Operation::emitError(), mlir::Operation::emitOpError(), mlir::Value::getType(), mlir::SymbolTable::lookupNearestSymbolFrom(), and success().
|
static |
Definition at line 627 of file OpenMPDialect.cpp.
References mlir::AsmParser::emitError(), mlir::AsmParser::getNameLoc(), and success().
Referenced by parseScheduleClause().
|
static |
Verifies a synchronization hint clause.
Definition at line 1877 of file OpenMPDialect.cpp.
References mlir::Operation::emitOpError(), and success().
| while | ( | succeeded(parser.parseOptionalComma()) | ) |
| do |
Definition at line 2079 of file OpenMPDialect.cpp.
| false |
Parses a map_entries map type from a string format back into its numeric value.
map-clause = map_clauses ( ( ( always, ? implicit, ? ompx_hold, ? / close, ? present, ? ( to | from | delete ) )+ )` ) static ParseResult parseMapClause(OpAsmParser &parser, ClauseMapFlagsAttr &mapType) { ClauseMapFlags mapTypeBits = ClauseMapFlags::none; This simply verifies the correct keyword is read in, the keyword itself is stored inside of the operation auto parseTypeAndMod = [&]() -> ParseResult { StringRef mapTypeMod; if (parser.parseKeyword(&mapTypeMod)) return failure();
if (mapTypeMod == "always") mapTypeBits |= ClauseMapFlags::always;
if (mapTypeMod == "implicit") mapTypeBits |= ClauseMapFlags::implicit;
if (mapTypeMod == "ompx_hold") mapTypeBits |= ClauseMapFlags::ompx_hold;
if (mapTypeMod == "close") mapTypeBits |= ClauseMapFlags::close;
if (mapTypeMod == "present") mapTypeBits |= ClauseMapFlags::present;
if (mapTypeMod == "to") mapTypeBits |= ClauseMapFlags::to;
if (mapTypeMod == "from") mapTypeBits |= ClauseMapFlags::from;
if (mapTypeMod == "tofrom") mapTypeBits |= ClauseMapFlags::to | ClauseMapFlags::from;
if (mapTypeMod == "delete") mapTypeBits |= ClauseMapFlags::del;
if (mapTypeMod == "storage") mapTypeBits |= ClauseMapFlags::storage;
if (mapTypeMod == "return_param") mapTypeBits |= ClauseMapFlags::return_param;
if (mapTypeMod == "private") mapTypeBits |= ClauseMapFlags::priv;
if (mapTypeMod == "literal") mapTypeBits |= ClauseMapFlags::literal;
if (mapTypeMod == "attach") mapTypeBits |= ClauseMapFlags::attach;
if (mapTypeMod == "attach_always") mapTypeBits |= ClauseMapFlags::attach_always;
if (mapTypeMod == "attach_never") mapTypeBits |= ClauseMapFlags::attach_never;
if (mapTypeMod == "attach_auto") mapTypeBits |= ClauseMapFlags::attach_auto;
if (mapTypeMod == "ref_ptr") mapTypeBits |= ClauseMapFlags::ref_ptr;
if (mapTypeMod == "ref_ptee") mapTypeBits |= ClauseMapFlags::ref_ptee;
if (mapTypeMod == "ref_ptr_ptee") mapTypeBits |= ClauseMapFlags::ref_ptr_ptee;
if (mapTypeMod == "is_device_ptr") mapTypeBits |= ClauseMapFlags::is_device_ptr;
return success(); };
if (parser.parseCommaSeparatedList(parseTypeAndMod)) return failure();
mapType = parser.getBuilder().getAttr<mlir::omp::ClauseMapFlagsAttr>(mapTypeBits);
return success(); }
/ Prints a map_entries map type from its numeric value out into its string / format. static void printMapClause(OpAsmPrinter &p, Operation *op, ClauseMapFlagsAttr mapType) { llvm::SmallVector<std::string, 4> mapTypeStrs; ClauseMapFlags mapFlags = mapType.getValue();
handling of always, close, present placed at the beginning of the string to aid readability if (mapTypeToBool(mapFlags, ClauseMapFlags::always)) mapTypeStrs.push_back("always"); if (mapTypeToBool(mapFlags, ClauseMapFlags::implicit)) mapTypeStrs.push_back("implicit"); if (mapTypeToBool(mapFlags, ClauseMapFlags::ompx_hold)) mapTypeStrs.push_back("ompx_hold"); if (mapTypeToBool(mapFlags, ClauseMapFlags::close)) mapTypeStrs.push_back("close"); if (mapTypeToBool(mapFlags, ClauseMapFlags::present)) mapTypeStrs.push_back("present");
special handling of to/from/tofrom/delete and release/alloc, release + alloc are the abscense of one of the other flags, whereas tofrom requires both the to and from flag to be set. bool to = mapTypeToBool(mapFlags, ClauseMapFlags::to); bool from = mapTypeToBool(mapFlags, ClauseMapFlags::from);
if (to && from) mapTypeStrs.push_back("tofrom"); else if (from) mapTypeStrs.push_back("from"); else if (to) mapTypeStrs.push_back("to");
if (mapTypeToBool(mapFlags, ClauseMapFlags::del)) mapTypeStrs.push_back("delete"); if (mapTypeToBool(mapFlags, ClauseMapFlags::return_param)) mapTypeStrs.push_back("return_param"); if (mapTypeToBool(mapFlags, ClauseMapFlags::storage)) mapTypeStrs.push_back("storage"); if (mapTypeToBool(mapFlags, ClauseMapFlags::priv)) mapTypeStrs.push_back("private"); if (mapTypeToBool(mapFlags, ClauseMapFlags::literal)) mapTypeStrs.push_back("literal"); if (mapTypeToBool(mapFlags, ClauseMapFlags::attach)) mapTypeStrs.push_back("attach"); if (mapTypeToBool(mapFlags, ClauseMapFlags::attach_always)) mapTypeStrs.push_back("attach_always"); if (mapTypeToBool(mapFlags, ClauseMapFlags::attach_never)) mapTypeStrs.push_back("attach_never"); if (mapTypeToBool(mapFlags, ClauseMapFlags::attach_auto)) mapTypeStrs.push_back("attach_auto"); if (mapTypeToBool(mapFlags, ClauseMapFlags::ref_ptr)) mapTypeStrs.push_back("ref_ptr"); if (mapTypeToBool(mapFlags, ClauseMapFlags::ref_ptee)) mapTypeStrs.push_back("ref_ptee"); if (mapTypeToBool(mapFlags, ClauseMapFlags::ref_ptr_ptee)) mapTypeStrs.push_back("ref_ptr_ptee"); if (mapTypeToBool(mapFlags, ClauseMapFlags::is_device_ptr)) mapTypeStrs.push_back("is_device_ptr"); if (mapFlags == ClauseMapFlags::none) mapTypeStrs.push_back("none");
for (unsigned int i = 0; i < mapTypeStrs.size(); ++i) { p << mapTypeStrs[i]; if (i + 1 < mapTypeStrs.size()) { p << ", "; } } }
static ParseResult parseMembersIndex(OpAsmParser &parser, ArrayAttr &membersIdx) { SmallVector<Attribute> values, memberIdxs;
auto parseIndices = [&]() -> ParseResult { int64_t value; if (parser.parseInteger(value)) return failure(); values.push_back(IntegerAttr::get(parser.getBuilder().getIntegerType(64), APInt(64, value, /*isSigned=
Definition at line 2075 of file OpenMPDialect.cpp.
Referenced by mlir::Operation::CloneOptions::CloneOptions(), mlir::LLVM::detail::DebugTranslation::DebugTranslation(), mlir::sparse_tensor::ir_detail::DimLvlMap::DimLvlMap(), mlir::SideEffects::EffectInstance< Effect >::EffectInstance(), mlir::SideEffects::EffectInstance< Effect >::EffectInstance(), mlir::SideEffects::EffectInstance< Effect >::EffectInstance(), mlir::SideEffects::EffectInstance< Effect >::EffectInstance(), mlir::SideEffects::EffectInstance< Effect >::EffectInstance(), mlir::SideEffects::EffectInstance< Effect >::EffectInstance(), mlir::LLVM::detail::LLVMStructTypeStorage::Key::Key(), mlir::sparse_tensor::LatPoint::LatPoint(), mlir::pdll::Lexer::Lexer(), mlir::sparse_tensor::Merger::Merger(), mlir::detail::OperandStorage::OperandStorage(), mlir::OpPrintingFlags::OpPrintingFlags(), mlir::detail::PassOptions::Option< DataType, OptionParser >::Option(), mlir::detail::PassExecutionState::PassExecutionState(), mlir::PassManager::PassManager(), mlir::PassManager::PassManager(), mlir::PassPipelineCLParser::PassPipelineCLParser(), mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::PyRewritePatternSet::PyRewritePatternSet(), mlir::python::MLIR_BINDINGS_PYTHON_DOMAIN::PyTypeConverter::PyTypeConverter(), mlir::ShapedTypeComponents::ShapedTypeComponents(), mlir::presburger::Simplex::Simplex(), mlir::presburger::SimplexBase::SimplexBase(), mlir::spirv::detail::StructTypeStorage::StructTypeStorage(), mlir::spirv::detail::StructTypeStorage::StructTypeStorage(), and mlir::tosa::ValueKnowledge::ValueKnowledge().