Skip to content

Commit af01c99

Browse files
committed
add skeleton
1 parent 777a403 commit af01c99

File tree

2 files changed

+13
-0
lines changed

2 files changed

+13
-0
lines changed

mlir/include/mlir/Dialect/XeGPU/Transforms/Passes.td

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -38,4 +38,16 @@ def XeGPUSubgroupDistribute : Pass<"xegpu-subgroup-distribute"> {
3838
];
3939
}
4040

41+
def XeGPUInstructionlize: Pass<"xegpu-instructionlize"> {
42+
let summary = "Instructionlize XeGPU ops";
43+
let description = [{
44+
The pass unrolls XeGPU ops working on large shapes into ops working on small shapes
45+
(given by the inst_data in the layout attr), such that each of them can be dispatch
46+
into a hardware instruction.
47+
}];
48+
let dependentDialects = [
49+
"memref::MemRefDialect", "xegpu::XeGPUDialect", "vector::VectorDialect"
50+
];
51+
}
52+
4153
#endif // MLIR_DIALECT_XEGPU_TRANSFORMS_PASSES_TD

mlir/lib/Dialect/XeGPU/Transforms/CMakeLists.txt

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,6 @@
11
add_mlir_dialect_library(MLIRXeGPUTransforms
22
XeGPUFoldAliasOps.cpp
3+
XeGPUInstructionlize.cpp
34
XeGPUSubgroupDistribute.cpp
45
XeGPUUnroll.cpp
56

0 commit comments

Comments
 (0)