[cuda] Add support for CUDA 12.4 in test channel

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