Skip to content

Commit

Permalink
Fix call to run_pass_pipeline!
Browse files Browse the repository at this point in the history
  • Loading branch information
Sergio Sánchez Ramírez committed Jul 22, 2024
1 parent 1d1cfba commit 9eabd69
Showing 1 changed file with 14 additions and 5 deletions.
19 changes: 14 additions & 5 deletions src/Compiler.jl
Original file line number Diff line number Diff line change
Expand Up @@ -184,12 +184,21 @@ function compile_to_module!(mod, f, args; optimize=true)

if optimize
run_pass_pipeline!(
opt_passes *
",enzyme-batch," *
opt_passes *
",enzyme,arith-raise{stablehlo=true},canonicalize, remove-unnecessary-enzyme-ops, enzyme-simplify-math," *
opt_passes,
mod,
join(
[
opt_passes,
"enzyme-batch",
opt_passes,
"enzyme",
"arith-raise{stablehlo=true}",
"canonicalize",
"remove-unnecessary-enzyme-ops",
"enzyme-simplify-math",
opt_passes,
],
',',
),
)
end

Expand Down

0 comments on commit 9eabd69

Please sign in to comment.