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

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

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

Public Member Functions

Type getColumnType () const
 
unsigned getNumRows () const
 Returns the number of rows. More...
 
unsigned getNumColumns () const
 Returns the number of columns. More...
 
unsigned getNumElements () const
 Returns total number of elements (rows*columns). More...
 
Type getElementType () const
 Returns the elements' type (i.e, single element type). 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 MatrixType get (Type columnType, uint32_t columnCount)
 
static MatrixType getChecked (function_ref< InFlightDiagnostic()> emitError, Type columnType, uint32_t columnCount)
 
static LogicalResult verify (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. More...
 
- 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 423 of file SPIRVTypes.h.

Member Function Documentation

◆ get()

MatrixType MatrixType::get ( Type  columnType,
uint32_t  columnCount 
)
static

Definition at line 1090 of file SPIRVTypes.cpp.

References mlir::Type::getContext().

Referenced by mlir::spirv::Deserializer::collect(), and parseMatrixType().

◆ getCapabilities()

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

Definition at line 1149 of file SPIRVTypes.cpp.

◆ getChecked()

MatrixType MatrixType::getChecked ( function_ref< InFlightDiagnostic()>  emitError,
Type  columnType,
uint32_t  columnCount 
)
static

Definition at line 1094 of file SPIRVTypes.cpp.

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

◆ getColumnType()

Type MatrixType::getColumnType ( ) const

Definition at line 1128 of file SPIRVTypes.cpp.

Referenced by mlir::spirv::CompositeType::getElementType(), and print().

◆ getElementType()

Type MatrixType::getElementType ( ) const

Returns the elements' type (i.e, single element type).

Definition at line 1130 of file SPIRVTypes.cpp.

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

Referenced by verifyCoopMatrixMulAdd().

◆ getExtensions()

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

Definition at line 1144 of file SPIRVTypes.cpp.

◆ getNumColumns()

unsigned MatrixType::getNumColumns ( ) const

Returns the number of columns.

Definition at line 1134 of file SPIRVTypes.cpp.

Referenced by print(), and verifyCoopMatrixMulAdd().

◆ getNumElements()

unsigned MatrixType::getNumElements ( ) const

Returns total number of elements (rows*columns).

Definition at line 1140 of file SPIRVTypes.cpp.

◆ getNumRows()

unsigned MatrixType::getNumRows ( ) const

Returns the number of rows.

Definition at line 1136 of file SPIRVTypes.cpp.

References getShape().

Referenced by verifyCoopMatrixMulAdd().

◆ isValidColumnType()

bool MatrixType::isValidColumnType ( Type  columnType)
static

Returns true if the matrix elements are vectors of float elements.

Definition at line 1120 of file SPIRVTypes.cpp.

References mlir::Type::dyn_cast(), and vectorType().

◆ verify()

LogicalResult MatrixType::verify ( function_ref< InFlightDiagnostic()>  emitError,
Type  columnType,
uint32_t  columnCount 
)
static

The underlying vectors (columns) must be of size 2, 3, or 4

Definition at line 1100 of file SPIRVTypes.cpp.

References mlir::Type::cast(), mlir::emitError(), getShape(), and mlir::success().


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