using isend instead of send for operation

modin-project · Mar 6, 2023 · 3d093cb · 3d093cb
1 parent 8abb067
commit 3d093cb
Show file tree

Hide file tree

Showing 5 changed files with 3 additions and 1,603 deletions.
diff --git a/unidist/core/backends/mpi/core/common.py b/unidist/core/backends/mpi/core/common.py
@@ -133,7 +133,7 @@ def base_data_id(self):
         return DataID(self._id)
 
 
-def get_logger(logger_name, file_name, activate=False):
+def get_logger(logger_name, file_name, activate=True):
     """
     Configure logger and get it's instance.
 

diff --git a/unidist/core/backends/mpi/core/controller/api.py b/unidist/core/backends/mpi/core/controller/api.py
@@ -237,6 +237,7 @@ def put(data):
         data_id = object_store.generate_data_id(garbage_collector)
         dest_rank = RoundRobin.get_instance().schedule_rank()
         object_store.put_data_owner(data_id, dest_rank)
+        object_store.put_data_size(data_id, data)
         push_data_directly_to_worker(dest_rank, data_id, data)
 
     logger.debug("PUT {} id".format(data_id._id))

diff --git a/unidist/core/backends/mpi/core/controller/common.py b/unidist/core/backends/mpi/core/controller/common.py
@@ -228,7 +228,7 @@ def push_data_directly_to_worker(dest_rank, data_id, value):
         "id": data_id,
         "data": value,
     }
-    communication.send_operation(
+    communication.isend_complex_operation(
         mpi_state.comm,
         operation_type,
         operation_data,