TastyPancakes/Flux.jl
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Make the changes backward compatible

Browse Source
This commit is contained in:
Avik Pal 2018-11-01 09:37:16 +05:30 committed by GitHub
parent 7804d980b2
commit c67e33f387
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/cuda/cuda.jl
View File

@ -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