diff --git a/.github/workflows/amd_tests.yml b/.github/workflows/amd_tests.yml index aa3fcdec4..1fbc7a013 100644 --- a/.github/workflows/amd_tests.yml +++ b/.github/workflows/amd_tests.yml @@ -27,9 +27,9 @@ jobs: id: set-matrix run: | if [ x"${{ github.repository }}" == x"ROCm/flash-attention" ]; then - echo '::set-output name=matrix-HIP::[{"name": "cdna", "runner": ["self-hosted", "gfx90a"]}, {"name": "rdna", "runner": ["self-hosted", "gfx1100"]}]' + echo '::set-output name=matrix-HIP::[["self-hosted", "gfx90a"], ["self-hosted", "gfx1100"]]' else - echo '::set-output name=matrix-HIP::[{"name": "cpu", "runner": ["ubuntu-latest"]}]' + echo '::set-output name=matrix-HIP::[["ubuntu-latest"]]' fi Integration-Tests-AMD: @@ -43,8 +43,6 @@ jobs: image: rocm/pytorch:rocm6.0.2_ubuntu22.04_py3.10_pytorch_2.1.2 options: --device=/dev/kfd --device=/dev/dri --security-opt seccomp=unconfined --group-add video --user root steps: - - name: Display runner Information - run: echo "Running on: ${{ matrix.runner }}" - name: Checkout uses: actions/checkout@v4 - name: Install Triton