diff --git a/create_matrix.py b/create_matrix.py index 446c225..bca4ea0 100644 --- a/create_matrix.py +++ b/create_matrix.py @@ -91,7 +91,7 @@ LINUX_ARM64_MATRIX = { # "2.6.0", # "2.7.1", # "2.8.0", - # "2.9.1", + "2.9.1", "2.10.0", ], "cuda-version": [ @@ -118,19 +118,18 @@ LINUX_SELF_HOSTED_MATRIX = { ], "torch-version": [ # "2.5.1", - # "2.6.0", - # "2.7.1", - # "2.8.0", - # "2.9.1", + "2.6.0", + "2.7.1", + "2.8.0", + "2.9.1", "2.10.0", ], "cuda-version": [ - # "12.4", + "12.4", "12.6", - # "12.8", + "12.8", # "12.9", - # "13.0", - # "13.1", + "13.0", ], } @@ -254,8 +253,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,