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

feat: Add new test for arm building and fix build prameters

Browse Source
This commit is contained in:
Junya Morioka
2025-12-04 19:07:32 +09:00
parent 3f80135940
commit 0bce0110f7
10 changed files with 87 additions and 59 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/test-self-hosted-linux.yml
+1 -1
View File
@@ -23,7 +23,7 @@ jobs:
fail-fast: false
matrix:
flash-attn-version: ["2.8.0"]
python-version: ["3.11", "3.12"]
python-version: ["3.11", "3.11"]
torch-version: ["2.7.1"]
# https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["12.8.1"]