From f1c9678d57d4225c2be36b1fb6dee1d5ee7ac9dd Mon Sep 17 00:00:00 2001 From: Pablo Andres-Martinez Date: Thu, 16 Nov 2023 03:42:12 -0800 Subject: [PATCH] Updated MPI example --- examples/mpi/mpi_overlap_bcast_mps.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/examples/mpi/mpi_overlap_bcast_mps.py b/examples/mpi/mpi_overlap_bcast_mps.py index a0d662d9..014a0e82 100644 --- a/examples/mpi/mpi_overlap_bcast_mps.py +++ b/examples/mpi/mpi_overlap_bcast_mps.py @@ -42,10 +42,10 @@ from pytket.circuit import Circuit, fresh_symbol -from pytket.extensions.cutensornet.mps import ( +from pytket.extensions.cutensornet.tnstate import ( simulate, - ConfigMPS, - ContractionAlg, + Config, + SimulationAlgorithm, CuTensorNetHandle, ) @@ -109,7 +109,7 @@ this_proc_mps = [] with CuTensorNetHandle(device_id) as libhandle: # Different handle for each process for circ in this_proc_circs: - mps = simulate(libhandle, circ, ContractionAlg.MPSxGate, ConfigMPS()) + mps = simulate(libhandle, circ, SimulationAlgorithm.MPSxGate, Config()) this_proc_mps.append(mps) if rank == root: