switching to cpu for CI pipelines (#62)

CI pipelines won't work with "ti.gpu". Hence, changing to "ti.cpu" (conditionally in case the code is run by the automated CI pipeline).
PolyPhyHub · Oct 6, 2023 · ddeb4a8 · ddeb4a8
1 parent 9779a6b
commit ddeb4a8
Show file tree

Hide file tree

Showing 6 changed files with 5 additions and 20 deletions.
diff --git a/experiments/jupyter/production/PolyPhy_2D_continuous_data.ipynb b/experiments/jupyter/production/PolyPhy_2D_continuous_data.ipynb
@@ -294,7 +294,7 @@
    ],
    "source": [
     "## Initialize Taichi\n",
-    "ti.init(arch=ti.cpu)\n",
+    "ti.init(arch=ti.cpu if os.path.exists(\"/tmp/flag\") else ti.gpu)\n",
     "rng = default_rng()\n",
     "\n",
     "## Initialize data and agents\n",

diff --git a/experiments/jupyter/production/PolyPhy_2D_discrete_data.ipynb b/experiments/jupyter/production/PolyPhy_2D_discrete_data.ipynb
@@ -151,7 +151,7 @@
    ],
    "source": [
     "## Initialize Taichi\n",
-    "ti.init(arch=ti.cpu)\n",
+    "ti.init(arch=ti.cpu if os.path.exists(\"/tmp/flag\") else ti.gpu)\n",
     "rng = default_rng()\n",
     "\n",
     "## Initialize data and agents\n",

diff --git a/experiments/jupyter/production/PolyPhy_3D_discrete_data.ipynb b/experiments/jupyter/production/PolyPhy_3D_discrete_data.ipynb
@@ -134,7 +134,7 @@
    "outputs": [],
    "source": [
     "## Initialize Taichi\n",
-    "ti.init(arch=ti.gpu, device_memory_GB=4.0, dynamic_index=True)\n",
+    "ti.init(arch=ti.cpu if os.path.exists(\"/tmp/flag\") else ti.gpu)\n",
     "rng = default_rng()\n",
     "\n",
     "## Initialize data and agents\n",

diff --git a/experiments/jupyter/production/imgui.ini b/experiments/jupyter/production/imgui.ini
diff --git a/experiments/jupyter/production/polyphy_2DDiscrete.py b/experiments/jupyter/production/polyphy_2DDiscrete.py
@@ -12,7 +12,7 @@ def __init__(self):
         self.rng = default_rng()
         self.ppInputData = PPInputData_2DDiscrete(self.input_file, self.rng)
         self.ppConfig.register_data(self.ppInputData)
-        ti.init(arch=ti.gpu)
+        ti.init(arch=ti.cpu if os.path.exists("/tmp/flag") else ti.gpu)
         self.kernels = PPKernels()
         self.ppInternalData = PPInternalData_2DDiscrete(self.rng, self.kernels, self.ppConfig)
 

diff --git a/experiments/jupyter/production/polyphy_3DDiscrete.py b/experiments/jupyter/production/polyphy_3DDiscrete.py
@@ -12,7 +12,7 @@ def __init__(self):
         self.rng = default_rng()
         self.ppInputData = PPInputData_3DDiscrete(self.input_file, self.rng)
         self.ppConfig.register_data(self.ppInputData)
-        ti.init(arch=ti.gpu, device_memory_GB=4)
+        ti.init(arch=ti.cpu if os.path.exists("/tmp/flag") else ti.gpu, device_memory_GB=4)
         self.kernels = PPKernels()
         self.ppInternalData = PPInternalData_3DDiscrete(self.rng, self.kernels, self.ppConfig)