#include "mlir/Dialect/AMX/AMXDialect.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/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.
|
static LogicalResult | verifyTileSize (Operation *op, amx::TileType tp) |
| Verify that AMX supports the implied tile shape. More...
|
|
static LogicalResult | verifyMultShape (Operation *op, amx::TileType atp, amx::TileType btp, amx::TileType ctp, unsigned scale) |
| Verify that AMX supports the multiplication. More...
|
|
◆ GET_OP_CLASSES
◆ GET_OP_LIST
◆ GET_TYPEDEF_CLASSES
#define GET_TYPEDEF_CLASSES |
◆ GET_TYPEDEF_LIST
◆ verifyMultShape()
static LogicalResult verifyMultShape |
( |
Operation * |
op, |
|
|
amx::TileType |
atp, |
|
|
amx::TileType |
btp, |
|
|
amx::TileType |
ctp, |
|
|
unsigned |
scale |
|
) |
| |
|
static |
◆ verifyTileSize()
static LogicalResult verifyTileSize |
( |
Operation * |
op, |
|
|
amx::TileType |
tp |
|
) |
| |
|
static |