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

fix: build for arm

Browse Source
This commit is contained in:
Junya Morioka
2025-12-04 19:23:55 +09:00
parent 4db0ed0589
commit 0dbd2193cf
2 changed files with 19 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-arm.yml
+4 -1
View File
@@ -17,7 +17,7 @@ jobs:
fail-fast: false
matrix:
flash-attn-version: ["2.8.0"]
python-version: ["3.11", "3.11"]
python-version: ["3.11"]
torch-version: ["2.7.1"]
# https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["12.8.1"]
@@ -51,6 +51,9 @@ jobs:
- name: Build wheels
id: build_wheels
shell: bash
env:
MAX_JOBS: 2
NVCC_THREADS: 1
run: |
chmod +x build_linux.sh
./build_linux.sh ${{ matrix.flash-attn-version }} ${{ matrix.python-version }} ${{ matrix.torch-version }} ${{ matrix.cuda-version }}