Merge pull request #941 from FluxML/tb/include_during_precompile

Don't include the CUDA module during precompilation.
FluxML · Nov 24, 2019 · fbb377a · fbb377a
2 parents 5839e16 + 4ece13c
commit fbb377a
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