Run Reactant tests with Julia v1.12 as well (#577)

* Run Reactant tests with Julia v1.12 as well

* [CI] Run Julia v1.12 tests with `--check-bounds=auto`

* Remove Reactant from test sources

* Move comment around

* Load Reactant also for quality assurance tests in v1.12

Author: giordano

.github/workflows/Benchmarks.yml

@@ -21,6 +21,7 @@ concurrency:
 env:
   # Don't make Reactant occupy all the GPU memory
   XLA_REACTANT_GPU_PREALLOCATE: 'false'
+  JULIA_PKG_SERVER_REGISTRY_PREFERENCE: 'eager'
 
 jobs:
   run-benchmarks:

.github/workflows/CI.yml

@@ -28,6 +28,7 @@ permissions:
 env:
   # Don't make Reactant occupy all the GPU memory
   XLA_REACTANT_GPU_PREALLOCATE: 'false'
+  JULIA_PKG_SERVER_REGISTRY_PREFERENCE: 'eager'
 
 jobs:
   test:
@@ -55,12 +56,12 @@ jobs:
           - version: "min"
             os: windows-latest
         include:
+          # `--check-bounds=auto` is necessary for testing Reactant workloads
+          # until https://github.com/EnzymeAD/Reactant.jl/issues/2347 is fixed.
           - version: "1.12.5"
             os: aws-linux-nvidia-gpu-l4
             arch: "default"
-            check-bounds: "yes"
-          # Necessary for testing Reactant workloads until
-          # https://github.com/EnzymeAD/Reactant.jl/issues/2347 is fixed.
+            check-bounds: "auto"
           - version: "1.11.9"
             os: aws-linux-nvidia-gpu-l4-4xl
             arch: "default"

.github/workflows/Documentation.yml

@@ -25,6 +25,7 @@ env:
   BREEZE_LABEL_BUILD_ALL_EXAMPLES: 'build all examples 🏗️'
   # Don't make Reactant occupy all the GPU memory
   XLA_REACTANT_GPU_PREALLOCATE: 'false'
+  JULIA_PKG_SERVER_REGISTRY_PREFERENCE: 'eager'
 
 jobs:
   build-docs:

test/Project.toml

@@ -39,7 +39,7 @@ NCDatasets = "0.14"
 Oceananigans = "0.105.1, 0.106"
 ParallelTestRunner = "2.0.1"
 RRTMGP = "0.21"
-Reactant = "0.2.210"
+Reactant = "0.2.238"
 SpecialFunctions = "2"
 Statistics = "<0.0.1, 1"
 Test = "<0.0.1, 1"

test/quality_assurance.jl

@@ -9,15 +9,15 @@ using ClimaComms: ClimaComms
 using Dates: Dates
 using RRTMGP: RRTMGP
 using SpecialFunctions: SpecialFunctions
-if VERSION < v"1.12"
-    using Reactant: Reactant
+using Reactant: Reactant
 
+allow_unanalyzable = if VERSION < v"1.12"
     # For some reason `BreezeReactantExt` is non-analyzable,
     # but only in Julia v1.11-.
     BreezeReactantExt = isdefined(Base, :get_extension) ? Base.get_extension(Breeze, :BreezeReactantExt) : Breeze.BreezeReactantExt
-    allow_unanalyzable = (BreezeReactantExt,)
+    (BreezeReactantExt,)
 else
-    allow_unanalyzable = ()
+    ()
 end
 
 @testset "Aqua" begin

test/runtests.jl

@@ -8,11 +8,11 @@ testsuite = find_tests(@__DIR__)
 # Parse arguments
 args = parse_args(ARGS)
 
-const REACTANT_COMPAT = !((VERSION >= v"1.12") || (Base.JLOptions().check_bounds == 1))
+const REACTANT_COMPAT = VERSION < v"1.13-" && Base.JLOptions().check_bounds != 1
 
 if filter_tests!(testsuite, args)
     # Reactant compilation tests require --check-bounds=auto (Reactant/Enzyme
-    # limitation) and Julia < 1.12 until upstream support is improved.
+    # limitation).
     if !REACTANT_COMPAT
         delete!(testsuite, "reactant_centered_compilation")
         delete!(testsuite, "reactant_weno_compilation")