From c67e33f387652835273b072af2764292b71afa4c Mon Sep 17 00:00:00 2001 From: Avik Pal Date: Thu, 1 Nov 2018 09:37:16 +0530 Subject: [PATCH] Make the changes backward compatible --- src/cuda/cuda.jl | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/cuda/cuda.jl b/src/cuda/cuda.jl index 15126aca..eb28abcf 100644 --- a/src/cuda/cuda.jl +++ b/src/cuda/cuda.jl @@ -2,6 +2,10 @@ module CUDA using ..CuArrays -CuArrays.libcudnn != nothing && include("cudnn.jl") +if isdefined(CuArrays, :libcudnn_handle) + handle() = CuArrays.libcudnn_handle[] +else + handle() = CuArrays.CUDNN.handle() +end end