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

old cuarrays compat

Browse Source
This commit is contained in:
Mike J Innes 2018-11-14 10:53:26 -05:00
parent a57f66e58a
commit cbc29c889a
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/cuda/cuda.jl
View File

@ -2,6 +2,10 @@ module CUDA
using ..CuArrays using ..CuArrays
if !applicable(CuArray{UInt8}, undef, 1)
(T::Type{<:CuArray})(::UndefInitializer, sz...) = T(sz...)
end
if CuArrays.libcudnn != nothing if CuArrays.libcudnn != nothing
include("cudnn.jl") include("cudnn.jl")
else else