mlir::OpBuilder Class Reference

This class helps build Operations. More...

#include "mlir/IR/Builders.h"

Inheritance diagram for mlir::OpBuilder:

Classes
class	InsertionGuard
	RAII guard to reset the insertion point of the builder when destroyed. More...
class	InsertPoint
	This class represents a saved insertion point. More...
struct	Listener
	This class represents a listener that may be used to hook into various actions within an OpBuilder. More...
struct	ListenerBase
	Base class for listeners. More...

Public Member Functions
	OpBuilder (MLIRContext *ctx, Listener *listener=nullptr)
	Create a builder with the given context. More...
	OpBuilder (Region *region, Listener *listener=nullptr)
	Create a builder and set the insertion point to the start of the region. More...
	OpBuilder (Region &region, Listener *listener=nullptr)
	OpBuilder (Operation *op, Listener *listener=nullptr)
	Create a builder and set insertion point to the given operation, which will cause subsequent insertions to go right before it. More...
	OpBuilder (Block *block, Block::iterator insertPoint, Listener *listener=nullptr)
void	setListener (Listener *newListener)
	Sets the listener of this builder to the one provided. More...
Listener *	getListener () const
	Returns the current listener of this builder, or nullptr if this builder doesn't have a listener. More...
void	clearInsertionPoint ()
	Reset the insertion point to no location. More...
InsertPoint	saveInsertionPoint () const
	Return a saved insertion point. More...
void	restoreInsertionPoint (InsertPoint ip)
	Restore the insert point to a previously saved point. More...
void	setInsertionPoint (Block *block, Block::iterator insertPoint)
	Set the insertion point to the specified location. More...
void	setInsertionPoint (Operation *op)
	Sets the insertion point to the specified operation, which will cause subsequent insertions to go right before it. More...
void	setInsertionPointAfter (Operation *op)
	Sets the insertion point to the node after the specified operation, which will cause subsequent insertions to go right after it. More...
void	setInsertionPointAfterValue (Value val)
	Sets the insertion point to the node after the specified value. More...
void	setInsertionPointToStart (Block *block)
	Sets the insertion point to the start of the specified block. More...
void	setInsertionPointToEnd (Block *block)
	Sets the insertion point to the end of the specified block. More...
Block *	getInsertionBlock () const
	Return the block the current insertion point belongs to. More...
Block::iterator	getInsertionPoint () const
	Returns the current insertion point of the builder. More...
Block *	getBlock () const
	Returns the current block of the builder. More...
Block *	createBlock (Region *parent, Region::iterator insertPt={}, TypeRange argTypes={}, ArrayRef< Location > locs={})
	Add new block with 'argTypes' arguments and set the insertion point to the end of it. More...
Block *	createBlock (Block *insertBefore, TypeRange argTypes={}, ArrayRef< Location > locs={})
	Add new block with 'argTypes' arguments and set the insertion point to the end of it. More...
Operation *	insert (Operation *op)
	Insert the given operation at the current insertion point and return it. More...
Operation *	create (const OperationState &state)
	Creates an operation given the fields represented as an OperationState. More...
Operation *	create (Location loc, StringAttr opName, ValueRange operands, TypeRange types={}, ArrayRef< NamedAttribute > attributes={}, BlockRange successors={}, MutableArrayRef< std::unique_ptr< Region >> regions={})
	Creates an operation with the given fields. More...
template<typename OpTy , typename... Args>
OpTy	create (Location location, Args &&...args)
	Create an operation of specific op type at the current insertion point. More...
template<typename OpTy , typename... Args>
void	createOrFold (SmallVectorImpl< Value > &results, Location location, Args &&...args)
	Create an operation of specific op type at the current insertion point, and immediately try to fold it. More...
template<typename OpTy , typename... Args>
std::enable_if_t< OpTy::template hasTrait< OpTrait::OneResult >), Value >	createOrFold (Location location, Args &&...args)
	Overload to create or fold a single result operation. More...
template<typename OpTy , typename... Args>
std::enable_if_t< OpTy::template hasTrait< OpTrait::ZeroResults >), OpTy >	createOrFold (Location location, Args &&...args)
	Overload to create or fold a zero result operation. More...
LogicalResult	tryFold (Operation *op, SmallVectorImpl< Value > &results, SmallVectorImpl< Operation * > *materializedConstants=nullptr)
	Attempts to fold the given operation and places new results within results. More...
Operation *	clone (Operation &op, IRMapping &mapper)
	Creates a deep copy of the specified operation, remapping any operands that use values outside of the operation using the map that is provided ( leaving them alone if no entry is present). More...
Operation *	clone (Operation &op)
Operation *	cloneWithoutRegions (Operation &op, IRMapping &mapper)
	Creates a deep copy of this operation but keep the operation regions empty. More...
Operation *	cloneWithoutRegions (Operation &op)
template<typename OpT >
OpT	cloneWithoutRegions (OpT op)
void	cloneRegionBefore (Region &region, Region &parent, Region::iterator before, IRMapping &mapping)
	Clone the blocks that belong to "region" before the given position in another region "parent". More...
void	cloneRegionBefore (Region &region, Region &parent, Region::iterator before)
void	cloneRegionBefore (Region &region, Block *before)
Public Member Functions inherited from mlir::Builder
	Builder (MLIRContext *context)
	Builder (Operation *op)
MLIRContext *	getContext () const
Location	getUnknownLoc ()
Location	getFusedLoc (ArrayRef< Location > locs, Attribute metadata=Attribute())
FloatType	getF8E8M0Type ()
FloatType	getBF16Type ()
FloatType	getF16Type ()
FloatType	getTF32Type ()
FloatType	getF32Type ()
FloatType	getF64Type ()
FloatType	getF80Type ()
FloatType	getF128Type ()
IndexType	getIndexType ()
IntegerType	getI1Type ()
IntegerType	getI2Type ()
IntegerType	getI4Type ()
IntegerType	getI8Type ()
IntegerType	getI16Type ()
IntegerType	getI32Type ()
IntegerType	getI64Type ()
IntegerType	getIntegerType (unsigned width)
IntegerType	getIntegerType (unsigned width, bool isSigned)
FunctionType	getFunctionType (TypeRange inputs, TypeRange results)
TupleType	getTupleType (TypeRange elementTypes)
NoneType	getNoneType ()
template<typename Ty , typename... Args>
Ty	getType (Args &&...args)
	Get or construct an instance of the type Ty with provided arguments. More...
template<typename Attr , typename... Args>
Attr	getAttr (Args &&...args)
	Get or construct an instance of the attribute Attr with provided arguments. More...
NamedAttribute	getNamedAttr (StringRef name, Attribute val)
UnitAttr	getUnitAttr ()
BoolAttr	getBoolAttr (bool value)
DictionaryAttr	getDictionaryAttr (ArrayRef< NamedAttribute > value)
IntegerAttr	getIntegerAttr (Type type, int64_t value)
IntegerAttr	getIntegerAttr (Type type, const APInt &value)
FloatAttr	getFloatAttr (Type type, double value)
FloatAttr	getFloatAttr (Type type, const APFloat &value)
StringAttr	getStringAttr (const Twine &bytes)
ArrayAttr	getArrayAttr (ArrayRef< Attribute > value)
TypedAttr	getZeroAttr (Type type)
TypedAttr	getOneAttr (Type type)
FloatAttr	getF16FloatAttr (float value)
FloatAttr	getF32FloatAttr (float value)
FloatAttr	getF64FloatAttr (double value)
IntegerAttr	getI8IntegerAttr (int8_t value)
IntegerAttr	getI16IntegerAttr (int16_t value)
IntegerAttr	getI32IntegerAttr (int32_t value)
IntegerAttr	getI64IntegerAttr (int64_t value)
IntegerAttr	getIndexAttr (int64_t value)
IntegerAttr	getSI32IntegerAttr (int32_t value)
	Signed and unsigned integer attribute getters. More...
IntegerAttr	getUI32IntegerAttr (uint32_t value)
DenseIntElementsAttr	getBoolVectorAttr (ArrayRef< bool > values)
	Vector-typed DenseIntElementsAttr getters. values must not be empty. More...
DenseIntElementsAttr	getI32VectorAttr (ArrayRef< int32_t > values)
DenseIntElementsAttr	getI64VectorAttr (ArrayRef< int64_t > values)
DenseIntElementsAttr	getIndexVectorAttr (ArrayRef< int64_t > values)
DenseFPElementsAttr	getF32VectorAttr (ArrayRef< float > values)
DenseFPElementsAttr	getF64VectorAttr (ArrayRef< double > values)
DenseIntElementsAttr	getI32TensorAttr (ArrayRef< int32_t > values)
	Tensor-typed DenseIntElementsAttr getters. More...
DenseIntElementsAttr	getI64TensorAttr (ArrayRef< int64_t > values)
DenseIntElementsAttr	getIndexTensorAttr (ArrayRef< int64_t > values)
DenseBoolArrayAttr	getDenseBoolArrayAttr (ArrayRef< bool > values)
	Tensor-typed DenseArrayAttr getters. More...
DenseI8ArrayAttr	getDenseI8ArrayAttr (ArrayRef< int8_t > values)
DenseI16ArrayAttr	getDenseI16ArrayAttr (ArrayRef< int16_t > values)
DenseI32ArrayAttr	getDenseI32ArrayAttr (ArrayRef< int32_t > values)
DenseI64ArrayAttr	getDenseI64ArrayAttr (ArrayRef< int64_t > values)
DenseF32ArrayAttr	getDenseF32ArrayAttr (ArrayRef< float > values)
DenseF64ArrayAttr	getDenseF64ArrayAttr (ArrayRef< double > values)
ArrayAttr	getAffineMapArrayAttr (ArrayRef< AffineMap > values)
ArrayAttr	getBoolArrayAttr (ArrayRef< bool > values)
ArrayAttr	getI32ArrayAttr (ArrayRef< int32_t > values)
ArrayAttr	getI64ArrayAttr (ArrayRef< int64_t > values)
ArrayAttr	getIndexArrayAttr (ArrayRef< int64_t > values)
ArrayAttr	getF32ArrayAttr (ArrayRef< float > values)
ArrayAttr	getF64ArrayAttr (ArrayRef< double > values)
ArrayAttr	getStrArrayAttr (ArrayRef< StringRef > values)
ArrayAttr	getTypeArrayAttr (TypeRange values)
AffineExpr	getAffineDimExpr (unsigned position)
AffineExpr	getAffineSymbolExpr (unsigned position)
AffineExpr	getAffineConstantExpr (int64_t constant)
AffineMap	getEmptyAffineMap ()
	Returns a zero result affine map with no dimensions or symbols: () -> (). More...
AffineMap	getConstantAffineMap (int64_t val)
	Returns a single constant result affine map with 0 dimensions and 0 symbols. More...
AffineMap	getDimIdentityMap ()
AffineMap	getMultiDimIdentityMap (unsigned rank)
AffineMap	getSymbolIdentityMap ()
AffineMap	getSingleDimShiftAffineMap (int64_t shift)
	Returns a map that shifts its (single) input dimension by 'shift'. More...
AffineMap	getShiftedAffineMap (AffineMap map, int64_t shift)
	Returns an affine map that is a translation (shift) of all result expressions in 'map' by 'shift'. More...

Static Public Member Functions
static OpBuilder	atBlockBegin (Block *block, Listener *listener=nullptr)
	Create a builder and set the insertion point to before the first operation in the block but still inside the block. More...
static OpBuilder	atBlockEnd (Block *block, Listener *listener=nullptr)
	Create a builder and set the insertion point to after the last operation in the block but still inside the block. More...
static OpBuilder	atBlockTerminator (Block *block, Listener *listener=nullptr)
	Create a builder and set the insertion point to before the block terminator. More...

Protected Attributes
Listener *	listener
	The optional listener for events of this builder. More...
Protected Attributes inherited from mlir::Builder
MLIRContext *	context

Detailed Description

This class helps build Operations.

Operations that are created are automatically inserted at an insertion point. The builder is copyable.

Definition at line 205 of file Builders.h.

Constructor & Destructor Documentation

OpBuilder() [1/5]

mlir::OpBuilder::OpBuilder ( MLIRContext *  ctx, Listener *  listener = nullptr  )

inlineexplicit

Create a builder with the given context.

Definition at line 211 of file Builders.h.

Referenced by atBlockBegin(), atBlockEnd(), and atBlockTerminator().

OpBuilder() [2/5]

mlir::OpBuilder::OpBuilder ( Region *  region, Listener *  listener = nullptr  )

inlineexplicit

Create a builder and set the insertion point to the start of the region.

Definition at line 215 of file Builders.h.

References mlir::Region::empty(), mlir::Region::front(), and setInsertionPointToStart().

OpBuilder() [3/5]

mlir::OpBuilder::OpBuilder ( Region &  region, Listener *  listener = nullptr  )

inlineexplicit

Definition at line 220 of file Builders.h.

OpBuilder() [4/5]

mlir::OpBuilder::OpBuilder ( Operation *  op, Listener *  listener = nullptr  )

inlineexplicit

Create a builder and set insertion point to the given operation, which will cause subsequent insertions to go right before it.

Definition at line 225 of file Builders.h.

References setInsertionPoint().

OpBuilder() [5/5]

mlir::OpBuilder::OpBuilder ( Block *  block, Block::iterator  insertPoint, Listener *  listener = nullptr  )

inline

Definition at line 230 of file Builders.h.

References setInsertionPoint().

Member Function Documentation

atBlockBegin()

static OpBuilder mlir::OpBuilder::atBlockBegin ( Block *  block, Listener *  listener = nullptr  )

inlinestatic

Create a builder and set the insertion point to before the first operation in the block but still inside the block.

Definition at line 238 of file Builders.h.

References mlir::Block::begin(), listener, and OpBuilder().

Referenced by createSingleEntryBlock(), createSingleExitingLatch(), legalizeBlockArguments(), mlir::affine::normalizeAffineParallel(), mlir::transformCFGToSCF(), and transformCyclesToSCFLoops().

atBlockEnd()

static OpBuilder mlir::OpBuilder::atBlockEnd ( Block *  block, Listener *  listener = nullptr  )

inlinestatic

Create a builder and set the insertion point to after the last operation in the block but still inside the block.

Definition at line 244 of file Builders.h.

References mlir::Block::end(), listener, and OpBuilder().

Referenced by mlir::FunctionCallBuilder::create(), and transformToStructuredCFBranches().

atBlockTerminator()

static OpBuilder mlir::OpBuilder::atBlockTerminator ( Block *  block, Listener *  listener = nullptr  )

inlinestatic

Create a builder and set the insertion point to before the block terminator.

Definition at line 250 of file Builders.h.

References mlir::Block::getTerminator(), listener, and OpBuilder().

Referenced by createFullTiles(), generatePointWiseCopy(), generateShiftedLoop(), generateUnrolledLoop(), stripmineSink(), and transformToStructuredCFBranches().

clearInsertionPoint()

void mlir::OpBuilder::clearInsertionPoint ( )

inline

Reset the insertion point to no location.

Creating an operation without a set insertion point is an error, but this can still be useful when the current insertion point a builder refers to is being removed.

Definition at line 377 of file Builders.h.

Referenced by mlir::bufferization::buildDeallocationLibraryFunction(), and restoreInsertionPoint().

clone() [1/2]

Operation * OpBuilder::clone

(

Operation &

op

)

Definition at line 570 of file Builders.cpp.

References clone().

clone() [2/2]

Operation * OpBuilder::clone	(	Operation &	op,
		IRMapping &	mapper
	)

Creates a deep copy of the specified operation, remapping any operands that use values outside of the operation using the map that is provided ( leaving them alone if no entry is present).

Replaces references to cloned sub-operations to the corresponding operation that is copied, and adds those mappings to the map.

Definition at line 548 of file Builders.cpp.

References mlir::Operation::clone(), mlir::Operation::getRegions(), insert(), listener, notifyBlockInsertions(), mlir::OpBuilder::Listener::notifyOperationInserted(), and mlir::PreOrder.

Referenced by mlir::sparse_tensor::Merger::buildExp(), buildPackingLoopNestImpl(), clone(), cloneAndFuseFirstUse(), cloneAsInsertSlice< tensor::InsertSliceOp >(), mlir::async::cloneConstantsIntoTheRegion(), cloneOpAndUpdateDestinationArgs(), createFullPartialVectorTransferRead(), createFullPartialVectorTransferWrite(), createFullTiles(), createParallelComputeFunction(), mlir::fuseIndependentSiblingForallLoops(), mlir::fuseIndependentSiblingForLoops(), mlir::affine::fuseLoops(), generateCleanupLoopForUnroll(), generateFusedElementwiseOpRegion(), genForeachOnSparseConstant(), hoistAffineIfOp(), inlinePayload(), inlineRegionAndEmitStore(), mlir::affine::insertBackwardComputationSlice(), mlir::loopUnrollByFactor(), mlir::affine::loopUnrollJamByFactor(), mlir::loopUnrollJamByFactor(), mlir::makeRegionIsolatedFromAbove(), CanonicalizeContractAdd< AddOpType >::matchAndRewrite(), AllocaScopeHoister::matchAndRewrite(), mlir::detail::CreateAlgebraicOpForEndomorphismSimplification::operator()(), mlir::outlineSingleBlockRegion(), peelForLoop(), reifyOpResultShapes(), mlir::sinkOperationsIntoLaunchOp(), mlir::vector::splitFullAndPartialTransfer(), mlir::linalg::splitReduction(), mlir::mesh::spmdizeFullyReplicatedOperation(), mlir::mesh::spmdizeOperation(), mlir::mesh::spmdizeTriviallyShardableOperation(), mlir::scf::tileAndFuseConsumerOfSlices(), tileAndFuseFirstExtractUse(), tileAndFuseFirstExtractUseThroughContainingOpBlockArgument(), tileLinalgOpImpl(), and mlir::linalg::tileReductionUsingForall().

cloneRegionBefore() [1/3]

void OpBuilder::cloneRegionBefore	(	Region &	region,
		Block *	before
	)

Definition at line 601 of file Builders.cpp.

References cloneRegionBefore(), and mlir::Block::getParent().

cloneRegionBefore() [2/3]

void OpBuilder::cloneRegionBefore	(	Region &	region,
		Region &	parent,
		Region::iterator	before
	)

Definition at line 595 of file Builders.cpp.

References cloneRegionBefore().

cloneRegionBefore() [3/3]

void OpBuilder::cloneRegionBefore	(	Region &	region,
		Region &	parent,
		Region::iterator	before,
		IRMapping &	mapping
	)

Clone the blocks that belong to "region" before the given position in another region "parent".

The two regions must be different. The caller is responsible for creating or updating the operation transferring flow of control to the region and passing it the correct block arguments.

Definition at line 575 of file Builders.cpp.

References mlir::Region::cloneInto(), mlir::Region::front(), listener, mlir::IRMapping::lookup(), mlir::OpBuilder::Listener::notifyBlockInserted(), notifyBlockInsertions(), mlir::OpBuilder::Listener::notifyOperationInserted(), and mlir::PreOrder.

Referenced by cloneRegionBefore(), createExpandedGenericOp(), genCoIterateBranchNest(), inlineIfCase(), and inlineWhileCase().

cloneWithoutRegions() [1/3]

Operation* mlir::OpBuilder::cloneWithoutRegions ( Operation &  op )

inline

Definition at line 588 of file Builders.h.

References mlir::Operation::cloneWithoutRegions(), and insert().

cloneWithoutRegions() [2/3]

Operation* mlir::OpBuilder::cloneWithoutRegions ( Operation &  op, IRMapping &  mapper  )

inline

Creates a deep copy of this operation but keep the operation regions empty.

Operands are remapped using mapper (if present), and mapper is updated to contain the results.

Definition at line 585 of file Builders.h.

References mlir::Operation::cloneWithoutRegions(), and insert().

Referenced by cloneWithoutRegions(), and wrapInExecuteRegion().

cloneWithoutRegions() [3/3]

template<typename OpT >

OpT mlir::OpBuilder::cloneWithoutRegions ( OpT  op )

inline

Definition at line 592 of file Builders.h.

References cloneWithoutRegions().

create() [1/3]

Operation * OpBuilder::create

(

const OperationState &

state

)

Creates an operation given the fields represented as an OperationState.

Create an operation given the fields represented as an OperationState.

Definition at line 452 of file Builders.cpp.

Referenced by mlir::ArithBuilder::_and(), mlir::ArithBuilder::add(), addAtomicRMW(), addBodyWithPayloadOp(), addInitOperandsToLoopNest(), adjustAccessChainForBitwidth(), mlir::affine::affineParallelize(), mlir::UnrankedMemRefDescriptor::alignedPtr(), mlir::sparse_tensor::allocaBuffer(), mlir::UnrankedMemRefDescriptor::allocatedPtr(), allocateSubviewGPUMemoryInAddressSpace(), mlir::sparse_tensor::allocDenseTensor(), allocSchemeForRank(), mlir::transform::gpu::alterGpuLaunch(), mlir::OpToFuncCallLowering< SourceOp >::appendOrGetFuncOp(), appendX1Dim(), applyPad(), atomicRMW(), bitAndAddrspaceCast(), bitcastSubByteVectorToI8(), broadcast(), broadcastDynamicDimension(), mlir::tensor::bubbleUpPadSlice(), mlir::linalg::bufferizeToAllocation(), mlir::MemRefDescriptor::bufferPtr(), HopperBuilder::buildAndInitBarrierInSharedMemory(), buildArithValue(), HopperBuilder::buildBarrierArriveTx(), buildBoolValue(), mlir::bufferization::buildDeallocationLibraryFunction(), buildDivOp(), mlir::sparse_tensor::Merger::buildExp(), HopperBuilder::buildGlobalMemRefDescriptor(), buildIdentityRegion(), mlir::tensor::buildIndependentOp(), buildLinearId(), mlir::scf::buildLoopNest(), buildMinMaxReductionSeq(), MmaSyncBuilder::buildMmaSync(), buildMultiDimReduce(), buildNumReadElements(), buildPackingLoopNestImpl(), HopperBuilder::buildPredicateLoadsOnThread0(), buildPredicates(), buildQuantifier(), mlir::sparse_tensor::buildRelu(), buildSubAndExpOp(), mlir::emitc::buildTerminatedBody(), mlir::scf::buildTerminatedBody(), mlir::vector::buildTerminatedBody(), HopperBuilder::buildTmaAsyncLoad(), HopperBuilder::buildTryWaitParity(), mlir::detail::ConversionPatternRewriterImpl::buildUnresolvedMaterialization(), mlir::vector::castAwayContractionLeadingOneDim(), castF32To(), castMFMAScaleOperand(), mlir::bufferization::castOrReallocMemRefValue(), castToCompatibleMemRefType(), castToF32(), ceilDivPositive(), checkAndNestUnderRewriteOp(), mlir::tosa::checkHasDynamicBatchDims(), mlir::tosa::clampFloatHelper(), clampInput(), mlir::tosa::clampIntHelper(), cloneAsInsertSlice< tensor::ParallelInsertSliceOp >(), cloneOpWithOperandsAndTypes(), cloneToCollapsedOp< GenericOp >(), mlir::affine::coalesceLoops(), collapseInnerDims(), mlir::linalg::collapseOpIterationDims(), mlir::collapseParallelLoops(), collapseValue(), mlir::spirv::combine(), common3DIdBuilderFn(), mlir::LLVM::composeValue(), mlir::UnrankedMemRefDescriptor::computeSizes(), computeStrides(), computeTargetSize(), concatSizesFromInputs(), mlir::sparse_tensor::constantI1(), mlir::sparse_tensor::constantI16(), mlir::sparse_tensor::constantI32(), mlir::sparse_tensor::constantI64(), mlir::sparse_tensor::constantI8(), mlir::sparse_tensor::constantIndex(), mlir::sparse_tensor::constantOne(), mlir::sparse_tensor::constantZero(), mlir::detail::constructContainerOpForParserIfNecessary(), constructTiledLoopNest(), convertBroadcastOp(), convertConstantOp(), convertConstantOpMmaSync(), convertContractOp(), convertContractOpToMmaSync(), convertCtlzOp(), mlir::complex::convertDivToLLVMUsingAlgebraic(), mlir::complex::convertDivToLLVMUsingRangeReduction(), mlir::complex::convertDivToStandardUsingAlgebraic(), mlir::complex::convertDivToStandardUsingRangeReduction(), convertElementwiseOp(), convertExtractStridedSlice(), convertFPowIOp(), mlir::convertFuncOpToLLVMFuncOp(), convertIntrinsicResult(), mlir::LLVM::ModuleImport::convertLinkerOptionsMetadata(), convertMFMAVectorOperand(), mlir::LLVM::ModuleImport::convertModuleFlagsMetadata(), mlir::convertOpResultTypes(), convertRsqrtOp(), convertTanhOp(), convertToLaunchFuncOp(), convertTransferReadOp(), convertTransferWriteOp(), convertTransferWriteToStores(), convertUnsignedToI32(), convertYieldOp(), convVals(), copy(), mlir::linalg::copyToGPUPrivateMemory(), mlir::linalg::copyToWorkgroupMemory(), mlir::ConvertToLLVMPattern::copyUnrankedDescriptors(), mlir::FunctionCallBuilder::create(), createAdd(), mlir::linalg::createAdd(), mlir::affine::createAffineComputationSlice(), mlir::bufferization::BufferizationOptions::createAlloc(), createAllocation(), createAllocFields(), createAsyncDispatchFunction(), mlir::nvgpu::createAsyncGroups(), createBinarySearchFunc(), mlir::affine::createCanonicalizedAffineForOp(), mlir::ControlFlowToSCFTransformation::createCFGSwitchOp(), createChoosePivot(), createCompareThenSwap(), createConst(), createConstantAllBitsSet(), createContractArithOp(), createCtlzFunc(), createDecl(), mlir::tensor::createDynamicDimValues(), createEqCompare(), createExpandedGenericOp(), createExpandedTransposeOp(), mlir::emitc::createExpression(), mlir::linalg::DecomposePadOpPattern::createFillOrGenerateOp(), createFloatConst(), createFor(), createForAllDimensions(), createFPConstant(), createFullPartialLinalgCopy(), createFullPartialVectorTransferRead(), createFullPartialVectorTransferWrite(), mlir::sparse_tensor::createFuncCall(), createGlobalVarForEntryPointArgument(), mlir::transform::gpu::createGpuLaunch(), createGroupReduceOpImpl(), createHeapSortFunc(), createI1Constant(), createI32Constant(), createI32ConstantOf(), createInBoundsCond(), createIndex(), mlir::ConvertToLLVMPattern::createIndexAttrConstant(), createIndexAttrConstant(), createInlinedCompareImplementation(), createInserts(), createIntConst(), createLessThanCompare(), createLinalgBodyCalculationForElementwiseOp(), createLinalgBodyCalculationForReduceOp(), mlir::arm_sme::createLoopOverTileSlices(), createMaskDropNonScalableUnitDims(), mlir::bufferization::BufferizationOptions::createMemCpy(), createMemcpy(), createMul(), mlir::linalg::createMul(), createNewDynamicSizes(), createNonLdMatrixLoads(), mlir::sparse_tensor::createOrFoldSliceOffsetOp(), mlir::sparse_tensor::createOrFoldSliceStrideOp(), mlir::tosa::createPadConstTensor(), mlir::tensor::createPadHighOp(), createPartitionFunc(), mlir::LLVM::createPrintStrCall(), createPrivateMemRef(), mlir::arith::createProduct(), createPushback(), createQuickSort(), createQuickSortFunc(), mlir::vector::createReadOrMaskedRead(), mlir::createScalarOrSplatConstant(), createScanLoop(), createSeparationCondition(), createShiftDownFunc(), createSortStableFunc(), createSPIRVBuiltinCall(), createSplitPart(), mlir::ControlFlowToSCFTransformation::createStructuredBranchRegionOp(), mlir::ControlFlowToSCFTransformation::createStructuredBranchRegionTerminatorOp(), mlir::ControlFlowToSCFTransformation::createStructuredDoWhileLoopOp(), createSubApply(), createSubTwoDividedByTwo(), createSubViewIntersection(), createSwap(), createSwitchOp(), createTypeCanonicalizedMemRefOperands(), mlir::ControlFlowToSCFTransformation::createUnreachableTerminator(), createVectorLoadForMaskedLoad(), mlir::xegpu::createVectorWithShapeFromValues(), mlir::tosa::createZeroPointTensor(), creatLdMatrixCompatibleLoads(), mlir::sparse_tensor::deallocDenseTensor(), mlir::affine::decompose(), mlir::LLVM::decomposeValue(), mlir::linalg::deduplicateOperandsAndRemoveDeadResults(), defaultDeallocBufferCallBack(), delinearizeInductionVariable(), mlir::gpu::WarpDistributionPattern::delinearizeLaneId(), mlir::denormalizeInductionVariable(), mlir::sparse_tensor::SparseIterator::deref(), deriveStaticUpperBound(), mlir::xegpu::doSCFStructuralTypeConversionWithTensorType(), doSequentialDispatch(), doubleBuffer(), downcastSelectAndUpcast(), mlir::bufferization::dropEquivalentBufferResults(), mlir::tensor::dropGivenUnitDims(), dropRefIfNoUses(), dropTrailingX1Dim(), mlir::linalg::dropUnitDims(), dumpIndexMemRef(), dynamicallyExtractSubVector(), dynamicallyInsertSubVector(), emitElementwiseComputation(), mlir::tensor::ExtractSliceFromCollapseHelper::emitLoopNestBody(), emitScalarImplementation(), emulatedVectorLoad(), endIf(), mlir::sparse_tensor::LoopEmitter::enterCoIterationOverTensorsAtLvls(), mlir::sparse_tensor::LoopEmitter::exitCurrentLoop(), expandValue(), extendMaskRank(), extendVectorRank(), extractConvFilterSlices(), extractConvInputSlices(), extractConvResultSlices(), mlir::extractFixedOuterLoops(), extractLastDimSlice(), extractNBitsPerByteAndExtendToI8(), extractNBitsPerByteAndSignExtendToI8(), extractOne(), mlir::StructBuilder::extractPtr(), extractSliceIntoByte(), mlir::xegpu::extractVectorsWithShapeFromValue(), finalizeWhileOp(), flattenVecToBits(), forEachIJPairInXs(), mlir::sparse_tensor::foreachInSparseConstant(), mlir::sparse_tensor::SparseIterator::forward(), mlir::sparse_tensor::SparseIterator::forwardIf(), mlir::linalg::fuseElementwiseOps(), fuseIfLegal(), mlir::fuseIndependentSiblingForallLoops(), mlir::fuseIndependentSiblingForLoops(), mlir::linalg::fuseProducerOfTensor(), fuseWithReshapeByExpansion(), mlir::sparse_tensor::genAlloca(), mlir::sparse_tensor::genAllocaScalar(), genBuffers(), mlir::sparse_tensor::FuncCallOrInlineGenerator< SubClass >::genCallOrInline(), mlir::sparse_tensor::genCast(), genCoIterateBranchNest(), mlir::sparse_tensor::genCoIteration(), genCompressed(), genConditionalInsert(), genConstFloat32From(), genConstInt32From(), genDemap(), genEndInsert(), mlir::linalg::generalizeNamedOp(), generateCollapsedIndexingRegion(), generateCopy(), generateFusedElementwiseOpRegion(), generateLoopNestUsingForallOp(), generateLoopNestUsingForOp(), mlir::linalg::generateParallelLoopNest(), generatePointWiseCopy(), generateShiftedLoop(), genExpand(), genIf(), mlir::sparse_tensor::genIndexLoad(), mlir::sparse_tensor::SparseIterator::genInit(), genInsertionLoad(), genInsertionLoadReduce(), genInsertionStore(), mlir::sparse_tensor::genIsNonzero(), genLoad(), genLoopWithIterator(), mlir::sparse_tensor::genMapBuffers(), mlir::sparse_tensor::SparseIterator::genNotEnd(), mlir::sparse_tensor::genReader(), genRemap(), mlir::sparse_tensor::genReshapeDstShape(), mlir::sparse_tensor::genScalarToTensor(), genSliceToSize(), genStmt(), genStore(), genTensorLoad(), genTensorStore(), mlir::sparse_tensor::genToMemref(), mlir::sparse_tensor::genTuple(), mlir::sparse_tensor::genValFromAttr(), genWhenInBound(), getAsLLVMValue(), mlir::ControlFlowToSCFTransformation::getCFGSwitchValue(), getCleanupLoopLowerBound(), getCollapsedIndices(), getCollapsedOpOperand(), getCompressedMaskOp(), mlir::tosa::getConstTensorInt(), getConvOrPoolOutputDim(), mlir::sparse_tensor::SparseTensorDescriptor::getCrdMemRefOrView(), getDynamicSharedMemorySymbol(), getFlatMemref(), getFlattenMemrefAndOffset(), mlir::sparse_tensor::getFunc(), mlir::bufferization::getGlobalFor(), getIndexedPtrs(), getIndicesForAccess(), mlir::sparse_tensor::SparseTensorSpecifier::getInitValue(), getLaneId(), getLibraryCallSymbolRef(), getLinearIndexI32(), getLinearizedSrcIndices(), getLocationToWriteFullVec(), getMangledSortHelperFunc(), mlir::ConvertToLLVMPattern::getMemRefDescriptorSizes(), mlir::bufferization::DeallocationState::getMemrefsAndConditionsToDeallocate(), mlir::bufferization::DeallocationState::getMemrefWithUniqueOwnership(), mlir::mesh::getMixedAsValues(), mlir::ConvertToLLVMPattern::getNumElements(), getNumRecords(), getOffsetForBitwidth(), mlir::spirv::getOpenCLElementPtr(), mlir::tensor::getOrCreateDestination(), getOrCreateOperandsMatchingResultTypes(), mlir::getOrCreateRanges(), getOrCreateStep(), mlir::getOrCreateStringConstant(), mlir::getOrDefineFunction(), getProductOfIntsOrIndexes(), mlir::spirv::getPushConstantValue(), getScalarOrVectorConstInt(), getScalarOrVectorI32Constant(), mlir::ConvertToLLVMPattern::getSizeInBytes(), getSlice(), mlir::sparse_tensor::SparseTensorSpecifier::getSpecifierField(), getStride(), mlir::LLVM::getStridedElementPtr(), getSubByteWriteMask(), getTensorDim(), getTileSizes(), mlir::ControlFlowToSCFTransformation::getUndefValue(), getValueFromOpFoldResult(), mlir::getValueOrCreateCastToIndexLike(), mlir::getValueOrCreateConstantIndexOp(), mlir::getValueOrCreateConstantIntOp(), mlir::vector::getVectorReductionOp(), mlir::spirv::getVulkanElementPtr(), getZero(), mlir::LLVM::detail::handleMultidimensionalVectors(), mlir::mesh::handlePartialAxesDuringResharding(), hoistAffineIfOp(), mlir::linalg::hoistPaddingOnTensors(), inferDynamicDimsForConv(), mlir::inferExpandShapeOutputShape(), mlir::sparse_tensor::LoopEmitter::initializeLoopEmit(), inlineIfCase(), inlinePayload(), inlineRegionAndEmitStore(), inlineWhileCase(), insertCasts(), insertConvResultSlices(), insertLastDimSlice(), insertOne(), mlir::linalg::insertSlicesBack(), mlir::sparse_tensor::insertYieldOp(), mlir::LLVM::detail::intrinsicRewrite(), invertCollapseShapeIndexing(), linalgBroadcastAndMaybeExt(), linalgIntBroadcastExtSIAdd(), mlir::LLVMTypeConverter::LLVMTypeConverter(), mlir::sparse_tensor::loadAll(), mlir::sparse_tensor::SparseIterator::locate(), lookupOrCreateSPIRVFn(), mlir::loopUnrollByFactor(), mlir::affine::loopUnrollJamByFactor(), mlir::affine::lowerAffineDelinearizeIndexOp(), mlir::affine::lowerAffineLinearizeIndexOp(), lowerAsEntryFunction(), lowerEntryPointABIAttr(), mlir::linalg::lowerPack(), mlir::scf::lowerToLoopsUsingSCFForOp(), mlir::linalg::lowerUnPack(), makeBufferRsrc(), makeCanonicalAffineApplies(), mlir::linalg::makeMemRefCopyOp(), mlir::sparse_tensor::makeSparseTensorLevel(), mlir::vector::makeVscaleConstantBuilder(), mlir::transform::gpu::mapForallToBlocksImpl(), mlir::affine::mapLoopToProcessorIds(), mlir::transform::gpu::mapNestedForallToThreadsImpl(), mlir::transform::gpu::mapOneForallToThreadsImpl(), matchAndReplaceDepthwiseConv(), YieldOpLowering::matchAndRewrite(), AssertOpLowering::matchAndRewrite(), mlir::ComposeCollapseOfExpandOp< CollapseOpTy, ExpandOpTy, CastOpTy, DimOpTy, TensorTy >::matchAndRewrite(), CollapseShapeOpMemRefCastFolder::matchAndRewrite(), FoldTensorCastProducerOp::matchAndRewrite(), WrapFuncInClass::matchAndRewrite(), MultiBlockExecuteInliner::matchAndRewrite(), DecomposeNDExtractStridedSlice::matchAndRewrite(), mlir::GPUDynamicSharedMemoryOpLowering::matchAndRewrite(), mlir::GPUFuncOpLowering::matchAndRewrite(), mlir::GPUPrintfOpToHIPLowering::matchAndRewrite(), mlir::GPUPrintfOpToLLVMCallLowering::matchAndRewrite(), mlir::GPUPrintfOpToVPrintfLowering::matchAndRewrite(), mlir::GPUReturnOpLowering::matchAndRewrite(), IndexSwitchOpLowering::matchAndRewrite(), DecomposeDifferentRankInsertStridedSlice::matchAndRewrite(), ConvertSameRankInsertStridedSliceIntoShuffle::matchAndRewrite(), FoldLaunchArguments::matchAndRewrite(), mlir::linalg::DecomposeOuterUnitDimsPackOpPattern::matchAndRewrite(), SwapTransposeWithBroadcast::matchAndRewrite(), mlir::linalg::DecomposeOuterUnitDimsUnPackOpPattern::matchAndRewrite(), MulExtendedFold< MulOp, IsSigned >::matchAndRewrite(), mlir::gpu::index_lowering::OpLowering< Op, XOp, YOp, ZOp >::matchAndRewrite(), mlir::OpWithOffsetSizesAndStridesConstantArgumentFolder< OpType, ResultTypeFn, CastOpFunc >::matchAndRewrite(), mlir::linalg::FoldTensorCastPackOp::matchAndRewrite(), mlir::OpToFuncCallLowering< SourceOp >::matchAndRewrite(), ConvertAccessChain::matchAndRewrite(), IAddCarryFold::matchAndRewrite(), ConvertLoad::matchAndRewrite(), ConvertStore::matchAndRewrite(), SwapExtractSliceOfFill::matchAndRewrite(), mlir::linalg::DecomposePadOpPattern::matchAndRewrite(), ConcatSliceOptimization::matchAndRewrite(), PadSliceOptimization::matchAndRewrite(), SliceDynamicSizeCanonicalization::matchAndRewrite(), mlir::linalg::FoldTensorCastUnPackOp::matchAndRewrite(), mlir::linalg::LinalgCopyVTRForwardingPattern::matchAndRewrite(), mlir::linalg::LinalgCopyVTWForwardingPattern::matchAndRewrite(), matchAndRewriteSortOp(), materializeBinaryNanCheckIfRequired(), mlir::affine::materializeComputedBound(), mlir::linalg::materializeTiledShape(), mlir::OpToFuncCallLowering< SourceOp >::maybeCast(), mlir::mesh::maybeInsertSourceShardingAnnotation(), maybeInsertTargetShardingAnnotationImpl(), movePaddingToFillOrGenericOp(), mlir::gpu::WarpDistributionPattern::moveRegionToNewWarpOpAndReplaceReturns(), mlir::ArithBuilder::mul(), mlir::memref::multiBuffer(), nonAtomicRMW(), mlir::affine::normalizeAffineFor(), normalizeForallLoopOp(), mlir::normalizeForallOp(), mlir::affine::normalizeMemRef(), mlir::MemRefDescriptor::offset(), mlir::UnrankedMemRefDescriptor::offset(), mlir::UnrankedMemRefDescriptor::offsetBasePtr(), mlir::LLVM::detail::oneToOneRewrite(), SliceCanonicalizer::operator()(), optionallyTruncateOrExtend(), outlineKernelFuncImpl(), mlir::outlineSingleBlockRegion(), mlir::linalg::pack(), padOperand(), padOperandToSmallestStaticBoundingBox(), padThroughLoopIterArg(), permuteVectorOffset(), mlir::MemRefDescriptor::poison(), mlir::UnrankedMemRefDescriptor::poison(), mlir::ComplexStructBuilder::poison(), mlir::bufferization::populateDynamicDimSizes(), mlir::arith::populateEmulateUnsupportedFloatsConversions(), mlir::math::populateExtendToSupportedTypesTypeConverter(), mlir::tosa::populateTosaTypeConversion(), mlir::LLVM::ModuleImport::processFunction(), processParallelLoop(), mlir::scf::promote(), mlir::affine::promoteIfSingleIteration(), mlir::LLVMTypeConverter::promoteOneMemRefDescriptor(), mlir::linalg::promoteSubviewAsNewBuffer(), propagateSubViewOp(), rankedMemRefMaterialization(), rankReducingSubviewDroppingUnitDims(), reduce(), reduceMatchAndRewriteHelper(), reifyOpResultShapes(), reifyOrComputeDynamicSizes(), mlir::arith::reifyValueBound(), relinkBranch(), remapValueRange(), mlir::affine::replaceAllMemRefUsesWith(), mlir::scf::replaceAndCastForOpIterArg(), replaceAndPropagateMemRefType(), replaceByPackingResult(), replaceConstantUsesOf(), mlir::tensor::replaceExtractSliceWithTiledProducer(), replaceForAllWithNewSignature(), replaceForOpWithNewSignature(), mlir::bufferization::replaceOpWithBufferizedValues(), mlir::bufferization::replaceOpWithNewBufferizedOp(), replaceOpWithPredicatedOp(), replaceUsesAndPropagateType(), replaceWithDifferentYield(), mlir::sparse_tensor::reshapeCvs(), reshapeLoad(), reshapeStore(), resolveSourceIndicesCollapseShape(), resolveSourceIndicesExpandShape(), restoreByValRefArgumentType(), mlir::linalg::DownscaleSizeOneWindowed2DConvolution< Conv2DOp, Conv1DOp >::returningMatchAndRewrite(), mlir::linalg::DownscaleConv2DOp::returningMatchAndRewrite(), mlir::linalg::DownscaleDepthwiseConv2DNhwcHwcOp::returningMatchAndRewrite(), mlir::linalg::rewriteAsPaddedOp(), rewriteI2ToI8Ext(), rewriteI4ToI8Ext(), rewriteI8ToI4Trunc(), mlir::linalg::rewriteInDestinationPassingStyle(), mlir::linalg::rewriteInIm2Col(), rewriteOneForallCommonImpl(), PadOpVectorizationWithInsertSlicePattern::rewriteUser(), scalarizeVectorOpHelper(), mlir::ArithBuilder::select(), mlir::UnrankedMemRefDescriptor::setAlignedPtr(), mlir::UnrankedMemRefDescriptor::setAllocatedPtr(), mlir::UnrankedMemRefDescriptor::setOffset(), mlir::MemRefDescriptor::setOffset(), mlir::StructBuilder::setPtr(), mlir::UnrankedMemRefDescriptor::setSize(), mlir::MemRefDescriptor::setSize(), mlir::sparse_tensor::SparseTensorSpecifier::setSpecifierField(), mlir::UnrankedMemRefDescriptor::setStride(), mlir::MemRefDescriptor::setStride(), mlir::linalg::LinalgTilingOptions::setTileSizes(), mlir::ArithBuilder::sgt(), shiftValue(), mlir::tensor::simplifyCollapseShapeWithRankReducingExtractSlice(), mlir::UnrankedMemRefDescriptor::size(), mlir::MemRefDescriptor::size(), mlir::UnrankedMemRefDescriptor::sizeBasePtr(), sizesForTensor(), sliceLoadStoreIndices(), sliceTransferIndices(), mlir::ArithBuilder::slt(), sourceMaterializationCallback(), mlir::SparseIterationTypeConverter::SparseIterationTypeConverter(), specializeForLoopForUnrolling(), specializeParallelLoopForUnrolling(), mlir::SPIRVTypeConverter::SPIRVTypeConverter(), mlir::vector::splitFullAndPartialTransfer(), mlir::linalg::splitReduction(), mlir::linalg::splitReductionByScaling(), staticallyExtractSubvector(), staticallyInsertSubvector(), mlir::sparse_tensor::storeAll(), mlir::UnrankedMemRefDescriptor::stride(), mlir::MemRefDescriptor::stride(), mlir::UnrankedMemRefDescriptor::strideBasePtr(), stripmineSink(), mlir::ArithBuilder::sub(), mlir::scf::tileAndFuseConsumerOfSlices(), mlir::linalg::tileReductionUsingForall(), mlir::linalg::transposeBatchMatmul(), mlir::linalg::transposeMatmul(), transposeOneLinalgOperandAndReplace(), transposeToShuffle1D(), truncOrExtToLLVMType(), unrankedMemRefMaterialization(), mlir::linalg::unrollIndex(), updateCalls(), updateExpandedGenericOpRegion(), updateReturnOps(), vectorizeAsInsertSliceOp(), vectorizeAsTensorPackOp(), vectorizeAsTensorPadOp(), vectorizeAsTensorUnpackOp(), mlir::linalg::vectorizeCopy(), mlir::LLVM::detail::vectorOneToOneRewrite(), wmmaPushInputOperand(), wmmaPushOutputOperand(), wrapExternalFunction(), wrapForExternalCallers(), wrapInExecuteRegion(), and mlir::scf::yieldReplacementForFusedProducer().

create() [2/3]

Operation * OpBuilder::create	(	Location	loc,
		StringAttr	opName,
		ValueRange	operands,
		TypeRange	types = {},
		ArrayRef< NamedAttribute >	attributes = {},
		BlockRange	successors = {},
		MutableArrayRef< std::unique_ptr< Region >>	regions = {}
	)

Creates an operation with the given fields.

Definition at line 457 of file Builders.cpp.

create() [3/3]

template<typename OpTy , typename... Args>

OpTy mlir::OpBuilder::create ( Location  location, Args &&...  args  )

inline

Create an operation of specific op type at the current insertion point.

Definition at line 503 of file Builders.h.

createBlock() [1/2]

Block * OpBuilder::createBlock	(	Block *	insertBefore,
		TypeRange	argTypes = {},
		ArrayRef< Location >	locs = {}
	)

Add new block with 'argTypes' arguments and set the insertion point to the end of it.

The block is placed before 'insertBefore'. locs contains the locations of the inserted arguments, and should match the size of argTypes.

The block is placed before 'insertBefore'.

Definition at line 444 of file Builders.cpp.

References createBlock(), and mlir::Block::getParent().

createBlock() [2/2]

Block * OpBuilder::createBlock	(	Region *	parent,
		Region::iterator	insertPt = {},
		TypeRange	argTypes = {},
		ArrayRef< Location >	locs = {}
	)

Add new block with 'argTypes' arguments and set the insertion point to the end of it.

The block is inserted at the provided insertion point of 'parent'. locs contains the locations of the inserted arguments, and should match the size of argTypes.

Definition at line 425 of file Builders.cpp.

References mlir::Block::addArguments(), mlir::Region::end(), mlir::Region::getBlocks(), listener, mlir::OpBuilder::Listener::notifyBlockInserted(), and setInsertionPointToEnd().

Referenced by addAtomicRMW(), mlir::detail::ConversionPatternRewriterImpl::applySignatureConversion(), buildGenericRegion(), buildQuantifier(), checkAndNestUnderRewriteOp(), createAsyncDispatchFunction(), createBinarySearchFunc(), createBlock(), createDecl(), createElementFPowIFunc(), createElementIPowIFunc(), createParallelComputeFunction(), createPartitionFunc(), createQuickSortFunc(), createScanLoop(), createShiftDownFunc(), mlir::ControlFlowToSCFTransformation::createStructuredDoWhileLoopOp(), ensureDistinctSuccessors(), mlir::impl::ensureRegionTerminator(), fillStructuredOpRegion(), mlir::sparse_tensor::genCoIteration(), generateFusedElementwiseOpRegion(), genLoopWithIterator(), mlir::makeRegionIsolatedFromAbove(), WrapFuncInClass::matchAndRewrite(), mlirLinalgFillBuiltinNamedOpRegion(), movePaddingToFillOrGenericOp(), mlir::outlineSingleBlockRegion(), mlir::LLVM::ModuleImport::processFunction(), mlir::linalg::rewriteInDestinationPassingStyle(), mlir::RewriterBase::splitBlock(), and mlir::mesh::spmdizeBlock().

createOrFold() [1/3]

template<typename OpTy , typename... Args>

std::enable_if_t<OpTy::template hasTrait<OpTrait::OneResult>), Value> mlir::OpBuilder::createOrFold ( Location  location, Args &&...  args  )

inline

Overload to create or fold a single result operation.

Definition at line 545 of file Builders.h.

createOrFold() [2/3]

template<typename OpTy , typename... Args>

std::enable_if_t<OpTy::template hasTrait<OpTrait::ZeroResults>), OpTy> mlir::OpBuilder::createOrFold ( Location  location, Args &&...  args  )

inline

Overload to create or fold a zero result operation.

Definition at line 554 of file Builders.h.

References tryFold().

createOrFold() [3/3]

template<typename OpTy , typename... Args>

void mlir::OpBuilder::createOrFold ( SmallVectorImpl< Value > &  results, Location  location, Args &&...  args  )

inline

Create an operation of specific op type at the current insertion point, and immediately try to fold it.

This functions populates 'results' with the results of the operation.

Definition at line 517 of file Builders.h.

Referenced by adjustAccessChainForBitwidth(), broadcastDynamicDimension(), broadcastIfNeeded(), buildLoopIterationCount(), canonicalizeAssociativeCommutativeBinaryOp(), mlir::vector::castAwayContractionLeadingOneDim(), castBoolToIntN(), castIntNToBool(), castIntValueToSameSizedType(), castSameSizedTypes(), castToSameSizedInt(), clampInput(), mlir::linalg::computeContinuousTileSizes(), computeStrides(), convertIntrinsicResult(), mlir::tensor::createCanonicalRankReducingExtractSliceOp(), mlir::tensor::createCanonicalRankReducingInsertSliceOp(), mlir::memref::createCanonicalRankReducingSubViewOp(), createExtractAndCast(), createInsertAndCast(), mlir::vector::createOrFoldDimOp(), mlir::linalg::createOrFoldDimOp(), mlir::createScalarOrSplatConstant(), doubleBuffer(), emitElementwiseComputation(), mlir::emitNormalizedLoopBounds(), expandValue(), generateCollapsedIndexingRegion(), getCollapsedOutputDimFromInputShape(), mlir::memref::getMixedSize(), mlir::tensor::getMixedSize(), getOffsetForBitwidth(), mlir::inferExpandShapeOutputShape(), mlir::spirv::linearizeIndex(), mlir::loopUnrollJamByFactor(), mlir::affine::lowerAffineDelinearizeIndexOp(), makeBufferRsrc(), mlir::linalg::DecomposePadOpPattern::matchAndRewrite(), peelForLoop(), permuteVectorOffset(), mlir::linalg::promoteSubviewAsNewBuffer(), shiftValue(), and wmmaPushInputOperand().

getBlock()

Block* mlir::OpBuilder::getBlock ( ) const

inline

Returns the current block of the builder.

Definition at line 446 of file Builders.h.

Referenced by mlir::sparse_tensor::Merger::buildExp(), mlir::FunctionCallBuilder::create(), and mlir::sparse_tensor::createFuncCall().

getInsertionBlock()

Block* mlir::OpBuilder::getInsertionBlock ( ) const

inline

Return the block the current insertion point belongs to.

Note that the insertion point is not necessarily the end of the block.

Definition at line 440 of file Builders.h.

Referenced by mlir::affine::affineParallelize(), checkAndNestUnderRewriteOp(), mlir::LLVM::createGlobalString(), mlir::sparse_tensor::LoopEmitter::enterCoIterationOverTensorsAtLvls(), finalizeWhileOp(), mlir::affine::makeComposedFoldedAffineApply(), makeComposedFoldedMinMax(), replaceWithConstant(), and saveInsertionPoint().

getInsertionPoint()

Block::iterator mlir::OpBuilder::getInsertionPoint ( ) const

inline

Returns the current insertion point of the builder.

Definition at line 443 of file Builders.h.

Referenced by mlir::affine::affineParallelize(), broadcastIfNeeded(), mlir::convertFuncOpToLLVMFuncOp(), mlir::ConvertToLLVMPattern::copyUnrankedDescriptors(), mlir::affine::makeComposedFoldedAffineApply(), makeComposedFoldedMinMax(), materializeConstant(), rewriteOneForallCommonImpl(), saveInsertionPoint(), mlir::scf::tileAndFuseConsumerOfSlices(), and mlir::scf::yieldReplacementForFusedProducer().

getListener()

Listener* mlir::OpBuilder::getListener ( ) const

inline

Returns the current listener of this builder, or nullptr if this builder doesn't have a listener.

Definition at line 318 of file Builders.h.

References listener.

Referenced by createAsyncDispatchFunction(), mlir::LLVM::createGlobalString(), createParallelComputeFunction(), mlir::affine::makeComposedFoldedAffineApply(), makeComposedFoldedMinMax(), mlir::PatternApplicator::matchAndRewrite(), mlir::affine::simplifyAffineMinMaxOps(), mlir::scf::tileConsumerAndFuseProducersUsingSCF(), and mlir::OperationFolder::tryToFold().

insert()

Operation * OpBuilder::insert

(

Operation *

op

)

Insert the given operation at the current insertion point and return it.

Definition at line 416 of file Builders.cpp.

References mlir::Block::getOperations(), listener, and mlir::OpBuilder::Listener::notifyOperationInserted().

Referenced by clone(), cloneWithoutRegions(), mlir::spirv::combine(), and mlir::impl::ensureRegionTerminator().

restoreInsertionPoint()

void mlir::OpBuilder::restoreInsertionPoint ( InsertPoint  ip )

inline

Restore the insert point to a previously saved point.

Definition at line 388 of file Builders.h.

References clearInsertionPoint(), mlir::OpBuilder::InsertPoint::getBlock(), mlir::OpBuilder::InsertPoint::getPoint(), mlir::OpBuilder::InsertPoint::isSet(), and setInsertionPoint().

Referenced by mlir::LLVM::createPrintStrCall(), and mlir::OpBuilder::InsertionGuard::~InsertionGuard().

saveInsertionPoint()

InsertPoint mlir::OpBuilder::saveInsertionPoint ( ) const

inline

Return a saved insertion point.

Definition at line 383 of file Builders.h.

References getInsertionBlock(), and getInsertionPoint().

Referenced by mlir::LLVM::createPrintStrCall().

setInsertionPoint() [1/2]

void mlir::OpBuilder::setInsertionPoint ( Block *  block, Block::iterator  insertPoint  )

inline

Set the insertion point to the specified location.

Definition at line 396 of file Builders.h.

Referenced by addInitOperandsToLoopNest(), mlir::memref::allocToAlloca(), applyTilingToAll(), mlir::linalg::bufferizeToAllocation(), HopperBuilder::buildGlobalMemRefDescriptor(), mlir::tensor::buildIndependentOp(), buildPackingLoopNestImpl(), mlir::bufferization::buildSubsetExtraction(), mlir::detail::ConversionPatternRewriterImpl::buildUnresolvedMaterialization(), canonicalizeMinMaxOp(), cloneAndFuseFirstUse(), mlir::coalesceLoops(), mlir::linalg::collapseOpIterationDims(), mlir::collapseParallelLoops(), convertAffineApply(), convertBroadcastOp(), convertConstantOp(), convertConstantOpMmaSync(), convertContractOp(), convertContractOpToMmaSync(), convertElementwiseOp(), convertExtractStridedSlice(), convertForOp(), convertTransferReadOp(), convertTransferReadToLoads(), convertTransferWriteOp(), convertTransferWriteToStores(), convertYieldOp(), mlir::nvgpu::createAsyncGroups(), createDecl(), createGlobalVarForEntryPointArgument(), createNonLdMatrixLoads(), creatLdMatrixCompatibleLoads(), declareReduction(), mlir::bufferization::dropEquivalentBufferResults(), mlir::sparse_tensor::LoopEmitter::exitCurrentLoop(), fuseIfLegal(), mlir::linalg::fuseProducerOfTensor(), fuseWithReshapeByExpansion(), mlir::sparse_tensor::FuncCallOrInlineGenerator< SubClass >::genCallOrInline(), generateLoopNestUsingForallOp(), getGenericOpLoopRange(), getLibraryCallSymbolRef(), getMangledSortHelperFunc(), mlir::tensor::getOrCreateDestination(), handleInlinedAllocas(), VectorizationState::initState(), inlineIfCase(), inlineWhileCase(), mlir::bufferization::insertTensorCopies(), mlir::linalg::interchangeGenericOp(), legalizeUnresolvedMaterialization(), mlir::loopUnrollJamByFactor(), mlir::linalg::lowerPack(), mlir::scf::lowerToLoopsUsingSCFForOp(), mlir::linalg::lowerUnPack(), mlir::affine::makeComposedFoldedAffineApply(), makeComposedFoldedMinMax(), AllocaScopeHoister::matchAndRewrite(), MultiBlockExecuteInliner::matchAndRewrite(), mlir::GPUDynamicSharedMemoryOpLowering::matchAndRewrite(), mlir::PatternApplicator::matchAndRewrite(), InsertSliceOfInsertSliceFolder< OpTy >::matchAndRewrite(), mlir::mesh::maybeInsertSourceShardingAnnotation(), mlir::gpu::WarpDistributionPattern::moveRegionToNewWarpOpAndReplaceReturns(), mlir::memref::multiBuffer(), mlir::normalizeForallOp(), OpBuilder(), mlir::nvgpu::optimizeSharedMemoryReadsAndWrites(), mlir::outlineSingleBlockRegion(), mlir::linalg::packTranspose(), peelForLoop(), pipelineForSharedCopies(), populateOpPayload(), processParallelLoop(), propagateSubViewOp(), relinkBranch(), mlir::scf::replaceAndCastForOpIterArg(), replaceByPackingResult(), replaceForAllWithNewSignature(), replaceForOpWithNewSignature(), replaceUnitDimIndexOps(), replaceUsesAndPropagateType(), replaceWithDifferentYield(), restoreInsertionPoint(), CopyBuilder::rewrite(), rewrite(), rewriteOneForallCommonImpl(), PadOpVectorizationWithInsertSlicePattern::rewriteUser(), PadOpVectorizationWithTransferWritePattern::rewriteUser(), setInsertionPoint(), setInsertionPointAfter(), setInsertionPointToEnd(), setInsertionPointToStart(), mlir::vector::splitFullAndPartialTransfer(), mlir::linalg::splitReduction(), mlir::linalg::splitReductionByScaling(), mlir::scf::tileAndFuseConsumerOfSlices(), tileAndFuseFirstExtractUse(), tileAndFuseFirstExtractUseThroughContainingOpBlockArgument(), mlir::scf::tileAndFuseProducerOfSlice(), tileLinalgOpImpl(), mlir::linalg::tileReductionUsingForall(), vectorizeAsInsertSliceOp(), vectorizeAsTensorPackOp(), vectorizeAsTensorPadOp(), vectorizeAsTensorUnpackOp(), wrapInExecuteRegion(), and mlir::scf::yieldReplacementForFusedProducer().

setInsertionPoint() [2/2]

void mlir::OpBuilder::setInsertionPoint ( Operation *  op )

inline

Sets the insertion point to the specified operation, which will cause subsequent insertions to go right before it.

Definition at line 404 of file Builders.h.

References mlir::Operation::getBlock(), and setInsertionPoint().

setInsertionPointAfter()

void mlir::OpBuilder::setInsertionPointAfter ( Operation *  op )

inline

Sets the insertion point to the node after the specified operation, which will cause subsequent insertions to go right after it.

Definition at line 410 of file Builders.h.

References mlir::Operation::getBlock(), and setInsertionPoint().

Referenced by mlir::linalg::blockPackMatmul(), createAsyncDispatchFunction(), createBinarySearchFunc(), createChoosePivot(), mlir::emitc::createExpression(), createHeapSortFunc(), createInlinedCompareImplementation(), createInsert3rd(), createPartitionFunc(), createQuickSort(), createQuickSortFunc(), createScanLoop(), createShiftDownFunc(), createSort3(), createSort5(), createSortStableFunc(), destructureSlot(), doubleBuffer(), dropRefIfNoUses(), mlir::sparse_tensor::LoopEmitter::exitCurrentLoop(), finalizeWhileOp(), mlir::sparse_tensor::SparseIterator::forwardIf(), mlir::fuseIndependentSiblingForallLoops(), mlir::fuseIndependentSiblingForLoops(), genCoIterateBranchNest(), genCompressed(), genConditionalInsert(), genEndInsert(), genInsertionStore(), genTensorStore(), genWhenInBound(), hoistAffineIfOp(), mlir::linalg::hoistPaddingOnTensors(), mlir::inlineCall(), mlir::loopUnrollByFactor(), mlir::affine::loopUnrollJamByFactor(), mlir::transform::gpu::mapOneForallToThreadsImpl(), IndexSwitchOpLowering::matchAndRewrite(), mlir::linalg::offsetIndices(), padThroughLoopIterArg(), peelForLoop(), mlir::scf::pipelineForLoop(), mlir::scf::promote(), promoteSubViews(), mlir::scf::replaceAndCastForOpIterArg(), replaceAndPropagateMemRefType(), mlir::bufferization::replaceOpWithBufferizedValues(), mlir::linalg::rewriteAsPaddedOp(), mlir::linalg::rewriteInDestinationPassingStyle(), setInsertionPointAfter(), setInsertionPointAfterValue(), setInsertionPointToStart(), mlir::linalg::tileReductionUsingForall(), mlir::scf::tileUsingSCF(), and updateExpandedGenericOpRegion().

setInsertionPointAfterValue()

void mlir::OpBuilder::setInsertionPointAfterValue ( Value  val )

inline

Sets the insertion point to the node after the specified value.

If value has a defining operation, sets the insertion point to the node after such defining operation. This will cause subsequent insertions to go right after it. Otherwise, value is a BlockArgument. Sets the insertion point to the start of its block.

Definition at line 419 of file Builders.h.

References mlir::Value::getDefiningOp(), setInsertionPointAfter(), and setInsertionPointToStart().

Referenced by mlir::transform::gpu::alterGpuLaunch(), createInlinedCompareImplementation(), mlir::mesh::handlePartialAxesDuringResharding(), maybeInsertTargetShardingAnnotationImpl(), mlir::mesh::moveLastSplitAxisInResharding(), and mlir::mesh::unsplitLastAxisInResharding().

setInsertionPointToEnd()

void mlir::OpBuilder::setInsertionPointToEnd ( Block *  block )

inline

Sets the insertion point to the end of the specified block.

Definition at line 434 of file Builders.h.

References mlir::Block::end(), and setInsertionPoint().

Referenced by addAtomicRMW(), mlir::scf::buildLoopNest(), buildPackingLoopNestImpl(), mlir::LLVM::ModuleImport::convertMetadata(), createAsyncDispatchFunction(), createBinarySearchFunc(), createBlock(), createDecl(), mlir::linalg::createDestinationPassingStyleInitOperand(), createElementFPowIFunc(), createElementIPowIFunc(), mlir::emitc::createExpression(), mlir::transform::gpu::createGpuLaunch(), createParallelComputeFunction(), createPartitionFunc(), createQuickSortFunc(), createScanLoop(), createShiftDownFunc(), mlir::ControlFlowToSCFTransformation::createStructuredDoWhileLoopOp(), mlir::impl::ensureRegionTerminator(), mlir::sparse_tensor::LoopEmitter::exitCurrentLoop(), fuseIfLegal(), genCoIterateBranchNest(), generateLoopNestUsingForallOp(), generateLoopNestUsingForOp(), genLoopWithIterator(), lowerEntryPointABIAttr(), AssertOpLowering::matchAndRewrite(), WrapFuncInClass::matchAndRewrite(), MultiBlockExecuteInliner::matchAndRewrite(), mlir::outlineSingleBlockRegion(), mlir::LLVM::ModuleImport::processFunction(), mlir::mesh::spmdizeBlock(), and mlir::linalg::tileReductionUsingForall().

setInsertionPointToStart()

void mlir::OpBuilder::setInsertionPointToStart ( Block *  block )

inline

Sets the insertion point to the start of the specified block.

Definition at line 429 of file Builders.h.

References mlir::Block::begin(), and setInsertionPoint().

Referenced by addBodyWithPayloadOp(), addComdat(), allocateSubviewGPUMemoryInAddressSpace(), atomicRMW(), buildAffineLoopNestImpl(), mlir::bufferization::buildDeallocationLibraryFunction(), mlir::scf::buildLoopNest(), buildPackingLoopNestImpl(), calculateTileOffsetsAndSizes(), mlir::async::cloneConstantsIntoTheRegion(), mlir::affine::coalesceLoops(), mlir::coalesceLoops(), mlir::collapseParallelLoops(), mlir::spirv::combine(), createBinarySearchFunc(), createChoosePivot(), createCompareThenSwap(), createCtlzFunc(), createEqCompare(), createFor(), createHeapSortFunc(), createLessThanCompare(), mlir::arm_sme::createLoopOverTileSlices(), createPartitionFunc(), mlir::LLVM::createPrintStrCall(), createQuickSort(), createQuickSortFunc(), createShiftDownFunc(), createSortStableFunc(), destructureSlot(), dropRefIfNoUses(), mlir::vector::eliminateVectorMasks(), endIf(), mlir::sparse_tensor::LoopEmitter::enterCoIterationOverTensorsAtLvls(), mlir::sparse_tensor::LoopEmitter::enterCurrentCoIterationCase(), fillStructuredOpRegion(), mlir::sparse_tensor::SparseIterator::forwardIf(), mlir::fuseIndependentSiblingForallLoops(), mlir::fuseIndependentSiblingForLoops(), mlir::sparse_tensor::FuncCallOrInlineGenerator< SubClass >::genCallOrInline(), genCoIterateBranchNest(), mlir::sparse_tensor::genCoIteration(), genCompressed(), genConditionalInsert(), genEndInsert(), generateCollapsedIndexingRegion(), genIf(), genInsertionStore(), genLoopWithIterator(), genTensorStore(), genWhenInBound(), getDynamicSharedMemorySymbol(), mlir::OperationFolder::getOrCreateConstant(), mlir::getOrCreateStringConstant(), mlir::getOrDefineFunction(), handleByValArgumentInit(), handleInlinedAllocas(), injectGpuIndexOperations(), mlir::LLVM::lookupOrCreateFn(), mlir::makeRegionIsolatedFromAbove(), mlir::transform::gpu::mapForallToBlocksImpl(), WrapFuncInClass::matchAndRewrite(), mlir::GPUFuncOpLowering::matchAndRewrite(), FoldLaunchArguments::matchAndRewrite(), mlirLinalgFillBuiltinNamedOpRegion(), mlir::LLVM::ModuleImport::ModuleImport(), movePaddingToFillOrGenericOp(), mlir::memref::multiBuffer(), mlir::affine::normalizeAffineFor(), normalizeForallLoopOp(), mlir::normalizeForallOp(), OpBuilder(), mlir::outlineSingleBlockRegion(), processParallelLoop(), mlir::scf::replaceAndCastForOpIterArg(), restoreByValRefArgumentType(), rewrite(), mlir::linalg::rewriteInDestinationPassingStyle(), setInsertionPointAfter(), setInsertionPointAfterValue(), setInsertionPointToStart(), mlir::linalg::LinalgTilingOptions::setTileSizes(), mlir::vector::splitFullAndPartialTransfer(), wrapExternalFunction(), wrapForExternalCallers(), and wrapInExecuteRegion().

setListener()

void mlir::OpBuilder::setListener ( Listener *  newListener )

inline

Sets the listener of this builder to the one provided.

Definition at line 314 of file Builders.h.

References listener.

Referenced by mlir::PatternApplicator::matchAndRewrite(), mlir::scf::tileConsumerAndFuseProducersUsingSCF(), mlir::transform::TransformRewriter::TransformRewriter(), and mlir::walkAndApplyPatterns().

tryFold()

LogicalResult OpBuilder::tryFold	(	Operation *	op,
		SmallVectorImpl< Value > &	results,
		SmallVectorImpl< Operation * > *	materializedConstants = nullptr
	)

Attempts to fold the given operation and places new results within results.

Returns success if the operation was folded, failure otherwise. If the fold was in-place, results will not be filled. Optionally, newly materialized constant operations can be returned to the caller.

Note: This function does not erase the operation on a successful fold.

Definition at line 468 of file Builders.cpp.

References mlir::Operation::getResults().

Referenced by createOrFold().

Member Data Documentation

listener

Listener* mlir::OpBuilder::listener

protected

The optional listener for events of this builder.

Definition at line 608 of file Builders.h.

Referenced by atBlockBegin(), mlir::ImplicitLocOpBuilder::atBlockBegin(), atBlockEnd(), mlir::ImplicitLocOpBuilder::atBlockEnd(), atBlockTerminator(), mlir::ImplicitLocOpBuilder::atBlockTerminator(), clone(), cloneRegionBefore(), createBlock(), mlir::RewriterBase::eraseBlock(), mlir::RewriterBase::eraseOp(), mlir::RewriterBase::finalizeOpModification(), getListener(), mlir::RewriterBase::inlineBlockBefore(), mlir::RewriterBase::inlineRegionBefore(), insert(), mlir::RewriterBase::moveBlockBefore(), mlir::RewriterBase::moveOpBefore(), mlir::RewriterBase::notifyMatchFailure(), mlir::RewriterBase::replaceAllOpUsesWith(), setListener(), and mlir::RewriterBase::splitBlock().

---

The documentation for this class was generated from the following files:

• include/mlir/IR/Builders.h
• lib/IR/Builders.cpp