Updated MPI example

CQCL · Nov 16, 2023 · f1c9678 · f1c9678
1 parent 193aed9
commit f1c9678
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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: