From 5c1b27fa148eff9945e2e86afdbbf202afb9ee23 Mon Sep 17 00:00:00 2001
From: nHackel <nHackel@users.noreply.github.com>
Date: Thu, 4 Jul 2024 10:42:20 +0200
Subject: [PATCH] Update GPU buildkite tests

---
 .buildkite/pipeline.yml | 23 +++++++++++++++++++++--
 test/gpu/cuda.jl        |  2 +-
 test/gpu/rocm.jl        |  5 +++++
 3 files changed, 27 insertions(+), 3 deletions(-)
 create mode 100644 test/gpu/rocm.jl

diff --git a/.buildkite/pipeline.yml b/.buildkite/pipeline.yml
index 691a002..cc8470b 100644
--- 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
\ No newline at end of file
diff --git a/test/gpu/cuda.jl b/test/gpu/cuda.jl
index 5f6349c..e7d0eb2 100644
--- 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
new file mode 100644
index 0000000..ebf32fb
--- /dev/null
+++ b/test/gpu/rocm.jl
@@ -0,0 +1,5 @@
+using AMDGPU
+
+arrayTypes = [ROCArray]
+
+include(joinpath(@__DIR__(), "..", "runtests.jl"))
\ No newline at end of file