From 75d894eeb2ebd046d150cc53e67c045f71806fda Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Ga=C3=ABtan=20LOUNES?= <gaetan.lounes@irisa.fr>
Date: Thu, 12 Dec 2024 20:46:15 +0100
Subject: [PATCH] apply CI check inside setup_workload

---
 src/Reactant.jl   | 5 +----
 src/precompile.jl | 4 ++++
 2 files changed, 5 insertions(+), 4 deletions(-)

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()