apply CI check inside setup_workload

EnzymeAD · Dec 12, 2024 · 75d894e · 75d894e
1 parent 483a52c
commit 75d894e
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git 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
@@ -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()