Make SUWeight axis order definite (#315)

* Make SUWeight axis order definite

* Update related tests

* Update docs on CTMRGEnv(wts)

* Update SU and bond truncation

* Test SU with non-standard virtual dualness

* Undo metadata changes

* Update docstring

* Refactor SUWeight rotation

* Apply suggestions and add SUWeight random init

* Streamline CTMRGEnv(wts)

* Correct a comment

* Apply suggestions

* Improve `rand!(wts)`

Author: Yue-Zhengyuan

docs/src/examples/hubbard_su/index.md

@@ -130,7 +130,7 @@ is achieved by using `trunc=truncrank(χ)` with different `χ`s in the CTMRG run
 χenv₀, χenv = 6, 16
 env_space = Vect[fℤ₂](0 => χenv₀ / 2, 1 => χenv₀ / 2)
 normalize!.(peps.A, Inf)
-env = CTMRGEnv(wts, peps)
+env = CTMRGEnv(wts)
 for χ in [χenv₀, χenv]
     global env, = leading_boundary(
         env, peps; alg = :sequential, tol = 1.0e-8, maxiter = 50, trunc = truncrank(χ)

docs/src/examples/hubbard_su/main.ipynb

@@ -140,7 +140,7 @@
     "χenv₀, χenv = 6, 16\n",
     "env_space = Vect[fℤ₂](0 => χenv₀ / 2, 1 => χenv₀ / 2)\n",
     "normalize!.(peps.A, Inf)\n",
-    "env = CTMRGEnv(wts, peps)\n",
+    "env = CTMRGEnv(wts)\n",
     "for χ in [χenv₀, χenv]\n",
     "    global env, = leading_boundary(\n",
     "        env, peps; alg = :sequential, tol = 1.0e-8, maxiter = 50, trunc = truncrank(χ)\n",

examples/hubbard_su/main.jl

@@ -83,7 +83,7 @@ is achieved by using `trunc=truncrank(χ)` with different `χ`s in the CTMRG run
 χenv₀, χenv = 6, 16
 env_space = Vect[fℤ₂](0 => χenv₀ / 2, 1 => χenv₀ / 2)
 normalize!.(peps.A, Inf)
-env = CTMRGEnv(wts, peps)
+env = CTMRGEnv(wts)
 for χ in [χenv₀, χenv]
     global env, = leading_boundary(
         env, peps; alg = :sequential, tol = 1.0e-8, maxiter = 50, trunc = truncrank(χ)

src/PEPSKit.jl

@@ -1,6 +1,7 @@
 module PEPSKit
 
 using LinearAlgebra, Statistics, Base.Threads, Base.Iterators, Printf
+using Random
 using Compat
 using Accessors: @set, @reset
 using VectorInterface

src/algorithms/time_evolution/evoltools.jl

@@ -233,7 +233,7 @@ function _apply_gate(
 
     a, b = absorb_s(a, s, b)
     if need_flip
-        a, s, b = flip_svd(a, s, b)
+        a, s, b = flip(a, numind(a)), _fliptwist_s(s), flip(b, 1)
     end
     return a, s, b, ϵ
 end

src/algorithms/time_evolution/simpleupdate3site.jl

@@ -479,7 +479,7 @@ function _su3site_se!(
             Ms = [first(_unfuse_physicalspace(M, Vphy)) for (M, Vphy) in zip(Ms, Vphys)]
         end
         for (wt, wt_idx, flip) in zip(wts, wt_idxs, flips)
-            env[CartesianIndex(wt_idx)] = normalize(flip ? _flip_s(wt) : wt, Inf)
+            env[CartesianIndex(wt_idx)] = normalize(flip ? _fliptwist_s(wt) : wt, Inf)
         end
     end
     # restore virtual arrows in `Ms`

src/algorithms/truncation/bond_truncation.jl

@@ -139,7 +139,7 @@ function bond_truncate(
     a, s, b = svd_trunc!(permute(_combine_ab(a, b), perm_ab); trunc = alg.trunc)
     a, b = absorb_s(a, s, b)
     if need_flip
-        a, s, b = flip_svd(a, s, b)
+        a, s, b = flip(a, numind(a)), _fliptwist_s(s), flip(b, 1)
     end
     return a, s, b, (; fid, Δfid, Δs)
 end
@@ -186,7 +186,7 @@ function bond_truncate(
     @tensor a[-1 -2; -3] := Qa[-1 -2 3] * u[3 -3]
     @tensor b[-1; -2 -3] := vh[-1 1] * Qb[1 -2 -3]
     if need_flip
-        a, s, b = flip_svd(a, s, b)
+        a, s, b = flip(a, numind(a)), _fliptwist_s(s), flip(b, 1)
     end
     return a, s, b, info
 end