From 1c9df2647264f4419fcf31f3a785bf58523e11ed Mon Sep 17 00:00:00 2001 From: tankya2 Date: Wed, 30 Aug 2023 17:25:04 +0800 Subject: [PATCH] Update with multi node code --- src/qibotn/cutn.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/src/qibotn/cutn.py b/src/qibotn/cutn.py index 5790772..7d9984f 100644 --- a/src/qibotn/cutn.py +++ b/src/qibotn/cutn.py @@ -19,7 +19,6 @@ def eval_tn_MPI(qibo_circ, datatype): root = 0 comm = MPI.COMM_WORLD rank = comm.Get_rank() - device_id = rank % getDeviceCount() cp.cuda.Device(device_id).use() @@ -31,7 +30,7 @@ def eval_tn_MPI(qibo_circ, datatype): operands_interleave = myconvertor.state_vector_operands() network = cutn.Network(*operands_interleave, options=network_opts) - network.contract_path(optimize={'samples': n_samples, 'threads': ncpu_threads}) # Calculate path, info + network.contract_path(optimize={'samples': n_samples, 'threads': ncpu_threads}) # Calculate optimal path, returns path and info result = network.contract()