Don't include the CUDA module during precompilation.

If we do, we could end up replacing it at runtime.
FluxML · Nov 22, 2019 · 4ece13c · 4ece13c
1 parent 2fa3e56
commit 4ece13c
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/src/Flux.jl b/src/Flux.jl
@@ -39,6 +39,12 @@ include("data/Data.jl")
 include("deprecations.jl")
 
 function __init__()
+  precompiling = ccall(:jl_generating_output, Cint, ()) != 0
+
+  # we don't want to include the CUDA module when precompiling,
+  # or we could end up replacing it at run time (triggering a warning)
+  precompiling && return
+
   if !CuArrays.functional()
     # nothing to do here, and either CuArrays or one of its dependencies will have warned
   else