debug log

unidist/core/backends/mpi/core/controller/api.py

@@ -380,6 +380,7 @@ def submit(task, *args, num_returns=1, **kwargs):
 
     # dest_rank = RoundRobin.get_instance().schedule_rank()
     dest_rank = choose_destination_rank(collected_data_ids)
+    print(dest_rank, collected_data_ids)
     push_data_owners(dest_rank, collected_data_ids)
     output_ids = object_store.generate_output_data_id(
         dest_rank, garbage_collector, num_returns

unidist/core/backends/mpi/core/worker/loop.py

@@ -4,6 +4,7 @@
 
 """Worker MPI process task processing functionality."""
 import asyncio
+import sys
 from functools import wraps, partial
 
 try:
@@ -29,7 +30,7 @@
 # When building documentation we do not have MPI initialized so
 # we use the condition to set "worker_0.log" in order to build it succesfully.
 log_file = "worker_{}.log".format(mpi_state.rank if mpi_state is not None else 0)
-w_logger = common.get_logger("worker", log_file)
+w_logger = common.get_logger("worker", sys.stdout, True)
 
 # Actors map {handle : actor}
 actor_map = {}