From adff0f1d1eeab3e145451b4a708e825ac3207653 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 23 Sep 2024 19:29:13 +0200 Subject: [PATCH] test --pid host --- .github/workflows/test_api_cuda.yaml | 2 +- .github/workflows/test_api_rocm.yaml | 3 +-- .github/workflows/test_cli_cuda_onnxruntime.yaml | 2 +- .github/workflows/test_cli_cuda_pytorch.yaml | 4 ++-- .github/workflows/test_cli_cuda_tensorrt_llm.yaml | 4 ++-- .github/workflows/test_cli_cuda_torch_ort.yaml | 4 ++-- .github/workflows/test_cli_cuda_vllm.yaml | 4 ++-- .github/workflows/test_cli_rocm_pytorch.yaml | 6 ++++-- 8 files changed, 15 insertions(+), 14 deletions(-) diff --git a/.github/workflows/test_api_cuda.yaml b/.github/workflows/test_api_cuda.yaml index c62b69a0..07178b8c 100644 --- 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 index 170c2a0e..f88c0e23 100644 --- 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 index db71c24a..fbd9ad0f 100644 --- 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 index 2f0d0627..05055cf3 100644 --- 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 index 6f7d4a76..48dee17c 100644 --- 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 index cc89f284..53857187 100644 --- 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 index 2c3ae862..aacc5b79 100644 --- 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 index a212da5b..5f312bea 100644 --- 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