optional debug mode to scheduler

hel-astro-lab · Mar 20, 2024 · 3eee850 · 3eee850
1 parent 605138d
commit 3eee850
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/pytools/scheduler.py b/pytools/scheduler.py
@@ -28,6 +28,8 @@ def __init__(self):
         self.master_rank       = 0
         self.example_work_rank = 0
 
+        self.debug = False # debug mode
+
     # swithc from all-in mode to task mode
     def switch_to_task_mode(self,):
         self.mpi_task_mode = True
@@ -56,6 +58,10 @@ def is_active_tile(self, tile):
         return True
 
     def operate(self, op):
+
+        if self.debug: # additional debug printing
+            print('R:', self.rank, op) # debug print
+            MPI.COMM_WORLD.Barrier() 
 
         #--------------------------------------------------
         # default values