[GPU][NFC] Add CUDA 12.6.1 redistribution.

This closes openxla/xla#17003 PiperOrigin-RevId: 677042671
google · Sep 21, 2024 · c8a70a2 · c8a70a2
1 parent 8606779
commit c8a70a2
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl b/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl
@@ -58,6 +58,10 @@ CUDA_REDIST_JSON_DICT = {
         "https://developer.download.nvidia.com/compute/cuda/redist/redistrib_12.6.0.json",
         "87740b01676b3d18982982ab96ec7fa1a626d03a96df070a6b0f258d01ff5fab",
     ],
+    "12.6.1": [
+        "https://developer.download.nvidia.com/compute/cuda/redist/redistrib_12.6.1.json",
+        "22ddfeb81a6f9cee4a708a2e3b4db1c36c7db0a1daa1f33f9c7f2f12a1e790de",
+    ],
 }
 
 CUDNN_REDIST_JSON_DICT = {
@@ -134,6 +138,7 @@ CUDA_NCCL_WHEELS = {
     "12.5.0": CUDA_12_NCCL_WHEEL_DICT,
     "12.5.1": CUDA_12_NCCL_WHEEL_DICT,
     "12.6.0": CUDA_12_NCCL_WHEEL_DICT,
+    "12.6.1": CUDA_12_NCCL_WHEEL_DICT,
 }
 
 REDIST_VERSIONS_TO_BUILD_TEMPLATES = {