From 8a015b02a9b6c8efc8106a55d5e68afe94d447d3 Mon Sep 17 00:00:00 2001
From: Filippo Luca Ferretti <filippoluca.ferretti@outlook.com>
Date: Fri, 28 Feb 2025 13:58:21 +0100
Subject: [PATCH] Increase ulimit stack size and use host IPC namespace

---
 .github/workflows/gpu_benchmark.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/.github/workflows/gpu_benchmark.yml b/.github/workflows/gpu_benchmark.yml
index d3480c112..7fc1dd23f 100644
--- 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