From 621a6dda164ae19bcf643e570ab309749e92be99 Mon Sep 17 00:00:00 2001
From: Alexander Barth <barth.alexander@gmail.com>
Date: Thu, 28 Nov 2024 15:47:10 +0200
Subject: [PATCH] extent test on AMD GPU

---
 test/test_reconstruct_points.jl           | 5 ++++-
 test/test_reconstruct_points_laplacian.jl | 3 +++
 2 files changed, 7 insertions(+), 1 deletion(-)

diff --git a/test/test_reconstruct_points.jl b/test/test_reconstruct_points.jl
index f5ed8fc..a6a6996 100644
--- a/test/test_reconstruct_points.jl
+++ b/test/test_reconstruct_points.jl
@@ -1,10 +1,11 @@
-ENV["CUDA_VISIBLE_DEVICES"]=""
+#ENV["CUDA_VISIBLE_DEVICES"]=""
 
 using DINCAE
 using NCDatasets
 using Random
 using Test
 using CUDA
+using AMDGPU
 
 T = Float32
 filename = "subset-sla-train.nc"
@@ -58,6 +59,8 @@ Random.seed!(seed)
 Atype =
     if CUDA.functional()
         CuArray{T}
+    elseif AMDGPU.functional()
+        ROCArray{T}
     else
         Array{T}
     end
diff --git a/test/test_reconstruct_points_laplacian.jl b/test/test_reconstruct_points_laplacian.jl
index 873bc59..f495a20 100644
--- a/test/test_reconstruct_points_laplacian.jl
+++ b/test/test_reconstruct_points_laplacian.jl
@@ -5,6 +5,7 @@ using NCDatasets
 using Random
 using Test
 using CUDA
+using AMDGPU
 
 T = Float32
 filename = "subset-sla-train.nc"
@@ -60,6 +61,8 @@ Random.seed!(seed)
 Atype =
     if CUDA.functional()
         CuArray{T}
+    elseif AMDGPU.functional()
+        ROCArray{T}
     else
         Array{T}
     end