From f814eb69c3f848953a6dd1f1f798dfdd6d175674 Mon Sep 17 00:00:00 2001 From: Ryan Li Date: Mon, 24 Aug 2026 09:24:00 +0000 Subject: [PATCH] Add cuDNN 9.23.0-9.25.0 and NVSHMEM 3.7.0-3.7.2 version pins Extend the cuDNN and NVSHMEM version registries with every release NVIDIA has published since the current newest pins (cuDNN 9.22.0, NVSHMEM 3.6.5): cuDNN 9.23.0, 9.23.1, 9.23.2, 9.24.0, 9.25.0 and NVSHMEM 3.7.0, 3.7.1, 3.7.2. Each manifest sha256 was computed from the manifest downloaded from developer.download.nvidia.com. The existing BUILD templates work unchanged with the new versions: the cuDNN sub-library set indexed from LIB_VERSIONS (ops, cnn, adv, graph, engines_precompiled, engines_runtime_compiled, heuristic) and the static archives are all present in the 9.23-9.25 linux-x86_64 and linux-sbsa archives for both cuda12 and cuda13 variants, as are libnvshmem_host.so and libnvshmem_device.a in the 3.7.x archives. The new top-level manifest entries (cudnn_jit, cudnn_samples, nvshmem_python) are ignored by the component registries. Bump the e2e pins for CUDA 13.2.1 to cuDNN 9.24.0 / NVSHMEM 3.7.1 and for CUDA 13.3.1 to cuDNN 9.25.0 / NVSHMEM 3.7.2 so the new pins are exercised, keeping older pins on CUDA 13.0.2 / 13.1.1 for coverage of existing versions. --- cudnn/cudnn_redist_versions.json | 7 ++++++- e2e/MODULE.bazel | 8 ++++---- nvshmem/nvshmem_redist_versions.json | 5 ++++- 3 files changed, 14 insertions(+), 6 deletions(-) diff --git a/cudnn/cudnn_redist_versions.json b/cudnn/cudnn_redist_versions.json index f2a8dbb..97e6592 100644 --- a/cudnn/cudnn_redist_versions.json +++ b/cudnn/cudnn_redist_versions.json @@ -42,5 +42,10 @@ "9.20.0": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.20.0.json", "488be2657edfaaa9eada30247de95e81e3ce767f504dccb847d422cd358226a0"], "9.21.0": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.21.0.json", "946dfcb95f1d5552546190eb68e72c2090a9a3849a46378c202b3a74c55da505"], "9.21.1": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.21.1.json", "3c51347afefe8bf2a051a1a3160fee48c46182b019435e7805f3c1126d0ee330"], - "9.22.0": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.22.0.json", "3dbb9002d52112ef69aa09187f523ef1ff07f8baf3892ee01e540af639d8f55f"] + "9.22.0": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.22.0.json", "3dbb9002d52112ef69aa09187f523ef1ff07f8baf3892ee01e540af639d8f55f"], + "9.23.0": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.23.0.json", "c15dc41bda2de1a3972af7769d9e1ae020a3cfb3c7346ef50db87670218a822b"], + "9.23.1": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.23.1.json", "51bc73671ac7d5c6710d66897e2fac5304c47a61b07809579b962c2bbd86dfa8"], + "9.23.2": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.23.2.json", "265add37bdd5e87abd18fd336502032a10e068a542e0ee5a9b59920a67b109be"], + "9.24.0": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.24.0.json", "89a0626f1925d9f4ceb8d6bc0b2de160b30e92afaae5471e6cfbb50234ae5488"], + "9.25.0": ["https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.25.0.json", "2e9e457fa3d4be5e0a3a002000d812a97e9935b544661342078bf3b351b65ff5"] } diff --git a/e2e/MODULE.bazel b/e2e/MODULE.bazel index 7d9affc..c8cceb1 100644 --- a/e2e/MODULE.bazel +++ b/e2e/MODULE.bazel @@ -55,15 +55,15 @@ cuda_ext.redist( cuda_ext.redist( name = "cuda_13_2_1", version = "13.2.1", - cudnn_version = "9.20.0", - nvshmem_version = "3.5.19", + cudnn_version = "9.24.0", + nvshmem_version = "3.7.1", nccl_version = "2.30.7", ) cuda_ext.redist( name = "cuda_13_3_1", version = "13.3.1", - cudnn_version = "9.20.0", - nvshmem_version = "3.5.19", + cudnn_version = "9.25.0", + nvshmem_version = "3.7.2", nccl_version = "2.29.3", ) diff --git a/nvshmem/nvshmem_redist_versions.json b/nvshmem/nvshmem_redist_versions.json index 47dceb9..071b648 100644 --- a/nvshmem/nvshmem_redist_versions.json +++ b/nvshmem/nvshmem_redist_versions.json @@ -7,5 +7,8 @@ "3.4.5": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.4.5.json", "a656614a6ec638d85922bc816e5e26063308c3905273a72a863cf0f24e188f38"], "3.5.19": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.5.19.json", "6dced4193eb728542504b346cfb768da6e3de2abca0cded95fda3a69729994d2"], "3.5.21": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.5.21.json", "c9a9c22766d1cfc64548205a526d9aabbdba46756e0dae60332e89a36a818350"], - "3.6.5": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.6.5.json", "afbf1ad5c4174c25a66cadd5b925d6814e679e8f836ef1fee1442afb6cb7fdd3"] + "3.6.5": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.6.5.json", "afbf1ad5c4174c25a66cadd5b925d6814e679e8f836ef1fee1442afb6cb7fdd3"], + "3.7.0": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.7.0.json", "45747435c4347ed9be8abbf876a99a6f18ab58987c6a87ded2e0d16a6bc74ddb"], + "3.7.1": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.7.1.json", "1a8466b994789e863d82865a746b0bb6fd5993e60d0d0fdd18f385b643a07f2a"], + "3.7.2": ["https://developer.download.nvidia.com/compute/nvshmem/redist/redistrib_3.7.2.json", "6dabe2b96f51b64fcf644f1dc1db5f4c8a59f309e6dfc6821502eaa6ccd4272b"] }