完善时间剪枝功能

benchmark_contract_sliced.py

@@ -3,19 +3,19 @@ import time, pickle, os
 import numpy as np
 from mpi4py import MPI
 
-NQUBITS, NLAYERS, NCORES = 25, 10, 96
+NQUBITS, NLAYERS, NCORES = 25, 10, 48
 
 comm = MPI.COMM_WORLD
 rank, size = comm.Get_rank(), comm.Get_size()
 
-os.environ['OMP_NUM_THREADS'] = str(max(1, NCORES // size))
-os.environ['MKL_NUM_THREADS'] = str(max(1, NCORES // size))
+os.environ['OMP_NUM_THREADS'] = str(NCORES)
+os.environ['MKL_NUM_THREADS'] = str(NCORES)
 
 import torch
 import qibo, quimb as qu
 from qibotn.observables import build_random_circuit
 
-torch.set_num_threads(max(1, NCORES // size))
+torch.set_num_threads(NCORES)
 
 circuit = build_random_circuit(NQUBITS, NLAYERS)
 qibo.set_backend("qibotn", platform="quimb")