MLIR  18.0.0git
Public Types | Public Member Functions | Static Public Member Functions | Static Public Attributes | List of all members
mlir::spirv::CooperativeMatrixNVType Class Reference

#include "mlir/Dialect/SPIRV/IR/SPIRVTypes.h"

+ Inheritance diagram for mlir::spirv::CooperativeMatrixNVType:

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...
 
unsigned getRows () const
 Returns the number of rows of the matrix. More...
 
unsigned getColumns () const
 Returns the number of columns of the 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 >
ImplTypegetImpl () const
 Utility for easy access to the storage instance. More...
 

Static Public Member Functions

static CooperativeMatrixNVType get (Type elementType, Scope scope, unsigned rows, unsigned columns)
 
- 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.NV.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 425 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()

CooperativeMatrixNVType CooperativeMatrixNVType::get ( Type  elementType,
Scope  scope,
unsigned  rows,
unsigned  columns 
)
static

Definition at line 297 of file SPIRVTypes.cpp.

References mlir::get(), and mlir::Type::getContext().

Referenced by parseCooperativeMatrixNVType().

◆ getCapabilities()

void CooperativeMatrixNVType::getCapabilities ( SPIRVType::CapabilityArrayRefVector capabilities,
std::optional< StorageClass >  storage = std::nullopt 
)

Definition at line 324 of file SPIRVTypes.cpp.

References getElementType().

◆ getColumns()

unsigned CooperativeMatrixNVType::getColumns ( ) const

Returns the number of columns of the matrix.

Definition at line 311 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by print().

◆ getElementType()

Type CooperativeMatrixNVType::getElementType ( ) const

◆ getExtensions()

void CooperativeMatrixNVType::getExtensions ( SPIRVType::ExtensionArrayRefVector extensions,
std::optional< StorageClass >  storage = std::nullopt 
)

Definition at line 315 of file SPIRVTypes.cpp.

References getElementType().

◆ getRows()

unsigned CooperativeMatrixNVType::getRows ( ) const

Returns the number of rows of the matrix.

Definition at line 309 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by print().

◆ getScope()

Scope CooperativeMatrixNVType::getScope ( ) const

Returns the scope of the matrix.

Definition at line 307 of file SPIRVTypes.cpp.

References mlir::detail::StorageUserBase< ConcreteT, BaseT, StorageT, UniquerT, Traits >::getImpl().

Referenced by print().

Member Data Documentation

◆ name

constexpr StringLiteral mlir::spirv::CooperativeMatrixNVType::name = "spirv.NV.coopmatrix"
staticconstexpr

Definition at line 431 of file SPIRVTypes.h.


The documentation for this class was generated from the following files: