jaunatisblue/qibotn
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

mpi完善
Some checks failed
Build wheels / build (ubuntu-latest, 3.11) (push) Has been cancelled
Details
Build wheels / build (ubuntu-latest, 3.12) (push) Has been cancelled
Details
Build wheels / build (ubuntu-latest, 3.13) (push) Has been cancelled
Details
Tests / check (push) Has been cancelled
Details
Tests / build (ubuntu-latest, 3.11) (push) Has been cancelled
Details
Tests / build (ubuntu-latest, 3.12) (push) Has been cancelled
Details
Tests / build (ubuntu-latest, 3.13) (push) Has been cancelled
Details

Browse Source
This commit is contained in:
jaunatisblue
2026-05-07 23:37:15 +08:00
parent 0a96553bd8
commit c818ac7a6e
2 changed files with 8 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
benchmark_contract_sliced.py
View File

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

2
hostfile Normal file
View File

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