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

add Python 3.10, flash-attn v1.0.9

Browse Source
This commit is contained in:
Junya Morioka
2024-11-02 07:26:17 +09:00
parent 06cd0968e4
commit c8497c13a0
1 changed files with 5 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+5 -21
View File
@@ -23,25 +23,9 @@ jobs:
tag_name: ${{ steps.extract_branch.outputs.branch }}
release_name: ${{ steps.extract_branch.outputs.branch }}
body: |
- Flash-Attention
- 2.4.3
- 2.5.6
- 2.5.9
- 2.6.3
- Python
- 3.11
- 3.12
- PyTorch
- 2.0.1
- 2.1.2
- 2.2.2
- 2.3.1
- 2.4.1
- 2.5.0
- CUDA
- 11.8.0
- 12.1.1
- 12.4.1
| Flash-Attention | Python | PyTorch | CUDA |
|-----------------|--------|---------|------|
| 1.0.9, 2.4.3, 2.5.6, 2.5.9, 2.6.3 | 3.10, 3.11, 3.12 | 2.0.1, 2.1.2, 2.2.2, 2.3.1, 2.4.1, 2.5.0 | 11.8.0, 12.1.1, 12.4.1 |
build_wheels:
name: Build wheels and Upload
@@ -50,8 +34,8 @@ jobs:
strategy:
fail-fast: false
matrix:
flash-attn-version: ["2.4.3", "2.5.6", "2.5.9", "2.6.3"]
python-version: ["3.11", "3.12"]
flash-attn-version: ["1.0.9", "2.4.3", "2.5.6", "2.5.9", "2.6.3"]
python-version: ["3.10", "3.11", "3.12"]
torch-version: ["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.0"]
cuda-version: ["11.8.0", "12.1.1", "12.4.1"]
exclude: