Commit de358ce5 authored by Martin Reinecke's avatar Martin Reinecke
Browse files

fix

parent 5abb59c6
Pipeline #63865 passed with stages
in 8 minutes and 47 seconds
......@@ -397,6 +397,8 @@ def from_global_data(arr, sum_up=False, distaxis=0):
raise TypeError
if sum_up:
arr = np_allreduce_sum(arr)
if arr.ndim == 0:
distaxis = -1
if distaxis == -1:
return data_object(arr.shape, arr, distaxis)
lo, hi = _shareRange(arr.shape[distaxis], ntask, rank)
......
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