Fix @local_config_cuda

PiperOrigin-RevId: 664903046
tensorflow · Aug 19, 2024 · e66d8e0 · e66d8e0
1 parent 748c121
commit e66d8e0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -114,6 +114,10 @@ python_init_repositories(
     default_python_version = "system",
 )
 
+load("@tsl//third_party/gpus/cuda/hermetic:cuda_configure.bzl", "cuda_configure")
+
+cuda_configure(name = "local_config_cuda")
+
 load("@org_tensorflow//third_party/py:python_init_toolchains.bzl", "python_init_toolchains")
 
 python_init_toolchains()