|
MLIR 23.0.0git
|
#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
Public Member Functions | |
| Type | getColumnType () const |
| unsigned | getNumRows () const |
| Returns the number of rows. | |
| unsigned | getNumColumns () const |
| Returns the number of columns. | |
| unsigned | getNumElements () const |
| Returns total number of elements (rows*columns). | |
| Type | getElementType () const |
| Returns the elements' type (i.e, single element type). | |
| operator ShapedType () const | |
| ArrayRef< int64_t > | getShape () const |
| bool | hasRank () const |
| MatrixType | cloneWith (std::optional< ArrayRef< int64_t > > shape, Type elementType) const |
| Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| ImplType * | getImpl () const |
| Utility for easy access to the storage instance. | |
Static Public Member Functions | |
| static MatrixType | get (Type columnType, uint32_t columnCount) |
| static MatrixType | getChecked (function_ref< InFlightDiagnostic()> emitError, Type columnType, uint32_t columnCount) |
| static LogicalResult | verifyInvariants (function_ref< InFlightDiagnostic()> emitError, Type columnType, uint32_t columnCount) |
| static bool | isValidColumnType (Type columnType) |
| Returns true if the matrix elements are vectors of float elements. | |
| Static Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| static TypeID | getTypeID () |
| Return a unique identifier for the concrete type. | |
| static bool | classof (T val) |
| Provide an implementation of 'classof' that compares the type id of the provided value with that of the concrete type. | |
| static detail::InterfaceMap | getInterfaceMap () |
| Returns an interface map for the interfaces registered to this storage user. | |
| static HasTraitFn | getHasTraitFn () |
| Returns the function that returns true if the given Trait ID matches the IDs of any of the traits defined by the storage user. | |
| static auto | getWalkImmediateSubElementsFn () |
| Returns a function that walks immediate sub elements of a given instance of the storage user. | |
| static auto | getReplaceImmediateSubElementsFn () |
| Returns a function that replaces immediate sub elements of a given instance of the storage user. | |
| static void | attachInterface (MLIRContext &context) |
| Attach the given models as implementations of the corresponding interfaces for the concrete storage user class. | |
| static ConcreteType | get (MLIRContext *ctx, Args &&...args) |
| Get or create a new ConcreteT instance within the ctx. | |
| static ConcreteType | getChecked (const Location &loc, Args &&...args) |
| Get or create a new ConcreteT instance within the ctx, defined at the given, potentially unknown, location. | |
| static ConcreteType | getChecked (function_ref< InFlightDiagnostic()> emitErrorFn, MLIRContext *ctx, Args... args) |
| Get or create a new ConcreteT instance within the ctx. | |
| static ConcreteType | getFromOpaquePointer (const void *ptr) |
| Get an instance of the concrete type from a void pointer. | |
Static Public Attributes | |
| static constexpr StringLiteral | name = "spirv.matrix" |
Additional Inherited Members | |
| Public Types inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| using | Base |
| Utility declarations for the concrete attribute class. | |
| using | ImplType |
| using | HasTraitFn |
| Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| LogicalResult | mutate (Args &&...args) |
| Mutate the current storage instance. | |
| Static Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... > | |
| static LogicalResult | verifyInvariants (Args... args) |
| Default implementation that just returns success. | |
Definition at line 428 of file SPIRVTypes.h.
|
inline |
Definition at line 468 of file SPIRVTypes.h.
References get(), and getNumColumns().
|
static |
Definition at line 1191 of file SPIRVTypes.cpp.
References mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >::get(), and mlir::Type::getContext().
Referenced by cloneWith(), parseMatrixType(), and mlir::spirv::Deserializer::processMatrixType().
|
static |
Definition at line 1195 of file SPIRVTypes.cpp.
References mlir::emitError(), mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >::getChecked(), and mlir::Type::getContext().
| Type MatrixType::getColumnType | ( | ) | const |
Definition at line 1230 of file SPIRVTypes.cpp.
Referenced by mlir::spirv::CompositeType::getElementType(), and print().
| Type MatrixType::getElementType | ( | ) | const |
Returns the elements' type (i.e, single element type).
Definition at line 1232 of file SPIRVTypes.cpp.
Referenced by mlir::spirv::CompositeType::getElementType().
| unsigned MatrixType::getNumColumns | ( | ) | const |
Returns the number of columns.
Definition at line 1236 of file SPIRVTypes.cpp.
Referenced by cloneWith(), mlir::spirv::CompositeType::getNumElements(), getNumElements(), and print().
| unsigned MatrixType::getNumElements | ( | ) | const |
Returns total number of elements (rows*columns).
Definition at line 1248 of file SPIRVTypes.cpp.
References getNumColumns(), and getNumRows().
| unsigned MatrixType::getNumRows | ( | ) | const |
Returns the number of rows.
Definition at line 1242 of file SPIRVTypes.cpp.
Referenced by getNumElements().
Definition at line 1252 of file SPIRVTypes.cpp.
|
inline |
Definition at line 466 of file SPIRVTypes.h.
Returns true if the matrix elements are vectors of float elements.
Definition at line 1222 of file SPIRVTypes.cpp.
Referenced by verifyInvariants().
|
inline |
Definition at line 462 of file SPIRVTypes.h.
|
static |
The underlying vectors (columns) must be of size 2, 3, or 4
Definition at line 1202 of file SPIRVTypes.cpp.
References mlir::emitError(), isValidColumnType(), and success().
|
staticconstexpr |
Definition at line 434 of file SPIRVTypes.h.