diff --git a/src/cuda/cuda.jl b/src/cuda/cuda.jl index eaa3fe00..764bb96f 100644 --- a/src/cuda/cuda.jl +++ b/src/cuda/cuda.jl @@ -2,6 +2,9 @@ module CUDA using CuArrays -CuArrays.cudnn_available() && include("cudnn.jl") +if CuArrays.cudnn_available() + include("cudnn.jl") + include("curnn.jl") +end end