|
| 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...
|
| |
| | operator::mlir::MemRefLayoutAttrInterface () const |
| |
| ::mlir::AffineMap | getAffineMap () const |
| | Get the MemRef layout as an AffineMap, the method must not return NULL. More...
|
| |
| bool | isIdentity () const |
| | Return true if this attribute represents the identity layout. More...
|
| |
| ::llvm::LogicalResult | verifyLayout (::llvm::ArrayRef< int64_t > shape, ::llvm::function_ref<::mlir::InFlightDiagnostic()> emitError) const |
| | Check if the current layout is applicable to the provided shape. More...
|
| |
| ::llvm::LogicalResult | getStridesAndOffset (::llvm::ArrayRef< int64_t > shape, ::llvm::SmallVectorImpl< int64_t > &strides, int64_t &offset) const |
| |
◆ getAffineMap()
| mlir::AffineMap mlir::tt::DeviceLayoutInterface::getAffineMap |
( |
| ) |
const |
Get the MemRef layout as an AffineMap, the method must not return NULL.
◆ getGridShape()
| llvm::ArrayRef< int64_t > mlir::tt::DeviceLayoutInterface::getGridShape |
( |
MemRefType |
memrefType | ) |
const |
Get the grid shape part of the memref shape (aka the first half)
◆ getShardNumElements()
| int64_t mlir::tt::DeviceLayoutInterface::getShardNumElements |
( |
MemRefType |
memrefType | ) |
const |
Get the number of elements for the shard shape part of the memref shape (aka the second half)
◆ getShardShape()
| llvm::ArrayRef< int64_t > mlir::tt::DeviceLayoutInterface::getShardShape |
( |
MemRefType |
memrefType | ) |
const |
Get the shard shape part of the memref shape (aka the second half)
◆ getStridesAndOffset()
| llvm::LogicalResult mlir::tt::DeviceLayoutInterface::getStridesAndOffset |
( |
::llvm::ArrayRef< int64_t > |
shape, |
|
|
::llvm::SmallVectorImpl< int64_t > & |
strides, |
|
|
int64_t & |
offset |
|
) |
| const |
Return the strides (using ShapedType::kDynamic for the dynamic case) that this layout corresponds to into strides and offset if such exist and can be determined from a combination of the layout and the given shape. If these strides cannot be inferred, return failure(). The values of strides and offset are undefined on failure.
◆ isIdentity()
| bool mlir::tt::DeviceLayoutInterface::isIdentity |
( |
| ) |
const |
Return true if this attribute represents the identity layout.
◆ operator::mlir::MemRefLayoutAttrInterface()
| mlir::tt::DeviceLayoutInterface::operator::mlir::MemRefLayoutAttrInterface |
( |
| ) |
const |
|
inline |
◆ verifyLayout()
| llvm::LogicalResult mlir::tt::DeviceLayoutInterface::verifyLayout |
( |
::llvm::ArrayRef< int64_t > |
shape, |
|
|
::llvm::function_ref<::mlir::InFlightDiagnostic()> |
emitError |
|
) |
| const |
Check if the current layout is applicable to the provided shape.
The documentation for this class was generated from the following files:
