#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"
Inheritance diagram for mlir::spirv::CooperativeMatrixType:
Public Types | |
| using | Base = StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits... > |
| Utility declarations for the concrete attribute class. More... | |
| Public Types inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits > | |
| using | Base = StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits... > |
| Utility declarations for the concrete attribute class. More... | |
| using | ImplType = StorageT |
| using | HasTraitFn = bool(*)(TypeID) |
Public Member Functions | |
| Type | getElementType () const |
| Scope | getScope () const |
| Returns the scope of the matrix. More... | |
| uint32_t | getRows () const |
| Returns the number of rows of the matrix. More... | |
| uint32_t | getColumns () const |
| Returns the number of columns of the matrix. More... | |
| CooperativeMatrixUseKHR | getUse () const |
| Returns the use parameter of the cooperative matrix. More... | |
| void | getExtensions (SPIRVType::ExtensionArrayRefVector &extensions, std::optional< StorageClass > storage=std::nullopt) |
| void | getCapabilities (SPIRVType::CapabilityArrayRefVector &capabilities, std::optional< StorageClass > storage=std::nullopt) |
| Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits > | |
| ImplType * | getImpl () const |
| Utility for easy access to the storage instance. More... | |
Static Public Member Functions | |
| static CooperativeMatrixType | get (Type elementType, uint32_t rows, uint32_t columns, Scope scope, CooperativeMatrixUseKHR use) |
| Static Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits > | |
| static TypeID | getTypeID () |
| Return a unique identifier for the concrete type. More... | |
| template<typename T > | |
| 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. More... | |
| static detail::InterfaceMap | getInterfaceMap () |
| Returns an interface map for the interfaces registered to this storage user. More... | |
| 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. More... | |
| static auto | getWalkImmediateSubElementsFn () |
| Returns a function that walks immediate sub elements of a given instance of the storage user. More... | |
| static auto | getReplaceImmediateSubElementsFn () |
| Returns a function that replaces immediate sub elements of a given instance of the storage user. More... | |
| template<typename... IfaceModels> | |
| static void | attachInterface (MLIRContext &context) |
| Attach the given models as implementations of the corresponding interfaces for the concrete storage user class. More... | |
| template<typename... Args> | |
| static ConcreteT | get (MLIRContext *ctx, Args &&...args) |
| Get or create a new ConcreteT instance within the ctx. More... | |
| template<typename... Args> | |
| static ConcreteT | getChecked (const Location &loc, Args &&...args) |
| Get or create a new ConcreteT instance within the ctx, defined at the given, potentially unknown, location. More... | |
| template<typename... Args> | |
| static ConcreteT | getChecked (function_ref< InFlightDiagnostic()> emitErrorFn, MLIRContext *ctx, Args... args) |
| Get or create a new ConcreteT instance within the ctx. More... | |
| static ConcreteT | getFromOpaquePointer (const void *ptr) |
| Get an instance of the concrete type from a void pointer. More... | |
Static Public Attributes | |
| static constexpr StringLiteral | name = "spirv.coopmatrix" |
Additional Inherited Members | |
| Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits > | |
| template<typename... Args> | |
| LogicalResult | mutate (Args &&...args) |
| Mutate the current storage instance. More... | |
| Static Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits > | |
| template<typename... Args> | |
| static LogicalResult | verify (Args... args) |
| Default implementation that just returns success. More... | |
Detailed Description
Definition at line 395 of file SPIRVTypes.h.
Member Typedef Documentation
◆ Base
| using mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::Base = StorageUserBase<ConcreteT, BaseT, StorageT, UniquerT, Traits...> |
Utility declarations for the concrete attribute class.
Definition at line 101 of file StorageUniquerSupport.h.
Member Function Documentation
◆ get()
|
static |
Definition at line 228 of file SPIRVTypes.cpp.
References mlir::get(), and mlir::Type::getContext().
Referenced by parseCooperativeMatrixType().
◆ getCapabilities()
| void CooperativeMatrixType::getCapabilities | ( | SPIRVType::CapabilityArrayRefVector & | capabilities, |
| std::optional< StorageClass > | storage = std::nullopt |
||
| ) |
Definition at line 260 of file SPIRVTypes.cpp.
References getElementType().
◆ getColumns()
| uint32_t CooperativeMatrixType::getColumns | ( | ) | const |
Returns the number of columns of the matrix.
Definition at line 242 of file SPIRVTypes.cpp.
References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().
Referenced by print().
◆ getElementType()
| Type CooperativeMatrixType::getElementType | ( | ) | const |
Definition at line 236 of file SPIRVTypes.cpp.
References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().
Referenced by getCapabilities(), getExtensions(), and print().
◆ getExtensions()
| void CooperativeMatrixType::getExtensions | ( | SPIRVType::ExtensionArrayRefVector & | extensions, |
| std::optional< StorageClass > | storage = std::nullopt |
||
| ) |
Definition at line 252 of file SPIRVTypes.cpp.
References getElementType().
◆ getRows()
| uint32_t CooperativeMatrixType::getRows | ( | ) | const |
Returns the number of rows of the matrix.
Definition at line 240 of file SPIRVTypes.cpp.
References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().
Referenced by print().
◆ getScope()
| Scope CooperativeMatrixType::getScope | ( | ) | const |
Returns the scope of the matrix.
Definition at line 246 of file SPIRVTypes.cpp.
References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().
Referenced by print().
◆ getUse()
| CooperativeMatrixUseKHR CooperativeMatrixType::getUse | ( | ) | const |
Returns the use parameter of the cooperative matrix.
Definition at line 248 of file SPIRVTypes.cpp.
References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().
Referenced by print().
Member Data Documentation
◆ name
|
staticconstexpr |
Definition at line 401 of file SPIRVTypes.h.
The documentation for this class was generated from the following files:
- include/mlir/Dialect/SPIRV/IR/SPIRVTypes.h
- lib/Dialect/SPIRV/IR/SPIRVTypes.cpp
