Update EnzymeAD/Enzyme-JAX to commit 04c15dc7c3736cca6f0a93c9934829fde489cc11 (#1980)

* Update EnzymeAD/Enzyme-JAX to commit 04c15dc7c3736cca6f0a93c9934829fde489cc11

Diff: EnzymeAD/Enzyme-JAX@ddea75d...04c15dc

* test: explicit options

---------

Co-authored-by: enzymead-bot[bot] <238314553+enzymead-bot[bot]@users.noreply.github.com>
Co-authored-by: Avik Pal <avikpal@mit.edu>

Author: enzymead-bot[bot]

deps/ReactantExtra/WORKSPACE

@@ -4,7 +4,7 @@ NSYNC_COMMIT = "82b118aa7ace3132e517e2c467f8732978cf4023"
 
 NSYNC_SHA256 = ""
 
-ENZYMEXLA_COMMIT = "ddea75dacb5b9ba708e0b44a0161d4acf9adae31"
+ENZYMEXLA_COMMIT = "04c15dc7c3736cca6f0a93c9934829fde489cc11"
 
 ENZYMEXLA_SHA256 = ""
 

test/batching.jl

@@ -49,7 +49,11 @@ function run_auto_batching_tests(f::F, args...) where {F}
             )
             @test occursin("stablehlo.while", hlo)
 
-            hlo = repr(@code_hlo f(args...))
+            hlo = repr(
+                @code_hlo compile_options = CompileOptions(;
+                    disable_auto_batching_passes=false
+                ) f(args...)
+            )
             @test !occursin("stablehlo.while", hlo)
         end
     end
@@ -115,9 +119,13 @@ end
     input1 = Reactant.to_rarray(Reactant.TestUtils.construct_test_array(Float32, 10))
     input2 = Reactant.to_rarray(Reactant.TestUtils.construct_test_array(Float32, 10))
 
-    hlo = @code_hlo optimize = false mctr(map_with_scalar_indexing, 1:8, input1, input2)
+    hlo = @code_hlo compile_options = CompileOptions(; disable_auto_batching_passes=true) mctr(
+        map_with_scalar_indexing, 1:8, input1, input2
+    )
     @test contains(repr(hlo), "stablehlo.while")
-    hlo = @code_hlo optimize = true mctr(map_with_scalar_indexing, 1:8, input1, input2)
+    hlo = @code_hlo compile_options = CompileOptions(; disable_auto_batching_passes=false) mctr(
+        map_with_scalar_indexing, 1:8, input1, input2
+    )
     @test !contains(repr(hlo), "stablehlo.while")
 
     res_ra = @jit mctr(map_with_scalar_indexing, 1:8, input1, input2)