Print HLO

Author: giordano

test/nn/lux.jl

@@ -45,6 +45,8 @@ end
     @info @__LINE__
     f = Reactant.compile((a, b, c, d) -> first(a(b, c, d)), (cmodel, cnoisy, cps, cst))
     @info @__LINE__
+    Reactant.Compiler.compile_mlir((a, b, c, d) -> first(a(b, c, d)), (cmodel, cnoisy, cps, cst))
+    @info @__LINE__
 
     comp = f(cmodel, cnoisy, cps, cst)
     @info @__LINE__
@@ -65,6 +67,10 @@ end
         gradient_loss_function, (cmodel, cnoisy, ctarget, cps, cst2)
     )
     @info @__LINE__
+    Reactant.Compiler.compile_mlir(
+        gradient_loss_function, (cmodel, cnoisy, ctarget, cps, cst2)
+    )
+    @info @__LINE__
 
     res_reactant, dps_reactant = compiled_gradient(cmodel, cnoisy, ctarget, cps, cst2)
     @info @__LINE__