Skip to content

GitLab

Commit 2d56144e authored by Martin Reinecke's avatar Martin Reinecke
Browse files

another fix

parent 57511c77
Pipeline #44094 passed with stages
in 9 minutes and 11 seconds
Hide whitespace changes
Inline Side-by-side
nifty5/minimization/metric_gaussian_kl_mpi.py
View file @ 2d56144e
...@@ -40,6 +40,7 @@ def _shareRange(nwork, nshares, myshare): ...@@ -40,6 +40,7 @@ def _shareRange(nwork, nshares, myshare):
def np_allreduce_sum(arr): def np_allreduce_sum(arr):
arr = np.array(arr)
res = np.empty_like(arr) res = np.empty_like(arr)
_comm.Allreduce(arr, res, MPI.SUM) _comm.Allreduce(arr, res, MPI.SUM)
return res return res
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment