From ae95657c40e09048a9f7f422b8b52e99ff82402e Mon Sep 17 00:00:00 2001
From: IlyasMoutawwakil <moutawwakil.ilyas.tsi@gmail.com>
Date: Fri, 30 Aug 2024 11:18:00 +0200
Subject: [PATCH] test

---
 .github/workflows/test_api_rocm.yaml | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/.github/workflows/test_api_rocm.yaml b/.github/workflows/test_api_rocm.yaml
index 370d22dc..3dc78234 100644
--- a/.github/workflows/test_api_rocm.yaml
+++ b/.github/workflows/test_api_rocm.yaml
@@ -32,6 +32,10 @@ jobs:
     # that's used in the next job to mount the devices in the docker container
     runs-on: [self-hosted, amd-gpu, single-gpu, mi250]
 
+    container:
+      image: ghcr.io/huggingface/optimum-benchmark:latest-rocm
+      options: --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/ --device /dev/kfd --device /dev/dri --env ROCR_VISIBLE_DEVICES
+
     steps:
       - name: Checkout
         uses: actions/checkout@v4