完善脚本功能，添加计时估计

2026-05-08 10:20:03 +08:00
parent 8b71ff96c8
commit cec0ba272a
4 changed files with 21 additions and 7 deletions
--- a/benchmark_slice.py
+++ b/benchmark_slice.py
@@ -8,7 +8,7 @@ with open(f"data/tree_q{NQUBITS}_l{NLAYERS}.pkl", 'rb') as f:

 print(f"Original peak: {tree.max_size() * 16 / 1e9:.2f} GB")

-tree_sliced = tree.slice_and_reconfigure(target_size=2**30)  # 2^29 = 8 GB
+tree_sliced = tree.slice_and_reconfigure(target_size=2**28)

 with open(f"data/tree_q{NQUBITS}_l{NLAYERS}_sliced.pkl", 'wb') as f:
    pickle.dump(tree_sliced, f)