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-14 15:53:26 +00:00
|
|
|
if !applicable(CuArray{UInt8}, undef, 1)
|
|
|
|
|
(T::Type{<:CuArray})(::UndefInitializer, sz...) = T(sz...)
|
|
|
|
|
end
|
|
|
|
|
|
2018-11-06 12:39:54 +00:00
|
|
|
if CuArrays.libcudnn != nothing
|
|
|
|
|
include("cudnn.jl")
|
2018-11-01 04:07:16 +00:00
|
|
|
else
|
2018-11-06 12:39:54 +00:00
|
|
|
@warn("CUDNN is not installed, some functionality will not be available.")
|
2018-11-01 04:07:16 +00:00
|
|
|
end
|
2018-01-24 18:45:24 +00:00
|
|
|
|
|
|
|
|
end
|
