Commit 005fbbb8 authored by theos's avatar theos
Browse files

Fixed a bug in slicing_distributor.consolidate_data.

parent 08631989
...@@ -1578,6 +1578,7 @@ class _slicing_distributor(distributor): ...@@ -1578,6 +1578,7 @@ class _slicing_distributor(distributor):
_gathered_data = np.empty(self.global_shape, dtype=self.dtype) _gathered_data = np.empty(self.global_shape, dtype=self.dtype)
_dim_list = self.all_local_slices[:, 3] _dim_list = self.all_local_slices[:, 3]
_dim_offset_list = self.all_local_slices[:, 4] _dim_offset_list = self.all_local_slices[:, 4]
data = np.ascontiguousarray(data)
if target_rank == 'all': if target_rank == 'all':
comm.Allgatherv([data, self.mpi_dtype], comm.Allgatherv([data, self.mpi_dtype],
[_gathered_data, _dim_list, _dim_offset_list, [_gathered_data, _dim_list, _dim_offset_list,
......
...@@ -20,4 +20,4 @@ ...@@ -20,4 +20,4 @@
# 1) we don't load dependencies by storing it in __init__.py # 1) we don't load dependencies by storing it in __init__.py
# 2) we can import it in setup.py for the same reason # 2) we can import it in setup.py for the same reason
# 3) we can import it into your module module # 3) we can import it into your module module
__version__ = '1.0.1' __version__ = '1.0.2'
\ No newline at end of file \ No newline at end of file
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