From 1889ccd316b7220e1f5f8db87f11f21b2a77ad8c Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Sat, 19 Aug 2017 11:06:19 +0100 Subject: [PATCH] tracker submodule --- src/Flux.jl | 4 +++- src/{grad/track.jl => Tracker/Tracker.jl} | 4 ++++ src/{grad => Tracker}/lib.jl | 0 3 files changed, 7 insertions(+), 1 deletion(-) rename src/{grad/track.jl => Tracker/Tracker.jl} (98%) rename src/{grad => Tracker}/lib.jl (100%) diff --git a/src/Flux.jl b/src/Flux.jl index 8c8ac7c2..1cdb48a5 100644 --- a/src/Flux.jl +++ b/src/Flux.jl @@ -19,8 +19,10 @@ export @net, unroll, unroll1, @shapes, include("core.jl") import .FluxCore: graph +include("Tracker/Tracker.jl") +using .Tracker + include("utils.jl") -include("grad/track.jl") include("params.jl") include("compiler/code.jl") diff --git a/src/grad/track.jl b/src/Tracker/Tracker.jl similarity index 98% rename from src/grad/track.jl rename to src/Tracker/Tracker.jl index b8125aa4..24a1950c 100644 --- a/src/grad/track.jl +++ b/src/Tracker/Tracker.jl @@ -1,3 +1,5 @@ +module Tracker + data(x) = x struct Call{F,As<:Tuple} @@ -54,3 +56,5 @@ function Base.showarray(io::IO, X::Var, repr::Bool = true; header = true) Base.showarray(io, data(X), false, header = false) end end + +end diff --git a/src/grad/lib.jl b/src/Tracker/lib.jl similarity index 100% rename from src/grad/lib.jl rename to src/Tracker/lib.jl