Use CUDA conditionally in tests

rwth-irt · Apr 23, 2024 · 16bcc27 · 16bcc27
1 parent c080902
commit 16bcc27
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -13,11 +13,14 @@ using Test
 # Setup a list of rngs to loop over
 cpurng = Random.default_rng()
 Random.seed!(cpurng, 42)
-curng = CUDA.default_rng()
-Random.seed!(curng, 42)
-# replace in test files
-# rngs = (cpurng,)
-rngs = (cpurng, curng)
+if CUDA.functional()
+    # Use CUDA only if available
+    curng = CUDA.default_rng()
+    Random.seed!(curng, 42)
+    rngs = (cpurng, curng)
+else
+    rngs = (cpurng,)
+end
 
 CUDA.allowscalar(false)