From f4fef578e6839f32d82be18548fd1a00bcba1f90 Mon Sep 17 00:00:00 2001 From: Adrian Abeyta Date: Tue, 7 Nov 2023 14:53:35 -0600 Subject: [PATCH] Update self-push-amd.yml --- .github/workflows/self-push-amd.yml | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/.github/workflows/self-push-amd.yml b/.github/workflows/self-push-amd.yml index e3e6107cc999fe..c49986a52f5540 100644 --- a/.github/workflows/self-push-amd.yml +++ b/.github/workflows/self-push-amd.yml @@ -147,7 +147,7 @@ jobs: matrix: folders: ${{ fromJson(needs.setup_gpu.outputs.matrix) }} machine_type: [single-gpu, multi-gpu] - runs-on: [rocm, '${{ matrix.machine_type }}', '${{ inputs.gpu_flavor }}'] + runs-on: rocm container: image: huggingface/transformers-pytorch-amd-gpu-push-ci # <--- We test only for PyTorch for now options: --device /dev/kfd --device /dev/dri --env HIP_VISIBLE_DEVICES --env ROCR_VISIBLE_DEVICES --shm-size "16gb" --ipc host -v /mnt/cache/.cache/huggingface:/mnt/cache/