CI aware precompilation

EnzymeAD · Dec 12, 2024 · b30b128 · b30b128
1 parent 1fda8c7
commit b30b128
Showing 1 changed file with 2 additions and 4 deletions.
diff --git a/src/Reactant.jl b/src/Reactant.jl
@@ -124,10 +124,8 @@ function set_default_backend(backend::String)
     return set_default_backend(XLA.backends[backend])
 end
 
-#HACK: Disable precompilation for test
-@info gethostname()
-@static if splitpath(Base.active_project())[3] != "runner"
-    @info "enable precompilation" Base.active_project()
+@static if !haskey(ENV, "REACTANT_TEST_GROUP")
+    @info "enable precompilation" gethostname() Base.active_project()
     include("precompile.jl")
 end