2018-01-24 18:45:24 +00:00
|
|
|
module CUDA
|
|
|
|
|
|
2018-08-20 12:08:04 +00:00
|
|
|
using ..CuArrays
|
2018-01-24 18:45:24 +00:00
|
|
|
|
2018-11-01 04:07:16 +00:00
|
|
|
if isdefined(CuArrays, :libcudnn_handle)
|
|
|
|
|
handle() = CuArrays.libcudnn_handle[]
|
|
|
|
|
else
|
|
|
|
|
handle() = CuArrays.CUDNN.handle()
|
|
|
|
|
end
|
2018-01-24 18:45:24 +00:00
|
|
|
|
|
|
|
|
end
|
