From 5dd3d00de34059b4372770638ef3eb0bfe7ba31c Mon Sep 17 00:00:00 2001
From: Raviteja Gorijala <gorijala@google.com>
Date: Mon, 19 Aug 2024 11:37:17 -0700
Subject: [PATCH] Fix @local_config_cuda

PiperOrigin-RevId: 664903046
---
 WORKSPACE | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/WORKSPACE b/WORKSPACE
index 740bdfa7a..5395727f5 100644
--- a/WORKSPACE
+++ b/WORKSPACE
@@ -83,6 +83,10 @@ http_archive(
     build_file = "//third_party/pybind11:BUILD.bzl",
 )
 
+load("@org_tensorflow//third_party/gpus/cuda/hermetic:cuda_configure.bzl", "cuda_configure")
+
+cuda_configure(name = "local_config_cuda")
+
 # Initialize hermetic Python
 load("@org_tensorflow//third_party/py:python_init_rules.bzl", "python_init_rules")