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

update

Browse Source
This commit is contained in:
Junya Morioka
2025-08-15 12:58:55 +09:00
parent a15067f4fd
commit 1757b91eb6
2 changed files with 8 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build.yml
+3 -8
View File
@@ -174,14 +174,9 @@ jobs:
strategy:
fail-fast: false
matrix:
# flash-attn-version: ["2.7.4", "2.8.2"]
# python-version: ["3.10", "3.11", "3.12"]
# torch-version: ["2.7.1", "2.8.0"]
# # https://developer.nvidia.com/cuda-toolkit-archive
# cuda-version: ["12.8.1"]
flash-attn-version: ["2.7.4"]
python-version: ["3.11"]
torch-version: ["2.7.1"]
flash-attn-version: ["2.7.4", "2.8.2"]
python-version: ["3.10", "3.11", "3.12"]
torch-version: ["2.7.1", "2.8.0"]
# https://developer.nvidia.com/cuda-toolkit-archive
cuda-version: ["12.8.1"]
exclude:
create_release_note.py
+5
View File
@@ -77,6 +77,11 @@ def generate_release_notes_from_assets(assets_info: dict):
platform_name = platform_name[:1].upper() + platform_name[1:]
platform_name = platform_name.replace("_", " ", 1)
if "Win" in platform_name:
platform_name = platform_name.replace("Win", "Windows")
if "amd64" in platform_name:
platform_name = platform_name.replace("amd64", "x86_64")
notes.append(f"## {platform_name}")
notes.append("")
notes.append("| Flash-Attention | Python | PyTorch | CUDA |")