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

rm jit from cuda

Browse Source
This commit is contained in:
Mike J Innes 2018-04-15 15:08:58 +01:00
parent 5cc681317a
commit b05e755068
1 changed files with 0 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
src/cuda/cuda.jl
View File

@ -4,11 +4,4 @@ using CuArrays
CuArrays.cudnn_available() && include("cudnn.jl")
import ..Flux.JIT: Shape, restructure
function restructure(sh::Shape{T}, buf::CuVector{UInt8}) where T
buf = buf[1:sizeof(sh)]
reshape(reinterpret(T, buf), size(sh))
end
end