r7849 - /branches/multi_processor_merge/multi/mpi4py_processor.py -- October 19, 2008 - 21:54

Author: bugman
Date: Sun Oct 19 21:54:52 2008
New Revision: 7849

URL: http://svn.gna.org/viewcvs/relax?rev=7849&view=rev
Log:
Some more fixes for MPI.COMM_WORLD.size and MPI.COMM_WORLD.rank.


Modified:
    branches/multi_processor_merge/multi/mpi4py_processor.py

Modified: branches/multi_processor_merge/multi/mpi4py_processor.py
URL: 
http://svn.gna.org/viewcvs/relax/branches/multi_processor_merge/multi/mpi4py_processor.py?rev=7849&r1=7848&r2=7849&view=diff
==============================================================================
--- branches/multi_processor_merge/multi/mpi4py_processor.py (original)
+++ branches/multi_processor_merge/multi/mpi4py_processor.py Sun Oct 19 
21:54:52 2008
@@ -59,13 +59,13 @@
 
 
 def broadcast_command(command):
-    for i in range(1, MPI.size):
+    for i in range(1, MPI.COMM_WORLD.size):
         if i != 0:
             MPI.COMM_WORLD.Send(buf=command, dest=i)
 
 
 def ditch_all_results():
-    for i in range(1, MPI.size):
+    for i in range(1, MPI.COMM_WORLD.size):
         if i != 0:
             while 1:
                 result = MPI.COMM_WORLD.Recv(source=i)
@@ -76,7 +76,7 @@
 # wrapper sys.exit function
 # CHECKME is status ok
 def exit(status=None):
-    if MPI.rank != 0:
+    if MPI.COMM_WORLD.rank != 0:
         if in_main_loop:
             raise Exception('sys.exit unexpectedley called on slave!')
         else:
@@ -96,7 +96,7 @@
 
 
 def exit_mpi():
-    if MPI.Is_initialized() and not MPI.Is_finalized() and MPI.rank == 0:
+    if MPI.Is_initialized() and not MPI.Is_finalized() and 
MPI.COMM_WORLD.rank == 0:
         broadcast_command(Exit_command())
         ditch_all_results()