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

Update with multi node code

Browse Source
This commit is contained in:
tankya2
2023-08-30 17:25:04 +08:00
parent 000c4a1b8e
commit 1c9df26472
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
src/qibotn/cutn.py
View File

@@ -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()