Further updates for GPU tests

Author: kshyatt

ext/TensorKitCUDAExt/TensorKitCUDAExt.jl

@@ -1,10 +1,7 @@
 module TensorKitCUDAExt
 
-using LinearAlgebra
-
-using CUDA
+using CUDA, CUDA.CUBLAS, LinearAlgebra
 using CUDA: @allowscalar
-using CUDA.CUBLAS # for LinearAlgebra tie-ins
 using cuTENSOR: cuTENSOR
 
 using TensorKit
@@ -81,4 +78,8 @@ function TensorKit.Factorizations.initialize_output(::typeof(eig_vals!), t::CuTe
     Tc = complex(scalartype(t))
     return D = CuDiagonalTensorMap{Tc}(undef, V_D)
 end
+
+
+# TODO
+# add VectorInterface extensions for proper CUDA promotion
 end

ext/TensorKitCUDAExt/cutensormap.jl

@@ -160,6 +160,8 @@ function TensorKit.scalar(t::CuTensorMap)
            first(blocks(t))[2][1, 1] : throw(DimensionMismatch())
 end
 
+TensorKit.scalartype(A::CuArray{T}) where {T} = T
+
 function TensorKit.similarstoragetype(TT::Type{<:CuTensorMap}, ::Type{T}) where {T}
     return CuVector{T}
 end

test/cuda.jl

@@ -445,7 +445,7 @@ for V in spacelist
                     d = LinearAlgebra.eigvals(t; sortby=nothing)
                     d′ = LinearAlgebra.diag(D)
                     for (c, b) in d
-                         @test b ≈ d′[c]
+                        @test b ≈ d′[c]
                     end
 
                     # Somehow moving these test before the previous one gives rise to errors