Accelerated Python/60 mpi4py: Fix MPIRUN and clean up notebook a little.

Author: brycelelbach

tutorials/accelerated-python/notebooks/distributed/60__mpi4py.ipynb

@@ -9,14 +9,12 @@
    },
    "outputs": [],
    "source": [
-    "## Setup on \n",
-    "MPIRUN=\"srun\"\n",
+    " MPIRUN=\"mpirun\"\n",
     "\n",
     "import os\n",
     "\n",
     "if os.getenv(\"COLAB_RELEASE_TAG\"): # If running in Google Colab\n",
-    "    !pip install mpi4py\n",
-    "    MPIRUN=\"mpirun --allow-run-as-root --oversubscribe\"\n"
+    "    !pip install mpi4py"
    ]
   },
   {
@@ -56,7 +54,7 @@
    },
    "outputs": [],
    "source": [
-    "!$MPIRUN  -n 4 -c 4 python hello_world.py"
+    "!$MPIRUN -n 4 -c 4 python hello_world.py"
    ]
   },
   {
@@ -77,7 +75,7 @@
     "name = MPI.Get_processor_name()\n",
     "pid = os.getpid()\n",
     "print(f\"Hello World! I am process {rank} of {size} on {name} \"\n",
-    "       \"with pid {pid}.\\n\")\n"
+    "       \"with pid {pid}.\\n\")"
    ]
   },
   {
@@ -110,7 +108,9 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "%%writefile fizzbuzz.py\n"
+    "%%writefile fizzbuzz.py\n",
+    "\n",
+    "# TODO"
    ]
   },
   {
@@ -144,7 +144,7 @@
     "    comm.send(data, dest=1, tag=11)\n",
     "elif rank == 1:\n",
     "    data = comm.recv(source=0, tag=11)\n",
-    "    print(data)\n"
+    "    print(data)"
    ]
   },
   {
@@ -169,6 +169,7 @@
    "outputs": [],
    "source": [
     "%%writefile send_recv_eff.py\n",
+    "\n",
     "import numpy as np\n",
     "from mpi4py import MPI\n",
     "comm = MPI.COMM_WORLD\n",
@@ -276,6 +277,7 @@
    "outputs": [],
    "source": [
     "%%writefile scatter.py\n",
+    "\n",
     "from mpi4py import MPI\n",
     "comm = MPI.COMM_WORLD\n",
     "rank = comm.Get_rank()\n",
@@ -314,6 +316,7 @@
    "outputs": [],
    "source": [
     "%%writefile gather.py\n",
+    "\n",
     "from mpi4py import MPI\n",
     "comm = MPI.COMM_WORLD\n",
     "rank = comm.Get_rank()\n",
@@ -355,6 +358,7 @@
    "outputs": [],
    "source": [
     "%%writefile reduce.py\n",
+    "\n",
     "from mpi4py import MPI\n",
     "comm = MPI.COMM_WORLD\n",
     "rank = comm.Get_rank()\n",
@@ -401,6 +405,7 @@
    "outputs": [],
    "source": [
     "%%writefile compute_pi.py\n",
+    "\n",
     "from mpi4py import MPI\n",
     "import math\n",
     "\n",
@@ -465,6 +470,7 @@
    "outputs": [],
    "source": [
     "%%writefile compute_mandelbot.py\n",
+    "\n",
     "import matplotlib.pyplot as plt\n",
     "import matplotlib.cm as cm\n",
     "\n",
@@ -554,16 +560,6 @@
     "from IPython.display import Image\n",
     "Image(filename='test.png')"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "id": "6db131ed-12dd-4cfb-9101-5f8cf00a8eb3",
-   "metadata": {
-    "id": "6db131ed-12dd-4cfb-9101-5f8cf00a8eb3"
-   },
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {