diff --git a/mlir/include/mlir/Dialect/AMX/AMX.td b/mlir/include/mlir/Dialect/AMX/AMX.td --- a/mlir/include/mlir/Dialect/AMX/AMX.td +++ b/mlir/include/mlir/Dialect/AMX/AMX.td @@ -38,6 +38,16 @@ def AMX_Dialect : Dialect { let name = "amx"; let cppNamespace = "::mlir::amx"; + let description = [{ + The Intel Advanced Matrix Extensions (AMX) provides 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 details of AMX. + The dialect is split into user-facing AMX ops (AMX_Op) and + backend-facing intrinsic ops (AMX_IntrOp). + }]; } //===----------------------------------------------------------------------===// @@ -203,7 +213,7 @@ Example: ```mlir - %0 = amx.tile_muli %a zext, %b zext, %c + %0 = amx.tile_muli %a zext, %b zext, %c : vector<16x64xi8>, vector<16x64xi8>, vector<16x16xi32> ``` }];