Update example tester

edan-bainglass · Feb 10, 2024 · 772f304 · 772f304
1 parent 6df84d5
commit 772f304
Showing 1 changed file with 29 additions and 2 deletions.
diff --git a/examples/coulomb_blockade/tester.ipynb b/examples/coulomb_blockade/tester.ipynb
@@ -235,9 +235,13 @@
     "                \"withmpi\": False,\n",
     "                \"resources\": {\n",
     "                    \"num_machines\": 1,\n",
-    "                    \"num_mpiprocs_per_machine\": 1,\n",
+    "                    \"num_mpiprocs_per_machine\": NPROCS,\n",
     "                },\n",
     "                \"max_wallclock_seconds\": h * m * s,\n",
+    "                \"environment_variables\": {\n",
+    "                    \"OMP_NUM_THREADS\": NPROCS,\n",
+    "                    \"NUMBA_NUM_THREADS\": NPROCS,\n",
+    "                },\n",
     "            }\n",
     "        },\n",
     "    },\n",
@@ -290,7 +294,30 @@
     "        \"code\": orm.load_code(\"trans-script\"),\n",
     "        \"metadata\": {\n",
     "            \"options\": {\n",
-    "                \"withmpi\": True,\n",
+    "                \"withmpi\": False,\n",
+    "                \"resources\": {\n",
+    "                    \"num_machines\": 1,\n",
+    "                    \"num_mpiprocs_per_machine\": NPROCS,\n",
+    "                },\n",
+    "                \"max_wallclock_seconds\": h * m * s,\n",
+    "                \"environment_variables\": {\n",
+    "                    \"OMP_NUM_THREADS\": NPROCS,\n",
+    "                    \"NUMBA_NUM_THREADS\": NPROCS,\n",
+    "                },\n",
+    "            }\n",
+    "        },\n",
+    "    },\n",
+    "    \"current\": {\n",
+    "        \"code\": orm.load_code(\"curr-script\"),\n",
+    "        # \"parameters\": {\n",
+    "        #     \"V_min\": -2.5,\n",
+    "        #     \"V_max\": 2.5,\n",
+    "        #     \"dV\": 0.1,\n",
+    "        #     \"temperature\": 9,\n",
+    "        # },\n",
+    "        \"metadata\": {\n",
+    "            \"options\": {\n",
+    "                \"withmpi\": False,\n",
     "                \"resources\": {\n",
     "                    \"num_machines\": 1,\n",
     "                    \"num_mpiprocs_per_machine\": NPROCS,\n",