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

Update return for all ranks

Browse Source
This commit is contained in:
tankya2
2023-08-30 17:29:08 +08:00
parent d12b8ab882
commit f59b1b0bc7
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/qibotn/cutn.py
View File

@@ -15,7 +15,6 @@ def eval_tn_MPI(qibo_circ, datatype):
ncpu_threads = multiprocessing.cpu_count() // 2 ncpu_threads = multiprocessing.cpu_count() // 2
n_samples = 8 n_samples = 8
root = 0
comm = MPI.COMM_WORLD comm = MPI.COMM_WORLD
rank = comm.Get_rank() rank = comm.Get_rank()
device_id = rank % getDeviceCount() device_id = rank % getDeviceCount()
@@ -37,8 +36,7 @@ def eval_tn_MPI(qibo_circ, datatype):
cutn.destroy(handle) cutn.destroy(handle)
if rank == root: return result, rank
return result, rank
if __name__ == "__main__": if __name__ == "__main__":