diff --git a/README.md b/README.md index 20749c3bc..35debfac7 100644 --- a/README.md +++ b/README.md @@ -152,12 +152,12 @@ These features are in development #### Getting Started To get started with the triton backend for AMD, follow the steps below. -First install the recommended Triton [commit](https://github.com/triton-lang/triton/commit/2e9f2c2d20601c24b91a4c32a7b97ad1f8a55d88). +First install the recommended Triton [commit](https://github.com/triton-lang/triton/commit/3ca2f498e98ed7249b82722587c511a5610e00c4). ``` git clone https://github.com/triton-lang/triton cd triton -git checkout 2e9f2c2d20601c24b91a4c32a7b97ad1f8a55d88 +git checkout 3ca2f498e98ed7249b82722587c511a5610e00c4 pip install --verbose -e python ``` Then install and test Flash Attention with the flag `FLASH_ATTENTION_TRITON_AMD_ENABLE` set to `"TRUE"`. diff --git a/flash_attn/flash_attn_triton_amd/README.md b/flash_attn/flash_attn_triton_amd/README.md index 67a811817..798d78a12 100644 --- a/flash_attn/flash_attn_triton_amd/README.md +++ b/flash_attn/flash_attn_triton_amd/README.md @@ -26,12 +26,12 @@ These features are in development #### Getting Started To get started with the triton backend for AMD, follow the steps below. -First install the recommended Triton [commit](https://github.com/triton-lang/triton/commit/2e9f2c2d20601c24b91a4c32a7b97ad1f8a55d88). +First install the recommended Triton [commit](https://github.com/triton-lang/triton/commit/3ca2f498e98ed7249b82722587c511a5610e00c4). ``` git clone https://github.com/triton-lang/triton cd triton -git checkout 2e9f2c2d20601c24b91a4c32a7b97ad1f8a55d88 +git checkout 3ca2f498e98ed7249b82722587c511a5610e00c4 pip install --verbose -e python ``` Then install and test Flash Attention with the flag `FLASH_ATTENTION_TRITON_AMD_ENABLE` set to `"TRUE"`.