fix: before_enzyme optimizations (#275)

avik-pal · web-flow · commit 2aceb8a7c23f · 2024-11-14T16:19:03.000-05:00
diff --git a/src/Compiler.jl b/src/Compiler.jl
@@ -333,7 +333,7 @@ function compile_mlir!(mod, f, args; optimize::Union{Bool,Symbol}=true)
             ),
         )
     elseif optimize === :before_enzyme
-        run_pass_pipeline!(mod, join([opt_passes, "enzyme-batch", opt_passes]))
+        run_pass_pipeline!(mod, join([opt_passes, "enzyme-batch", opt_passes], ","))
         run_pass_pipeline!(mod, "enzyme,arith-raise{stablehlo=true}"; enable_verifier=false)
         run_pass_pipeline!(
             mod, "canonicalize,remove-unnecessary-enzyme-ops,enzyme-simplify-math"

Original file line number	Diff line number	Diff line change
`@@ -333,7 +333,7 @@ function compile_mlir!(mod, f, args; optimize::Union{Bool,Symbol}=true)`
`333`	`333`	`),`
`334`	`334`	`)`
`335`	`335`	`elseif optimize === :before_enzyme`
`336`		`- run_pass_pipeline!(mod, join([opt_passes, "enzyme-batch", opt_passes]))`
	`336`	`+ run_pass_pipeline!(mod, join([opt_passes, "enzyme-batch", opt_passes], ","))`
`337`	`337`	`run_pass_pipeline!(mod, "enzyme,arith-raise{stablehlo=true}"; enable_verifier=false)`
`338`	`338`	`run_pass_pipeline!(`
`339`	`339`	`mod, "canonicalize,remove-unnecessary-enzyme-ops,enzyme-simplify-math"`