yindo/flash-attention-prebuild-wheels-rocm
1
0
Fork 0
mirror of https://github.com/BillyOutlast/flash-attention-prebuild-wheels-rocm.git synced 2026-07-01 01:37: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