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

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

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

Public Member Functions

Type getElementType () const
 
Scope getScope () const
 Return the scope of the cooperative matrix. More...
 
unsigned getRows () const
 return the number of rows of the matrix. More...
 
unsigned getColumns () const
 return the number of columns of the matrix. More...
 
void getExtensions (SPIRVType::ExtensionArrayRefVector &extensions, Optional< StorageClass > storage=llvm::None)
 
void getCapabilities (SPIRVType::CapabilityArrayRefVector &capabilities, Optional< StorageClass > storage=llvm::None)
 

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...
 
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...
 

Additional Inherited Members

- 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)
 
- 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...
 
ImplTypegetImpl () const
 Utility for easy access to the 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 399 of file SPIRVTypes.h.

Member Function Documentation

◆ get()

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

◆ getCapabilities()

void CooperativeMatrixNVType::getCapabilities ( SPIRVType::CapabilityArrayRefVector capabilities,
Optional< StorageClass >  storage = llvm::None 
)

Definition at line 249 of file SPIRVTypes.cpp.

References mlir::Type::cast(), and getElementType().

◆ getColumns()

unsigned CooperativeMatrixNVType::getColumns ( ) const

return the number of columns of the matrix.

Definition at line 236 of file SPIRVTypes.cpp.

Referenced by print(), and verifyCoopMatrixMulAdd().

◆ getElementType()

Type CooperativeMatrixNVType::getElementType ( ) const

Definition at line 228 of file SPIRVTypes.cpp.

Referenced by print(), and verifyCoopMatrixMulAdd().

◆ getExtensions()

void CooperativeMatrixNVType::getExtensions ( SPIRVType::ExtensionArrayRefVector extensions,
Optional< StorageClass >  storage = llvm::None 
)

Definition at line 240 of file SPIRVTypes.cpp.

References mlir::Type::cast(), and getElementType().

◆ getRows()

unsigned CooperativeMatrixNVType::getRows ( ) const

return the number of rows of the matrix.

Definition at line 234 of file SPIRVTypes.cpp.

Referenced by print(), and verifyCoopMatrixMulAdd().

◆ getScope()

Scope CooperativeMatrixNVType::getScope ( ) const

Return the scope of the cooperative matrix.

Definition at line 232 of file SPIRVTypes.cpp.

Referenced by print(), and verifyCoopMatrixMulAdd().


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