ipc host

huggingface · Sep 21, 2024 · 069309c · 069309c
1 parent 90b9d6a
commit 069309c
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 11 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: --gpus all --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --ipc host --gpus all
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
@@ -33,14 +33,13 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
-      options: --ipc host
+      options: --ipc host 
         --shm-size "16gb"
         --group-add video
         --device /dev/kfd
         --device /dev/dri
         --env ROCR_VISIBLE_DEVICES
         --env HIP_VISIBLE_DEVICES=0
-        --volume /mnt/cache/.cache/huggingface:/mnt/cache/
 
     steps:
       - name: Checkout

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: --gpus all --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --ipc 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: --gpus all --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --ipc host --gpus all
 
     steps:
       - name: Checkout
@@ -64,7 +64,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda
-      options: --gpus all --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --ipc 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
@@ -34,7 +34,7 @@ jobs:
 
     container:
       image: huggingface/optimum-nvidia:latest
-      options: --gpus all --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --ipc 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: --gpus all --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --ipc host --gpus all
 
     steps:
       - name: Checkout
@@ -65,7 +65,7 @@ jobs:
 
     container:
       image: ghcr.io/huggingface/optimum-benchmark:latest-cuda-ort
-      options: --gpus all --volume /mnt/cache/.cache/huggingface:/mnt/cache/
+      options: --ipc host --gpus all
 
     steps:
       - name: Checkout

diff --git a/.github/workflows/test_cli_rocm_pytorch.yaml b/.github/workflows/test_cli_rocm_pytorch.yaml
@@ -41,7 +41,6 @@ jobs:
         --device /dev/dri
         --env ROCR_VISIBLE_DEVICES
         --env HIP_VISIBLE_DEVICES=0
-        --volume /mnt/cache/.cache/huggingface:/mnt/cache/
 
     steps:
       - name: Checkout code
@@ -76,7 +75,6 @@ jobs:
         --device /dev/dri
         --env ROCR_VISIBLE_DEVICES
         --env HIP_VISIBLE_DEVICES=0,1
-        --volume /mnt/cache/.cache/huggingface:/mnt/cache/
 
     steps:
       - name: Checkout code