Add PDL dialect dependency on the pass

j2kun · Jul 22, 2024 · e418182 · e418182
1 parent 44617af
commit e418182
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/lib/Transform/Arith/BUILD b/lib/Transform/Arith/BUILD
@@ -25,6 +25,8 @@ gentbl_cc_library(
     td_file = "Passes.td",
     deps = [
         "@llvm-project//mlir:OpBaseTdFiles",
+        "@llvm-project//mlir:PDLDialectTdFiles",
+        "@llvm-project//mlir:PDLInterpOpsTdFiles",
         "@llvm-project//mlir:PassBaseTdFiles",
     ],
 )

diff --git a/lib/Transform/Arith/Passes.td b/lib/Transform/Arith/Passes.td
@@ -1,6 +1,8 @@
 #ifndef LIB_TRANSFORM_ARITH_PASSES_TD_
 #define LIB_TRANSFORM_ARITH_PASSES_TD_
 
+include "mlir/Dialect/PDL/IR/PDLDialect.td"
+include "mlir/Dialect/PDLInterp/IR/PDLInterpOps.td"
 include "mlir/Pass/PassBase.td"
 
 def MulToAdd : Pass<"mul-to-add"> {
@@ -15,6 +17,10 @@ def MulToAddPdll : Pass<"mul-to-add-pdll"> {
   let description = [{
     Convert multiplications to repeated additions (using pdll).
   }];
+  let dependentDialects = [
+    "mlir::pdl::PDLDialect",
+    "mlir::pdl_interp::PDLInterpDialect",
+];
 }
 
 #endif  // LIB_TRANSFORM_ARITH_PASSES_TD_