MLIR  16.0.0git
Public Types | 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 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 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=std::nullopt)
 
void getCapabilities (SPIRVType::CapabilityArrayRefVector &capabilities, 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 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

- 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 452 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 99 of file StorageUniquerSupport.h.

Member Function Documentation

◆ get()

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

Definition at line 1175 of file SPIRVTypes.cpp.

References mlir::Type::getContext().

Referenced by parseMatrixType().

◆ getCapabilities()

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

Definition at line 1234 of file SPIRVTypes.cpp.

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

◆ getChecked()

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

Definition at line 1179 of file SPIRVTypes.cpp.

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

◆ getColumnType()

Type MatrixType::getColumnType ( ) const

◆ getElementType()

Type MatrixType::getElementType ( ) const

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

Definition at line 1215 of file SPIRVTypes.cpp.

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

◆ getExtensions()

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

Definition at line 1229 of file SPIRVTypes.cpp.

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

◆ getNumColumns()

unsigned MatrixType::getNumColumns ( ) const

Returns the number of columns.

Definition at line 1219 of file SPIRVTypes.cpp.

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

Referenced by print().

◆ getNumElements()

unsigned MatrixType::getNumElements ( ) const

Returns total number of elements (rows*columns).

Definition at line 1225 of file SPIRVTypes.cpp.

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

◆ getNumRows()

unsigned MatrixType::getNumRows ( ) const

Returns the number of rows.

Definition at line 1221 of file SPIRVTypes.cpp.

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

Referenced by getNumElements().

◆ isValidColumnType()

bool MatrixType::isValidColumnType ( Type  columnType)
static

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

Definition at line 1205 of file SPIRVTypes.cpp.

References mlir::Type::dyn_cast().

Referenced by verify().

◆ 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 1185 of file SPIRVTypes.cpp.

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


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