Merge pull request #13 from dhairyagandhi96/master

Add reshaped array wrapper
JuliaGPU · Jan 16, 2019 · 74a0499 · 74a0499
2 parents 2f5b711 + e2fdd66
commit 74a0499
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/base.jl b/src/base.jl
@@ -36,7 +36,8 @@ const wrappers = (
   :(LinearAlgebra.UnitLowerTriangular{<:Any,AT})  => (A,mut)->LinearAlgebra.UnitLowerTriangular(mut(parent(A))),
   :(LinearAlgebra.UpperTriangular{<:Any,AT})      => (A,mut)->LinearAlgebra.UpperTriangular(mut(parent(A))),
   :(LinearAlgebra.UnitUpperTriangular{<:Any,AT})  => (A,mut)->LinearAlgebra.UnitUpperTriangular(mut(parent(A))),
-  :(LinearAlgebra.Diagonal{<:Any,AT})             => (A,mut)->LinearAlgebra.Diagonal(mut(parent(A)))
+  :(LinearAlgebra.Diagonal{<:Any,AT})             => (A,mut)->LinearAlgebra.Diagonal(mut(parent(A))),
+  :(Base.ReshapedArray{<:Any,<:Any,AT,<:Any})     => (A,mut)->Base.reshape(mut(parent(A)), size(A))
 )
 
 for (W, ctor) in wrappers