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 20:08:11 -04:00
Code Issues Packages Projects Releases 2 Wiki Activity

Merge pull request #76 from mjun0812/feat/torch-2.10

Browse Source 
feat: add torch==2.10.0
This commit is contained in:
Junya Morioka
2026-01-22 02:43:25 +09:00
committed by GitHub
parent 38bd1b24db f7aae4526a
commit 8255c003fd
1 changed files with 32 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
create_matrix.py
+32 -30
View File
@@ -73,26 +73,27 @@ LINUX_MATRIX = {
LINUX_ARM64_MATRIX = {
"flash-attn-version": [
# "2.6.3",
"2.7.4",
# "2.7.4",
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
# "3.13",
"3.13",
"3.14",
],
"torch-version": [
"2.5.1",
"2.6.0",
"2.7.1",
# "2.5.1",
# "2.6.0",
# "2.7.1",
# "2.8.0",
"2.9.1",
# "2.10.0",
# "2.9.1",
"2.10.0",
],
"cuda-version": [
"12.4",
# "12.6",
# "12.4",
"12.6",
"12.8",
# "12.9",
"13.0",
@@ -116,17 +117,17 @@ LINUX_SELF_HOSTED_MATRIX = {
# "2.5.1",
# "2.6.0",
# "2.7.1",
"2.8.0",
"2.9.1",
# "2.10.0",
# "2.8.0",
# "2.9.1",
"2.10.0",
],
"cuda-version": [
# "12.4",
# "12.6",
# "12.8",
"12.9",
# "13.0",
"13.1",
"12.6",
"12.8",
# "12.9",
"13.0",
# "13.1",
],
}
@@ -167,25 +168,26 @@ WINDOWS_MATRIX = {
"2.8.3",
],
"python-version": [
# "3.10",
# "3.11",
"3.10",
"3.11",
"3.12",
# "3.13",
"3.13",
"3.14",
],
"torch-version": [
# "2.5.1",
# "2.6.0",
# "2.7.1",
# "2.8.0",
"2.9.1",
# "2.10.0",
# "2.9.1",
"2.10.0",
],
"cuda-version": [
# "12.4",
# "12.6",
"12.6",
"12.8",
# "12.9",
# "13.0",
"13.0",
],
}
@@ -248,8 +250,8 @@ def main():
"linux": False,
# "linux": LINUX_MATRIX,
#
"linux_arm64": False,
# "linux_arm64": LINUX_ARM64_MATRIX,
# "linux_arm64": False,
"linux_arm64": LINUX_ARM64_MATRIX,
#
# "linux_self_hosted": False,
"linux_self_hosted": LINUX_SELF_HOSTED_MATRIX,
@@ -257,11 +259,11 @@ def main():
"linux_arm64_self_hosted": False,
# "linux_arm64_self_hosted": LINUX_ARM64_SELF_HOSTED_MATRIX,
#
"windows": False,
# "windows": WINDOWS_MATRIX,
# "windows": False,
"windows": WINDOWS_MATRIX,
#
# "windows_self_hosted": False,
"windows_self_hosted": WINDOWS_SELF_HOSTED_MATRIX,
"windows_self_hosted": False,
# "windows_self_hosted": WINDOWS_SELF_HOSTED_MATRIX,
#
"windows_code_build": False,
# "windows_code_build": WINDOWS_CODEBUILD_MATRIX,