From 9b18fd639a94cc33968f99fcf0ab967697fd8255 Mon Sep 17 00:00:00 2001 From: Mike J Innes Date: Fri, 9 Jun 2017 14:29:47 +0100 Subject: [PATCH] mxarray ctx --- src/backend/mxnet/mxarray.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/backend/mxnet/mxarray.jl b/src/backend/mxnet/mxarray.jl index f89ed734..ecfe7439 100644 --- a/src/backend/mxnet/mxarray.jl +++ b/src/backend/mxnet/mxarray.jl @@ -32,8 +32,8 @@ end Base.copy(mx::MXArray) = copy!(Array{Float32}(size(mx)), mx) -function MXArray(xs::AbstractArray) - mx = MXArray(size(xs)) +function MXArray(xs::AbstractArray, ctx = mx.cpu()) + mx = MXArray(size(xs), ctx) copy!(mx, xs) end