Fix @local_config_cuda

PiperOrigin-RevId: 664903046
tensorflow · Aug 19, 2024 · 5dd3d00 · 5dd3d00
1 parent 748c121
commit 5dd3d00
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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")