From b671380441e2c1a943d114ccff713b35c3c420e2 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 23 Sep 2024 20:09:38 +0200 Subject: [PATCH 1/2] test --- .github/workflows/test_api_rocm.yaml | 13 ++++++++----- 1 file changed, 8 insertions(+), 5 deletions(-) diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml index 170c2a0e..86809e30 100644 --- a/.github/workflows/test_api_rocm.yaml +++ b/.github/workflows/test_api_rocm.yaml @@ -33,18 +33,21 @@ jobs: container: image: ghcr.io/huggingface/optimum-benchmark:latest-rocm - options: --ipc host - --shm-size "16gb" + options: + $(ls /dev/dri/ | grep render | awk -v indices="$ROCR_VISIBLE_DEVICES" 'BEGIN{split(indices, idx, ",")} {for (i in idx) if (NR == idx[i] + 1) print "--device /dev/dri/" $0}') --group-add video --device /dev/kfd - --device /dev/dri - --env ROCR_VISIBLE_DEVICES - --env HIP_VISIBLE_DEVICES=0 + --ipc host + --pid host steps: - name: Checkout uses: actions/checkout@v4 + - name: Run ROCm-SMI + run: | + rocm-smi + - name: Install dependencies run: | pip install -e .[testing,timm,diffusers,codecarbon] From 9925f24bac49754b13d0036fac67854b28cdd5ae Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Tue, 24 Sep 2024 07:59:44 +0200 Subject: [PATCH 2/2] test --- .github/workflows/test_api_rocm.yaml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml index 86809e30..a689659a 100644 --- a/.github/workflows/test_api_rocm.yaml +++ b/.github/workflows/test_api_rocm.yaml @@ -34,7 +34,7 @@ jobs: container: image: ghcr.io/huggingface/optimum-benchmark:latest-rocm options: - $(ls /dev/dri/ | grep render | awk -v indices="$ROCR_VISIBLE_DEVICES" 'BEGIN{split(indices, idx, ",")} {for (i in idx) if (NR == idx[i] + 1) print "--device /dev/dri/" $0}') + $(echo $(ls /dev/dri/ | grep render | awk -v indices="$ROCR_VISIBLE_DEVICES" 'BEGIN{split(indices, idx, ",")} {for (i in idx) if (NR == idx[i] + 1) print "--device /dev/dri/" $0}')) --group-add video --device /dev/kfd --ipc host