The pipeline pass is now functional

2022-05-15 22:29:27 +08:00
parent 7e0e7ec365
commit 7027af9666
4 changed files with 119 additions and 98 deletions
--- a/include/triton/Dialect/TritonGPU/Transforms/Passes.td
+++ b/include/triton/Dialect/TritonGPU/Transforms/Passes.td
@@ -21,7 +21,9 @@ def TritonGPUPipeline : Pass<"tritongpu-pipeline", "mlir::ModuleOp"> {

  let constructor = "mlir::triton::gpu::createPipelinePass";

-  let dependentDialects = ["mlir::triton::gpu::TritonGPUDialect"];
+  let dependentDialects = ["mlir::triton::gpu::TritonGPUDialect",
+                           "mlir::scf::SCFDialect",
+                           "mlir::arith::ArithmeticDialect"];
 }

 #endif