From 104caff5b0d7fb1b405e3cac7772c7c760286a96 Mon Sep 17 00:00:00 2001 From: Ilia Sergachev Date: Mon, 13 Jan 2025 03:32:13 -0800 Subject: [PATCH] PR #20948: [GPU] Fix NCCL with CUDA 12.6.3. Imported from GitHub PR https://github.com/openxla/xla/pull/20948 Copybara import of the project: -- bc566f263b10c2fffe14e0f8007f60983509e02e by Ilia Sergachev : [GPU] Fix NCCL with CUDA 12.6.3. Merging this change closes #20948 PiperOrigin-RevId: 714902545 --- third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl | 1 + 1 file changed, 1 insertion(+) diff --git a/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl b/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl index 721fcdabc..e4049a408 100644 --- a/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl +++ b/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl @@ -158,6 +158,7 @@ CUDA_NCCL_WHEELS = { "12.6.0": CUDA_12_NCCL_WHEEL_DICT, "12.6.1": CUDA_12_NCCL_WHEEL_DICT, "12.6.2": CUDA_12_NCCL_WHEEL_DICT, + "12.6.3": CUDA_12_NCCL_WHEEL_DICT, } REDIST_VERSIONS_TO_BUILD_TEMPLATES = {