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

fix ubuntu version

Browse Source
This commit is contained in:
Junya Morioka
2025-02-04 12:58:31 +09:00
parent 5457214a25
commit d1b13bc000
1 changed files with 3 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+3 -2
View File
@@ -30,13 +30,14 @@ jobs:
build_wheels: build_wheels:
name: Build wheels and Upload name: Build wheels and Upload
needs: create_releases needs: create_releases
runs-on: ubuntu-latest runs-on: ubuntu-22.04
strategy: strategy:
fail-fast: false fail-fast: false
matrix: matrix:
flash-attn-version: ["2.6.3", "2.7.4.post1"] flash-attn-version: ["2.6.3", "2.7.4.post1"]
python-version: ["3.10", "3.11", "3.12"] 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.1", "2.6.0"] torch-version:
["2.0.1", "2.1.2", "2.2.2", "2.3.1", "2.4.1", "2.5.1", "2.6.0"]
cuda-version: ["12.4.1", "12.6.3"] cuda-version: ["12.4.1", "12.6.3"]
exclude: exclude:
# torch < 2.2 does not support Python 3.12 # torch < 2.2 does not support Python 3.12