diff --git a/src/tracker/Tracker.jl b/src/tracker/Tracker.jl index 190837ab..94f9a94c 100644 --- a/src/tracker/Tracker.jl +++ b/src/tracker/Tracker.jl @@ -108,10 +108,8 @@ param(xs::AbstractArray) = TrackedArray(float.(xs)) param(x::TrackedReal) = track(identity, x) param(x::TrackedArray) = track(identity, x) -import NNlib.cudata import Adapt.adapt -cudata(x::TrackedArray) = data(x) adapt(T, xs::TrackedArray) = param(adapt(T, data(xs))) end