Update return for all ranks

2023-08-30 17:29:08 +08:00
parent d12b8ab882
commit f59b1b0bc7
1 changed files with 1 additions and 3 deletions
--- a/src/qibotn/cutn.py
+++ b/src/qibotn/cutn.py
@@ -15,7 +15,6 @@ def eval_tn_MPI(qibo_circ, datatype):
    ncpu_threads = multiprocessing.cpu_count() // 2
    n_samples = 8

-    root = 0
    comm = MPI.COMM_WORLD
    rank = comm.Get_rank()
    device_id = rank % getDeviceCount()
@@ -37,8 +36,7 @@ def eval_tn_MPI(qibo_circ, datatype):

    cutn.destroy(handle)

-    if rank == root:
-        return result, rank
+    return result, rank


 if __name__ == "__main__":