Update mpi4py_ex.md

WonyoungCho · Jun 18, 2019 · 2cb2f07 · 2cb2f07
1 parent a2e2ec6
commit 2cb2f07
Showing 1 changed file with 8 additions and 5 deletions.
diff --git a/docs/mpi4py_ex.md b/docs/mpi4py_ex.md
@@ -3,15 +3,18 @@
 ```
 from mpi4py import MPI
 
-print 'Hello World!'
+comm = MPI.COMM_WORLD
+rank = comm.Get_rank()
+
+print rank, 'Hello World!'
 ```
 
 ```sh
 $ mpiexec -n 4 python hello.py
- Hello World!
- Hello World!
- Hello World!
- Hello World!
+0 Hello World!
+1 Hello World!
+2 Hello World!
+3 Hello World!
 ```
 
 # Send & Recv