test CI 2

EnzymeAD · Jan 3, 2025 · d4c71bb · d4c71bb
1 parent 9d82b9f
commit d4c71bb
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/src/Precompile.jl b/src/Precompile.jl
@@ -2,12 +2,9 @@ using PrecompileTools: @setup_workload, @compile_workload
 
 @setup_workload begin
     initialize_dialect()
-    cpu = XLA.CPUClient()
     @compile_workload begin
-        x = Reactant.ConcreteRArray(randn(Float64, 2, 2); client=cpu)
-        @code_hlo optimize = false sum(x)
+        interp = Reactant.ReactantInterpreter()
+        Base.code_ircode(sum, (Reactant.TracedRArray{Float64,2},); interp)
     end
-    XLA.free_client(cpu)
     deinitialize_dialect()
-    XLA.cpuclientcount[] = 0
 end