mlir::tt::MetalLayoutAttr Class Reference

Inheritance diagram for mlir::tt::MetalLayoutAttr:

Public Member Functions
llvm::SmallVector< int64_t >	getShardStride (RankedTensorType tensorType) const
llvm::SmallVector< int64_t >	getNormalizedIntervals () const
void	print (::mlir::AsmPrinter &odsPrinter) const
::llvm::ArrayRef< int64_t >	getLogicalShape () const
::llvm::ArrayRef< int64_t >	getDimAlignments () const
DenseIntElementsAttr	getCollapsedIntervals () const
OOBVal	getOobVal () const
MemorySpace	getMemorySpace () const

Static Public Member Functions
static MetalLayoutAttr	get (::mlir::MLIRContext *context, ArrayRef< int64_t > logicalShape, uint64_t deviceGridRank, OOBVal oobVal, MemorySpace memorySpace, DenseIntElementsAttr collapseIntervals={}, ArrayRef< int64_t > dimAlignments={})
static llvm::SmallVector< int64_t >	derivePhysicalShape (ArrayRef< int64_t > logicalShape, ArrayRef< int64_t > gridShape, ArrayRef< int64_t > tileShape, DenseIntElementsAttr collapseIntervals, ArrayRef< int64_t > dimAlignments)
static MemRefType	getMemRefType (RankedTensorType tensorType)
static MetalLayoutAttr	get (::mlir::MLIRContext *context, ::llvm::ArrayRef< int64_t > logical_shape, ::llvm::ArrayRef< int64_t > dim_alignments, DenseIntElementsAttr collapsed_intervals, OOBVal oob_val, MemorySpace memory_space)
static constexpr ::llvm::StringLiteral	getMnemonic ()
::mlir::Attribute	parse (::mlir::AsmParser &odsParser, ::mlir::Type odsType)

Static Public Attributes
static constexpr ::llvm::StringLiteral	name = "ttcore.metal_layout"
static constexpr ::llvm::StringLiteral	dialectName = "ttcore"

Member Function Documentation

derivePhysicalShape()

static llvm::SmallVector<int64_t> mlir::tt::MetalLayoutAttr::derivePhysicalShape ( ArrayRef< int64_t >  logicalShape, ArrayRef< int64_t >  gridShape, ArrayRef< int64_t >  tileShape, DenseIntElementsAttr  collapseIntervals, ArrayRef< int64_t >  dimAlignments  )

static

get() [1/2]

MetalLayoutAttr mlir::tt::MetalLayoutAttr::get ( ::mlir::MLIRContext *  context, ::llvm::ArrayRef< int64_t >  logical_shape, ::llvm::ArrayRef< int64_t >  dim_alignments, DenseIntElementsAttr  collapsed_intervals, OOBVal  oob_val, MemorySpace  memory_space  )

static

get() [2/2]

static MetalLayoutAttr mlir::tt::MetalLayoutAttr::get ( ::mlir::MLIRContext *  context, ArrayRef< int64_t >  logicalShape, uint64_t  deviceGridRank, OOBVal  oobVal, MemorySpace  memorySpace, DenseIntElementsAttr  collapseIntervals = {}, ArrayRef< int64_t >  dimAlignments = {}  )

static

getCollapsedIntervals()

DenseIntElementsAttr mlir::tt::MetalLayoutAttr::getCollapsedIntervals

(

)

const

getDimAlignments()

llvm::ArrayRef< int64_t > mlir::tt::MetalLayoutAttr::getDimAlignments

(

)

const

getLogicalShape()

llvm::ArrayRef< int64_t > mlir::tt::MetalLayoutAttr::getLogicalShape

(

)

const

getMemorySpace()

MemorySpace mlir::tt::MetalLayoutAttr::getMemorySpace

(

)

const

getMemRefType()

static MemRefType mlir::tt::MetalLayoutAttr::getMemRefType ( RankedTensorType  tensorType )

static

getMnemonic()

static constexpr ::llvm::StringLiteral mlir::tt::MetalLayoutAttr::getMnemonic ( )

inlinestatic

getNormalizedIntervals()

llvm::SmallVector<int64_t> mlir::tt::MetalLayoutAttr::getNormalizedIntervals

(

)

const

getOobVal()

OOBVal mlir::tt::MetalLayoutAttr::getOobVal

(

)

const

getShardStride()

llvm::SmallVector<int64_t> mlir::tt::MetalLayoutAttr::getShardStride

(

RankedTensorType

tensorType

)

const

parse()

mlir::Attribute mlir::tt::MetalLayoutAttr::parse ( ::mlir::AsmParser &  odsParser, ::mlir::Type  odsType  )

static

print()

void mlir::tt::MetalLayoutAttr::print

(

::mlir::AsmPrinter &

odsPrinter

)

const

Member Data Documentation

dialectName

constexpr ::llvm::StringLiteral mlir::tt::MetalLayoutAttr::dialectName = "ttcore"

static

name

constexpr ::llvm::StringLiteral mlir::tt::MetalLayoutAttr::name = "ttcore.metal_layout"

static

---

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

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTCore/IR/TTCoreOpsAttrDefs.h.inc
• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TTCore/IR/TTCoreOpsAttrDefs.cpp.inc