materialize!(::KnetArray, ...) returns KnetArray.

denizyuret · Jul 23, 2021 · e9bec84 · e9bec84
1 parent 1394098
commit e9bec84
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/knetarrays/getindex.jl b/src/knetarrays/getindex.jl
@@ -54,24 +54,28 @@ function materialize!(A::SubArray{T,N,<:KnetArray}, B) where {T,N}
     _A = view(CuArray(A.parent), A.indices...)
     _B = (B isa KnetArray || B isa AbstractArray ? CuArray(B) : B)
     materialize!(_A, _B)
+    return A
 end
 
 # For contiguous I, dotview(A, I...) gives a shared-memory KnetArray rather than a view
 function materialize!(A::KnetArray, B) where {T,N}
     _A = CuArray(A)
     _B = (B isa KnetArray || B isa AbstractArray ? CuArray(B) : B)
     materialize!(_A, _B)
+    return A
 end
 
 # Ambiguity fixes:
 function materialize!(A::SubArray{T,N,<:KnetArray}, B::Broadcasted{S}) where {T,N,S}
     _A = view(CuArray(A.parent), A.indices...)
     materialize!(_A, B)
+    return A
 end
 
 function materialize!(A::KnetArray{T,N}, B::Broadcasted{S}) where {T,N,S}
     _A = CuArray(A)
     materialize!(_A, B)
+    return A
 end