Tensors tests passing

Author: kshyatt

ext/TensorKitCUDAExt/cutensormap.jl

@@ -11,6 +11,10 @@ function TensorKit.tensormaptype(S::Type{<:IndexSpace}, N₁, N₂, TorA::Type{<
     end
 end
 
+function TensorKit.TensorMap{T, S, N₁, N₂, <:CuVector{T}}(t::TensorMap{T, S, N₁, N₂, A}) where {T, S, N₁, N₂, A}
+    return CuTensorMap{T, S, N₁, N₂}(CuArray(t.data), t.space)
+end
+
 function CuTensorMap{T}(::UndefInitializer, V::TensorMapSpace{S, N₁, N₂}) where {T, S, N₁, N₂}
     return CuTensorMap{T, S, N₁, N₂}(undef, V)
 end

src/tensors/tensor.jl

@@ -361,19 +361,14 @@ function TensorMap(
     end
 
     arraydata = reshape(data, arraysize)
-    if data isa DenseVector
-        t = TensorMap{T, S, N₁, N₂, typeof(data)}(undef, codom ← dom) # to ensure the correct storage type
-        t = project_symmetric!(t, arraydata)
-        if !isapprox(Array(arraydata), convert(Array, t); atol = tol)
-            throw(ArgumentError("Data has non-zero elements at incompatible positions"))
-        end
-    elseif data isa DenseArray # can be reshaped into a DenseVector
+    if data isa DenseArray # can be reshaped into a DenseVector
         A = densevectortype(typeof(data))
-        t = TensorMap{T, S, N₁, N₂, A}(undef, codom ← dom) # to ensure the correct storage type
-        t = project_symmetric!(t, arraydata)
+        t = TensorMap{T, S, N₁, N₂, Vector{T}}(undef, codom ← dom) # to ensure the correct storage type
+        t = project_symmetric!(t, Array(arraydata))
         if !isapprox(Array(arraydata), convert(Array, t); atol = tol)
             throw(ArgumentError("Data has non-zero elements at incompatible positions"))
         end
+        t = TensorMap(A(t.data), t.space)
     else
         t = TensorMap{T}(undef, codom ← dom) # to ensure the correct storage type
         t = project_symmetric!(t, arraydata)