diff --git a/WORKSPACE b/WORKSPACE index 740bdfa7a..b5b8956b1 100644 --- a/WORKSPACE +++ b/WORKSPACE @@ -83,6 +83,10 @@ http_archive( build_file = "//third_party/pybind11:BUILD.bzl", ) +load("@local_tsl//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")