From 6481ee9ef33a7d6f2e471deba7e1788f202d43b6 Mon Sep 17 00:00:00 2001 From: Lucas C Wilcox Date: Fri, 10 May 2024 09:39:53 -0700 Subject: [PATCH] Run precompile workload with CUDA when functional As suggested by @vchuravy, we guard the CUDA precompilation with `CUDA.functional()`. --- ext/RavenCUDAExt.jl | 8 +++++--- 1 file changed, 5 insertions(+), 3 deletions(-) diff --git a/ext/RavenCUDAExt.jl b/ext/RavenCUDAExt.jl index 34733f4..1a611b4 100644 --- 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