Fix tests

test/test-activation.jl

@@ -24,7 +24,9 @@ cutoff = 0.1
 down_factor = 6
 ds = create_CNOdownsampler(T, D, N0, down_factor, cutoff, force_cpu = true)
 u = ds(u0)
-ugpu = CuArray(u)
+if !CUDA.functional()
+    ugpu = CuArray(u)
+end
 N = size(u, 1)
 # Define some activation layers
 # (1) Identity activation

test/test-training.jl

@@ -29,7 +29,9 @@ down_factor0 = 8
 cutoff = 0.1
 ds = create_CNOdownsampler(T, D, N0, down_factor0, cutoff, force_cpu = true)
 u = ds(u0)
-ug = CuArray(u)
+if !CUDA.functional()
+    ug = CuArray(u)
+end
 N = size(u)[1]
 # Model configuration
 ch_ = [2]

test/test-updown.jl

@@ -21,7 +21,9 @@ T = Float32
 D = 2
 u0 = zeros(T, N0, N0, D, 1)
 u0[:, :, 1, 1] .= testimage("cameraman")
-gu0 = CuArray(u0)
+if !CUDA.functional()
+    gu0 = CuArray(u0)
+end
 cutoff = 0.1
 # first scale down to work with smaller images
 down_factor0 = 8