Move enzyme tests to dedicated group (#3347)

CliMA · Oct 20, 2023 · d0b7ec8 · d0b7ec8
1 parent 00006c1
commit d0b7ec8
Show file tree

Hide file tree

Showing 2 changed files with 34 additions and 1 deletion.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -553,6 +553,33 @@ steps:
       architecture: CPU
     depends_on: "init_cpu"
 
+#####
+##### Enzyme extension tests
+#####
+
+  - label: "👺 gpu Enzyme extension tests"
+    env:
+      JULIA_DEPOT_PATH: "$SVERDRUP_HOME/.julia-$BUILDKITE_BUILD_NUMBER"
+      TEST_GROUP: "enzyme"
+    commands:
+      - "$SVERDRUP_HOME/julia-$JULIA_VERSION/bin/julia -O0 --color=yes --project -e 'using Pkg; Pkg.test()'"
+    agents:
+      queue: Oceananigans
+      architecture: GPU
+    depends_on: "init_gpu"
+
+  - label: "👹 cpu Enzyme extension tests"
+    env:
+      JULIA_DEPOT_PATH: "$TARTARUS_HOME/.julia-$BUILDKITE_BUILD_NUMBER"
+      TEST_GROUP: "enzyme"
+      CUDA_VISIBLE_DEVICES: "-1"
+    commands:
+      - "$TARTARUS_HOME/julia-$JULIA_VERSION/bin/julia -O0 --color=yes --project -e 'using Pkg; Pkg.test()'"
+    agents:
+      queue: Oceananigans
+      architecture: CPU
+    depends_on: "init_cpu"
+
 #####
 ##### Documentation
 #####

diff --git a/test/runtests.jl b/test/runtests.jl
@@ -12,7 +12,6 @@ CUDA.allowscalar() do
     # Core Oceananigans
     if group == :unit || group == :all
         @testset "Unit tests" begin
-            include("test_enzyme.jl")
             include("test_grids.jl")
             include("test_operators.jl")
             include("test_boundary_conditions.jl")
@@ -168,6 +167,13 @@ CUDA.allowscalar() do
         end
     end
 
+    # Tests for Enzyme extension
+    if group == :enzyme || group == :all
+        @testset "Enzyme extension tests" begin
+            include("test_enzyme.jl")
+        end
+    end
+
     if group == :convergence
         include("test_convergence.jl")
     end