TT-MLIR
|
Public Types | |
using | Adaptor = GenericOpAdaptor |
template<typename RangeT > | |
using | GenericAdaptor = GenericOpGenericAdaptor< RangeT > |
using | FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute > > |
using | Properties = FoldAdaptor::Properties |
Public Member Functions | |
static ::llvm::ArrayRef<::llvm::StringRef > | getAttributeNames () |
::mlir::StringAttr | getGridAttrName () |
::mlir::StringAttr | getIndexingMapsAttrName () |
::mlir::StringAttr | getIteratorTypesAttrName () |
::mlir::StringAttr | getThreadsAttrName () |
::mlir::StringAttr | getOperandSegmentSizesAttrName () |
std::pair< unsigned, unsigned > | getODSOperandIndexAndLength (unsigned index) |
::mlir::Operation::operand_range | getODSOperands (unsigned index) |
::mlir::Operation::operand_range | getInputs () |
::mlir::Operation::operand_range | getOutputs () |
::mlir::MutableOperandRange | getInputsMutable () |
::mlir::MutableOperandRange | getOutputsMutable () |
std::pair< unsigned, unsigned > | getODSResultIndexAndLength (unsigned index) |
::mlir::Operation::result_range | getODSResults (unsigned index) |
::mlir::Operation::result_range | getResults () |
::mlir::MutableArrayRef<::mlir::Region > | getRegions () |
void | writeProperties (::mlir::DialectBytecodeWriter &writer) |
::mlir::tt::GridAttr | getGridAttr () |
::mlir::tt::GridAttr | getGrid () |
::mlir::ArrayAttr | getIndexingMapsAttr () |
::mlir::ArrayAttr | getIndexingMaps () |
::mlir::ArrayAttr | getIteratorTypesAttr () |
::mlir::ArrayAttr | getIteratorTypes () |
::mlir::ArrayAttr | getThreadsAttr () |
::mlir::ArrayAttr | getThreads () |
void | setGridAttr (::mlir::tt::GridAttr attr) |
void | setIndexingMapsAttr (::mlir::ArrayAttr attr) |
void | setIteratorTypesAttr (::mlir::ArrayAttr attr) |
void | setThreadsAttr (::mlir::ArrayAttr attr) |
::llvm::LogicalResult | verifyInvariantsImpl () |
::llvm::LogicalResult | verifyInvariants () |
::llvm::LogicalResult | verify () |
bool | bufferizesToMemoryRead (::mlir::OpOperand &opOperand, const ::mlir::bufferization::AnalysisState &state) |
bool | bufferizesToMemoryWrite (::mlir::OpOperand &opOperand, const ::mlir::bufferization::AnalysisState &state) |
::mlir::bufferization::AliasingValueList | getAliasingValues (::mlir::OpOperand &opOperand, const ::mlir::bufferization::AnalysisState &state) |
::llvm::LogicalResult | bufferize (::mlir::RewriterBase &rewriter, const ::mlir::bufferization::BufferizationOptions &options) |
::mlir::FailureOr<::mlir::BaseMemRefType > | getBufferType (::mlir::Value value, const ::mlir::bufferization::BufferizationOptions &options, ::llvm::SmallVector<::mlir::Value > &invocationStack) |
void | getEffects (::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> &effects) |
void | getAsmBlockArgumentNames (::mlir::Region ®ion, ::mlir::OpAsmSetValueNameFn setNameFn) |
void | getAsmBlockNames (::mlir::OpAsmSetBlockNameFn setNameFn) |
void | print (::mlir::OpAsmPrinter &_odsPrinter) |
MutableOperandRange | getDpsInitsMutable () |
unsigned | getNumLoops () |
unsigned | getNumDims () |
SmallVector< int64_t > | getLoopBounds () |
SmallVector< AffineMap > | getIndexingMapsValue () |
SmallVector< IteratorType > | getIteratorTypesValue () |
SmallVector< SmallVector< int64_t > > | getOperandGridShapes () |
ThreadType | getRegionThreadType (unsigned regionIndex) |
bool | isAffineMapForm () |
bool | isLoweredLoopForm () |
bool | isExternalSymbolForm () |
Static Public Member Functions | |
::mlir::StringAttr | getGridAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getIndexingMapsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getIteratorTypesAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getThreadsAttrName (::mlir::OperationName name) |
::mlir::StringAttr | getOperandSegmentSizesAttrName (::mlir::OperationName name) |
static constexpr ::llvm::StringLiteral | getOperationName () |
::llvm::LogicalResult | setPropertiesFromAttr (Properties &prop, ::mlir::Attribute attr, ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError) |
::mlir::Attribute | getPropertiesAsAttr (::mlir::MLIRContext *ctx, const Properties &prop) |
static llvm::hash_code | computePropertiesHash (const Properties &prop) |
static std::optional< mlir::Attribute > | getInherentAttr (::mlir::MLIRContext *ctx, const Properties &prop, llvm::StringRef name) |
static void | setInherentAttr (Properties &prop, llvm::StringRef name, mlir::Attribute value) |
static void | populateInherentAttrs (::mlir::MLIRContext *ctx, const Properties &prop, ::mlir::NamedAttrList &attrs) |
::llvm::LogicalResult | verifyInherentAttrs (::mlir::OperationName opName, ::mlir::NamedAttrList &attrs, llvm::function_ref<::mlir::InFlightDiagnostic()> emitError) |
::llvm::LogicalResult | readProperties (::mlir::DialectBytecodeReader &reader, ::mlir::OperationState &state) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ValueRange inputs, ValueRange outputs, ArrayAttr indexingMaps, ArrayAttr iteratorTypes) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ValueRange inputs, ValueRange outputs, ArrayAttr indexingMaps, ArrayAttr iteratorTypes, llvm::function_ref< void(OpBuilder &, Location, ValueRange)> computeRegionBuilder) |
static void | build (::mlir::OpBuilder &odsBuilder, ::mlir::OperationState &odsState, ::mlir::TypeRange results, ::mlir::ValueRange inputs, ::mlir::ValueRange outputs, ::mlir::tt::GridAttr grid, ::mlir::ArrayAttr indexing_maps, ::mlir::ArrayAttr iterator_types, ::mlir::ArrayAttr threads, unsigned regionsCount) |
static void | build (::mlir::OpBuilder &, ::mlir::OperationState &odsState, ::mlir::TypeRange resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute > attributes, unsigned numRegions) |
::mlir::ParseResult | parse (::mlir::OpAsmParser &parser, ::mlir::OperationState &result) |
Private Member Functions | |
::mlir::StringAttr | getAttributeNameForIndex (unsigned index) |
Static Private Member Functions | |
::mlir::StringAttr | getAttributeNameForIndex (::mlir::OperationName name, unsigned index) |
using mlir::tt::ttir::GenericOp::FoldAdaptor = GenericAdaptor<::llvm::ArrayRef<::mlir::Attribute> > |
using mlir::tt::ttir::GenericOp::GenericAdaptor = GenericOpGenericAdaptor<RangeT> |
::llvm::LogicalResult mlir::tt::ttir::GenericOp::bufferize | ( | ::mlir::RewriterBase & | rewriter, |
const ::mlir::bufferization::BufferizationOptions & | options | ||
) |
bool mlir::tt::ttir::GenericOp::bufferizesToMemoryRead | ( | ::mlir::OpOperand & | opOperand, |
const ::mlir::bufferization::AnalysisState & | state | ||
) |
bool mlir::tt::ttir::GenericOp::bufferizesToMemoryWrite | ( | ::mlir::OpOperand & | opOperand, |
const ::mlir::bufferization::AnalysisState & | state | ||
) |
|
static |
|
static |
|
static |
|
static |
|
static |
bufferization::AliasingValueList mlir::tt::ttir::GenericOp::getAliasingValues | ( | ::mlir::OpOperand & | opOperand, |
const ::mlir::bufferization::AnalysisState & | state | ||
) |
void mlir::tt::ttir::GenericOp::getAsmBlockArgumentNames | ( | ::mlir::Region & | region, |
::mlir::OpAsmSetValueNameFn | setNameFn | ||
) |
void mlir::tt::ttir::GenericOp::getAsmBlockNames | ( | ::mlir::OpAsmSetBlockNameFn | setNameFn | ) |
|
inlinestaticprivate |
|
inlineprivate |
|
inline |
FailureOr< mlir::BaseMemRefType > mlir::tt::ttir::GenericOp::getBufferType | ( | ::mlir::Value | value, |
const ::mlir::bufferization::BufferizationOptions & | options, | ||
::llvm::SmallVector<::mlir::Value > & | invocationStack | ||
) |
|
inline |
void mlir::tt::ttir::GenericOp::getEffects | ( | ::llvm::SmallVectorImpl<::mlir::SideEffects::EffectInstance<::mlir::MemoryEffects::Effect >> & | effects | ) |
mlir::tt::GridAttr mlir::tt::ttir::GenericOp::getGrid | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
mlir::ArrayAttr mlir::tt::ttir::GenericOp::getIndexingMaps | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
SmallVector<AffineMap> mlir::tt::ttir::GenericOp::getIndexingMapsValue | ( | ) |
|
static |
|
inline |
mlir::MutableOperandRange mlir::tt::ttir::GenericOp::getInputsMutable | ( | ) |
mlir::ArrayAttr mlir::tt::ttir::GenericOp::getIteratorTypes | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
SmallVector<IteratorType> mlir::tt::ttir::GenericOp::getIteratorTypesValue | ( | ) |
SmallVector<int64_t> mlir::tt::ttir::GenericOp::getLoopBounds | ( | ) |
unsigned mlir::tt::ttir::GenericOp::getNumDims | ( | ) |
unsigned mlir::tt::ttir::GenericOp::getNumLoops | ( | ) |
std::pair< unsigned, unsigned > mlir::tt::ttir::GenericOp::getODSOperandIndexAndLength | ( | unsigned | index | ) |
|
inline |
std::pair< unsigned, unsigned > mlir::tt::ttir::GenericOp::getODSResultIndexAndLength | ( | unsigned | index | ) |
|
inline |
SmallVector<SmallVector<int64_t> > mlir::tt::ttir::GenericOp::getOperandGridShapes | ( | ) |
|
inline |
|
inlinestatic |
|
inlinestatic |
|
inline |
mlir::MutableOperandRange mlir::tt::ttir::GenericOp::getOutputsMutable | ( | ) |
|
static |
|
inline |
|
inline |
|
inline |
mlir::ArrayAttr mlir::tt::ttir::GenericOp::getThreads | ( | ) |
|
inline |
|
inline |
|
inlinestatic |
|
inline |
|
inline |
|
inline |
|
static |
|
static |
void mlir::tt::ttir::GenericOp::print | ( | ::mlir::OpAsmPrinter & | _odsPrinter | ) |
|
static |
|
inline |
|
inline |
|
static |
|
inline |
|
static |
|
inline |
::llvm::LogicalResult mlir::tt::ttir::GenericOp::verify | ( | ) |
|
static |
llvm::LogicalResult mlir::tt::ttir::GenericOp::verifyInvariants | ( | ) |
llvm::LogicalResult mlir::tt::ttir::GenericOp::verifyInvariantsImpl | ( | ) |
void mlir::tt::ttir::GenericOp::writeProperties | ( | ::mlir::DialectBytecodeWriter & | writer | ) |