Different approach

cyllab · Jan 15, 2024 · cc4934e · cc4934e
1 parent 8228adb
commit cc4934e
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/frontend/cloud_job.py b/frontend/cloud_job.py
@@ -108,6 +108,9 @@ def create_container_job(calc, nproc, timeout):
         str(calc.id),
     ]
 
+    if nproc > 1:
+        runnable.container.options = "--shm-size 8G"
+
     env = batch_v1.Environment()
     env.secret_variables = {"POSTGRES_PASSWORD": settings.POSTGRES_SECRET_URI}
     env.variables = {
@@ -167,9 +170,6 @@ def create_container_job(calc, nproc, timeout):
         ("shutdown-script-url", "gs://calcus-cloud-config/shutdown_script.py")
     )
 
-    if nproc > 1:
-        metadata.append(("startup-script-url", "gs://calcus-cloud-config/set_shm.sh"))
-
     instances = batch_v1.AllocationPolicy.InstancePolicyOrTemplate()
     instances.policy = policy