probprog postpasses

Author: sbrantq

src/Compiler.jl

@@ -1183,6 +1183,7 @@ end
 # TODO we want to be able to run the more advanced passes via transform dialect as an enzyme intermediate
 # However, this errs as we cannot attach the transform with to the funcop itself [as we run a functionpass].
 const enzyme_pass::String = "enzyme{postpasses=\"arith-raise{stablehlo=true},canonicalize,cse,canonicalize,remove-unnecessary-enzyme-ops,enzyme-simplify-math,canonicalize,cse,canonicalize\"}"
+const probprog_pass::String = "probprog{postpasses=\"arith-raise{stablehlo=true},canonicalize,cse,canonicalize,remove-unnecessary-enzyme-ops,enzyme-simplify-math,canonicalize,cse,canonicalize\"}"
 
 function run_pass_pipeline!(mod, pass_pipeline, key=""; enable_verifier=true)
     pm = MLIR.IR.PassManager()
@@ -1753,7 +1754,7 @@ function compile_mlir!(
                         "enzyme-batch",
                         opt_passes2,
                         enzyme_pass,
-                        "probprog",
+                        probprog_pass,
                         opt_passes2,
                         "canonicalize",
                         "remove-unnecessary-enzyme-ops",
@@ -1767,7 +1768,7 @@ function compile_mlir!(
                         "enzyme-batch",
                         opt_passes2,
                         enzyme_pass,
-                        "probprog",
+                        probprog_pass,
                         opt_passes2,
                         "canonicalize",
                         "remove-unnecessary-enzyme-ops",
@@ -1794,7 +1795,7 @@ function compile_mlir!(
                         "enzyme-batch",
                         opt_passes2,
                         enzyme_pass,
-                        "probprog",
+                        probprog_pass,
                         opt_passes2,
                         "canonicalize",
                         "remove-unnecessary-enzyme-ops",
@@ -1811,7 +1812,7 @@ function compile_mlir!(
                         "enzyme-batch",
                         opt_passes2,
                         enzyme_pass,
-                        "probprog",
+                        probprog_pass,
                         opt_passes2,
                         "canonicalize",
                         "remove-unnecessary-enzyme-ops",