MLIR 23.0.0git
mlir::spirv::MatrixType Class Reference

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

Inheritance diagram for mlir::spirv::MatrixType:

Public Member Functions

Type getColumnType () const
unsigned getNumRows () const
 Returns the number of rows.
unsigned getNumColumns () const
 Returns the number of columns.
unsigned getNumElements () const
 Returns total number of elements (rows*columns).
Type getElementType () const
 Returns the elements' type (i.e, single element type).
 operator ShapedType () const
ArrayRef< int64_tgetShape () const
bool hasRank () const
MatrixType cloneWith (std::optional< ArrayRef< int64_t > > shape, Type elementType) const
Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >
ImplTypegetImpl () const
 Utility for easy access to the storage instance.

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 verifyInvariants (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.
Static Public Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >
static TypeID getTypeID ()
 Return a unique identifier for the concrete type.
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.
static detail::InterfaceMap getInterfaceMap ()
 Returns an interface map for the interfaces registered to this storage user.
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.
static auto getWalkImmediateSubElementsFn ()
 Returns a function that walks immediate sub elements of a given instance of the storage user.
static auto getReplaceImmediateSubElementsFn ()
 Returns a function that replaces immediate sub elements of a given instance of the storage user.
static void attachInterface (MLIRContext &context)
 Attach the given models as implementations of the corresponding interfaces for the concrete storage user class.
static ConcreteType get (MLIRContext *ctx, Args &&...args)
 Get or create a new ConcreteT instance within the ctx.
static ConcreteType getChecked (const Location &loc, Args &&...args)
 Get or create a new ConcreteT instance within the ctx, defined at the given, potentially unknown, location.
static ConcreteType getChecked (function_ref< InFlightDiagnostic()> emitErrorFn, MLIRContext *ctx, Args... args)
 Get or create a new ConcreteT instance within the ctx.
static ConcreteType getFromOpaquePointer (const void *ptr)
 Get an instance of the concrete type from a void pointer.

Static Public Attributes

static constexpr StringLiteral name = "spirv.matrix"

Additional Inherited Members

Public Types inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >
using Base
 Utility declarations for the concrete attribute class.
using ImplType
using HasTraitFn
Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >
LogicalResult mutate (Args &&...args)
 Mutate the current storage instance.
Static Protected Member Functions inherited from mlir::detail::StorageUserBase< ConcreteType, BaseType, StorageType, detail::TypeUniquer, Traits... >
static LogicalResult verifyInvariants (Args... args)
 Default implementation that just returns success.

Detailed Description

Definition at line 428 of file SPIRVTypes.h.

Member Function Documentation

◆ cloneWith()

MatrixType mlir::spirv::MatrixType::cloneWith ( std::optional< ArrayRef< int64_t > > shape,
Type elementType ) const
inline

Definition at line 468 of file SPIRVTypes.h.

References get(), and getNumColumns().

◆ get()

◆ getChecked()

◆ getColumnType()

◆ getElementType()

Type MatrixType::getElementType ( ) const

◆ getNumColumns()

◆ getNumElements()

unsigned MatrixType::getNumElements ( ) const

Returns total number of elements (rows*columns).

Definition at line 1248 of file SPIRVTypes.cpp.

References getNumColumns(), and getNumRows().

◆ getNumRows()

unsigned MatrixType::getNumRows ( ) const

◆ getShape()

◆ hasRank()

bool mlir::spirv::MatrixType::hasRank ( ) const
inline

Definition at line 466 of file SPIRVTypes.h.

◆ isValidColumnType()

bool MatrixType::isValidColumnType ( Type columnType)
static

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

Definition at line 1222 of file SPIRVTypes.cpp.

Referenced by verifyInvariants().

◆ operator ShapedType()

mlir::spirv::MatrixType::operator ShapedType ( ) const
inline

Definition at line 462 of file SPIRVTypes.h.

◆ verifyInvariants()

LogicalResult MatrixType::verifyInvariants ( 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 1202 of file SPIRVTypes.cpp.

References mlir::emitError(), isValidColumnType(), and success().

Member Data Documentation

◆ name

StringLiteral mlir::spirv::MatrixType::name = "spirv.matrix"
staticconstexpr

Definition at line 434 of file SPIRVTypes.h.


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