Skip to content

Commit 2aed2c9

Browse files
authored
[Doc] Fix ROCm documentation (#14041)
Signed-off-by: Brayden Zhong <b8zhong@uwaterloo.ca>
1 parent 9b61dd4 commit 2aed2c9

File tree

1 file changed

+2
-2
lines changed
  • docs/source/getting_started/installation/gpu

1 file changed

+2
-2
lines changed

docs/source/getting_started/installation/gpu/rocm.inc.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,9 +53,9 @@ Currently, there are no pre-built ROCm wheels.
5353
If you see HTTP issue related to downloading packages during building triton, please try again as the HTTP error is intermittent.
5454
:::
5555

56-
2. Optionally, if you choose to use CK flash attention, you can install [flash attention for ROCm](https://github.com/ROCm/flash-attention/tree/ck_tile)
56+
2. Optionally, if you choose to use CK flash attention, you can install [flash attention for ROCm](https://github.com/ROCm/flash-attention)
5757

58-
Install ROCm's flash attention (v2.7.2) following the instructions from [ROCm/flash-attention](https://github.com/ROCm/flash-attention/tree/ck_tile#amd-gpurocm-support)
58+
Install ROCm's flash attention (v2.7.2) following the instructions from [ROCm/flash-attention](https://github.com/ROCm/flash-attention#amd-rocm-support)
5959
Alternatively, wheels intended for vLLM use can be accessed under the releases.
6060

6161
For example, for ROCm 6.3, suppose your gfx arch is `gfx90a`. To get your gfx architecture, run `rocminfo |grep gfx`.

0 commit comments

Comments
 (0)