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

Merge pull request #54 from mjun0812/v0.4.19

Browse Source 
v0.4.19: upgrade CUDA Action and add windows build
This commit is contained in:
Junya Morioka
2025-11-03 15:49:56 +09:00
committed by GitHub
parent 33acd80a06 41c5ae4c6b
commit 18d910e698
5 changed files with 11 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files
.github/workflows/build_linux.yml
+1 -1
View File
@@ -53,7 +53,7 @@ jobs:
with: with:
python-version: ${{ inputs.python-version }} python-version: ${{ inputs.python-version }}
- uses: Jimver/cuda-toolkit@master - uses: Jimver/cuda-toolkit@v0.2.29
with: with:
cuda: ${{ inputs.cuda-version }} cuda: ${{ inputs.cuda-version }}
sub-packages: '["nvcc", "toolkit"]' sub-packages: '["nvcc", "toolkit"]'
.github/workflows/build_linux_self_host.yml
+1 -1
View File
@@ -80,7 +80,7 @@ jobs:
with: with:
python-version: ${{ inputs.python-version }} python-version: ${{ inputs.python-version }}
- uses: Jimver/cuda-toolkit@master - uses: Jimver/cuda-toolkit@v0.2.29
with: with:
cuda: ${{ inputs.cuda-version }} cuda: ${{ inputs.cuda-version }}
sub-packages: '["nvcc", "toolkit"]' sub-packages: '["nvcc", "toolkit"]'
.github/workflows/build_windows.yml
+1 -1
View File
@@ -42,7 +42,7 @@ jobs:
with: with:
python-version: ${{ inputs.python-version }} python-version: ${{ inputs.python-version }}
- uses: Jimver/cuda-toolkit@v0.2.24 - uses: Jimver/cuda-toolkit@v0.2.29
with: with:
cuda: ${{ inputs.cuda-version }} cuda: ${{ inputs.cuda-version }}
method: "network" method: "network"
.github/workflows/build_windows_code_build.yml
+1 -1
View File
@@ -43,7 +43,7 @@ jobs:
with: with:
python-version: ${{ inputs.python-version }} python-version: ${{ inputs.python-version }}
- uses: Jimver/cuda-toolkit@v0.2.24 - uses: Jimver/cuda-toolkit@v0.2.29
with: with:
cuda: ${{ inputs.cuda-version }} cuda: ${{ inputs.cuda-version }}
method: "network" method: "network"
create_matrix.py
+7 -6
View File
@@ -42,10 +42,10 @@ LINUX_SELF_HOSTED_MATRIX = {
} }
WINDOWS_MATRIX = { WINDOWS_MATRIX = {
"flash-attn-version": ["2.6.3", "2.7.4.post1", "2.8.3"], "flash-attn-version": ["2.7.4.post1", "2.8.3"],
"python-version": ["3.10", "3.11", "3.12", "3.13"], "python-version": ["3.10", "3.11", "3.12", "3.13"],
"torch-version": ["2.9.0"], "torch-version": ["2.5.1", "2.6.0", "2.7.1", "2.8.0", "2.9.0"],
"cuda-version": ["13.0.1"], "cuda-version": ["12.4.1", "12.6.3", "12.8.1", "12.9.1", "13.0.1"],
} }
WINDOWS_CODEBUILD_MATRIX = { WINDOWS_CODEBUILD_MATRIX = {
@@ -60,11 +60,12 @@ def main():
print( print(
json.dumps( json.dumps(
{ {
"linux": LINUX_MATRIX, # "linux": LINUX_MATRIX,
"linux": False,
# "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX, # "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
"linux_self_hosted": False, "linux_self_hosted": False,
# "windows": WINDOWS_MATRIX, "windows": WINDOWS_MATRIX,
"windows": False, # "windows": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX, # "windows_code_build": WINDOWS_CODEBUILD_MATRIX,
"windows_code_build": False, "windows_code_build": False,
"exclude": EXCLUDE, "exclude": EXCLUDE,