赛前稳定版

tools/baseline_mps_expectation.py

@@ -19,6 +19,22 @@ from qibotn.backends.qmatchatea import QMatchaTeaBackend
 from qibotn.backends.vidal_tebd import run_vidal_ring_xz
 
 
+def optional_int(text):
+    if isinstance(text, str) and text.lower() in {"none", "null", "inf", "unlimited"}:
+        return None
+    return int(text)
+
+
+def optional_float(text):
+    if isinstance(text, str) and text.lower() in {"none", "null", "inf", "unlimited"}:
+        return None
+    return float(text)
+
+
+def format_optional(value, fmt="g"):
+    return "None" if value is None else format(value, fmt)
+
+
 def build_circuit(nqubits, nlayers, seed):
     return build_benchmark_circuit("brickwall_cnot", nqubits, nlayers, seed)
 
@@ -35,7 +51,8 @@ def main():
     parser = argparse.ArgumentParser()
     parser.add_argument("--nqubits", type=int, default=40)
     parser.add_argument("--nlayers", type=int, default=30)
-    parser.add_argument("--bond", "--bonds", dest="bond", type=int, default=512)
+    parser.add_argument("--bond", "--bonds", dest="bond", type=optional_int, default=512)
+    parser.add_argument("--cut-ratio", type=optional_float, default=1e-12)
     parser.add_argument("--seed", type=int, default=42)
     parser.add_argument("--tensor-module", choices=("numpy", "torch"), default="torch")
     parser.add_argument("--torch-threads", type=int, default=32)
@@ -109,7 +126,8 @@ def main():
             mpi_label = f"MPIMPS/{size}" if args.mpi_ct else "SR"
         print(
             f"nqubits={args.nqubits} nlayers={args.nlayers} "
-            f"bond={args.bond} seed={args.seed} "
+            f"bond={format_optional(args.bond)} "
+            f"cut_ratio={format_optional(args.cut_ratio)} seed={args.seed} "
             f"tensor_module={args.tensor_module} svd_control=E! "
             f"compile_circuit=True mpi={mpi_label} executor={args.executor}"
         )
@@ -128,7 +146,7 @@ def main():
             value, timings = run_segment_vidal_mpi_ring_xz(
                 circuit,
                 max_bond=args.bond,
-                cut_ratio=1e-12,
+                cut_ratio=args.cut_ratio,
                 tensor_module=args.tensor_module,
                 comm=MPI.COMM_WORLD,
             )
@@ -136,7 +154,7 @@ def main():
             value = run_vidal_ring_xz(
                 circuit,
                 max_bond=args.bond,
-                cut_ratio=1e-12,
+                cut_ratio=args.cut_ratio,
                 tensor_module=args.tensor_module,
             )
     else:
@@ -144,7 +162,7 @@ def main():
         backend.configure_tn_simulation(
             ansatz="MPS",
             max_bond_dimension=args.bond,
-            cut_ratio=1e-12,
+            cut_ratio=args.cut_ratio,
             svd_control="E!",
             tensor_module=args.tensor_module,
             compile_circuit=True,