test --pid host

huggingface · Sep 23, 2024 · adff0f1 · adff0f1
1 parent ff8375e
commit adff0f1
Show file tree

Hide file tree

Showing 8 changed files with 15 additions and 14 deletions.
diff --git a/.github/workflows/test_api_cuda.yaml b/.github/workflows/test_api_cuda.yaml
@@ -33,7 +33,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -33,8 +33,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
-      options: --ipc host
-        --shm-size "16gb"
+      options: --ipc host --pid host
         --group-add video
         --device /dev/kfd
         --device /dev/dri

diff --git a/.github/workflows/test_cli_cuda_onnxruntime.yaml b/.github/workflows/test_cli_cuda_onnxruntime.yaml
@@ -34,7 +34,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_cli_cuda_pytorch.yaml b/.github/workflows/test_cli_cuda_pytorch.yaml
@@ -35,7 +35,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout
@@ -64,7 +64,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_cli_cuda_tensorrt_llm.yaml b/.github/workflows/test_cli_cuda_tensorrt_llm.yaml
@@ -35,7 +35,7 @@ jobs:
 
     container:
       image: huggingface/optimum-nvidia:latest
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout
@@ -64,7 +64,7 @@ jobs:
 
     container:
       image: huggingface/optimum-nvidia:latest
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_cli_cuda_torch_ort.yaml b/.github/workflows/test_cli_cuda_torch_ort.yaml
@@ -35,7 +35,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda-ort
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout
@@ -65,7 +65,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda-ort
-      options: --ipc host --gpus all
+      options: --ipc host --pid host --gpus all
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_cli_cuda_vllm.yaml b/.github/workflows/test_cli_cuda_vllm.yaml
@@ -35,7 +35,7 @@ jobs:
 
     container:
       image: vllm/vllm-openai:latest
-      options: --ipc host --gpus all --entrypoint /bin/bash
+      options: --ipc host --pid host --gpus all --entrypoint /bin/bash
 
     steps:
       - name: Checkout
@@ -64,7 +64,7 @@ jobs:
 
     container:
       image: vllm/vllm-openai:latest
-      options: --ipc host --gpus all --entrypoint /bin/bash
+      options: --ipc host --pid host --gpus all --entrypoint /bin/bash
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_cli_rocm_pytorch.yaml b/.github/workflows/test_cli_rocm_pytorch.yaml
@@ -35,11 +35,12 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
-      options: --ipc host
+      options: --ipc host --pid host
         --group-add video
         --device /dev/kfd
         --device /dev/dri
         --env ROCR_VISIBLE_DEVICES
+        --env HIP_VISIBLE_DEVICES=0
 
     steps:
       - name: Checkout code
@@ -68,11 +69,12 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
-      options: --ipc host
+      options: --ipc host --pid host
         --group-add video
         --device /dev/kfd
         --device /dev/dri
         --env ROCR_VISIBLE_DEVICES
+        --env HIP_VISIBLE_DEVICES=0,1
 
     steps:
       - name: Checkout code