From 932a4a3dcb370d8e2457caf875e2b2d6ee884194 Mon Sep 17 00:00:00 2001
From: Richard Stotz <richardstotz@google.com>
Date: Fri, 25 Oct 2024 00:28:30 -0700
Subject: [PATCH] Add Hermetic CUDA rules.

PiperOrigin-RevId: 689675919
---
 WORKSPACE | 47 +++++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/WORKSPACE b/WORKSPACE
index 5a292f9..0c06a8c 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -88,6 +88,53 @@ load("@org_tensorflow//tensorflow:workspace0.bzl", tf4 = "workspace")
 
 tf4()
 
+load(
+    "@local_tsl//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl",
+    "cuda_json_init_repository",
+)
+
+cuda_json_init_repository()
+
+load(
+    "@cuda_redist_json//:distributions.bzl",
+    "CUDA_REDISTRIBUTIONS",
+    "CUDNN_REDISTRIBUTIONS",
+)
+load(
+    "@local_tsl//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl",
+    "cuda_redist_init_repositories",
+    "cudnn_redist_init_repository",
+)
+
+cuda_redist_init_repositories(
+    cuda_redistributions = CUDA_REDISTRIBUTIONS,
+)
+
+cudnn_redist_init_repository(
+    cudnn_redistributions = CUDNN_REDISTRIBUTIONS,
+)
+
+load(
+    "@local_tsl//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
+    "cuda_configure",
+)
+
+cuda_configure(name = "local_config_cuda")
+
+load(
+    "@local_tsl//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
+    "nccl_redist_init_repository",
+)
+
+nccl_redist_init_repository()
+
+load(
+    "@local_tsl//third_party/nccl/hermetic:nccl_configure.bzl",
+    "nccl_configure",
+)
+
+nccl_configure(name = "local_config_nccl")
+
 # ========================================
 #  End of TensorFlow and its dependencies
 # ========================================