Run precompile workload with CUDA when functional

As suggested by @vchuravy, we guard the CUDA precompilation with `CUDA.functional()`.
HorribleSanity · May 10, 2024 · 6481ee9 · 6481ee9
1 parent 52e0cc1
commit 6481ee9
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/ext/RavenCUDAExt.jl b/ext/RavenCUDAExt.jl
@@ -71,9 +71,11 @@ CUDA.@device_override function Base.checkbounds(A::StaticArrays.MArray, I...)
 end
 
 PrecompileTools.@compile_workload let
-    for FT in (Float32, Float64)
-        AT = CuArray
-        Raven.precompile_workload(FT, AT)
+    if CUDA.functional()
+        for FT in (Float32, Float64)
+            AT = CuArray
+            Raven.precompile_workload(FT, AT)
+        end
     end
 end