mpi完善

benchmark_contract_sliced.py

@@ -24,9 +24,12 @@ backend.configure_tn_simulation(ansatz="tn")
 qc = backend._qibo_circuit_to_quimb(circuit, backend.circuit_ansatz)
 tn = qc.local_expectation(qu.pauli('x') & qu.pauli('z'), (0, 1), rehearse='tn')
 
-with open(f"data/tree_q{NQUBITS}_l{NLAYERS}_sliced.pkl", 'rb') as f:
-#with open(f"data/tree_q{NQUBITS}_l{NLAYERS}.pkl", 'rb') as f:
-    tree = pickle.load(f)
+if rank == 0:
+    with open(f"data/tree_q{NQUBITS}_l{NLAYERS}_sliced.pkl", 'rb') as f:
+        tree = pickle.load(f)
+else:
+    tree = None
+tree = comm.bcast(tree, root=0)
 
 arrays = [torch.from_numpy(np.ascontiguousarray(t._data, dtype=np.complex128)) for t in tn.tensors]
 n_slices = tree.multiplicity

hostfile

@@ -0,0 +1,2 @@
+10.20.6.74:1
+10.20.6.102:1