Skip to content

Commit 35f25ab

Browse files
authored
Call parent when unwrapping arrays. (#388)
1 parent 4cdb50b commit 35f25ab

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/host/abstractarray.jl

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -125,7 +125,7 @@ end
125125
function Base.copyto!(dest::Array, dstart::Integer,
126126
src::WrappedGPUArray, sstart::Integer, n::Integer)
127127
n == 0 && return dest
128-
temp = similar(src, n)
128+
temp = similar(parent(src), n)
129129
copyto!(temp, 1, src, sstart, n)
130130
copyto!(dest, dstart, temp, 1, n)
131131
return dest
@@ -134,7 +134,7 @@ end
134134
function Base.copyto!(dest::WrappedGPUArray, dstart::Integer,
135135
src::Array, sstart::Integer, n::Integer)
136136
n == 0 && return dest
137-
temp = similar(dest, n)
137+
temp = similar(parent(dest), n)
138138
copyto!(temp, 1, src, sstart, n)
139139
copyto!(dest, dstart, temp, 1, n)
140140
return dest

0 commit comments

Comments
 (0)