Move extension testing to their own pipeline

JuliaGPU · Jul 11, 2023 · 46a1d1b · 46a1d1b
1 parent 0a5f753
commit 46a1d1b
Show file tree

Hide file tree

Showing 4 changed files with 37 additions and 12 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -6,6 +6,7 @@ steps:
           - "1.6"
           - "1.7"
           - "1.8"
+          - "1.9"
     plugins:
       - JuliaCI/julia#v1:
           version: "{{matrix.version}}"
@@ -27,6 +28,35 @@ steps:
     soft_fail:
       - exit_status: 3
 
+  - label: "CUDA Enzyme Julia {{matrix.version}}"
+    matrix:
+      setup:
+        version:
+          - "1.8"
+          - "1.9"
+    plugins:
+      - JuliaCI/julia#v1:
+          version: "{{matrix.version}}"
+      - JuliaCI/julia-coverage#v1:
+          codecov: true
+    command: |
+      julia -e 'println("--- :julia: Instantiating project")
+                using Pkg
+                Pkg.develop(; path=pwd())
+                Pkg.develop(; name="CUDA")
+                Pkg.add("Enzyme")' || exit 3
+
+      julia -e 'println("+++ :julia: Running tests")
+                using CUDA
+                include("test/extensions/enzyme.jl")
+                enzyme_testsuite(CUDABackend, CuArray)'
+    agents:
+      queue: "juliagpu"
+      cuda: "*"
+    timeout_in_minutes: 120
+    soft_fail:
+      - exit_status: 3
+
   - label: "Metal Julia {{matrix.version}}"
     matrix:
       setup:

diff --git a/test/enzyme.jl → test/extensions/enzyme.jl b/test/enzyme.jl → test/extensions/enzyme.jl
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -47,3 +47,10 @@ struct NewBackend <: KernelAbstractions.GPU end
     kernel = my_no_cpu_kernel(backend)
     @test_throws MethodError kernel()
 end
+
+include("extensions/enzyme.jl")
+@static if VERSION >= v"1.7.0"
+    @testset "Enzyme" begin
+        enzyme_testsuite(CPU, Array)
+    end
+end
diff --git a/test/testsuite.jl b/test/testsuite.jl
@@ -19,15 +19,9 @@ macro conditional_testset(name, skip_tests, expr)
     end)
 end
 
-const Enzyme_AVAILABLE = haskey(
-    Pkg.dependencies(),
-    Base.UUID("7da242da-08ed-463a-9acd-ee780be4f1d9")) # Enzyme
 
 include("test.jl")
 include("localmem.jl")
-if Enzyme_AVAILABLE
-    include("enzyme.jl")
-end
 include("private.jl")
 include("unroll.jl")
 include("nditeration.jl")
@@ -52,12 +46,6 @@ function testsuite(backend, backend_str, backend_mod, AT, DAT; skip_tests = Set{
         localmem_testsuite(backend, AT)
     end
 
-    @static if VERSION >= v"1.7.0" && Enzyme_AVAILABLE
-        @conditional_testset "Enzyme" skip_tests begin
-                enzyme_testsuite(backend, AT)
-        end
-    end
-
     @conditional_testset "Private" skip_tests begin
         private_testsuite(backend, AT)
     end