run tests without testing anything to precompile the right things.

Author: jumerckx

.github/workflows/CI.yml

@@ -46,6 +46,19 @@ jobs:
         with:
           cache-name: julia-cache;workflow=${{ github.workflow }};version=${{ matrix.version }};os=${{ matrix.os }};runtime=${{ matrix.runtime }};assertions=${{ matrix.assertions }}
           include-matrix: false
+      - name: "Run Tests to warm cache"
+        run: |
+          import Pkg
+          Pkg.Registry.update()
+          Pkg.test()
+        shell: julia --color=yes --depwarn=yes --project=. {0}
+        env:
+          JULIA_PKG_SERVER_REGISTRY_PREFERENCE: eager
+          ENABLE_PJRT_COMPATIBILITY: 1
+          REACTANT_TEST_GROUP: "none"
+          XLA_FLAGS: "--xla_force_host_platform_device_count=12"
+          JULIA_DEBUG: "Reactant,Reactant_jll"
+
       - name: Warm cache
         run: |
           import Pkg

test/runtests.jl

@@ -6,6 +6,13 @@ end
 
 const REACTANT_TEST_GROUP = lowercase(get(ENV, "REACTANT_TEST_GROUP", "all"))
 
+# this is for the CI to precompile everything before running tests
+if REACTANT_TEST_GROUP == "none"
+    println("Skipping all tests as REACTANT_TEST_GROUP is set to 'none'")
+    exit(0)
+end
+
+
 @testset "Reactant.jl Tests" begin
     if REACTANT_TEST_GROUP == "all" || REACTANT_TEST_GROUP == "core"
         if Sys.isapple() && haskey(Reactant.XLA.global_backend_state.clients, "metal")