Test setup for CUDA

JuliaImageRecon · Jun 6, 2024 · 2475c89 · 2475c89
1 parent c278627
commit 2475c89
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/test/gpu/cuda.jl b/test/gpu/cuda.jl
@@ -0,0 +1,5 @@
+using CUDA, CuNFFT
+
+arrayTypes = [CuArray]
+
+include(joinpath(@__DIR__(), "..", "runtests.jl"))
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -8,7 +8,8 @@ using NFFT
 using RadonKA
 using JLArrays
 
-arrayTypes = [Array, JLArray]
+areTypesDefined = @isdefined arrayTypes
+arrayTypes = areTypesDefined ? arrayTypes : [Array, JLArray]
 
 @testset "LinearOperatorCollection" begin
   include("testNormalOp.jl")

diff --git a/test/testOperators.jl b/test/testOperators.jl
@@ -234,7 +234,7 @@ function testNFFT2d(N=16;arrayType = Array)
   F_nfft.toeplitz = false
   AHA = normalOperator(F_nfft)
   y_AHA_nfft = Array(AHA * xop)
-  y_AHA = F' * F * xop
+  y_AHA = F' * F * vec(x)
   @test y_AHA ≈ y_AHA_nfft   rtol = 1e-2
 
   # test AHA with Toeplitz