convert(TensorMap, t) retains storagetype

Author: lkdvos

src/tensors/tensor.jl

@@ -532,7 +532,8 @@ end
 #---------------------------
 Base.convert(::Type{TensorMap}, t::TensorMap) = t
 function Base.convert(::Type{TensorMap}, t::AbstractTensorMap)
-    return copy!(TensorMap{scalartype(t)}(undef, space(t)), t)
+    A = storagetype(t)
+    return copy!(TensorMapWithStorage{scalartype(A), A}(undef, space(t)), t)
 end
 
 function Base.convert(::Type{TensorMapWithStorage{T, A}}, t::TensorMap) where {T, A}

test/cuda/tensors.jl

@@ -258,14 +258,14 @@ for V in spacelist
                 end
             end
         end
-        @timedtestset "Tensor conversion" begin # TODO adjoint conversion methods don't work yet
+        @timedtestset "Tensor conversion" begin
             W = V1 ⊗ V2
             t = @constinferred CUDA.randn(W ← W)
-            #@test typeof(convert(TensorMap, t')) == typeof(t) # TODO Adjoint not supported yet
+            @test typeof(convert(TensorMap, t')) == typeof(t)
             tc = complex(t)
             @test convert(typeof(tc), t) == tc
             @test typeof(convert(typeof(tc), t)) == typeof(tc)
-            # @test typeof(convert(typeof(tc), t')) == typeof(tc) # TODO Adjoint not supported yet
+            @test typeof(convert(typeof(tc), t')) == typeof(tc)
             @test Base.promote_typeof(t, tc) == typeof(tc)
             @test Base.promote_typeof(tc, t) == typeof(tc + t)
         end