Skip to content

Commit

Permalink
PR #20948: [GPU] Fix NCCL with CUDA 12.6.3.
Browse files Browse the repository at this point in the history
Imported from GitHub PR openxla/xla#20948

Copybara import of the project:

--
bc566f263b10c2fffe14e0f8007f60983509e02e by Ilia Sergachev <[email protected]>:

[GPU] Fix NCCL with CUDA 12.6.3.

Merging this change closes #20948

PiperOrigin-RevId: 714902545
  • Loading branch information
sergachev authored and copybara-github committed Jan 13, 2025
1 parent 0f6571f commit 104caff
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl
Original file line number Diff line number Diff line change
Expand Up @@ -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 = {
Expand Down

0 comments on commit 104caff

Please sign in to comment.