mlir::tt::DeviceAttr Class Reference

Inheritance diagram for mlir::tt::DeviceAttr:

Public Member Functions
AffineMap	getMapForMemorySpace (MemorySpace memorySpace) const
uint64_t	getLayoutSizeBytes (ArrayRef< int64_t > tensorShape, LayoutAttr layout, MemorySpace memorySpace) const
uint64_t	getTensorSizeBytes (RankedTensorType tensorType, MemorySpace memorySpace) const
void	print (::mlir::AsmPrinter &odsPrinter) const
::mlir::tt::GridAttr	getWorkerGrid () const
AffineMap	getL1Map () const
AffineMap	getDramMap () const
::llvm::ArrayRef< int64_t >	getMeshShape () const
::llvm::ArrayRef< unsigned >	getChipIds () const

Static Public Member Functions
static DeviceAttr	get (::mlir::MLIRContext *context, SystemDescAttr systemDesc, ArrayRef< int64_t > meshShape, ArrayRef< unsigned > chipIds)
static DeviceAttr	get (::mlir::MLIRContext *context, SystemDescAttr systemDesc, ArrayRef< int64_t > meshShape={})
static DeviceAttr	get (::mlir::MLIRContext *context, ::mlir::tt::GridAttr workerGrid, AffineMap l1Map, AffineMap dramMap, ::llvm::ArrayRef< int64_t > meshShape, ::llvm::ArrayRef< unsigned > chipIds)
static DeviceAttr	getChecked (::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::MLIRContext *context, ::mlir::tt::GridAttr workerGrid, AffineMap l1Map, AffineMap dramMap, ::llvm::ArrayRef< int64_t > meshShape, ::llvm::ArrayRef< unsigned > chipIds)
::llvm::LogicalResult	verify (::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError, ::mlir::tt::GridAttr workerGrid, AffineMap l1Map, AffineMap dramMap, ::llvm::ArrayRef< int64_t > meshShape, ::llvm::ArrayRef< unsigned > chipIds)
static constexpr ::llvm::StringLiteral	getMnemonic ()
::mlir::Attribute	parse (::mlir::AsmParser &odsParser, ::mlir::Type odsType)

Static Public Attributes
static constexpr ::llvm::StringLiteral	name = "tt.device"
static constexpr ::llvm::StringLiteral	dialectName = "tt"

Member Function Documentation

get() [1/3]

DeviceAttr mlir::tt::DeviceAttr::get ( ::mlir::MLIRContext *  context, ::mlir::tt::GridAttr  workerGrid, AffineMap  l1Map, AffineMap  dramMap, ::llvm::ArrayRef< int64_t >  meshShape, ::llvm::ArrayRef< unsigned >  chipIds  )

static

get() [2/3]

static DeviceAttr mlir::tt::DeviceAttr::get ( ::mlir::MLIRContext *  context, SystemDescAttr  systemDesc, ArrayRef< int64_t >  meshShape, ArrayRef< unsigned >  chipIds  )

static

get() [3/3]

static DeviceAttr mlir::tt::DeviceAttr::get ( ::mlir::MLIRContext *  context, SystemDescAttr  systemDesc, ArrayRef< int64_t >  meshShape = {}  )

static

getChecked()

DeviceAttr mlir::tt::DeviceAttr::getChecked ( ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError, ::mlir::MLIRContext *  context, ::mlir::tt::GridAttr  workerGrid, AffineMap  l1Map, AffineMap  dramMap, ::llvm::ArrayRef< int64_t >  meshShape, ::llvm::ArrayRef< unsigned >  chipIds  )

static

getChipIds()

llvm::ArrayRef< unsigned > mlir::tt::DeviceAttr::getChipIds

(

)

const

getDramMap()

AffineMap mlir::tt::DeviceAttr::getDramMap

(

)

const

getL1Map()

AffineMap mlir::tt::DeviceAttr::getL1Map

(

)

const

getLayoutSizeBytes()

uint64_t mlir::tt::DeviceAttr::getLayoutSizeBytes	(	ArrayRef< int64_t >	tensorShape,
		LayoutAttr	layout,
		MemorySpace	memorySpace
	)		const

getMapForMemorySpace()

AffineMap mlir::tt::DeviceAttr::getMapForMemorySpace ( MemorySpace  memorySpace ) const

inline

getMeshShape()

llvm::ArrayRef< int64_t > mlir::tt::DeviceAttr::getMeshShape

(

)

const

getMnemonic()

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

inlinestatic

getTensorSizeBytes()

uint64_t mlir::tt::DeviceAttr::getTensorSizeBytes	(	RankedTensorType	tensorType,
		MemorySpace	memorySpace
	)		const

getWorkerGrid()

mlir::tt::GridAttr mlir::tt::DeviceAttr::getWorkerGrid

(

)

const

parse()

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

static

print()

void mlir::tt::DeviceAttr::print

(

::mlir::AsmPrinter &

odsPrinter

)

const

verify()

::llvm::LogicalResult mlir::tt::DeviceAttr::verify ( ::llvm::function_ref<::mlir::InFlightDiagnostic()>  emitError, ::mlir::tt::GridAttr  workerGrid, AffineMap  l1Map, AffineMap  dramMap, ::llvm::ArrayRef< int64_t >  meshShape, ::llvm::ArrayRef< unsigned >  chipIds  )

static

Member Data Documentation

dialectName

constexpr ::llvm::StringLiteral mlir::tt::DeviceAttr::dialectName = "tt"

static

name

constexpr ::llvm::StringLiteral mlir::tt::DeviceAttr::name = "tt.device"

static

---

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

• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TT/IR/TTOpsAttrDefs.h.inc
• /__w/tt-mlir/tt-mlir/build/include/ttmlir/Dialect/TT/IR/TTOpsAttrDefs.cpp.inc