[cuda] Add support for CUDA 12.4 in test channel

- Add support for CUDA 12.4 in `test` channel
pytorch · Jun 12, 2024 · 1ca62cf · 1ca62cf
1 parent 85f3293
commit 1ca62cf
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tools/scripts/generate_binary_build_matrix.py b/tools/scripts/generate_binary_build_matrix.py
@@ -25,8 +25,8 @@
 }
 CUDA_ARCHES_DICT = {
     "nightly": ["11.8", "12.1", "12.4"],
-    "test": ["11.8", "12.1"],
-    "release": ["11.8", "12.1"],
+    "test": ["11.8", "12.1", "12.4"],
+    "release": ["11.8", "12.1", "12.4"],
 }
 ROCM_ARCHES_DICT = {
     "nightly": ["6.0", "6.1"],