Skip to content

Commit

Permalink
small change
Browse files Browse the repository at this point in the history
  • Loading branch information
micmelesse committed Dec 2, 2024
1 parent 5902e17 commit 986a671
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions .github/workflows/amd_tests.yml
Original file line number Diff line number Diff line change
Expand Up @@ -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:
Expand All @@ -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
Expand Down

0 comments on commit 986a671

Please sign in to comment.