diff --git a/ocean-climate-simulation/ocean_climate_simulation_mlir.jl b/ocean-climate-simulation/ocean_climate_simulation_mlir.jl index d9981ec..59187c3 100644 --- a/ocean-climate-simulation/ocean_climate_simulation_mlir.jl +++ b/ocean-climate-simulation/ocean_climate_simulation_mlir.jl @@ -1,9 +1,11 @@ ENV["run"] = false ENV["use-reactant"] = true +Reactant.Compiler.DUMP_LLVMIR[] = true + include("data_free_ocean_climate_simulation.jl") -unopt = @code_hlo optimize=false raise=true run!(simulation) +unopt = @code_hlo optimize=false raise=false run!(simulation) # Unoptimized HLO open("unopt_ocean_climate_simulation.mlir", "w") do io @@ -11,7 +13,7 @@ open("unopt_ocean_climate_simulation.mlir", "w") do io end # Optimized HLO -opt = @code_hlo optimize=:before_jit raise=true run!(simulation) +opt = @code_hlo optimize=:before_jit raise=false run!(simulation) open("opt_ocean_climate_simulation.mlir", "w") do io write(io, string(opt))