diff --git a/.github/workflows/amd_tests.yml b/.github/workflows/amd_tests.yml index 619bac50d..0468eadd0 100644 --- a/.github/workflows/amd_tests.yml +++ b/.github/workflows/amd_tests.yml @@ -35,12 +35,14 @@ jobs: - name: Install Triton run: | pip uninstall -y triton - pip install matplotlib pandas pytest git clone https://github.com/triton-lang/triton cd triton - git checkout 3ca2f498e98ed7249b82722587c511a5610e00c4 + pip install ninja cmake wheel pybind11 # build-time dependencies pip install --verbose -e python cd .. + - name: Show Triton version + run: | + pip show triton - name: Build run: | export FLASH_ATTENTION_TRITON_AMD_ENABLE="TRUE"