AMXDialect.cpp File Reference

#include "mlir/Dialect/AMX/AMXDialect.h"
#include "mlir/Conversion/LLVMCommon/Pattern.h"
#include "mlir/Dialect/LLVMIR/LLVMDialect.h"
#include "mlir/Dialect/LLVMIR/LLVMTypes.h"
#include "mlir/IR/Builders.h"
#include "mlir/IR/DialectImplementation.h"
#include "mlir/IR/OpImplementation.h"
#include "mlir/IR/TypeUtilities.h"
#include "llvm/ADT/TypeSwitch.h"
#include "mlir/Dialect/AMX/AMXInterfaces.cpp.inc"
#include "mlir/Dialect/AMX/AMXDialect.cpp.inc"
#include "mlir/Dialect/AMX/AMXTypes.cpp.inc"
#include "mlir/Dialect/AMX/AMX.cpp.inc"

Go to the source code of this file.

Macros
#define	GET_TYPEDEF_LIST
#define	GET_OP_LIST
#define	GET_OP_CLASSES
#define	GET_TYPEDEF_CLASSES

Functions
static LogicalResult	verifyTileSize (Operation *op, amx::TileType tp)
	Verify that AMX supports the implied tile shape.
static LogicalResult	verifyMultShape (Operation *op, amx::TileType atp, amx::TileType btp, amx::TileType ctp, unsigned scale)
	Verify that AMX supports the multiplication.
static SmallVector< Value >	getTileSizes (Location loc, amx::TileType tType, RewriterBase &rewriter)
	Maps the 2-dim vector shape to the two 16-bit tile sizes.
static Value	computeStrideInBytes (Location loc, MemRefType mType, Value elementStride, RewriterBase &rewriter)
	Returns stride expressed in number of bytes for the given elementStride stride encoded in number of elements of the type mType.
static Value	inferStride (Location loc, MemRefType mType, Value base, RewriterBase &rewriter)
	Maps the 2-dim memref shape to the 64-bit stride.
template<typename OpTy, typename = std::enable_if_t<std::is_same_v<OpTy, amx::TileLoadOp> || std::is_same_v<OpTy, amx::TileStoreOp>>>
static LogicalResult	tileTransferVerifier (OpTy op)

Macro Definition Documentation

GET_OP_CLASSES

#define GET_OP_CLASSES

Definition at line 314 of file AMXDialect.cpp.

GET_OP_LIST

#define GET_OP_LIST

GET_TYPEDEF_CLASSES

#define GET_TYPEDEF_CLASSES

Definition at line 317 of file AMXDialect.cpp.

GET_TYPEDEF_LIST

#define GET_TYPEDEF_LIST

Function Documentation

computeStrideInBytes()

Value computeStrideInBytes ( Location loc, MemRefType mType, Value elementStride, RewriterBase & rewriter )

static

Returns stride expressed in number of bytes for the given elementStride stride encoded in number of elements of the type mType.

Definition at line 85 of file AMXDialect.cpp.

References mlir::Builder::getI64IntegerAttr(), and mlir::Builder::getIntegerType().

Referenced by inferStride().

getTileSizes()

SmallVector< Value > getTileSizes ( Location loc, amx::TileType tType, RewriterBase & rewriter )

static

Maps the 2-dim vector shape to the two 16-bit tile sizes.

The first dimension directly translates into the number of rows of the tiles. The second dimensions needs to be scaled by the number of bytes.

Definition at line 70 of file AMXDialect.cpp.

References mlir::Builder::getI16IntegerAttr(), and mlir::Builder::getIntegerType().

inferStride()

Value inferStride ( Location loc, MemRefType mType, Value base, RewriterBase & rewriter )

static

Maps the 2-dim memref shape to the 64-bit stride.

Note that the buffer shape may "envelop" the actual tile shape, and may be dynamically sized.

Definition at line 97 of file AMXDialect.cpp.

References computeStrideInBytes(), mlir::Builder::getI64IntegerAttr(), mlir::Builder::getIntegerType(), and mlir::MemRefDescriptor::stride().

tileTransferVerifier()

template<typename OpTy, typename = std::enable_if_t<std::is_same_v<OpTy, amx::TileLoadOp> || std::is_same_v<OpTy, amx::TileStoreOp>>>

LogicalResult tileTransferVerifier ( OpTy op )

static

Definition at line 132 of file AMXDialect.cpp.

References success(), and verifyTileSize().

verifyMultShape()

LogicalResult verifyMultShape ( Operation * op, amx::TileType atp, amx::TileType btp, amx::TileType ctp, unsigned scale )

static

Verify that AMX supports the multiplication.

Definition at line 55 of file AMXDialect.cpp.

References mlir::Operation::emitOpError(), and success().

verifyTileSize()

LogicalResult verifyTileSize ( Operation * op, amx::TileType tp )

static

Verify that AMX supports the implied tile shape.

Definition at line 43 of file AMXDialect.cpp.

References mlir::Operation::emitOpError(), and success().

Referenced by tileTransferVerifier().