From 6710e1843ca33d8b58a1b8951054c54d86a7fe8f Mon Sep 17 00:00:00 2001 From: Ilia Sergachev Date: Tue, 24 Sep 2024 12:00:11 -0700 Subject: [PATCH] [GPU] Fix compilation with NVIDIA driver 560. This closes https://github.com/openxla/xla/pull/17549 PiperOrigin-RevId: 678341835 --- .../tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl | 1 + 1 file changed, 1 insertion(+) 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 417a237053c8e..140879d7e271e 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 @@ -145,6 +145,7 @@ REDIST_VERSIONS_TO_BUILD_TEMPLATES = { "nvidia_driver": { "repo_name": "cuda_driver", "version_to_template": { + "560": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl", "555": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl", "550": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl", "545": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl",