[Mlir-commits] [mlir] [mlir][xegpu] Add definitons of MatrixDescType and related ops. (PR #153273)
Jianhui Li
llvmlistbot at llvm.org
Wed Aug 13 19:20:50 PDT 2025
================
@@ -201,4 +201,26 @@ def XeGPU_Nbarrier: XeGPUTypeDef<"Nbarrier", "nbarrier", [], "mlir::Type"> {
}];
}
+def XeGPU_MatrixDesc: XeGPUTypeDef<"MatrixDesc", "matrix_desc", [ShapedTypeInterface], "mlir::Type"> {
+ let summary = "MatrixDesc describing the data in SLM";
+ let description = [{
+ MatrixDesc represents a block of data stored in shared local memory.
+ By default, unless a layout attribute is provided, the data is stored
+ contiguously in row-major order within the region.
+ }];
+ let parameters = (ins ArrayRefParameter<"int64_t">: $shape,
+ "mlir::Type": $elementType,
+ OptionalParameter<"mlir::Attribute">: $layout);
----------------
Jianhui-Li wrote:
consider using mem_layout instead of layout, to differentiate with XeGPU.layout which describes the mapping between sg/lane ids to the data.
https://github.com/llvm/llvm-project/pull/153273
More information about the Mlir-commits
mailing list