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

reverted back the weights changes in rnndesc

Browse Source
This commit is contained in:
Tejan Karmali 2018-10-10 10:29:15 -04:00
parent 7b3e9c35ad
commit 6b4bbd4fce
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
src/cuda/cudnn.jl
View File

@ -87,11 +87,8 @@ function RNNDesc{T}(mode::Int, input::Int, hidden::Int; layers = 1) where T
libcudnn_handle[],d[],hidden,layers,dropoutDesc,inputMode,direction,mode,algo,cudnnDataType(T))
w = cuzeros(T, rnnParamSize(T, d[], input))
(wx, wh), bias = params(w, input, hidden, ngates(mode))
w_ = vcat(wx[:], wh[:], bias)
w[1:length(w_)] .= w_
# TODO: avoid reserve allocation here
rd = RNNDesc{T}(mode, input, hidden, w, (wx, wh), bias, d[])
rd = RNNDesc{T}(mode, input, hidden, w, params(w, input, hidden, ngates(mode))..., d[])
finalizer(rd) do x
@check ccall((:cudnnDestroyRNNDescriptor,libcudnn),cudnnStatus_t,(Ptr{Nothing},),x)
end