mirror of
https://github.com/triqs/dft_tools
synced 2024-11-06 22:23:52 +01:00
[PATCH] fixed a missing mpi.reduce_all in spaghetti function from sumk_dft_tools
This commit is contained in:
parent
cff294e1b2
commit
b3e62640da
@ -377,6 +377,11 @@ class SumkDFTTools(SumkDFT):
|
|||||||
for sp in spn:
|
for sp in spn:
|
||||||
Akw[sp][ish,ik,iom] = G_loc[sp].data[iom,ish,ish].imag/(-1.0*numpy.pi)
|
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 save_to_file and mpi.is_master_node():
|
||||||
if ishell is None:
|
if ishell is None:
|
||||||
for sp in spn: # loop over GF blocs:
|
for sp in spn: # loop over GF blocs:
|
||||||
|
Loading…
Reference in New Issue
Block a user