TT-MLIR
Namespaces | Classes | Enumerations | Functions
mlir::tt Namespace Reference

Namespaces

 detail
 
 ttir
 
 ttkernel
 
 ttmetal
 
 ttnn
 

Classes

class  GridAttr
 
class  ArchAttr
 
class  DataTypeAttr
 
class  CoreCoordAttr
 
class  TileSizeAttr
 
class  ChipPhysicalCoresAttr
 
class  ChipDescAttr
 
class  ChipCoordAttr
 
class  ChipChannelAttr
 
class  SystemDescAttr
 
class  LayoutAttr
 
class  BufferAccessAttr
 
class  BufferAttr
 
class  DeviceAttr
 
class  MemorySpaceAttr
 
class  TensorMemoryLayoutAttr
 
class  OOBValAttr
 
class  IteratorTypeAttr
 
class  OperandConstraintAttr
 
class  ArgumentAllocationAttr
 
class  TTDialect
 
class  ChipCapabilityAttr
 
class  TileType
 
class  DeviceType
 

Enumerations

enum class  Arch : uint32_t { Grayskull = 0 , WormholeB0 = 1 , Blackhole = 2 }
 
enum class  BufferAccess : uint32_t { Alias = 1 , Stream = 2 }
 
enum class  ChipCapability : uint32_t { PCIE = 1 , HostMMIO = 2 }
 
enum class  DataType : uint32_t {
  Float32 = 0 , Float16 = 1 , BFloat16 = 2 , BFP_Float8 = 3 ,
  BFP_BFloat8 = 4 , BFP_Float4 = 5 , BFP_BFloat4 = 6 , BFP_Float2 = 7 ,
  BFP_BFloat2 = 8 , UInt32 = 9 , UInt16 = 10 , UInt8 = 11
}
 
enum class  IteratorType : uint32_t { Parallel = 0 , Systolic = 1 , Broadcast = 2 }
 
enum class  MemorySpace : uint32_t { System = 0 , SystemMMIO = 1 , DeviceDRAM = 2 , DeviceL1 = 3 }
 
enum class  OOBVal : uint32_t {
  Undef = 0 , Zero = 1 , One = 2 , Inf = 3 ,
  NegInf = 4
}
 
enum class  OperandConstraint : uint32_t {
  System = 1 , DRAM = 2 , L1 = 4 , Scalar = 8 ,
  Tile = 16 , None = 32 , Interleaved = 64 , SingleBank = 128 ,
  HeightSharded = 256 , WidthSharded = 512 , BlockSharded = 1024 , AnyLayout = 2016 ,
  Any = 2047 , AnyDevice = 2046 , AnyDeviceTile = 2038 , L1BlockSharded = 1052
}
 
enum class  TensorMemoryLayout : uint32_t {
  None = 0 , Interleaved = 1 , SingleBank = 2 , HeightSharded = 3 ,
  WidthSharded = 4 , BlockSharded = 5
}
 

Functions

::llvm::StringRef stringifyArch (Arch val)
 
::std::optional< ArchsymbolizeArch (::llvm::StringRef str)
 
::std::optional< ArchsymbolizeArch (uint32_t value)
 
std::string stringifyBufferAccess (BufferAccess symbol)
 
::std::optional< BufferAccesssymbolizeBufferAccess (::llvm::StringRef str)
 
::std::optional< BufferAccesssymbolizeBufferAccess (uint32_t value)
 
std::string stringifyChipCapability (ChipCapability symbol)
 
::std::optional< ChipCapabilitysymbolizeChipCapability (::llvm::StringRef str)
 
::std::optional< ChipCapabilitysymbolizeChipCapability (uint32_t value)
 
::llvm::StringRef DataTypeEnumToString (DataType val)
 
::std::optional< DataTypeDataTypeStringToEnum (::llvm::StringRef str)
 
::std::optional< DataTypesymbolizeDataType (uint32_t value)
 
::llvm::StringRef stringifyIteratorType (IteratorType val)
 
::std::optional< IteratorTypesymbolizeIteratorType (::llvm::StringRef str)
 
::std::optional< IteratorTypesymbolizeIteratorType (uint32_t value)
 
::llvm::StringRef stringifyMemorySpace (MemorySpace val)
 
::std::optional< MemorySpacesymbolizeMemorySpace (::llvm::StringRef str)
 
::std::optional< MemorySpacesymbolizeMemorySpace (uint32_t value)
 
::llvm::StringRef stringifyOOBVal (OOBVal val)
 
::std::optional< OOBValsymbolizeOOBVal (::llvm::StringRef str)
 
::std::optional< OOBValsymbolizeOOBVal (uint32_t value)
 
std::string stringifyOperandConstraint (OperandConstraint symbol)
 
::std::optional< OperandConstraintsymbolizeOperandConstraint (::llvm::StringRef str)
 
::std::optional< OperandConstraintsymbolizeOperandConstraint (uint32_t value)
 
::llvm::StringRef stringifyTensorMemoryLayout (TensorMemoryLayout val)
 
::std::optional< TensorMemoryLayoutsymbolizeTensorMemoryLayout (::llvm::StringRef str)
 
::std::optional< TensorMemoryLayoutsymbolizeTensorMemoryLayout (uint32_t value)
 
constexpr unsigned getMaxEnumValForArch ()
 
inline ::llvm::StringRef stringifyEnum (Arch enumValue)
 
template<typename EnumType >
::std::optional< EnumType > symbolizeEnum (::llvm::StringRef)
 
template<>
inline ::std::optional< ArchsymbolizeEnum< Arch > (::llvm::StringRef str)
 
constexpr BufferAccess operator| (BufferAccess a, BufferAccess b)
 
constexpr BufferAccess operator& (BufferAccess a, BufferAccess b)
 
constexpr BufferAccess operator^ (BufferAccess a, BufferAccess b)
 
constexpr BufferAccess operator~ (BufferAccess bits)
 
constexpr bool bitEnumContainsAll (BufferAccess bits, BufferAccess bit)
 
constexpr bool bitEnumContainsAny (BufferAccess bits, BufferAccess bit)
 
constexpr BufferAccess bitEnumClear (BufferAccess bits, BufferAccess bit)
 
constexpr BufferAccess bitEnumSet (BufferAccess bits, BufferAccess bit, bool value=true)
 
std::string stringifyEnum (BufferAccess enumValue)
 
template<>
inline ::std::optional< BufferAccesssymbolizeEnum< BufferAccess > (::llvm::StringRef str)
 
constexpr ChipCapability operator| (ChipCapability a, ChipCapability b)
 
constexpr ChipCapability operator& (ChipCapability a, ChipCapability b)
 
constexpr ChipCapability operator^ (ChipCapability a, ChipCapability b)
 
constexpr ChipCapability operator~ (ChipCapability bits)
 
constexpr bool bitEnumContainsAll (ChipCapability bits, ChipCapability bit)
 
constexpr bool bitEnumContainsAny (ChipCapability bits, ChipCapability bit)
 
constexpr ChipCapability bitEnumClear (ChipCapability bits, ChipCapability bit)
 
constexpr ChipCapability bitEnumSet (ChipCapability bits, ChipCapability bit, bool value=true)
 
std::string stringifyEnum (ChipCapability enumValue)
 
template<>
inline ::std::optional< ChipCapabilitysymbolizeEnum< ChipCapability > (::llvm::StringRef str)
 
constexpr unsigned getMaxEnumValForDataType ()
 
inline ::llvm::StringRef stringifyEnum (DataType enumValue)
 
template<>
inline ::std::optional< DataTypesymbolizeEnum< DataType > (::llvm::StringRef str)
 
constexpr unsigned getMaxEnumValForIteratorType ()
 
inline ::llvm::StringRef stringifyEnum (IteratorType enumValue)
 
template<>
inline ::std::optional< IteratorTypesymbolizeEnum< IteratorType > (::llvm::StringRef str)
 
constexpr unsigned getMaxEnumValForMemorySpace ()
 
inline ::llvm::StringRef stringifyEnum (MemorySpace enumValue)
 
template<>
inline ::std::optional< MemorySpacesymbolizeEnum< MemorySpace > (::llvm::StringRef str)
 
constexpr unsigned getMaxEnumValForOOBVal ()
 
inline ::llvm::StringRef stringifyEnum (OOBVal enumValue)
 
template<>
inline ::std::optional< OOBValsymbolizeEnum< OOBVal > (::llvm::StringRef str)
 
constexpr OperandConstraint operator| (OperandConstraint a, OperandConstraint b)
 
constexpr OperandConstraint operator& (OperandConstraint a, OperandConstraint b)
 
constexpr OperandConstraint operator^ (OperandConstraint a, OperandConstraint b)
 
constexpr OperandConstraint operator~ (OperandConstraint bits)
 
constexpr bool bitEnumContainsAll (OperandConstraint bits, OperandConstraint bit)
 
constexpr bool bitEnumContainsAny (OperandConstraint bits, OperandConstraint bit)
 
constexpr OperandConstraint bitEnumClear (OperandConstraint bits, OperandConstraint bit)
 
constexpr OperandConstraint bitEnumSet (OperandConstraint bits, OperandConstraint bit, bool value=true)
 
std::string stringifyEnum (OperandConstraint enumValue)
 
template<>
inline ::std::optional< OperandConstraintsymbolizeEnum< OperandConstraint > (::llvm::StringRef str)
 
constexpr unsigned getMaxEnumValForTensorMemoryLayout ()
 
inline ::llvm::StringRef stringifyEnum (TensorMemoryLayout enumValue)
 
template<>
inline ::std::optional< TensorMemoryLayoutsymbolizeEnum< TensorMemoryLayout > (::llvm::StringRef str)
 

Enumeration Type Documentation

◆ Arch

enum mlir::tt::Arch : uint32_t
strong
Enumerator
Grayskull 
WormholeB0 
Blackhole 

◆ BufferAccess

enum mlir::tt::BufferAccess : uint32_t
strong
Enumerator
Alias 
Stream 

◆ ChipCapability

enum mlir::tt::ChipCapability : uint32_t
strong
Enumerator
PCIE 
HostMMIO 

◆ DataType

enum mlir::tt::DataType : uint32_t
strong
Enumerator
Float32 
Float16 
BFloat16 
BFP_Float8 
BFP_BFloat8 
BFP_Float4 
BFP_BFloat4 
BFP_Float2 
BFP_BFloat2 
UInt32 
UInt16 
UInt8 

◆ IteratorType

enum mlir::tt::IteratorType : uint32_t
strong
Enumerator
Parallel 
Systolic 
Broadcast 

◆ MemorySpace

enum mlir::tt::MemorySpace : uint32_t
strong
Enumerator
System 
SystemMMIO 
DeviceDRAM 
DeviceL1 

◆ OOBVal

enum mlir::tt::OOBVal : uint32_t
strong
Enumerator
Undef 
Zero 
One 
Inf 
NegInf 

◆ OperandConstraint

enum mlir::tt::OperandConstraint : uint32_t
strong
Enumerator
System 
DRAM 
L1 
Scalar 
Tile 
None 
Interleaved 
SingleBank 
HeightSharded 
WidthSharded 
BlockSharded 
AnyLayout 
Any 
AnyDevice 
AnyDeviceTile 
L1BlockSharded 

◆ TensorMemoryLayout

enum mlir::tt::TensorMemoryLayout : uint32_t
strong
Enumerator
None 
Interleaved 
SingleBank 
HeightSharded 
WidthSharded 
BlockSharded 

Function Documentation

◆ bitEnumClear() [1/3]

constexpr BufferAccess mlir::tt::bitEnumClear ( BufferAccess  bits,
BufferAccess  bit 
)
inlineconstexpr

◆ bitEnumClear() [2/3]

constexpr ChipCapability mlir::tt::bitEnumClear ( ChipCapability  bits,
ChipCapability  bit 
)
inlineconstexpr

◆ bitEnumClear() [3/3]

constexpr OperandConstraint mlir::tt::bitEnumClear ( OperandConstraint  bits,
OperandConstraint  bit 
)
inlineconstexpr

◆ bitEnumContainsAll() [1/3]

constexpr bool mlir::tt::bitEnumContainsAll ( BufferAccess  bits,
BufferAccess  bit 
)
inlineconstexpr

◆ bitEnumContainsAll() [2/3]

constexpr bool mlir::tt::bitEnumContainsAll ( ChipCapability  bits,
ChipCapability  bit 
)
inlineconstexpr

◆ bitEnumContainsAll() [3/3]

constexpr bool mlir::tt::bitEnumContainsAll ( OperandConstraint  bits,
OperandConstraint  bit 
)
inlineconstexpr

◆ bitEnumContainsAny() [1/3]

constexpr bool mlir::tt::bitEnumContainsAny ( BufferAccess  bits,
BufferAccess  bit 
)
inlineconstexpr

◆ bitEnumContainsAny() [2/3]

constexpr bool mlir::tt::bitEnumContainsAny ( ChipCapability  bits,
ChipCapability  bit 
)
inlineconstexpr

◆ bitEnumContainsAny() [3/3]

constexpr bool mlir::tt::bitEnumContainsAny ( OperandConstraint  bits,
OperandConstraint  bit 
)
inlineconstexpr

◆ bitEnumSet() [1/3]

constexpr BufferAccess mlir::tt::bitEnumSet ( BufferAccess  bits,
BufferAccess  bit,
bool  value = true 
)
inlineconstexpr

◆ bitEnumSet() [2/3]

constexpr ChipCapability mlir::tt::bitEnumSet ( ChipCapability  bits,
ChipCapability  bit,
bool  value = true 
)
inlineconstexpr

◆ bitEnumSet() [3/3]

constexpr OperandConstraint mlir::tt::bitEnumSet ( OperandConstraint  bits,
OperandConstraint  bit,
bool  value = true 
)
inlineconstexpr

◆ DataTypeEnumToString()

llvm::StringRef mlir::tt::DataTypeEnumToString ( DataType  val)

◆ DataTypeStringToEnum()

std::optional< DataType > mlir::tt::DataTypeStringToEnum ( ::llvm::StringRef  str)

◆ getMaxEnumValForArch()

constexpr unsigned mlir::tt::getMaxEnumValForArch ( )
inlineconstexpr

◆ getMaxEnumValForDataType()

constexpr unsigned mlir::tt::getMaxEnumValForDataType ( )
inlineconstexpr

◆ getMaxEnumValForIteratorType()

constexpr unsigned mlir::tt::getMaxEnumValForIteratorType ( )
inlineconstexpr

◆ getMaxEnumValForMemorySpace()

constexpr unsigned mlir::tt::getMaxEnumValForMemorySpace ( )
inlineconstexpr

◆ getMaxEnumValForOOBVal()

constexpr unsigned mlir::tt::getMaxEnumValForOOBVal ( )
inlineconstexpr

◆ getMaxEnumValForTensorMemoryLayout()

constexpr unsigned mlir::tt::getMaxEnumValForTensorMemoryLayout ( )
inlineconstexpr

◆ operator&() [1/3]

constexpr BufferAccess mlir::tt::operator& ( BufferAccess  a,
BufferAccess  b 
)
inlineconstexpr

◆ operator&() [2/3]

constexpr ChipCapability mlir::tt::operator& ( ChipCapability  a,
ChipCapability  b 
)
inlineconstexpr

◆ operator&() [3/3]

constexpr OperandConstraint mlir::tt::operator& ( OperandConstraint  a,
OperandConstraint  b 
)
inlineconstexpr

◆ operator^() [1/3]

constexpr BufferAccess mlir::tt::operator^ ( BufferAccess  a,
BufferAccess  b 
)
inlineconstexpr

◆ operator^() [2/3]

constexpr ChipCapability mlir::tt::operator^ ( ChipCapability  a,
ChipCapability  b 
)
inlineconstexpr

◆ operator^() [3/3]

constexpr OperandConstraint mlir::tt::operator^ ( OperandConstraint  a,
OperandConstraint  b 
)
inlineconstexpr

◆ operator|() [1/3]

constexpr BufferAccess mlir::tt::operator| ( BufferAccess  a,
BufferAccess  b 
)
inlineconstexpr

◆ operator|() [2/3]

constexpr ChipCapability mlir::tt::operator| ( ChipCapability  a,
ChipCapability  b 
)
inlineconstexpr

◆ operator|() [3/3]

constexpr OperandConstraint mlir::tt::operator| ( OperandConstraint  a,
OperandConstraint  b 
)
inlineconstexpr

◆ operator~() [1/3]

constexpr BufferAccess mlir::tt::operator~ ( BufferAccess  bits)
inlineconstexpr

◆ operator~() [2/3]

constexpr ChipCapability mlir::tt::operator~ ( ChipCapability  bits)
inlineconstexpr

◆ operator~() [3/3]

constexpr OperandConstraint mlir::tt::operator~ ( OperandConstraint  bits)
inlineconstexpr

◆ stringifyArch()

llvm::StringRef mlir::tt::stringifyArch ( Arch  val)

◆ stringifyBufferAccess()

std::string mlir::tt::stringifyBufferAccess ( BufferAccess  symbol)

◆ stringifyChipCapability()

std::string mlir::tt::stringifyChipCapability ( ChipCapability  symbol)

◆ stringifyEnum() [1/9]

inline ::llvm::StringRef mlir::tt::stringifyEnum ( Arch  enumValue)

◆ stringifyEnum() [2/9]

std::string mlir::tt::stringifyEnum ( BufferAccess  enumValue)
inline

◆ stringifyEnum() [3/9]

std::string mlir::tt::stringifyEnum ( ChipCapability  enumValue)
inline

◆ stringifyEnum() [4/9]

inline ::llvm::StringRef mlir::tt::stringifyEnum ( DataType  enumValue)

◆ stringifyEnum() [5/9]

inline ::llvm::StringRef mlir::tt::stringifyEnum ( IteratorType  enumValue)

◆ stringifyEnum() [6/9]

inline ::llvm::StringRef mlir::tt::stringifyEnum ( MemorySpace  enumValue)

◆ stringifyEnum() [7/9]

inline ::llvm::StringRef mlir::tt::stringifyEnum ( OOBVal  enumValue)

◆ stringifyEnum() [8/9]

std::string mlir::tt::stringifyEnum ( OperandConstraint  enumValue)
inline

◆ stringifyEnum() [9/9]

inline ::llvm::StringRef mlir::tt::stringifyEnum ( TensorMemoryLayout  enumValue)

◆ stringifyIteratorType()

llvm::StringRef mlir::tt::stringifyIteratorType ( IteratorType  val)

◆ stringifyMemorySpace()

llvm::StringRef mlir::tt::stringifyMemorySpace ( MemorySpace  val)

◆ stringifyOOBVal()

llvm::StringRef mlir::tt::stringifyOOBVal ( OOBVal  val)

◆ stringifyOperandConstraint()

std::string mlir::tt::stringifyOperandConstraint ( OperandConstraint  symbol)

◆ stringifyTensorMemoryLayout()

llvm::StringRef mlir::tt::stringifyTensorMemoryLayout ( TensorMemoryLayout  val)

◆ symbolizeArch() [1/2]

std::optional< Arch > mlir::tt::symbolizeArch ( ::llvm::StringRef  str)

◆ symbolizeArch() [2/2]

std::optional< Arch > mlir::tt::symbolizeArch ( uint32_t  value)

◆ symbolizeBufferAccess() [1/2]

std::optional< BufferAccess > mlir::tt::symbolizeBufferAccess ( ::llvm::StringRef  str)

◆ symbolizeBufferAccess() [2/2]

std::optional< BufferAccess > mlir::tt::symbolizeBufferAccess ( uint32_t  value)

◆ symbolizeChipCapability() [1/2]

std::optional< ChipCapability > mlir::tt::symbolizeChipCapability ( ::llvm::StringRef  str)

◆ symbolizeChipCapability() [2/2]

std::optional< ChipCapability > mlir::tt::symbolizeChipCapability ( uint32_t  value)

◆ symbolizeDataType()

std::optional< DataType > mlir::tt::symbolizeDataType ( uint32_t  value)

◆ symbolizeEnum()

template<typename EnumType >
::std::optional<EnumType> mlir::tt::symbolizeEnum ( ::llvm::StringRef  )

◆ symbolizeEnum< Arch >()

template<>
inline ::std::optional<Arch> mlir::tt::symbolizeEnum< Arch > ( ::llvm::StringRef  str)

◆ symbolizeEnum< BufferAccess >()

template<>
inline ::std::optional<BufferAccess> mlir::tt::symbolizeEnum< BufferAccess > ( ::llvm::StringRef  str)

◆ symbolizeEnum< ChipCapability >()

template<>
inline ::std::optional<ChipCapability> mlir::tt::symbolizeEnum< ChipCapability > ( ::llvm::StringRef  str)

◆ symbolizeEnum< DataType >()

template<>
inline ::std::optional<DataType> mlir::tt::symbolizeEnum< DataType > ( ::llvm::StringRef  str)

◆ symbolizeEnum< IteratorType >()

template<>
inline ::std::optional<IteratorType> mlir::tt::symbolizeEnum< IteratorType > ( ::llvm::StringRef  str)

◆ symbolizeEnum< MemorySpace >()

template<>
inline ::std::optional<MemorySpace> mlir::tt::symbolizeEnum< MemorySpace > ( ::llvm::StringRef  str)

◆ symbolizeEnum< OOBVal >()

template<>
inline ::std::optional<OOBVal> mlir::tt::symbolizeEnum< OOBVal > ( ::llvm::StringRef  str)

◆ symbolizeEnum< OperandConstraint >()

template<>
inline ::std::optional<OperandConstraint> mlir::tt::symbolizeEnum< OperandConstraint > ( ::llvm::StringRef  str)

◆ symbolizeEnum< TensorMemoryLayout >()

template<>
inline ::std::optional<TensorMemoryLayout> mlir::tt::symbolizeEnum< TensorMemoryLayout > ( ::llvm::StringRef  str)

◆ symbolizeIteratorType() [1/2]

std::optional< IteratorType > mlir::tt::symbolizeIteratorType ( ::llvm::StringRef  str)

◆ symbolizeIteratorType() [2/2]

std::optional< IteratorType > mlir::tt::symbolizeIteratorType ( uint32_t  value)

◆ symbolizeMemorySpace() [1/2]

std::optional< MemorySpace > mlir::tt::symbolizeMemorySpace ( ::llvm::StringRef  str)

◆ symbolizeMemorySpace() [2/2]

std::optional< MemorySpace > mlir::tt::symbolizeMemorySpace ( uint32_t  value)

◆ symbolizeOOBVal() [1/2]

std::optional< OOBVal > mlir::tt::symbolizeOOBVal ( ::llvm::StringRef  str)

◆ symbolizeOOBVal() [2/2]

std::optional< OOBVal > mlir::tt::symbolizeOOBVal ( uint32_t  value)

◆ symbolizeOperandConstraint() [1/2]

std::optional< OperandConstraint > mlir::tt::symbolizeOperandConstraint ( ::llvm::StringRef  str)

◆ symbolizeOperandConstraint() [2/2]

std::optional< OperandConstraint > mlir::tt::symbolizeOperandConstraint ( uint32_t  value)

◆ symbolizeTensorMemoryLayout() [1/2]

std::optional< TensorMemoryLayout > mlir::tt::symbolizeTensorMemoryLayout ( ::llvm::StringRef  str)

◆ symbolizeTensorMemoryLayout() [2/2]

std::optional< TensorMemoryLayout > mlir::tt::symbolizeTensorMemoryLayout ( uint32_t  value)