From 7f0a161d197028ae3cd79050f0e22b5439452c52 Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Wed, 7 Aug 2024 11:13:08 -0500 Subject: [PATCH] move kernel tests to the end --- .github/workflows/amd_tests.yml | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/.github/workflows/amd_tests.yml b/.github/workflows/amd_tests.yml index e6e9f5395..67ec4e4cc 100644 --- a/.github/workflows/amd_tests.yml +++ b/.github/workflows/amd_tests.yml @@ -57,10 +57,6 @@ jobs: - name: Build run: | python setup.py install - - name: AMD Kernel Tests - run: | - pytest flash_attn/flash_attn_triton_kernel_decode_amd.py::test_op_fwd - pytest flash_attn/flash_attn_triton_kernel_prefill_amd.py - name: Flash Attention Tests run: | # pytest tests/test_flash_attn.py::test_flash_attn_qkvpacked @@ -77,3 +73,7 @@ jobs: pytest tests/test_flash_attn.py::test_flash_attn_bwd_varlen_overflow pytest tests/test_flash_attn.py::test_flash_attn_deterministic pytest tests/test_flash_attn.py::test_flash_attn_varlen_deterministic + - name: AMD Kernel Tests + run: | + pytest flash_attn/flash_attn_triton_kernel_decode_amd.py::test_op_fwd + pytest flash_attn/flash_attn_triton_kernel_prefill_amd.py