diff --git a/python/sumk_dft_tools.py b/python/sumk_dft_tools.py index bee0054b..0e880559 100644 --- a/python/sumk_dft_tools.py +++ b/python/sumk_dft_tools.py @@ -377,6 +377,11 @@ class SumkDFTTools(SumkDFT): for sp in spn: Akw[sp][ish,ik,iom] = G_loc[sp].data[iom,ish,ish].imag/(-1.0*numpy.pi) + # Collect data from mpi + for sp in spn: + Akw[sp] = mpi.all_reduce(mpi.world, Akw[sp], lambda x,y : x+y) + mpi.barrier() + if save_to_file and mpi.is_master_node(): if ishell is None: for sp in spn: # loop over GF blocs: