Skip to content

Commit d0b7ec8

Browse files
authored
Move enzyme tests to dedicated group (#3347)
1 parent 00006c1 commit d0b7ec8

File tree

2 files changed

+34
-1
lines changed

2 files changed

+34
-1
lines changed

.buildkite/pipeline.yml

+27
Original file line numberDiff line numberDiff line change
@@ -553,6 +553,33 @@ steps:
553553
architecture: CPU
554554
depends_on: "init_cpu"
555555

556+
#####
557+
##### Enzyme extension tests
558+
#####
559+
560+
- label: "👺 gpu Enzyme extension tests"
561+
env:
562+
JULIA_DEPOT_PATH: "$SVERDRUP_HOME/.julia-$BUILDKITE_BUILD_NUMBER"
563+
TEST_GROUP: "enzyme"
564+
commands:
565+
- "$SVERDRUP_HOME/julia-$JULIA_VERSION/bin/julia -O0 --color=yes --project -e 'using Pkg; Pkg.test()'"
566+
agents:
567+
queue: Oceananigans
568+
architecture: GPU
569+
depends_on: "init_gpu"
570+
571+
- label: "👹 cpu Enzyme extension tests"
572+
env:
573+
JULIA_DEPOT_PATH: "$TARTARUS_HOME/.julia-$BUILDKITE_BUILD_NUMBER"
574+
TEST_GROUP: "enzyme"
575+
CUDA_VISIBLE_DEVICES: "-1"
576+
commands:
577+
- "$TARTARUS_HOME/julia-$JULIA_VERSION/bin/julia -O0 --color=yes --project -e 'using Pkg; Pkg.test()'"
578+
agents:
579+
queue: Oceananigans
580+
architecture: CPU
581+
depends_on: "init_cpu"
582+
556583
#####
557584
##### Documentation
558585
#####

test/runtests.jl

+7-1
Original file line numberDiff line numberDiff line change
@@ -12,7 +12,6 @@ CUDA.allowscalar() do
1212
# Core Oceananigans
1313
if group == :unit || group == :all
1414
@testset "Unit tests" begin
15-
include("test_enzyme.jl")
1615
include("test_grids.jl")
1716
include("test_operators.jl")
1817
include("test_boundary_conditions.jl")
@@ -168,6 +167,13 @@ CUDA.allowscalar() do
168167
end
169168
end
170169

170+
# Tests for Enzyme extension
171+
if group == :enzyme || group == :all
172+
@testset "Enzyme extension tests" begin
173+
include("test_enzyme.jl")
174+
end
175+
end
176+
171177
if group == :convergence
172178
include("test_convergence.jl")
173179
end

0 commit comments

Comments
 (0)