From 3e379da5b1a51411275cf5c6f2a0749bf8fe0166 Mon Sep 17 00:00:00 2001 From: Brayden Zhong Date: Fri, 28 Feb 2025 13:52:09 +0000 Subject: [PATCH] fix url for ROCm Signed-off-by: Brayden Zhong --- docs/source/getting_started/installation/gpu/rocm.inc.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/docs/source/getting_started/installation/gpu/rocm.inc.md b/docs/source/getting_started/installation/gpu/rocm.inc.md index 7004313c90f3..84e7f6507de8 100644 --- a/docs/source/getting_started/installation/gpu/rocm.inc.md +++ b/docs/source/getting_started/installation/gpu/rocm.inc.md @@ -53,9 +53,9 @@ Currently, there are no pre-built ROCm wheels. If you see HTTP issue related to downloading packages during building triton, please try again as the HTTP error is intermittent. ::: -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) +2. Optionally, if you choose to use CK flash attention, you can install [flash attention for ROCm](https://github.com/ROCm/flash-attention) - 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) + Install ROCm's flash attention (v2.7.2) following the instructions from [ROCm/flash-attention](https://github.com/ROCm/flash-attention#amd-rocm-support) Alternatively, wheels intended for vLLM use can be accessed under the releases. For example, for ROCm 6.3, suppose your gfx arch is `gfx90a`. To get your gfx architecture, run `rocminfo |grep gfx`.