MLIR 23.0.0git
OpenMPDialect.cpp File Reference
#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/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)
 linear ::= linear ( linear-list ) linear-list := linear-val | linear-val linear-list linear-val := ssa-id-and-type = ssa-id-and-type
static void printLinearClause (OpAsmPrinter &p, Operation *op, ValueRange linearVars, TypeRange linearTypes, ValueRange linearStepVars, TypeRange stepVarTypes)
 Print Linear Clause.
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 > &regionPrivateArgs, 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 &region, AllRegionParseArgs args)
static ParseResult parseTargetOpRegion (OpAsmParser &parser, Region &region, 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 &region, 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 &region, 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 &region, llvm::SmallVectorImpl< OpAsmParser::UnresolvedOperand > &privateVars, llvm::SmallVectorImpl< Type > &privateTypes, ArrayAttr &privateSyms, UnitAttr &privateNeedsBarrier)
static ParseResult parsePrivateReductionRegion (OpAsmParser &parser, Region &region, 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 &region, SmallVectorImpl< OpAsmParser::UnresolvedOperand > &taskReductionVars, SmallVectorImpl< Type > &taskReductionTypes, DenseBoolArrayAttr &taskReductionByref, ArrayAttr &taskReductionSyms)
static ParseResult parseUseDeviceAddrUseDevicePtrRegion (OpAsmParser &parser, Region &region, 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 &region, const AllRegionPrintArgs &args)
static void printTargetOpRegion (OpAsmPrinter &p, Operation *op, Region &region, 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 &region, ValueRange inReductionVars, TypeRange inReductionTypes, DenseBoolArrayAttr inReductionByref, ArrayAttr inReductionSyms, ValueRange privateVars, TypeRange privateTypes, ArrayAttr privateSyms, UnitAttr privateNeedsBarrier)
static void printInReductionPrivateReductionRegion (OpAsmPrinter &p, Operation *op, Region &region, 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 &region, ValueRange privateVars, TypeRange privateTypes, ArrayAttr privateSyms, UnitAttr privateNeedsBarrier)
static void printPrivateReductionRegion (OpAsmPrinter &p, Operation *op, Region &region, 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 &region, ValueRange taskReductionVars, TypeRange taskReductionTypes, DenseBoolArrayAttr taskReductionByref, ArrayAttr taskReductionSyms)
static void printUseDeviceAddrUseDevicePtrRegion (OpAsmPrinter &p, Operation *op, Region &region, 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 > &copyprivateVars, SmallVectorImpl< Type > &copyprivateTypes, ArrayAttr &copyprivateSyms)
 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)
 depend-entry-list ::= depend-entry | depend-entry-list , depend-entry depend-entry ::= depend-kind -> ssa-id : type
static void printDependVarList (OpAsmPrinter &p, Operation *op, OperandRange dependVars, TypeRange dependTypes, std::optional< ArrayAttr > dependKinds)
 Print Depend clause.
static LogicalResult verifyDependVarList (Operation *op, std::optional< ArrayAttr > dependKinds, OperandRange dependVars)
 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 OperationfindCapturedOmpOp (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 OperationgetParentInSameDialect (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 &region, 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 &region, 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

Macro Definition Documentation

◆ GET_ATTRDEF_CLASSES

#define GET_ATTRDEF_CLASSES

Definition at line 4820 of file OpenMPDialect.cpp.

◆ GET_ATTRDEF_LIST

#define GET_ATTRDEF_LIST

◆ GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 4823 of file OpenMPDialect.cpp.

◆ GET_OP_LIST

#define GET_OP_LIST

◆ GET_TYPEDEF_CLASSES

#define GET_TYPEDEF_CLASSES

Definition at line 4826 of file OpenMPDialect.cpp.

◆ GET_TYPEDEF_LIST

#define GET_TYPEDEF_LIST

Function Documentation

◆ canPromoteToNoLoop()

bool canPromoteToNoLoop ( Operation * capturedOp,
TeamsOp teamsOp,
WsloopOp * wsLoopOp )
static

Check if we can promote SPMD kernel to No-Loop kernel.

Definition at line 2423 of file OpenMPDialect.cpp.

References mlir::Operation::getParentOfType().

◆ checkApplyeesNesting()

LogicalResult checkApplyeesNesting ( TileOp op)
static

Check properties of the loop nest consisting of the transformation's applyees:

  1. They are nested inside each other
  2. They are perfectly nested (no code with side-effects in-between the loops)
  3. They are rectangular (loop bounds are invariant in respect to the outer loops)

TODO: Generalize for LoopTransformationInterface.

Definition at line 3814 of file OpenMPDialect.cpp.

References mlir::omp::decodeCli(), and success().

◆ clear()

◆ findCapturedOmpOp()

◆ generateLoopNestingName()

std::string generateLoopNestingName ( StringRef prefix,
CanonicalLoopOp op )
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 94 of file OpenMPDialect.cpp.

References b, mlir::Region::getBlocks(), mlir::Region::getParentOp(), mlir::Operation::getParentRegion(), mlir::Operation::getRegions(), mlir::Operation::hasTrait(), and isUnique().

◆ getParentInSameDialect()

Operation * getParentInSameDialect ( Operation * thisOp)
static

◆ getPrivateNeedsBarrierSpelling()

constexpr StringRef getPrivateNeedsBarrierSpelling ( )
inlinestaticconstexpr

Definition at line 855 of file OpenMPDialect.cpp.

Referenced by printClauseWithRegionArgs().

◆ if() [1/3]

if ( !memberIdxs. empty())

◆ if() [2/3]

if ( failed(parser.parseRSquare()) )

◆ if() [3/3]

if ( parser. parseCommaSeparatedListparseIndices)

◆ makeArrayAttr()

ArrayAttr makeArrayAttr ( MLIRContext * context,
llvm::ArrayRef< Attribute > attrs )
static

Definition at line 49 of file OpenMPDialect.cpp.

References ArrayAttr(), and nullptr.

◆ makeDenseBoolArrayAttr()

DenseBoolArrayAttr makeDenseBoolArrayAttr ( MLIRContext * ctx,
const ArrayRef< bool > boolArray )
static

Definition at line 55 of file OpenMPDialect.cpp.

References mlir::detail::DenseArrayAttrImpl< bool >::get(), and nullptr.

◆ makeDenseI64ArrayAttr()

DenseI64ArrayAttr makeDenseI64ArrayAttr ( MLIRContext * ctx,
const ArrayRef< int64_t > intArray )
static

◆ mapTypeToBool()

bool mapTypeToBool ( ClauseMapFlags value,
ClauseMapFlags flag )
static

Definition at line 1795 of file OpenMPDialect.cpp.

Referenced by convertClauseMapFlags(), and verifyMapClause().

◆ opInGlobalImplicitParallelRegion()

bool opInGlobalImplicitParallelRegion ( Operation * op)
static

◆ parseAffinityClause()

ParseResult parseAffinityClause ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & iterated,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & affinityVars,
SmallVectorImpl< Type > & iteratedTypes,
SmallVectorImpl< Type > & affinityVarTypes )
static

Definition at line 4676 of file OpenMPDialect.cpp.

References parseSplitIteratedList(), and success().

◆ parseAlignedClause()

ParseResult parseAlignedClause ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & alignedVars,
SmallVectorImpl< Type > & alignedTypes,
ArrayAttr & alignmentsAttr )
static

aligned ::= aligned ( aligned-list ) aligned-list := aligned-val | aligned-val aligned-list aligned-val := ssa-id-and-type -> alignment

Definition at line 518 of file OpenMPDialect.cpp.

References ArrayAttr(), mlir::AsmParser::parseCommaSeparatedList(), and success().

◆ parseAllocateAndAllocator()

ParseResult parseAllocateAndAllocator ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & allocateVars,
SmallVectorImpl< Type > & allocateTypes,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & allocatorVars,
SmallVectorImpl< Type > & allocatorTypes )
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 354 of file OpenMPDialect.cpp.

References mlir::AsmParser::parseArrow(), mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), mlir::OpAsmParser::parseOperand(), and success().

◆ parseBlockArgClause() [1/3]

ParseResult parseBlockArgClause ( OpAsmParser & parser,
llvm::SmallVectorImpl< OpAsmParser::Argument > & entryBlockArgs,
StringRef keyword,
std::optional< MapParseArgs > mapArgs )
static

◆ parseBlockArgClause() [2/3]

ParseResult parseBlockArgClause ( OpAsmParser & parser,
llvm::SmallVectorImpl< OpAsmParser::Argument > & entryBlockArgs,
StringRef keyword,
std::optional< PrivateParseArgs > privateArgs )
static

◆ parseBlockArgClause() [3/3]

ParseResult parseBlockArgClause ( OpAsmParser & parser,
llvm::SmallVectorImpl< OpAsmParser::Argument > & entryBlockArgs,
StringRef keyword,
std::optional< ReductionParseArgs > reductionArgs )
static

◆ parseBlockArgRegion()

◆ parseCaptureType()

ParseResult parseCaptureType ( OpAsmParser & parser,
VariableCaptureKindAttr & mapCaptureType )
static

◆ parseClauseAttr()

template<typename ClauseAttr>
ParseResult parseClauseAttr ( AsmParser & parser,
ClauseAttr & attr )
static

◆ parseClauseWithRegionArgs()

◆ parseCopyprivate()

ParseResult parseCopyprivate ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & copyprivateVars,
SmallVectorImpl< Type > & copyprivateTypes,
ArrayAttr & copyprivateSyms )
static

copyprivate-entry-list ::= copyprivate-entry | copyprivate-entry-list , copyprivate-entry copyprivate-entry ::= ssa-id -> symbol-ref : type

Definition at line 1540 of file OpenMPDialect.cpp.

References ArrayAttr(), mlir::AsmParser::getContext(), mlir::AsmParser::parseCommaSeparatedList(), and success().

◆ parseDependVarList()

ParseResult parseDependVarList ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & dependVars,
SmallVectorImpl< Type > & dependTypes,
ArrayAttr & dependKinds )
static

depend-entry-list ::= depend-entry | depend-entry-list , depend-entry depend-entry ::= depend-kind -> ssa-id : type

Definition at line 1644 of file OpenMPDialect.cpp.

References ArrayAttr(), mlir::AsmParser::getContext(), mlir::AsmParser::parseCommaSeparatedList(), and success().

◆ parseGrainsizeClause()

ParseResult parseGrainsizeClause ( OpAsmParser & parser,
ClauseGrainsizeTypeAttr & grainsizeMod,
std::optional< OpAsmParser::UnresolvedOperand > & grainsize,
Type & grainsizeType )
static

Definition at line 767 of file OpenMPDialect.cpp.

References parseGranularityClause().

◆ parseGranularityClause()

template<typename ClauseTypeAttr, typename ClauseType>
ParseResult parseGranularityClause ( OpAsmParser & parser,
ClauseTypeAttr & prescriptiveness,
std::optional< OpAsmParser::UnresolvedOperand > & operand,
Type & operandType,
std::optional< ClauseType >(* symbolizeClause )(StringRef),
StringRef clauseName )
static

◆ parseInReductionPrivateReductionRegion()

ParseResult parseInReductionPrivateReductionRegion ( OpAsmParser & parser,
Region & region,
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

Definition at line 1106 of file OpenMPDialect.cpp.

References ArrayAttr(), and parseBlockArgRegion().

◆ parseInReductionPrivateRegion()

ParseResult parseInReductionPrivateRegion ( OpAsmParser & parser,
Region & region,
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

Definition at line 1090 of file OpenMPDialect.cpp.

References ArrayAttr(), and parseBlockArgRegion().

◆ parseIteratorHeader()

◆ parseLinearClause()

ParseResult parseLinearClause ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & linearVars,
SmallVectorImpl< Type > & linearTypes,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & linearStepVars,
SmallVectorImpl< Type > & linearStepTypes )
static

linear ::= linear ( linear-list ) linear-list := linear-val | linear-val linear-list linear-val := ssa-id-and-type = ssa-id-and-type

Definition at line 422 of file OpenMPDialect.cpp.

References mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), mlir::AsmParser::parseEqual(), mlir::OpAsmParser::parseOperand(), and success().

◆ parseLoopTransformClis()

◆ parseNumTasksClause()

ParseResult parseNumTasksClause ( OpAsmParser & parser,
ClauseNumTasksTypeAttr & numTasksMod,
std::optional< OpAsmParser::UnresolvedOperand > & numTasks,
Type & numTasksType )
static

Definition at line 789 of file OpenMPDialect.cpp.

References parseGranularityClause().

◆ parseOrderClause()

ParseResult parseOrderClause ( OpAsmParser & parser,
ClauseOrderKindAttr & order,
OrderModifierAttr & orderMod )
static

◆ parsePrivateReductionRegion()

ParseResult parsePrivateReductionRegion ( OpAsmParser & parser,
Region & region,
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

Definition at line 1138 of file OpenMPDialect.cpp.

References ArrayAttr(), and parseBlockArgRegion().

◆ parsePrivateRegion()

ParseResult parsePrivateRegion ( OpAsmParser & parser,
Region & region,
llvm::SmallVectorImpl< OpAsmParser::UnresolvedOperand > & privateVars,
llvm::SmallVectorImpl< Type > & privateTypes,
ArrayAttr & privateSyms,
UnitAttr & privateNeedsBarrier )
static

Definition at line 1127 of file OpenMPDialect.cpp.

References ArrayAttr(), and parseBlockArgRegion().

◆ parseScheduleClause()

ParseResult parseScheduleClause ( OpAsmParser & parser,
ClauseScheduleKindAttr & scheduleAttr,
ScheduleModifierAttr & scheduleMod,
UnitAttr & scheduleSimd,
std::optional< OpAsmParser::UnresolvedOperand > & chunkSize,
Type & chunkType )
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 596 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().

◆ parseSplitIteratedList()

template<typename ParsePrefixFn>
ParseResult parseSplitIteratedList ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & iteratedVars,
SmallVectorImpl< Type > & iteratedTypes,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & plainVars,
SmallVectorImpl< Type > & plainTypes,
ParsePrefixFn && parsePrefix )
static

◆ parseSynchronizationHint()

ParseResult parseSynchronizationHint ( OpAsmParser & parser,
IntegerAttr & hintAttr )
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 1708 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().

◆ parseTargetAllocMemOp()

◆ parseTargetOpRegion()

ParseResult parseTargetOpRegion ( OpAsmParser & parser,
Region & region,
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

Definition at line 1065 of file OpenMPDialect.cpp.

References ArrayAttr(), and parseBlockArgRegion().

◆ parseTaskReductionRegion()

ParseResult parseTaskReductionRegion ( OpAsmParser & parser,
Region & region,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & taskReductionVars,
SmallVectorImpl< Type > & taskReductionTypes,
DenseBoolArrayAttr & taskReductionByref,
ArrayAttr & taskReductionSyms )
static

Definition at line 1154 of file OpenMPDialect.cpp.

References ArrayAttr(), and parseBlockArgRegion().

◆ parseUniformClause()

ParseResult parseUniformClause ( OpAsmParser & parser,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & uniformVars,
SmallVectorImpl< Type > & uniformTypes )
static

uniform ::= uniform ( uniform-list ) uniform-list := uniform-val (, uniform-val)* uniform-val := ssa-id : type */

Definition at line 4651 of file OpenMPDialect.cpp.

References mlir::AsmParser::parseColonType(), mlir::AsmParser::parseCommaSeparatedList(), and mlir::OpAsmParser::parseOperand().

◆ parseUseDeviceAddrUseDevicePtrRegion()

ParseResult parseUseDeviceAddrUseDevicePtrRegion ( OpAsmParser & parser,
Region & region,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & useDeviceAddrVars,
SmallVectorImpl< Type > & useDeviceAddrTypes,
SmallVectorImpl< OpAsmParser::UnresolvedOperand > & useDevicePtrVars,
SmallVectorImpl< Type > & useDevicePtrTypes )
static

Definition at line 1165 of file OpenMPDialect.cpp.

References parseBlockArgRegion().

◆ printAffinityClause()

void printAffinityClause ( OpAsmPrinter & p,
Operation * op,
ValueRange iterated,
ValueRange affinityVars,
TypeRange iteratedTypes,
TypeRange affinityVarTypes )
static

Definition at line 4689 of file OpenMPDialect.cpp.

References printSplitIteratedList().

◆ printAlignedClause()

void printAlignedClause ( OpAsmPrinter & p,
Operation * op,
ValueRange alignedVars,
TypeRange alignedTypes,
std::optional< ArrayAttr > alignments )
static

Print Aligned Clause.

Definition at line 539 of file OpenMPDialect.cpp.

References mlir::ValueRange::getType().

◆ printAllocateAndAllocator()

void printAllocateAndAllocator ( OpAsmPrinter & p,
Operation * op,
OperandRange allocateVars,
TypeRange allocateTypes,
OperandRange allocatorVars,
TypeRange allocatorTypes )
static

Print allocate clause.

Definition at line 380 of file OpenMPDialect.cpp.

◆ printBlockArgClause() [1/3]

void printBlockArgClause ( OpAsmPrinter & p,
MLIRContext * ctx,
StringRef clauseName,
ValueRange argsSubrange,
std::optional< MapPrintArgs > mapArgs )
static

Definition at line 1273 of file OpenMPDialect.cpp.

References printClauseWithRegionArgs().

Referenced by printBlockArgRegion().

◆ printBlockArgClause() [2/3]

void printBlockArgClause ( OpAsmPrinter & p,
MLIRContext * ctx,
StringRef clauseName,
ValueRange argsSubrange,
std::optional< PrivatePrintArgs > privateArgs )
static

Definition at line 1281 of file OpenMPDialect.cpp.

References printClauseWithRegionArgs().

◆ printBlockArgClause() [3/3]

void printBlockArgClause ( OpAsmPrinter & p,
MLIRContext * ctx,
StringRef clauseName,
ValueRange argsSubrange,
std::optional< ReductionPrintArgs > reductionArgs )
static

Definition at line 1292 of file OpenMPDialect.cpp.

References printClauseWithRegionArgs().

◆ printBlockArgRegion()

◆ printCaptureType()

void printCaptureType ( OpAsmPrinter & p,
Operation * op,
VariableCaptureKindAttr mapCaptureType )
static

Definition at line 2007 of file OpenMPDialect.cpp.

◆ printClauseAttr()

template<typename ClauseAttr>
void printClauseAttr ( OpAsmPrinter & p,
Operation * op,
ClauseAttr attr )
static

Definition at line 411 of file OpenMPDialect.cpp.

◆ printClauseWithRegionArgs()

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

◆ printCopyprivate()

void printCopyprivate ( OpAsmPrinter & p,
Operation * op,
OperandRange copyprivateVars,
TypeRange copyprivateTypes,
std::optional< ArrayAttr > copyprivateSyms )
static

Print Copyprivate clause.

Definition at line 1560 of file OpenMPDialect.cpp.

◆ printDependVarList()

void printDependVarList ( OpAsmPrinter & p,
Operation * op,
OperandRange dependVars,
TypeRange dependTypes,
std::optional< ArrayAttr > dependKinds )
static

Print Depend clause.

Definition at line 1669 of file OpenMPDialect.cpp.

◆ printGrainsizeClause()

void printGrainsizeClause ( OpAsmPrinter & p,
Operation * op,
ClauseGrainsizeTypeAttr grainsizeMod,
Value grainsize,
mlir::Type grainsizeType )
static

Definition at line 775 of file OpenMPDialect.cpp.

References printGranularityClause().

◆ printGranularityClause()

template<typename ClauseTypeAttr, typename ClauseType>
void printGranularityClause ( OpAsmPrinter & p,
Operation * op,
ClauseTypeAttr prescriptiveness,
Value operand,
mlir::Type operandType,
StringRef(* stringifyClauseType )(ClauseType) )
static

Definition at line 749 of file OpenMPDialect.cpp.

Referenced by printGrainsizeClause(), and printNumTasksClause().

◆ printInReductionPrivateReductionRegion()

void printInReductionPrivateReductionRegion ( OpAsmPrinter & p,
Operation * op,
Region & region,
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

Definition at line 1368 of file OpenMPDialect.cpp.

References ArrayAttr(), and printBlockArgRegion().

◆ printInReductionPrivateRegion()

void printInReductionPrivateRegion ( OpAsmPrinter & p,
Operation * op,
Region & region,
ValueRange inReductionVars,
TypeRange inReductionTypes,
DenseBoolArrayAttr inReductionByref,
ArrayAttr inReductionSyms,
ValueRange privateVars,
TypeRange privateTypes,
ArrayAttr privateSyms,
UnitAttr privateNeedsBarrier )
static

Definition at line 1354 of file OpenMPDialect.cpp.

References ArrayAttr(), and printBlockArgRegion().

◆ printIteratorHeader()

◆ printLinearClause()

void printLinearClause ( OpAsmPrinter & p,
Operation * op,
ValueRange linearVars,
TypeRange linearTypes,
ValueRange linearStepVars,
TypeRange stepVarTypes )
static

Print Linear Clause.

Definition at line 446 of file OpenMPDialect.cpp.

◆ printLoopTransformClis() [1/2]

void printLoopTransformClis ( OpAsmPrinter & p,
FuseOp op,
OperandRange generatees,
OperandRange applyees )
static

Definition at line 3917 of file OpenMPDialect.cpp.

◆ printLoopTransformClis() [2/2]

void printLoopTransformClis ( OpAsmPrinter & p,
TileOp op,
OperandRange generatees,
OperandRange applyees )
static

Definition at line 3767 of file OpenMPDialect.cpp.

◆ printMembersIndex()

void printMembersIndex ( OpAsmPrinter & p,
MapInfoOp op,
ArrayAttr membersIdx )
static

Definition at line 1992 of file OpenMPDialect.cpp.

References ArrayAttr().

◆ printNumTasksClause()

void printNumTasksClause ( OpAsmPrinter & p,
Operation * op,
ClauseNumTasksTypeAttr numTasksMod,
Value numTasks,
mlir::Type numTasksType )
static

Definition at line 797 of file OpenMPDialect.cpp.

References printGranularityClause().

◆ printOrderClause()

void printOrderClause ( OpAsmPrinter & p,
Operation * op,
ClauseOrderKindAttr order,
OrderModifierAttr orderMod )
static

Definition at line 702 of file OpenMPDialect.cpp.

◆ printPrivateReductionRegion()

void printPrivateReductionRegion ( OpAsmPrinter & p,
Operation * op,
Region & region,
ValueRange privateVars,
TypeRange privateTypes,
ArrayAttr privateSyms,
UnitAttr privateNeedsBarrier,
ReductionModifierAttr reductionMod,
ValueRange reductionVars,
TypeRange reductionTypes,
DenseBoolArrayAttr reductionByref,
ArrayAttr reductionSyms )
static

Definition at line 1398 of file OpenMPDialect.cpp.

References ArrayAttr(), and printBlockArgRegion().

◆ printPrivateRegion()

void printPrivateRegion ( OpAsmPrinter & p,
Operation * op,
Region & region,
ValueRange privateVars,
TypeRange privateTypes,
ArrayAttr privateSyms,
UnitAttr privateNeedsBarrier )
static

Definition at line 1387 of file OpenMPDialect.cpp.

References ArrayAttr(), and printBlockArgRegion().

◆ printScheduleClause()

void printScheduleClause ( OpAsmPrinter & p,
Operation * op,
ClauseScheduleKindAttr scheduleKind,
ScheduleModifierAttr scheduleMod,
UnitAttr scheduleSimd,
Value scheduleChunk,
Type scheduleChunkType )
static

Print schedule clause.

Definition at line 658 of file OpenMPDialect.cpp.

References mlir::Value::getType().

◆ printSplitIteratedList()

template<typename PrintPrefixFn>
void printSplitIteratedList ( OpAsmPrinter & p,
ValueRange iteratedVars,
TypeRange iteratedTypes,
ValueRange plainVars,
TypeRange plainTypes,
PrintPrefixFn && printPrefixForPlain,
PrintPrefixFn && printPrefixForIterated )
static

Definition at line 1466 of file OpenMPDialect.cpp.

References emit().

Referenced by printAffinityClause().

◆ printSynchronizationHint()

void printSynchronizationHint ( OpAsmPrinter & p,
Operation * op,
IntegerAttr hintAttr )
static

Prints a Synchronization Hint clause.

Definition at line 1739 of file OpenMPDialect.cpp.

◆ printTargetOpRegion()

void printTargetOpRegion ( OpAsmPrinter & p,
Operation * op,
Region & region,
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

Definition at line 1334 of file OpenMPDialect.cpp.

References ArrayAttr(), and printBlockArgRegion().

◆ printTaskReductionRegion()

void printTaskReductionRegion ( OpAsmPrinter & p,
Operation * op,
Region & region,
ValueRange taskReductionVars,
TypeRange taskReductionTypes,
DenseBoolArrayAttr taskReductionByref,
ArrayAttr taskReductionSyms )
static

Definition at line 1413 of file OpenMPDialect.cpp.

References ArrayAttr(), and printBlockArgRegion().

◆ printUniformClause()

void printUniformClause ( OpAsmPrinter & p,
Operation * op,
ValueRange uniformVars,
TypeRange uniformTypes )
static

Print Uniform Clauses.

Definition at line 4663 of file OpenMPDialect.cpp.

◆ printUseDeviceAddrUseDevicePtrRegion()

void printUseDeviceAddrUseDevicePtrRegion ( OpAsmPrinter & p,
Operation * op,
Region & region,
ValueRange useDeviceAddrVars,
TypeRange useDeviceAddrTypes,
ValueRange useDevicePtrVars,
TypeRange useDevicePtrTypes )
static

Definition at line 1425 of file OpenMPDialect.cpp.

References printBlockArgRegion().

◆ push_back()

memberIdxs push_back ( ArrayAttr::get(parser.getContext(), values) )

◆ success()

◆ verifyAlignedClause()

LogicalResult verifyAlignedClause ( Operation * op,
std::optional< ArrayAttr > alignments,
OperandRange alignedVars )
static

Definition at line 478 of file OpenMPDialect.cpp.

References mlir::Operation::emitOpError(), and success().

◆ verifyCopyprivateVarList()

LogicalResult verifyCopyprivateVarList ( Operation * op,
OperandRange copyprivateVars,
std::optional< ArrayAttr > copyprivateSyms )
static

Verifies CopyPrivate Clause.

Definition at line 1576 of file OpenMPDialect.cpp.

References mlir::Operation::emitOpError(), mlir::SymbolTable::lookupNearestSymbolFrom(), and success().

◆ verifyDependVarList()

LogicalResult verifyDependVarList ( Operation * op,
std::optional< ArrayAttr > dependKinds,
OperandRange dependVars )
static

Verifies Depend clause.

Definition at line 1684 of file OpenMPDialect.cpp.

References mlir::Operation::emitOpError(), and success().

◆ verifyMapClause()

LogicalResult verifyMapClause ( Operation * op,
OperandRange mapVars )
static

◆ verifyMapInfoDefinedArgs()

LogicalResult verifyMapInfoDefinedArgs ( Operation * op,
StringRef clauseName,
OperandRange vars )
static

Definition at line 2135 of file OpenMPDialect.cpp.

References mlir::Operation::emitOpError(), and success().

◆ verifyNontemporalClause()

LogicalResult verifyNontemporalClause ( Operation * op,
OperandRange nontemporalVars )
static

Definition at line 463 of file OpenMPDialect.cpp.

References mlir::Operation::emitOpError(), and success().

◆ verifyNumTeamsClause()

LogicalResult verifyNumTeamsClause ( Operation * op,
Value numTeamsLower,
OperandRange numTeamsUpperVars )
static

◆ verifyOrderedParent()

◆ verifyPrivateVarList()

template<typename OpType>
LogicalResult verifyPrivateVarList ( OpType & op)
static

Definition at line 2582 of file OpenMPDialect.cpp.

References mlir::SymbolTable::lookupNearestSymbolFrom(), and success().

◆ verifyPrivateVarsMapping()

LogicalResult verifyPrivateVarsMapping ( TargetOp targetOp)
static

Definition at line 2113 of file OpenMPDialect.cpp.

References mlir::emitError(), and success().

◆ verifyReductionVarList()

LogicalResult verifyReductionVarList ( Operation * op,
std::optional< ArrayAttr > reductionSyms,
OperandRange reductionVars,
std::optional< ArrayRef< bool > > reductionByref )
static

◆ verifyScheduleModifiers()

ParseResult verifyScheduleModifiers ( OpAsmParser & parser,
SmallVectorImpl< SmallString< 12 > > & modifiers )
static

◆ verifySynchronizationHint()

LogicalResult verifySynchronizationHint ( Operation * op,
uint64_t hint )
static

Verifies a synchronization hint clause.

Definition at line 1770 of file OpenMPDialect.cpp.

References mlir::Operation::emitOpError(), and success().

◆ while()

while ( succeeded(parser.parseOptionalComma()) )

Variable Documentation

◆ do

do
Initial value:
{
if (failed(parser.parseLSquare()))
return failure()

Definition at line 1972 of file OpenMPDialect.cpp.

◆ false

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 1968 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().