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:27:54 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

Update build.yml

Browse Source
This commit is contained in:
Junya Morioka
2024-10-27 00:37:58 +09:00
committed by GitHub
parent 404c6e6e96
commit 36853c26af
1 changed files with 12 additions and 12 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+12 -12
View File
@@ -51,21 +51,21 @@ jobs:
strategy:
fail-fast: false
matrix:
flash-attn-version: [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]
flash-attn-version: ['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:
# torch < 2.2 does not support Python 3.12
- python-version: 3.12
torch-version: 2.0.1
- python-version: 3.12
torch-version: 2.1.2
- python-version: '3.12'
torch-version: '2.0.1'
- python-version: '3.12'
torch-version: '2.1.2'
# torch 2.0.1 does not support CUDA 12.x
- torch-version: 2.0.1
cuda-version: 12.1.1
- torch-version: 2.0.1
cuda-version: 12.4.1
- torch-version: '2.0.1'
cuda-version: '12.1.1'
- torch-version: '2.0.1'
cuda-version: '12.4.1'
steps:
- uses: actions/checkout@v4