Fix @local_config_cuda

PiperOrigin-RevId: 664903046
tensorflow · Aug 19, 2024 · 6941d47 · 6941d47
1 parent 748c121
commit 6941d47
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/WORKSPACE b/WORKSPACE
@@ -56,10 +56,10 @@ http_archive(
 
 http_archive(
     name = "org_tensorflow",
-    strip_prefix = "tensorflow-9d2929b69fdddc1e8d5de70a706556b15d8ed37b",
-    sha256 = "a4f965340ea11d49c8897df59a24822c2b367e4cebb5908b7ca8c3b607097af9",
+    strip_prefix = "tensorflow-bf48f263a2d1e7298627ac5f6f3c9e238f99540c",
+    sha256 = "28c759cbf5e54ed39bbbb9d1e3f8d73b25142f62ba97e6b6067a22f566aa515f",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/9d2929b69fdddc1e8d5de70a706556b15d8ed37b.zip"
+        "https://github.com/tensorflow/tensorflow/archive/bf48f263a2d1e7298627ac5f6f3c9e238f99540c.zip"
     ],
 )
 
@@ -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")