diff --git a/src/Precompile.jl b/src/Precompile.jl index a125f0ca5..3cbb964e2 100644 --- 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