ci: try fixing more tests

LuxDL · Nov 4, 2024 · 33d6dac · 33d6dac
1 parent 2ffa2f7
commit 33d6dac
Show file tree

Hide file tree

Showing 6 changed files with 13 additions and 8 deletions.
diff --git a/.buildkite/testing.yml b/.buildkite/testing.yml
@@ -1,7 +1,7 @@
 steps:
   - group: ":julia: (Lux) CUDA GPU"
     steps:
-      - label: ":julia: Julia {{matrix.julia}} + CUDA GPU"
+      - label: ":julia: Julia {{matrix.julia}} + CUDA GPU + {{matrix.group}}"
         plugins:
           - JuliaCI/julia#v1:
               version: "{{matrix.julia}}"
@@ -45,13 +45,17 @@ steps:
               include(joinpath(dir, "../test/runtests.jl"))'
         env:
           BACKEND_GROUP: "CUDA"
+          LUX_TEST_GROUP: "{{matrix.group}}"
         if: build.message !~ /\[skip tests\]/ && build.message !~ /\[skip ci\]/
         timeout_in_minutes: 60
         matrix:
           setup:
             julia:
               - "1.10"
               - "1"
+            group:
+              - "!reactant"
+              - "reactant"
 
   - group: ":julia: (Lux) AMD GPU"
     steps:

diff --git a/docs/Project.toml b/docs/Project.toml
@@ -48,7 +48,7 @@ Lux = "1"
 LuxCUDA = "0.3.2"
 LuxCore = "1"
 LuxLib = "1.3.4"
-LuxTestUtils = "1.2"
+LuxTestUtils = "1.4"
 MLDataDevices = "1.4"
 Optimisers = "0.3.3"
 Pkg = "1.10"

diff --git a/lib/LuxLib/test/Project.toml b/lib/LuxLib/test/Project.toml
@@ -47,7 +47,7 @@ Hwloc = "3.2"
 InteractiveUtils = "<0.0.1, 1"
 JLArrays = "0.1.5"
 LoopVectorization = "0.12.171"
-LuxTestUtils = "1.2.1"
+LuxTestUtils = "1.4"
 MKL = "0.7"
 MLDataDevices = "1.0.0"
 NNlib = "0.9.21"

diff --git a/lib/LuxTestUtils/Project.toml b/lib/LuxTestUtils/Project.toml
@@ -1,7 +1,7 @@
 name = "LuxTestUtils"
 uuid = "ac9de150-d08f-4546-94fb-7472b5760531"
 authors = ["Avik Pal <[email protected]>"]
-version = "1.3.1"
+version = "1.4.0"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

diff --git a/lib/LuxTestUtils/src/LuxTestUtils.jl b/lib/LuxTestUtils/src/LuxTestUtils.jl
@@ -24,7 +24,7 @@ const FD = FiniteDiff
 # Check if JET will work
 try
     using JET: JET, JETTestFailure, get_reports, report_call, report_opt
-    global JET_TESTING_ENABLED = true
+    global JET_TESTING_ENABLED = v"1.10-" ≤ VERSION < v"1.11-"
 catch err
     @error "`JET.jl` did not successfully precompile on $(VERSION). All `@jet` tests will \
             be skipped." maxlog=1 err=err
@@ -36,14 +36,15 @@ try
     using Enzyme: Enzyme
     __ftest(x) = x
     Enzyme.autodiff(Enzyme.Reverse, __ftest, Enzyme.Active, Enzyme.Active(2.0))
-    global ENZYME_TESTING_ENABLED = length(VERSION.prerelease) == 0
+    # XXX: Lift this once Enzyme supports 1.11 properly
+    global ENZYME_TESTING_ENABLED = v"1.10-" ≤ VERSION < v"1.11-"
 catch err
     global ENZYME_TESTING_ENABLED = false
 end
 
 if !ENZYME_TESTING_ENABLED
     @warn "`Enzyme.jl` is currently not functional on $(VERSION) either because it errored \
-           of the current version is a prerelease. Enzyme tests will be skipped..."
+           or the current version is a prerelease. Enzyme tests will be skipped..."
 end
 
 include("test_softfail.jl")

diff --git a/test/Project.toml b/test/Project.toml
@@ -59,7 +59,7 @@ Logging = "1.10"
 LoopVectorization = "0.12.171"
 LuxCore = "1.0"
 LuxLib = "1.3.4"
-LuxTestUtils = "1.3"
+LuxTestUtils = "1.4"
 MLDataDevices = "1.3"
 MLUtils = "0.4.3"
 NNlib = "0.9.24"