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-06-20 13:16:42 +00:00
|
|
|
if CuArrays.cudnn_available()
|
|
|
|
|
include("curnn.jl")
|
2018-06-28 08:51:17 +00:00
|
|
|
include("cudnn.jl")
|
2018-06-20 13:16:42 +00:00
|
|
|
end
|
2018-01-24 18:45:24 +00:00
|
|
|
|
|
|
|
|
end
|
