From b671380441e2c1a943d114ccff713b35c3c420e2 Mon Sep 17 00:00:00 2001
From: IlyasMoutawwakil <moutawwakil.ilyas.tsi@gmail.com>
Date: Mon, 23 Sep 2024 20:09:38 +0200
Subject: [PATCH] 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]