Update for ITensors.ITensorMPS

ITensor · Mar 26, 2024 · a010bb3 · a010bb3
1 parent 3622d19
commit a010bb3
Show file tree

Hide file tree

Showing 5 changed files with 26 additions and 23 deletions.
diff --git a/src/alternating_update.jl b/src/alternating_update.jl
@@ -1,4 +1,5 @@
-using ITensors:
+using ITensors: permute
+using ITensors.ITensorMPS:
   AbstractObserver,
   MPO,
   MPS,
@@ -9,7 +10,6 @@ using ITensors:
   disk,
   linkind,
   maxlinkdim,
-  permute,
   siteinds
 
 function _compute_nsweeps(t; time_step=default_time_step(t), nsweeps=default_nsweeps())

diff --git a/src/projmpo_apply.jl b/src/projmpo_apply.jl
@@ -1,4 +1,5 @@
-using ITensors: ITensors, AbstractProjMPO, ITensor, MPO, MPS
+using ITensors: ITensors, ITensor
+using ITensors.ITensorMPS: ITensorMPS, AbstractProjMPO, MPO, MPS
 
 """
 A ProjMPOApply represents the application of an
@@ -35,12 +36,12 @@ function Base.copy(P::ProjMPOApply)
   return ProjMPOApply(P.lpos, P.rpos, P.nsite, copy(P.psi0), copy(P.H), copy(P.LR))
 end
 
-function ITensors.set_nsite!(P::ProjMPOApply, nsite)
+function ITensorMPS.set_nsite!(P::ProjMPOApply, nsite)
   P.nsite = nsite
   return P
 end
 
-function ITensors.makeL!(P::ProjMPOApply, psi::MPS, k::Int)
+function ITensorMPS.makeL!(P::ProjMPOApply, psi::MPS, k::Int)
   # Save the last `L` that is made to help with caching
   # for DiskProjMPO
   ll = P.lpos
@@ -64,7 +65,7 @@ function ITensors.makeL!(P::ProjMPOApply, psi::MPS, k::Int)
   return P
 end
 
-function ITensors.makeR!(P::ProjMPOApply, psi::MPS, k::Int)
+function ITensorMPS.makeR!(P::ProjMPOApply, psi::MPS, k::Int)
   # Save the last `R` that is made to help with caching
   # for DiskProjMPO
   rl = P.rpos

diff --git a/src/projmpo_mps2.jl b/src/projmpo_mps2.jl
@@ -1,4 +1,5 @@
-using ITensors: AbstractProjMPO, ProjMPO, contract, makeL!, makeR!, nsite, set_nsite!
+using ITensors: contract
+using ITensors.ITensorMPS: AbstractProjMPO, ProjMPO, makeL!, makeR!, nsite, set_nsite!
 
 mutable struct ProjMPO_MPS2 <: AbstractProjMPO
   PH::ProjMPO
@@ -15,25 +16,25 @@ end
 
 Base.copy(P::ProjMPO_MPS2) = ProjMPO_MPS2(copy(P.PH), copy(P.Ms))
 
-ITensors.nsite(P::ProjMPO_MPS2) = nsite(P.PH)
+ITensorMPS.nsite(P::ProjMPO_MPS2) = nsite(P.PH)
 
-function ITensors.set_nsite!(P::ProjMPO_MPS2, nsite)
+function ITensorMPS.set_nsite!(P::ProjMPO_MPS2, nsite)
   set_nsite!(P.PH, nsite)
   for m in P.Ms
     set_nsite!(m, nsite)
   end
   return P
 end
 
-function ITensors.makeL!(P::ProjMPO_MPS2, psi::MPS, k::Int)
+function ITensorMPS.makeL!(P::ProjMPO_MPS2, psi::MPS, k::Int)
   makeL!(P.PH, psi, k)
   for m in P.Ms
     makeL!(m, psi, k)
   end
   return P
 end
 
-function ITensors.makeR!(P::ProjMPO_MPS2, psi::MPS, k::Int)
+function ITensorMPS.makeR!(P::ProjMPO_MPS2, psi::MPS, k::Int)
   makeR!(P.PH, psi, k)
   for m in P.Ms
     makeR!(m, psi, k)

diff --git a/src/projmps2.jl b/src/projmps2.jl
@@ -1,5 +1,5 @@
-using ITensors:
-  AbstractProjMPO, ITensor, OneITensor, dag, dim, lproj, prime, rproj, site_range
+using ITensors: ITensor, dag, dim, prime
+using ITensors.ITensorMPS: ITensorMPS, AbstractProjMPO, OneITensor, lproj, rproj, site_range
 
 """
 Holds the following data where psi
@@ -29,12 +29,12 @@ function Base.copy(P::ProjMPS2)
   return ProjMPS2(P.lpos, P.rpos, P.nsite, copy(P.M), copy(P.LR))
 end
 
-function ITensors.set_nsite!(P::ProjMPS2, nsite)
+function ITensorMPS.set_nsite!(P::ProjMPS2, nsite)
   P.nsite = nsite
   return P
 end
 
-function ITensors.makeL!(P::ProjMPS2, psi::MPS, k::Int)
+function ITensorMPS.makeL!(P::ProjMPS2, psi::MPS, k::Int)
   # Save the last `L` that is made to help with caching
   # for DiskProjMPO
   ll = P.lpos
@@ -58,7 +58,7 @@ function ITensors.makeL!(P::ProjMPS2, psi::MPS, k::Int)
   return P
 end
 
-function ITensors.makeR!(P::ProjMPS2, psi::MPS, k::Int)
+function ITensorMPS.makeR!(P::ProjMPS2, psi::MPS, k::Int)
   # Save the last `R` that is made to help with caching
   # for DiskProjMPO
   rl = P.rpos
@@ -82,7 +82,7 @@ function ITensors.makeR!(P::ProjMPS2, psi::MPS, k::Int)
   return P
 end
 
-function ITensors.contract(P::ProjMPS2, v::ITensor)
+function ITensorMPS.contract(P::ProjMPS2, v::ITensor)
   itensor_map = Union{ITensor,OneITensor}[lproj(P)]
   append!(itensor_map, [prime(t, "Link") for t in P.M[site_range(P)]])
   push!(itensor_map, rproj(P))

diff --git a/src/sweep_update.jl b/src/sweep_update.jl
@@ -1,5 +1,6 @@
-using ITensors:
-  MPS, isortho, orthocenter, orthogonalize!, position!, replacebond!, set_nsite!, uniqueinds
+using ITensors: uniqueinds
+using ITensors.ITensorMPS:
+  ITensorMPS, MPS, isortho, orthocenter, orthogonalize!, position!, replacebond!, set_nsite!
 using LinearAlgebra: norm, normalize!, svd
 using Observers: update!
 using Printf: @printf
@@ -270,9 +271,9 @@ function region_update!(
     psi[b] = U
     phi0 = S * V
     if isforward(direction)
-      ITensors.setleftlim!(psi, b)
+      ITensorMPS.setleftlim!(psi, b)
     elseif isreverse(direction)
-      ITensors.setrightlim!(psi, b)
+      ITensorMPS.setrightlim!(psi, b)
     end
     set_nsite!(PH, nsite - 1)
     position!(PH, psi, b1)
@@ -281,9 +282,9 @@ function region_update!(
     normalize && (phi0 ./= norm(phi0))
     psi[b + Δ] = phi0 * psi[b + Δ]
     if isforward(direction)
-      ITensors.setrightlim!(psi, b + Δ + 1)
+      ITensorMPS.setrightlim!(psi, b + Δ + 1)
     elseif isreverse(direction)
-      ITensors.setleftlim!(psi, b + Δ - 1)
+      ITensorMPS.setleftlim!(psi, b + Δ - 1)
     end
     set_nsite!(PH, nsite)
   end