Bug fixes

varipeps/expectation/three_sites.py

@@ -399,7 +399,7 @@ def calc_three_sites_triangle_without_bottom_right_multiple_gates(
     The order of the PEPS sequence have to be
     [top-left, top-right, bottom-left, bottom-left].
 
-    The gate is applied in the order [top-left, top-right, bottom-right].
+    The gate is applied in the order [top-left, top-right, bottom-left].
 
     Args:
       peps_tensors (:term:`sequence` of :obj:`jax.numpy.ndarray`):
@@ -430,7 +430,7 @@ def calc_three_sites_triangle_without_bottom_right_multiple_gates(
     )
 
     density_matrix_bottom_left = apply_contraction(
-        "density_matrix_four_sites_left",
+        "density_matrix_four_sites_bottom_left",
         [peps_tensors[2]],
         [peps_tensor_objs[2]],
         [],

varipeps/optimization/optimizer.py

@@ -463,6 +463,9 @@ def optimize_peps_network(
 
     As convergence criterion the norm of the gradient is used.
 
+    If the first CTMRG calculation does not converge. OptimizeResult is returned with Success=False. 
+    This should be handled by the script calling this function.
+
     Args:
       input_tensors (:obj:`~varipeps.peps.PEPS_Unit_Cell` or :term:`sequence` of :obj:`jax.numpy.ndarray`):
         The PEPS unitcell to work on or the tensors which should be mapped by

varipeps/utils/svd.py

@@ -336,17 +336,28 @@ def gauge_fixed_svd(
       :obj:`tuple`\\ (:obj:`jnp.ndarray`, :obj:`jnp.ndarray`, :obj:`jnp.ndarray`):
         Tuple with sign-fixed U, S and Vh of the SVD.
     """
-    if only_u_or_vh is None:
+    if any(d.platform == "gpu" for d in jax.devices()):
         U, S, Vh = svd_wrapper(matrix, use_qr=use_qr)
-        gauge_unitary = U
-    elif only_u_or_vh == "U":
-        U, S = svd_only_u(matrix, use_qr=use_qr)
-        gauge_unitary = U
-    elif only_u_or_vh == "Vh":
-        S, Vh = svd_only_vt(matrix, use_qr=use_qr)
-        gauge_unitary = Vh.T.conj()
+        if only_u_or_vh is None:
+            gauge_unitary = U
+        elif only_u_or_vh == "U":
+            gauge_unitary = U
+        elif only_u_or_vh == "Vh":
+            gauge_unitary = Vh.T.conj()
+        else:
+            raise ValueError("Invalid value for parameter 'only_u_or_vh'.")
     else:
-        raise ValueError("Invalid value for parameter 'only_u_or_vh'.")
+        if only_u_or_vh is None:
+            U, S, Vh = svd_wrapper(matrix, use_qr=use_qr)
+            gauge_unitary = U
+        elif only_u_or_vh == "U":
+            U, S = svd_only_u(matrix, use_qr=use_qr)
+            gauge_unitary = U
+        elif only_u_or_vh == "Vh":
+            S, Vh = svd_only_vt(matrix, use_qr=use_qr)
+            gauge_unitary = Vh.T.conj()
+        else:
+            raise ValueError("Invalid value for parameter 'only_u_or_vh'.")
 
     # Fix the gauge of the SVD
     abs_gauge_unitary = jnp.abs(gauge_unitary)