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

fixup

Browse Source
This commit is contained in:
Mike J Innes 2018-11-27 18:44:07 -05:00
parent 7992de5cba
commit 1c36504768
2 changed files with 4 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
src/cuda/curnn.jl
View File

@ -231,6 +231,9 @@ function LinearAlgebra.copy_transpose!(dst::CuArray, src::CuArray)
dst[I...] = src[reverse(I)...] dst[I...] = src[reverse(I)...]
return return
end end
blk, thr = cudims(dst)
@cuda blocks=blk threads=thr kernel(dst, src)
return dst
end end
CuParam{T,N} = Union{CuArray{T,N},TrackedArray{T,N,CuArray{T,N}}} CuParam{T,N} = Union{CuArray{T,N},TrackedArray{T,N,CuArray{T,N}}}

3
test/cuda/cuda.jl
View File

@ -37,8 +37,7 @@ Flux.back!(sum(l))
end end
if CuArrays.libcudnn != nothing if CuArrays.libcudnn != nothing
@info "Testing Flux/CUDNN BatchNorm" @info "Testing Flux/CUDNN"
include("cudnn.jl") include("cudnn.jl")
@info "Testing Flux/CUDNN RNN"
include("curnn.jl") include("curnn.jl")
end end