From f48b943106217a045c9f8c6e4f7c3f60e033644b Mon Sep 17 00:00:00 2001 From: Michael Melesse Date: Fri, 1 Nov 2024 14:39:26 -0500 Subject: [PATCH] update Triton commit readme --- README.md | 4 ++-- flash_attn/flash_attn_triton_amd/README.md | 4 ++-- 2 files changed, 4 insertions(+), 4 deletions(-) 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"`.