diff --git a/src/Reactant.jl b/src/Reactant.jl index 60e2d5f8..6922cf6e 100644 --- a/src/Reactant.jl +++ b/src/Reactant.jl @@ -124,9 +124,6 @@ function set_default_backend(backend::String) return set_default_backend(XLA.backends[backend]) end -@static if !haskey(ENV, "REACTANT_TEST_GROUP") - @info "enable precompilation" gethostname() Base.active_project() - include("precompile.jl") -end +include("precompile.jl") end # module diff --git a/src/precompile.jl b/src/precompile.jl index 83d70962..dbd36437 100644 --- a/src/precompile.jl +++ b/src/precompile.jl @@ -1,6 +1,10 @@ using PrecompileTools: @setup_workload, @compile_workload @setup_workload begin + @static if haskey(ENV, "REACTANT_TEST_GROUP") + return + end + @info "enable precompilation" gethostname() Base.active_project() @compile_workload begin Reactant.__init__() cpu = XLA.CPUClient()