From cab235a57863558aa060a28776f8934d5a0a0ed4 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Thu, 30 Nov 2017 13:51:31 +0000 Subject: [PATCH] gpu compat --- src/tracker/Tracker.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tracker/Tracker.jl b/src/tracker/Tracker.jl index 3a64fcb7..74ed2d75 100644 --- a/src/tracker/Tracker.jl +++ b/src/tracker/Tracker.jl @@ -40,7 +40,7 @@ TrackedArray(x::AbstractArray) = TrackedArray(Call(nothing), x, zeros(x)) isleaf(x::TrackedArray) = x.f == Call(nothing) -param(xs) = TrackedArray(AbstractFloat.(xs)) +param(xs) = TrackedArray(map(x -> AbstractFloat(x), xs)) param(xs::Real) = param(fill(xs)) istracked(x::TrackedArray) = true