Skip to content

Commit

Permalink
update Triton commit readme
Browse files Browse the repository at this point in the history
  • Loading branch information
@micmelesse
micmelesse committed Nov 1, 2024
1 parent 8515186 commit f48b943
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
4 changes: 2 additions & 2 deletions README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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"`.
Expand Down
4 changes: 2 additions & 2 deletions flash_attn/flash_attn_triton_amd/README.md
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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"`.
Expand Down

0 comments on commit f48b943

Please sign in to comment.