diff --git a/src/cuda/curnn.jl b/src/cuda/curnn.jl index 09f6d43c..4990599f 100644 --- a/src/cuda/curnn.jl +++ b/src/cuda/curnn.jl @@ -63,7 +63,7 @@ function RNNDesc{T}(mode::Int, input::Int, hidden::Int; layers = 1) where T @check ccall((:cudnnSetRNNDescriptor_v6,libcudnn), cudnnStatus_t, (Ptr{Nothing},Ptr{Nothing},Cint,Cint,Ptr{Nothing},Cint,Cint,Cint,Cint,Cint), handle(),d[],hidden,layers,dropoutDesc,inputMode,direction,mode,algo,cudnnDataType(T)) - w = cuzeros(T, rnnParamSize(T, d[], input)) + w = CuArrays.zeros(T, rnnParamSize(T, d[], input)) # TODO: avoid reserve allocation here rd = RNNDesc{T}(mode, input, hidden, w, params(w, input, hidden, ngates(mode))..., d[]) finalizer(rd) do x