Add support for cuda driver versions 520 and 530 in hermetic cuda

PiperOrigin-RevId: 717060115
openxla · Jan 18, 2025 · b21aaff · b21aaff
1 parent e141532
commit b21aaff
Showing 1 changed file with 2 additions and 0 deletions.
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
@@ -169,6 +169,8 @@ REDIST_VERSIONS_TO_BUILD_TEMPLATES = {
             "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",
+            "530": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl",
+            "520": "//third_party/gpus/cuda/hermetic:cuda_driver.BUILD.tpl",
         },
     },
     "cuda_nccl": {