mxarray ctx

2017-06-09 14:29:47 +01:00 · 2017-06-09 14:29:47 +01:00 · 9b18fd639a
commit 9b18fd639a
parent 0f433137d0
1 changed files with 2 additions and 2 deletions
--- 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