决赛现场脚本

tools/tn_contest_runner.py

@@ -47,7 +47,7 @@ CASES = {
     "main1": CaseSpec(
         circuit_kind="rxx_rzz_chain",
         observables=("ring_xz",),
-        nqubits=34,
+        nqubits=37,
         nlayers=20,
         seed=31001,
         target_slices=None,
@@ -205,6 +205,8 @@ def build_parallel_opts(args, tree_file=None, search_only=False):
         opts["search_backend"] = args.tn_search_backend
     if args.dask_address is not None:
         opts["dask_address"] = args.dask_address
+    if args.dask_expected_workers is not None:
+        opts["dask_expected_workers"] = args.dask_expected_workers
     if args.dask_close_workers:
         opts["dask_close_workers"] = True
     if args.tn_debug_trials:
@@ -378,7 +380,7 @@ def main():
     parser.add_argument("--quimb-backend", choices=("numpy", "torch"), default="torch")
     parser.add_argument("--dtype", choices=("complex128", "complex64"), default="complex64")
     parser.add_argument("--tn-target-slices", type=int)
-    parser.add_argument("--tn-target-size", type=int, default=2**32)
+    parser.add_argument("--tn-target-size", type=int, default=2**34)
     parser.add_argument("--tn-search-workers", type=int)
     parser.add_argument("--tn-search-repeats", type=int, default=2048)
     parser.add_argument("--tn-search-time", type=float, default=300.0)
@@ -392,6 +394,7 @@ def main():
         ),
     )
     parser.add_argument("--dask-address")
+    parser.add_argument("--dask-expected-workers", type=int)
     parser.add_argument("--dask-close-workers", action="store_true")
     parser.add_argument(
         "--keep-dask",