From e1a4c88fac8cb771af4f321a56c3052ac0d3c8ed Mon Sep 17 00:00:00 2001 From: Junya Morioka Date: Tue, 4 Nov 2025 20:19:53 +0900 Subject: [PATCH] feat: add flash_attn 2.7 and PyTorch 2.9 --- create_matrix.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) diff --git a/create_matrix.py b/create_matrix.py index fbc741c..29f93b8 100644 --- a/create_matrix.py +++ b/create_matrix.py @@ -35,10 +35,10 @@ LINUX_MATRIX = { } LINUX_SELF_HOSTED_MATRIX = { - "flash-attn-version": ["2.7.4.post1", "2.8.3"], + "flash-attn-version": ["2.7.4"], "python-version": ["3.10", "3.11", "3.12", "3.13"], "torch-version": ["2.9.0"], - "cuda-version": ["13.0.1"], + "cuda-version": ["12.8.1", "13.0.1"], } WINDOWS_MATRIX = { @@ -62,10 +62,10 @@ def main(): { # "linux": LINUX_MATRIX, "linux": False, - # "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX, - "linux_self_hosted": False, - "windows": WINDOWS_MATRIX, - # "windows": False, + "linux_self_hosted": LINUX_SELF_HOSTED_MATRIX, + # "linux_self_hosted": False, + # "windows": WINDOWS_MATRIX, + "windows": False, # "windows_code_build": WINDOWS_CODEBUILD_MATRIX, "windows_code_build": False, "exclude": EXCLUDE,