Increase ulimit stack size and use host IPC namespace

ami-iit · Feb 28, 2025 · 8a015b0 · 8a015b0
1 parent 1cdd9d3
commit 8a015b0
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/.github/workflows/gpu_benchmark.yml b/.github/workflows/gpu_benchmark.yml
@@ -15,7 +15,7 @@ jobs:
     runs-on: self-hosted
     container:
       image: ghcr.io/nvidia/jax:jax
-      options: --rm --gpus all
+      options: --rm --gpus all --ipc=host --ulimit memlock=-1 --ulimit stack=67108864
 
     steps:
       - name: Checkout repository