TT-MLIR
Public Member Functions | List of all members
mlir::tt::detail::DeviceLayoutInterfaceTrait< ConcreteAttr > Struct Template Reference
Inheritance diagram for mlir::tt::detail::DeviceLayoutInterfaceTrait< ConcreteAttr >:
mlir::tt::DeviceLayoutInterface::Trait< ConcreteAttr >

Public Member Functions

llvm::ArrayRef< int64_t > getGridShape (MemRefType memrefType) const
 Get the grid shape part of the memref shape (aka the first half) More...
 
llvm::ArrayRef< int64_t > getShardShape (MemRefType memrefType) const
 Get the shard shape part of the memref shape (aka the second half) More...
 
int64_t getShardNumElements (MemRefType memrefType) const
 Get the number of elements for the shard shape part of the memref shape (aka the second half) More...
 

Member Function Documentation

◆ getGridShape()

template<typename ConcreteAttr >
llvm::ArrayRef<int64_t> mlir::tt::detail::DeviceLayoutInterfaceTrait< ConcreteAttr >::getGridShape ( MemRefType  memrefType) const
inline

Get the grid shape part of the memref shape (aka the first half)

◆ getShardNumElements()

template<typename ConcreteAttr >
int64_t mlir::tt::detail::DeviceLayoutInterfaceTrait< ConcreteAttr >::getShardNumElements ( MemRefType  memrefType) const
inline

Get the number of elements for the shard shape part of the memref shape (aka the second half)

◆ getShardShape()

template<typename ConcreteAttr >
llvm::ArrayRef<int64_t> mlir::tt::detail::DeviceLayoutInterfaceTrait< ConcreteAttr >::getShardShape ( MemRefType  memrefType) const
inline

Get the shard shape part of the memref shape (aka the second half)


The documentation for this struct was generated from the following file: