|
11 | 11 |
|
12 | 12 | include "mlir/Pass/PassBase.td"
|
13 | 13 |
|
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 | + ]; |
18 | 22 | }
|
19 | 23 |
|
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.}]; |
25 | 29 | 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" |
31 | 32 | ];
|
32 | 33 | }
|
33 | 34 |
|
|
0 commit comments