Skip to content

Commit 3ad5f04

Browse files
committed
More dumb fixes
1 parent 2c66082 commit 3ad5f04

1 file changed

Lines changed: 11 additions & 7 deletions

File tree

test/cuda/operators.jl

Lines changed: 11 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,10 @@ MPSKit.Defaults.alg_svd() = CUSOLVER_QRIteration()
1717
O₁ = rand(T, V^L, V^L)
1818
O₂ = rand(T, space(O₁))
1919
O₃ = rand(real(T), space(O₁))
20+
21+
dO₁ = adapt(CuArray, O₁)
22+
dO₂ = adapt(CuArray, O₂)
23+
dO₃ = adapt(CuArray, O₃)
2024

2125
mpo₁ = adapt(CuVector{T, CUDA.DeviceMemory}, FiniteMPO(O₁))
2226
mpo₂ = adapt(CuVector{T, CUDA.DeviceMemory}, FiniteMPO(O₂))
@@ -65,14 +69,14 @@ MPSKit.Defaults.alg_svd() = CUSOLVER_QRIteration()
6569

6670
@test @constinferred GeometryStyle(mps₁, mpo₁, mps₁) == GeometryStyle(mps₁)
6771

68-
#@test convert(TM, mpo₁ * mps₁) ≈ O₁ * ψ₁
69-
@test mpo₁ * ψ₁ O* ψ₁
70-
#@test convert(TM, mpo₃ * mps₁) ≈ O₃ * ψ₁
71-
@test mpo₃ * ψ₁ O* ψ₁
72-
#@test convert(TM, mpo₁ * mps₂) ≈ O₁ * ψ₂
73-
#@test mpo₁ * ψ₂ ≈ O₁ * ψ₂
72+
#@test convert(TM, mpo₁ * mps₁) ≈ dO₁ * ψ₁
73+
@test mpo₁ * ψ₁ dO* ψ₁
74+
#@test convert(TM, mpo₃ * mps₁) ≈ dO₃ * ψ₁
75+
@test mpo₃ * ψ₁ dO* ψ₁
76+
#@test convert(TM, mpo₁ * mps₂) ≈ dO₁ * ψ₂
77+
#@test mpo₁ * ψ₂ ≈ dO₁ * ψ₂
7478

75-
@test dot(mps₁, mpo₁, mps₁) dot(ψ₁, O₁, ψ₁)
79+
@test dot(mps₁, mpo₁, mps₁) dot(ψ₁, dO₁, ψ₁)
7680
@test dot(mps₁, mpo₁, mps₁) dot(mps₁, mpo₁ * mps₁)
7781
# test conversion to and from mps
7882
mpomps₁ = convert(FiniteMPS, mpo₁)

0 commit comments

Comments
 (0)