fix mpi load proj mps

block-hczhai · Aug 9, 2022 · ca04e60 · ca04e60
1 parent 5b4a96d
commit ca04e60
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pyblock2/driver/block2main b/pyblock2/driver/block2main
@@ -1819,6 +1819,8 @@ def get_mps_from_tags(iroot, proj_mps=False, ref_center=0):
     if smps_info.bond_dim < max_bdim:
         smps_info.bond_dim = max_bdim
     smps.load_data()
+    if MPI is not None:
+        MPI.barrier()
     if smps.dot == 2 and smps.center == smps.n_sites - 1 and dot == 1:
         smps.dot = 1
     elif smps.dot == 2 and smps.center == 0 and dot == 1: