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

Merge #837

Browse Source 
837: Use `CuArrays.ones` instead `cuones` which is deprecated r=dhairyagandhi96 a=mimadrid

I

Co-authored-by: Miguel Madrid Mencía <miguel.madrid.mencia@gmail.com>
This commit is contained in:
bors[bot] 2019-08-12 05:36:29 +00:00
commit aab3c4e052
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
src/cuda/curnn.jl
View File

@ -130,8 +130,8 @@ end
# TODO: can we just manipulate strides here? # TODO: can we just manipulate strides here?
# TODO: should use repmat, but this isn't implemented. # TODO: should use repmat, but this isn't implemented.
hBatch(x::AbstractVector, h::CuVector) = h hBatch(x::AbstractVector, h::CuVector) = h
hBatch(x::AbstractMatrix, h::CuVector) = h .* cuones(1, size(x, 2)) hBatch(x::AbstractMatrix, h::CuVector) = h .* CuArrays.ones(1, size(x, 2))
hBatch(x::AbstractMatrix, h::CuMatrix) = h .* cuones(1, size(h,2) == 1 ? size(x,2) : 1) hBatch(x::AbstractMatrix, h::CuMatrix) = h .* CuArrays.ones(1, size(h,2) == 1 ? size(x,2) : 1)
function forward(rnn::RNNDesc{T}, x::CuArray{T}, h_::CuArray{T}, c_ = nothing, train = Val{false}) where T function forward(rnn::RNNDesc{T}, x::CuArray{T}, h_::CuArray{T}, c_ = nothing, train = Val{false}) where T
h = hBatch(x, h_) h = hBatch(x, h_)