initial cuarrays integration

2017-08-24 17:00:48 +01:00 · 2017-08-24 17:00:48 +01:00 · 52f5f4a4c0
parent e7f26370d7
commit 52f5f4a4c0
2 changed files with 8 additions and 0 deletions
--- a/1
+++ b/1
@ -4,3 +4,4 @@ Juno
 MacroTools 0.3.3
 NNlib
 ForwardDiff
+Requires
--- a/src/tracker/Tracker.jl
+++ b/src/tracker/Tracker.jl
@ -84,4 +84,11 @@ end
 include("lib.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