Update GPU buildkite tests

JuliaImageRecon · Jul 4, 2024 · 5c1b27f · 5c1b27f
1 parent b835266
commit 5c1b27f
Show file tree

Hide file tree

Showing 3 changed files with 27 additions and 3 deletions.
diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
@@ -1,5 +1,5 @@
 steps:
-  - label: "Nvidia GPUs -- LinearOperators.jl"
+  - label: "Nvidia GPUs -- LinearOperatorCollection.jl"
     plugins:
       - JuliaCI/julia#v1:
           version: "1.10"
@@ -13,7 +13,26 @@ steps:
       using TestEnv
       TestEnv.activate();
       Pkg.add("CUDA")
-      Pkg.add("CuNFFT")
       Pkg.instantiate()
       include("test/gpu/cuda.jl")'
+    timeout_in_minutes: 30
+
+steps:
+  - label: "AMD GPUs -- LinearOperatorCollection.jl"
+    plugins:
+      - JuliaCI/julia#v1:
+          version: "1.10"
+    agents:
+      queue: "juliagpu"
+      rocm: "*"
+      rocmgpu: "*"
+    command: |
+      julia --color=yes --project -e '
+      using Pkg
+      Pkg.add("TestEnv")
+      using TestEnv
+      TestEnv.activate();
+      Pkg.add("AMDGPU")
+      Pkg.instantiate()
+      include("test/gpu/rocm.jl")'
     timeout_in_minutes: 30
diff --git a/test/gpu/cuda.jl b/test/gpu/cuda.jl
@@ -1,4 +1,4 @@
-using CUDA, CuNFFT
+using CUDA
 
 arrayTypes = [CuArray]
 

diff --git a/test/gpu/rocm.jl b/test/gpu/rocm.jl
@@ -0,0 +1,5 @@
+using AMDGPU
+
+arrayTypes = [ROCArray]
+
+include(joinpath(@__DIR__(), "..", "runtests.jl"))