add automate onGPU

euro-hpc-pl · Sep 18, 2024 · 7a949ad · 7a949ad
1 parent a04f40b
commit 7a949ad
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -8,10 +8,13 @@ using LinearAlgebra
 using TensorCast
 using Statistics
 using MetaGraphs
+using CUDA
 
 disable_logging(LogLevel(1))
 
-onGPU = true
+user_onGPU = true  # or false, based on user's preference
+gpu_available = CUDA.functional()
+onGPU = user_onGPU && gpu_available
 
 using Test
 my_tests = []