Skip to content

Commit 33f6207

Browse files
committed
add deep tile pass for matmul and tests
1 parent 0225874 commit 33f6207

File tree

5 files changed

+808
-64
lines changed

5 files changed

+808
-64
lines changed

include/gc/Transforms/Passes.td

Lines changed: 15 additions & 14 deletions
Original file line numberDiff line numberDiff line change
@@ -11,23 +11,24 @@
1111

1212
include "mlir/Pass/PassBase.td"
1313

14-
def TileLinalgNamed : Pass<"tile-named-linalg", "func::FuncOp"> {
15-
let summary = "Tile linalg named operations.";
16-
let dependentDialects =
17-
["linalg::LinalgDialect", "scf::SCFDialect", "tensor::TensorDialect"];
14+
def ConvertOneDNNGraphToLinalg : Pass<"convert-onednn-graph-to-linalg"> {
15+
let summary =
16+
"Lower the operations from the oneDNN Graph dialect into Linalg";
17+
let description = [{Lowers the `onednn_graph` ops to `linalg` ops.}];
18+
let dependentDialects = [
19+
"func::FuncDialect", "math::MathDialect", "arith::ArithDialect",
20+
"tensor::TensorDialect", "linalg::LinalgDialect"
21+
];
1822
}
1923

20-
def ConvertOneDNNGraphToLinalg : Pass<"convert-onednn-graph-to-linalg"> {
21-
let summary = "Lower the operations from the oneDNN Graph dialect into Linalg";
22-
let description = [{
23-
Lowers the `onednn_graph` ops to `linalg` ops.
24-
}];
24+
def DeepTileContractionNamedOp
25+
: Pass<"deep-tile-contraction-named-op", "func::FuncOp"> {
26+
let summary = "Tile linalg contraction named operation deeply";
27+
let description =
28+
[{The pass tries to tile the linalg contraction named op deeply.}];
2529
let dependentDialects = [
26-
"func::FuncDialect",
27-
"math::MathDialect",
28-
"arith::ArithDialect",
29-
"tensor::TensorDialect",
30-
"linalg::LinalgDialect"
30+
"func::FuncDialect", "linalg::LinalgDialect", "scf::SCFDialect",
31+
"tensor::TensorDialect"
3132
];
3233
}
3334

lib/gc/Transforms/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,7 +6,7 @@ gc_set_mlir_link_components(MLIR_LINK_COMPONENTS
66

77
add_mlir_library(GCPasses
88
OneDNNGraphToLinalg.cpp
9-
TileNamed.cpp
9+
DeepTileContractionNamedOp.cpp
1010

1111
ADDITIONAL_HEADER_DIRS
1212
${PROJECT_SOURCE_DIR}/include

0 commit comments

Comments
 (0)