test

huggingface · Aug 30, 2024 · ae95657 · ae95657
1 parent 4d182c3
commit ae95657
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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