From f59b1b0bc7346bd1aa4d6ee086c06c350f748238 Mon Sep 17 00:00:00 2001
From: tankya2 <tankya2@ihpc.a-star.edu.sg>
Date: Wed, 30 Aug 2023 17:29:08 +0800
Subject: [PATCH] Update return for all ranks

---
 src/qibotn/cutn.py | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/src/qibotn/cutn.py b/src/qibotn/cutn.py
index 2fc9079..9bc1d67 100644
--- 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__":