Update WORKSPACE

PiperOrigin-RevId: 682562648
google-ai-edge · Oct 5, 2024 · cdcd675 · cdcd675
1 parent b33a61c
commit cdcd675
Showing 1 changed file with 49 additions and 10 deletions.
diff --git a/src/builtin-adapter/WORKSPACE b/src/builtin-adapter/WORKSPACE
@@ -28,9 +28,9 @@ http_archive(
     ],
 )
 
-TENSORFLOW_COMMIT = "9e0f7896da725cdfb04d5d2160433b4a699e3362"
+TENSORFLOW_COMMIT = "32f376e44ba50279a98bf131108782f408d0a496"
 
-TENSORFLOW_SHA256 = "f147c9d84d75aad0fb276eca96d588e768be251c4a4723d61b06297fa2d61b5d"
+TENSORFLOW_SHA256 = "14c2e18223107a312b4a801ddcbdcd6fb9dca1d9ebed59149f2830334c3ec7e2"
 
 http_archive(
     name = "org_tensorflow",
@@ -42,14 +42,6 @@ http_archive(
     ],
 )
 
-# ===== protobuf =====
-http_archive(
-    name = "com_google_protobuf",
-    sha256 = "a0b6b02197b4df8d55492841b605295e187a3c45dad7ec187a668d396e48bb75",
-    strip_prefix = "protobuf-28.2",
-    url = "https://github.com/protocolbuffers/protobuf/archive/9fff46d7327c699ef970769d5c9fd0e44df08fc7.tar.gz",
-)
-
 http_archive(
     name = "rules_python",
     sha256 = "9d04041ac92a0985e344235f5d946f71ac543f1b1565f2cdbc9a2aaee8adf55b",
@@ -121,3 +113,50 @@ tf_workspace1()
 load("@org_tensorflow//tensorflow:workspace0.bzl", "tf_workspace0")
 
 tf_workspace0()
+
+load(
+    "@local_tsl//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl",
+    "cuda_json_init_repository",
+)
+
+cuda_json_init_repository()
+
+load(
+    "@cuda_redist_json//:distributions.bzl",
+    "CUDA_REDISTRIBUTIONS",
+    "CUDNN_REDISTRIBUTIONS",
+)
+load(
+    "@local_tsl//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl",
+    "cuda_redist_init_repositories",
+    "cudnn_redist_init_repository",
+)
+
+cuda_redist_init_repositories(
+    cuda_redistributions = CUDA_REDISTRIBUTIONS,
+)
+
+cudnn_redist_init_repository(
+    cudnn_redistributions = CUDNN_REDISTRIBUTIONS,
+)
+
+load(
+    "@local_tsl//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
+    "cuda_configure",
+)
+
+cuda_configure(name = "local_config_cuda")
+
+load(
+    "@local_tsl//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
+    "nccl_redist_init_repository",
+)
+
+nccl_redist_init_repository()
+
+load(
+    "@local_tsl//third_party/nccl/hermetic:nccl_configure.bzl",
+    "nccl_configure",
+)
+
+nccl_configure(name = "local_config_nccl")