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

chore: add coverage matrix definitions

Browse Source 
Co-authored-by: mjun0812 <77187490+mjun0812@users.noreply.github.com>
This commit is contained in:
copilot-swe-agent[bot]
2026-02-01 05:09:32 +00:00
parent 2087780f2a
commit 0001aa1574
2 changed files with 79 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
coverage_matrix.py
+74
View File
@@ -0,0 +1,74 @@
"""Coverage matrix definitions for wheel availability checks."""
LINUX_MATRIX = {
"flash-attn-version": [
"2.6.3",
"2.7.4",
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
"3.13",
"3.14",
],
"torch-version": [
"2.5.1",
"2.6.0",
"2.7.1",
"2.8.0",
"2.9.1",
"2.10.0",
],
"cuda-version": [
"12.4",
"12.6",
"12.8",
"13.0",
],
}
LINUX_ARM64_MATRIX = {
"flash-attn-version": [
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
"3.13",
"3.14",
],
"torch-version": [
"2.9.1",
"2.10.0",
],
"cuda-version": [
"12.6",
"12.8",
"13.0",
],
}
WINDOWS_MATRIX = {
"flash-attn-version": [
"2.8.3",
],
"python-version": [
"3.10",
"3.11",
"3.12",
"3.13",
"3.14",
],
"torch-version": [
"2.9.1",
"2.10.0",
],
"cuda-version": [
"12.6",
"12.8",
"13.0",
],
}