Forgot about cuTENSOR whoops

Author: kshyatt

Project.toml

@@ -79,3 +79,4 @@ test = ["Aqua", "Adapt", "CUDA", "cuTENSOR", "Pkg", "Test", "TestExtras", "Plots
 
 [sources]
 TensorKit = {url="https://github.com/QuantumKitHub/TensorKit.jl", rev="ksh/cuda_tweaks"}
+TensorOperations = {path="/home/kshyatt/.julia/dev/TensorOperations"}

test/cuda/operators.jl

@@ -4,7 +4,7 @@ using MPSKit
 using MPSKit: GeometryStyle, FiniteChainStyle, InfiniteChainStyle, OperatorStyle, MPOStyle
 using TensorKit
 using TensorKit: ℙ
-using Adapt, CUDA
+using Adapt, CUDA, cuTENSOR
 
 @testset "CuFiniteMPO" begin
     # start from random operators
@@ -34,51 +34,45 @@ using Adapt, CUDA
             @test Vright[i] == right_virtualspace(mpo₁, i)
         end
 
-        # needs tensoradd support for CuStridedView
-        #=@test convert(TensorMap, mpo₁) ≈ O₁
+        @test convert(TensorMap, mpo₁) ≈ O₁
         @test convert(TensorMap, -mpo₂) ≈ -O₂
-        @test convert(TensorMap, @constinferred complex(mpo₃)) ≈ complex(O₃)=#
+        @test convert(TensorMap, @constinferred complex(mpo₃)) ≈ complex(O₃)
 
         # test scalar multiplication
         α = rand(T)
-        # needs tensoradd support for CuStridedView
-        #@test convert(TensorMap, α * mpo₁) ≈ α * O₁
-        #@test convert(TensorMap, mpo₁ * α) ≈ O₁ * α
-        #@test α * mpo₃ ≈ α * complex(mpo₃) atol = 1.0e-6
+        @test convert(TensorMap, α * mpo₁) ≈ α * O₁
+        @test convert(TensorMap, mpo₁ * α) ≈ O₁ * α
+        @test α * mpo₃ ≈ α * complex(mpo₃) atol = 1.0e-6
 
         # test addition and multiplication
-        # needs tensoradd support for CuStridedView
-        #=@test convert(TensorMap, mpo₁ + mpo₂) ≈ O₁ + O₂
+        @test convert(TensorMap, mpo₁ + mpo₂) ≈ O₁ + O₂
         @test convert(TensorMap, mpo₁ + mpo₃) ≈ O₁ + O₃
         @test convert(TensorMap, mpo₁ * mpo₂) ≈ O₁ * O₂
-        @test convert(TensorMap, mpo₁ * mpo₃) ≈ O₁ * O₃=#
+        @test convert(TensorMap, mpo₁ * mpo₃) ≈ O₁ * O₃
 
         # test application to a state
         ψ₁ = adapt(CuArray, rand(T, domain(O₁)))
         ψ₂ = adapt(CuArray, rand(real(T), domain(O₂)))
-        #mps₁ = adapt(CuArray, FiniteMPS(ψ₁))
-        #mps₂ = adapt(CuArray, FiniteMPS(ψ₂))
+        mps₁ = adapt(CuArray, FiniteMPS(ψ₁))
+        mps₂ = adapt(CuArray, FiniteMPS(ψ₂))
 
-        #@test @constinferred GeometryStyle(mps₁, mpo₁, mps₁) == GeometryStyle(mps₁)
+        @test @constinferred GeometryStyle(mps₁, mpo₁, mps₁) == GeometryStyle(mps₁)
 
-        # needs tensoradd support for CuStridedView
-        #@test convert(TensorMap, mpo₁ * mps₁) ≈ O₁ * ψ₁
-        #@test mpo₁ * ψ₁ ≈ O₁ * ψ₁
-        #@test convert(TensorMap, mpo₃ * mps₁) ≈ O₃ * ψ₁
-        #@test mpo₃ * ψ₁ ≈ O₃ * ψ₁
-        #@test convert(TensorMap, mpo₁ * mps₂) ≈ O₁ * ψ₂
-        #@test mpo₁ * ψ₂ ≈ O₁ * ψ₂
+        @test convert(TensorMap, mpo₁ * mps₁) ≈ O₁ * ψ₁
+        @test mpo₁ * ψ₁ ≈ O₁ * ψ₁
+        @test convert(TensorMap, mpo₃ * mps₁) ≈ O₃ * ψ₁
+        @test mpo₃ * ψ₁ ≈ O₃ * ψ₁
+        @test convert(TensorMap, mpo₁ * mps₂) ≈ O₁ * ψ₂
+        @test mpo₁ * ψ₂ ≈ O₁ * ψ₂
 
-        # needs tensoradd support for CuStridedView
-        #@test dot(mps₁, mpo₁, mps₁) ≈ dot(ψ₁, O₁, ψ₁)
-        #@test dot(mps₁, mpo₁, mps₁) ≈ dot(mps₁, mpo₁ * mps₁)
+        @test dot(mps₁, mpo₁, mps₁) ≈ dot(ψ₁, O₁, ψ₁)
+        @test dot(mps₁, mpo₁, mps₁) ≈ dot(mps₁, mpo₁ * mps₁)
         # test conversion to and from mps
-        #mpomps₁ = convert(FiniteMPS, mpo₁)
-        #mpompsmpo₁ = convert(FiniteMPO, mpomps₁)
+        mpomps₁ = convert(FiniteMPS, mpo₁)
+        mpompsmpo₁ = convert(FiniteMPO, mpomps₁)
 
-        #@test convert(FiniteMPO, mpomps₁) ≈ mpo₁ rtol = 1.0e-6
+        @test convert(FiniteMPO, mpomps₁) ≈ mpo₁ rtol = 1.0e-6
 
-        # needs tensoradd support for CuStridedView
-        #@test dot(mpomps₁, mpomps₁) ≈ dot(mpo₁, mpo₁)
+        @test dot(mpomps₁, mpomps₁) ≈ dot(mpo₁, mpo₁)
     end
 end