Expand description
amx dialect.
The Intel Advanced Matrix Extensions (AMX) provide a tile matrix multiply unit (TMUL), a tile control register (TILECFG), and eight tile registers TMM0 through TMM7 (TILEDATA).
This AMX dialect provides a bridge between MLIR concepts such as
vectors and memrefs and the lower level LLVM IR support of AMX.
The dialect is split into user-facing AMX ops (AMX_Op) and
backend-facing intrinsic ops (AMX_IntrOp).
Note that since configuration changes (implicit at dialect level) are costly, it is highly recommended to use the AMX dialect on same-shaped vectors, at least within a single method.
For details, see the Intel documentation: https://software.intel.com/content/www/us/en/develop/articles/intel-sdm.html
Structs§
- A
tile_loadoperation. Tile load operation. - A builder for a
tile_loadoperation. - A
tile_mulfoperation. Tile multiplication operation (floating-point). - A builder for a
tile_mulfoperation. - A
tile_mulioperation. Tile multiplication operation (integer). - A builder for a
tile_mulioperation. - A
tile_storeoperation. Tile store operation. - A builder for a
tile_storeoperation. - A
tile_zerooperation. Tile zero operation. - A builder for a
tile_zerooperation. - A
tdpbf16psoperation. - A builder for a
tdpbf16psoperation. - A
tdpbssdoperation. - A builder for a
tdpbssdoperation. - A
tdpbsudoperation. - A builder for a
tdpbsudoperation. - A
tdpbusdoperation. - A builder for a
tdpbusdoperation. - A
tdpbuudoperation. - A builder for a
tdpbuudoperation. - A
tileloadd64operation. - A builder for a
tileloadd64operation. - A
tilestored64operation. - A builder for a
tilestored64operation. - A
tilezerooperation. - A builder for a
tilezerooperation.
Functions§
- Creates a
tdpbf16psoperation. - Creates a
tdpbssdoperation. - Creates a
tdpbsudoperation. - Creates a
tdpbusdoperation. - Creates a
tdpbuudoperation. - Creates a
tile_loadoperation. - Creates a
tile_mulfoperation. - Creates a
tile_mulioperation. - Creates a
tile_storeoperation. - Creates a
tile_zerooperation. - Creates a
tileloadd64operation. - Creates a
tilestored64operation. - Creates a
tilezerooperation.