File tree Expand file tree Collapse file tree 2 files changed +13
-0
lines changed
include/mlir/Dialect/XeGPU/Transforms
lib/Dialect/XeGPU/Transforms Expand file tree Collapse file tree 2 files changed +13
-0
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff line change 11add_mlir_dialect_library(MLIRXeGPUTransforms
22 XeGPUFoldAliasOps.cpp
3+ XeGPUInstructionlize.cpp
34 XeGPUSubgroupDistribute.cpp
45 XeGPUUnroll.cpp
56
You can’t perform that action at this time.
0 commit comments