MLIR
18.0.0git
|
MMAMatrixType storage and uniquing. More...
#include "mlir/Dialect/GPU/IR/GPUDialect.h"
Public Types | |
using | KeyTy = std::tuple< ArrayRef< int64_t >, Type, StringRef > |
The hash key for uniquing. More... | |
Public Member Functions | |
MMAMatrixStorageType (unsigned numDims, const int64_t *dimShapes, Type elementType, StringRef operand) | |
bool | operator== (const KeyTy &key) const |
ArrayRef< int64_t > | getShape () const |
StringRef | getOperand () const |
![]() | |
const AbstractType & | getAbstractType () |
Return the abstract type descriptor for this type. More... | |
Static Public Member Functions | |
static MMAMatrixStorageType * | construct (TypeStorageAllocator &allocator, const KeyTy &key) |
Construction. More... | |
Public Attributes | |
const int64_t * | dimShapes |
Reference to the shape of the MMA matrix. More... | |
unsigned | numDims |
Number of dimensions in the MMA matrix. More... | |
Type | elementType |
Element type of elements held in the MMA matrix. More... | |
StringRef | operand |
MMA operand that this MMAMatrix holds. More... | |
Additional Inherited Members | |
![]() | |
TypeStorage () | |
This constructor is used by derived classes as part of the TypeUniquer. More... | |
![]() | |
BaseStorage ()=default | |
MMAMatrixType storage and uniquing.
Array is uniqued based on its shape and type.
Definition at line 52 of file GPUDialect.h.
using mlir::gpu::MMAMatrixStorageType::KeyTy = std::tuple<ArrayRef<int64_t>, Type, StringRef> |
The hash key for uniquing.
Definition at line 59 of file GPUDialect.h.
|
inline |
Definition at line 53 of file GPUDialect.h.
Referenced by construct().
|
inlinestatic |
Construction.
Definition at line 65 of file GPUDialect.h.
References mlir::StorageUniquer::StorageAllocator::allocate(), mlir::StorageUniquer::StorageAllocator::copyInto(), MMAMatrixStorageType(), and operand.
|
inline |
Definition at line 79 of file GPUDialect.h.
References operand.
|
inline |
Definition at line 75 of file GPUDialect.h.
References dimShapes, and numDims.
Referenced by operator==().
|
inline |
Definition at line 60 of file GPUDialect.h.
References elementType, getShape(), and operand.
const int64_t* mlir::gpu::MMAMatrixStorageType::dimShapes |
Reference to the shape of the MMA matrix.
Definition at line 82 of file GPUDialect.h.
Referenced by getShape().
Type mlir::gpu::MMAMatrixStorageType::elementType |
Element type of elements held in the MMA matrix.
Definition at line 88 of file GPUDialect.h.
Referenced by operator==().
unsigned mlir::gpu::MMAMatrixStorageType::numDims |
Number of dimensions in the MMA matrix.
Definition at line 85 of file GPUDialect.h.
Referenced by getShape().
StringRef mlir::gpu::MMAMatrixStorageType::operand |
MMA operand that this MMAMatrix holds.
The general form of operation this type supports is given by the equation C += A*B. This field specifies which operand in the given equation is held by this type. The valid values are "AOp", "BOp" and "COp".
Definition at line 94 of file GPUDialect.h.
Referenced by construct(), getOperand(), and operator==().