2018-01-24 18:45:24 +00:00
|
|
|
module CUDA
|
|
|
|
|
|
|
|
|
|
using CuArrays
|
|
|
|
|
|
2018-06-20 13:16:42 +00:00
|
|
|
if CuArrays.cudnn_available()
|
2018-06-28 08:51:17 +00:00
|
|
|
CuParam{T,N} = Union{CuArray{T,N},TrackedArray{T,N,CuArray{T,N}}}
|
2018-06-20 13:16:42 +00:00
|
|
|
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
|
