Fix more in-place operations

QuantumKitHub · Jan 8, 2025 · 9f38380 · 9f38380
1 parent 52d8d28
commit 9f38380
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/environments/ctmrg_environments.jl b/src/environments/ctmrg_environments.jl
@@ -556,14 +556,14 @@ function LinearAlgebra.mul!(edst::CTMRGEnv, esrc::CTMRGEnv, α::Number)
 end
 
 function LinearAlgebra.rmul!(e::CTMRGEnv, α::Number)
-    rmul!(e.corners, α)
-    rmul!(e.edges, α)
+    rmul!.(e.corners, α)
+    rmul!.(e.edges, α)
     return e
 end
 
 function LinearAlgebra.axpy!(α::Number, e₁::CTMRGEnv, e₂::CTMRGEnv)
-    e₂.corners .+= α * e₁.corners
-    e₂.edges .+= α * e₁.edges
+    axpy!.(α, e₁.corners, e₂.corners)
+    axpy!.(α, e₁.edges, e₂.edges)
     return e₂
 end
 

diff --git a/src/states/infinitepeps.jl b/src/states/infinitepeps.jl
@@ -155,13 +155,13 @@ end
 
 # Used in _scale during OptimKit.optimize
 function LinearAlgebra.rmul!(ψ::InfinitePEPS, α::Number)
-    rmul!(ψ.A, α)
+    rmul!.(ψ.A, α)
     return ψ
 end
 
 # Used in _add during OptimKit.optimize
 function LinearAlgebra.axpy!(α::Number, ψ₁::InfinitePEPS, ψ₂::InfinitePEPS)
-    ψ₂.A .+= α * ψ₁.A
+    axpy!.(α, ψ₁.A, ψ₂.A)
     return ψ₂
 end