Uno reverse on _copyto

Author: kshyatt

ext/TensorKitCUDAExt/TensorKitCUDAExt.jl

@@ -18,6 +18,5 @@ using Random
 
 include("cutensormap.jl")
 include("truncation.jl")
-include("auxiliary.jl")
 
 end

ext/TensorKitCUDAExt/auxiliary.jl

@@ -57,9 +57,11 @@ function _interleave(a::NTuple{N}, b::NTuple{N}) where {N}
     return (a[1], b[1], _interleave(tail(a), tail(b))...)
 end
 
+_copyto!(A, B) = copyto!(A, B)
+
 # Low-overhead implementation of `copyto!` for specific case of `stride(B, 1) < stride(B, 2)`
-# used in indexmanipulations: avoids the overhead of Strided.jl
-function _copyto!(A::StridedView{<:Any, 1}, B::StridedView{<:Any, 2})
+# for CPU-hosted Arrays # used in indexmanipulations: avoids the overhead of Strided.jl
+function _copyto!(A::StridedView{TA, 1, AA}, B::StridedView{TB, 2, BB}) where {TA <: Number, TB <: Number, AA <: DenseArray{TA}, BB <: DenseArray{TB}}
     length(A) == length(B) || throw(DimensionMismatch(lazy"length of A ($(length(A))) does not match length of B ($(length(B))"))
 
     Adata = parent(A)