diff --git a/create_matrix.py b/create_matrix.py index 38228d0..50bf729 100644 --- a/create_matrix.py +++ b/create_matrix.py @@ -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,