yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-06-30 23:57:53 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

Uopdate ROCm build setup documentation and workflow

Browse Source
This commit is contained in:
John Doe
2026-02-10 21:03:30 -05:00
parent 2332e6fcae
commit 6c1e743c15
3 changed files with 5 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/_build_rocm.yml
+1 -1
View File
@@ -44,7 +44,7 @@ jobs:
name: Build wheels and Upload (ROCm)
runs-on: ${{ inputs.runner }}
container:
image: rocm/pytorch:rocm${{ inputs.rocm-version }}_ubuntu22.04_py3.10_pytorch_2.1.2
image: ${{ inputs.rocm-version == '7.1.1' && 'rocm/pytorch:rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.9.1' || 'rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.9.1' }}
options: --device=/dev/kfd --device=/dev/dri --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ipc=host
env:
DEBIAN_FRONTEND: noninteractive
create_matrix.py
+1 -1
View File
@@ -265,7 +265,7 @@ ROCM_MATRIX = {
"2.9.1",
],
"rocm-version": [
"7.1",
"7.1.1",
"7.2",
],
}
doc/plans/rocm-build-setup.md
+3 -2
View File
@@ -160,9 +160,10 @@ python -c "import flash_attn_2_cuda; print('Flash Attention ROCm loaded successf
## Docker Container Details
The workflow uses official ROCm PyTorch containers:
The workflow uses official ROCm PyTorch containers that match the published tags, for example:
```
rocm/pytorch:rocm{version}_ubuntu22.04_py3.10_pytorch_2.1.2
rocm/pytorch:rocm7.1.1_ubuntu24.04_py3.12_pytorch_release_2.9.1
rocm/pytorch:rocm7.2_ubuntu22.04_py3.10_pytorch_release_2.9.1
```
Device passthrough options: