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,