Flush needed for "nice" printing

Author: jussienko

demos/mpi-alltoall.py

@@ -1,5 +1,6 @@
 from mpi4py import MPI
 import numpy as np
+import sys
 
 comm = MPI.COMM_WORLD
 rank = comm.Get_rank()
@@ -17,6 +18,7 @@
 for r in range(size):
     if rank == r:
         print("rank ", rank, data)
+        sys.stdout.flush()
     comm.Barrier()
 
 comm.Alltoall(data, recv_buf)
@@ -25,11 +27,13 @@
 if rank == 0:
     print()
     print("Final data")
+    sys.stdout.flush()
 comm.Barrier()
 
 for r in range(size):
     if rank == r:
         print("rank ", rank, recv_buf)
+        sys.stdout.flush()
     comm.Barrier()
 
 

demos/mpi-bcast.py

@@ -1,5 +1,6 @@
 from mpi4py import MPI
 import numpy as np
+import sys
 
 comm = MPI.COMM_WORLD
 rank = comm.Get_rank()
@@ -14,16 +15,19 @@
 
 if rank == 0:
     print("Original data")
+    sys.stdout.flush()
 comm.Barrier()
 
 print("rank ", rank, data)
+sys.stdout.flush()
 
 comm.Bcast(data, root=0)
 
 comm.Barrier()
 if rank == 0:
     print()
     print("Final data")
+    sys.stdout.flush()
 comm.Barrier()
 
 print("rank ", rank, data)

demos/mpi-gather.py

@@ -1,5 +1,6 @@
 from mpi4py import MPI
 import numpy as np
+import sys
 
 comm = MPI.COMM_WORLD
 rank = comm.Get_rank()
@@ -16,16 +17,19 @@
 
 if rank == 0:
     print("Original data")
+    sys.stdout.flush()
 comm.Barrier()
 
 print("rank ", rank, data)
+sys.stdout.flush()
 
 comm.Gather(data, recv_buf, root=0)
 
 comm.Barrier()
 if rank == 0:
     print()
     print("Final data")
+    sys.stdout.flush()
 comm.Barrier()
 
 print("rank ", rank, recv_buf)

demos/mpi-scatter.py

@@ -1,5 +1,6 @@
 from mpi4py import MPI
 import numpy as np
+import sys
 
 comm = MPI.COMM_WORLD
 rank = comm.Get_rank()
@@ -16,16 +17,19 @@
 
 if rank == 0:
     print("Original data")
+    sys.stdout.flush()
 comm.Barrier()
 
 print("rank ", rank, data)
+sys.stdout.flush()
 
 comm.Scatter(data, recv_buf, root=0)
 
 comm.Barrier()
 if rank == 0:
     print()
     print("Final data")
+    sys.stdout.flush()
 comm.Barrier()
 
 print("rank ", rank, recv_buf)