From 81194d8204aa0091f4adaf437e32b682923ab809 Mon Sep 17 00:00:00 2001 From: Ilia Sergachev Date: Wed, 12 Feb 2025 04:00:15 -0800 Subject: [PATCH] PR #22537: [GPU] Add cuDNN 9.7.1. Imported from GitHub PR https://github.com/openxla/xla/pull/22537 Copybara import of the project: -- cdecd418a3159a798c7f2c6ac1e17a8f8ce6d862 by Ilia Sergachev : [GPU] Add cuDNN 9.7.1. Merging this change closes #22537 COPYBARA_INTEGRATE_REVIEW=https://github.com/openxla/xla/pull/22537 from openxla:cudnn_971 cdecd418a3159a798c7f2c6ac1e17a8f8ce6d862 PiperOrigin-RevId: 725993273 --- .../third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl b/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl index d373a4a21a2d1..2c1c5d68cacb1 100644 --- a/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl +++ b/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl @@ -129,6 +129,10 @@ CUDNN_REDIST_JSON_DICT = { "https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.7.0.json", "e715c1d028585d228c4678c2cdc5ad9a34fde54515a1c52aa60e36021a90dd90", ], + "9.7.1": [ + "https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.7.1.json", + "f9bc411a4908f0931e7323f89049e3a38453632c4ac5f4aa3220af69ddded9dc", + ], } CUDA_12_NCCL_WHEEL_DICT = {