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

initial cuarrays integration

Browse Source
This commit is contained in:
Mike J Innes 2017-08-24 17:00:48 +01:00
parent e7f26370d7
commit 52f5f4a4c0
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
REQUIRE
View File

@ -4,3 +4,4 @@ Juno
MacroTools 0.3.3 MacroTools 0.3.3
NNlib NNlib
ForwardDiff ForwardDiff
Requires

7
src/tracker/Tracker.jl
View File

@ -84,4 +84,11 @@ end
include("lib.jl") include("lib.jl")
include("numeric.jl") include("numeric.jl")
using Requires
@require CuArrays begin
import CuArrays: cu
cu(xs::TrackedArray) = TrackedArray(xs.f, cu(xs.x), cu(xs.Δ))
end
end end