Remove Tensorflow Dependency from XProf

PiperOrigin-RevId: 751518904

Author: Matt-Hurd

WORKSPACE

@@ -177,52 +177,57 @@ load("@io_bazel_rules_sass//:defs.bzl", "sass_repositories")
 sass_repositories()
 
 http_archive(
-    name = "org_tensorflow",
+    name = "xla",
     patch_args = ["-p1"],
     patches = [
-        "//third_party:tensorflow.patch",
-        "//third_party:tensorflow_add_grpc_cares_darwin_arm64_support.patch",
-        "//third_party:tensorflow_rules_closure.patch",
+        "//third_party:xla.patch",
     ],
-    strip_prefix = "tensorflow-2495c486e8711a0e69e8e232a1e9f25c6d78f152",
+    sha256 = "722ef9af3e9084c3c6c549162734cc2343434a27323a8430478a04e6aa7df8e9",
+    strip_prefix = "xla-d1887c9bf970d864ceba39472dc0f3aec2be3612",
     urls = [
-        "https://github.com/tensorflow/tensorflow/archive/2495c486e8711a0e69e8e232a1e9f25c6d78f152.zip",
+        "https://github.com/openxla/xla/archive/d1887c9bf970d864ceba39472dc0f3aec2be3612.zip",
     ],
 )
 
-load(
-    "@org_tensorflow//tensorflow/tools/toolchains/python:python_repo.bzl",
-    "python_repository",
+http_archive(
+    name = "tsl",
+    sha256 = "8cf1e1285c7b1843a7f5f787465c1ef80304b3400ed837870bc76d74ce04f5af",
+    strip_prefix = "tsl-d71df2f7612583617d359c36243695097dd63726",
+    urls = [
+        "https://github.com/google/tsl/archive/d71df2f7612583617d359c36243695097dd63726.zip",
+    ],
 )
 
+load("@xla//tools/toolchains/python:python_repo.bzl", "python_repository")
+
 python_repository(name = "python_version_repo")
 
-# Initialize TensorFlow's external dependencies.
-load("@org_tensorflow//tensorflow:workspace3.bzl", "tf_workspace3")
+# Initialize XLA's external dependencies.
+load("@xla//:workspace3.bzl", "xla_workspace3")
 
-tf_workspace3()
+xla_workspace3()
 
-load("@org_tensorflow//tensorflow:workspace2.bzl", "tf_workspace2")
+load("@xla//:workspace2.bzl", "xla_workspace2")
 
-tf_workspace2()
+xla_workspace2()
 
-load("@org_tensorflow//tensorflow:workspace1.bzl", "tf_workspace1")
+load("@xla//:workspace1.bzl", "xla_workspace1")
 
-tf_workspace1()
+xla_workspace1()
 
-load("@org_tensorflow//tensorflow:workspace0.bzl", "tf_workspace0")
+load("@xla//:workspace0.bzl", "xla_workspace0")
 
-tf_workspace0()
+xla_workspace0()
 
 load(
-    "@local_xla//third_party/py:python_wheel.bzl",
+    "@xla//third_party/py:python_wheel.bzl",
     "python_wheel_version_suffix_repository",
 )
 
 python_wheel_version_suffix_repository(name = "tf_wheel_version_suffix")
 
 load(
-    "@local_xla//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl",
+    "@xla//third_party/gpus/cuda/hermetic:cuda_json_init_repository.bzl",
     "cuda_json_init_repository",
 )
 
@@ -234,7 +239,7 @@ load(
     "CUDNN_REDISTRIBUTIONS",
 )
 load(
-    "@local_xla//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl",
+    "@xla//third_party/gpus/cuda/hermetic:cuda_redist_init_repositories.bzl",
     "cuda_redist_init_repositories",
     "cudnn_redist_init_repository",
 )
@@ -248,21 +253,21 @@ cudnn_redist_init_repository(
 )
 
 load(
-    "@local_xla//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
+    "@xla//third_party/gpus/cuda/hermetic:cuda_configure.bzl",
     "cuda_configure",
 )
 
 cuda_configure(name = "local_config_cuda")
 
 load(
-    "@local_xla//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
+    "@xla//third_party/nccl/hermetic:nccl_redist_init_repository.bzl",
     "nccl_redist_init_repository",
 )
 
 nccl_redist_init_repository()
 
 load(
-    "@local_xla//third_party/nccl/hermetic:nccl_configure.bzl",
+    "@xla//third_party/nccl/hermetic:nccl_configure.bzl",
     "nccl_configure",
 )
 

plugin/tensorboard_plugin_profile/protobuf/BUILD

@@ -1,7 +1,8 @@
 # Description:
 #  Protobufs used by TensorBoard profile plugin.
 
-load("@local_xla//xla/tsl/platform:build_config.bzl", xprof_proto_library = "tf_proto_library")
+load("@com_google_protobuf//:protobuf.bzl", "py_proto_library")
+load("@xla//xla/tsl/platform:build_config.bzl", xprof_proto_library = "tf_proto_library")
 
 package(
     default_visibility = ["//visibility:public"],

third_party/tensorflow.patch

@@ -0,0 +1,69 @@
+diff --git a/tsl_workspace1.bzl b/tsl_workspace1.bzl
+index aead122980..66b8eef868 100644
+--- a/tsl_workspace1.bzl
++++ b/tsl_workspace1.bzl
+@@ -3,7 +3,7 @@
+ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+ load("@com_github_grpc_grpc//bazel:grpc_deps.bzl", "grpc_deps")
+ load("@com_github_nelhage_rules_boost//:boost/boost.bzl", "boost_deps")
+-load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
++load("@io_bazel_rules_closure//closure:repositories.bzl", "rules_closure_dependencies")
+ load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
+
+ # buildifier: disable=unnamed-macro
+@@ -16,7 +16,7 @@ def workspace(with_rules_cc = True):
+     native.register_toolchains("@local_config_python//:py_toolchain")
+     rules_pkg_dependencies()
+
+-    closure_repositories()
++    rules_closure_dependencies()
+
+     boost_deps()
+
+diff --git a/workspace1.bzl b/workspace1.bzl
+index b4c4461958..b73cfcd5e1 100644
+--- a/workspace1.bzl
++++ b/workspace1.bzl
+@@ -3,7 +3,7 @@
+ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
+ load("@com_github_grpc_grpc//bazel:grpc_deps.bzl", "grpc_deps")
+ load("@com_github_nelhage_rules_boost//:boost/boost.bzl", "boost_deps")
+-load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
++load("@io_bazel_rules_closure//closure:repositories.bzl", "rules_closure_dependencies")
+ load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
+ load("//:tsl_workspace1.bzl", "tsl_workspace1")
+
+@@ -15,7 +15,7 @@ def workspace():
+     native.register_toolchains("@local_config_python//:py_toolchain")
+     rules_pkg_dependencies()
+
+-    closure_repositories()
++    rules_closure_dependencies()
+
+     boost_deps()
+
+diff --git a/tools/toolchains/python/python_repo.bzl b/tools/toolchains/python/python_repo.bzl
+index 47fe64d7b7b..a01a1f19c8b 100644
+--- a/tools/toolchains/python/python_repo.bzl
++++ b/tools/toolchains/python/python_repo.bzl
+@@ -21,6 +21,7 @@ TF_PYTHON_VERSION = "{}"
+ HERMETIC_PYTHON_VERSION = "{}"
+ WHEEL_NAME = "{}"
+ WHEEL_COLLAB = "{}"
++USE_PYWRAP_RULES = "False"
+ """
+
+ def _python_repository_impl(repository_ctx):
+diff --git a/xla/tsl/platform/default/build_config.bzl b/xla/tsl/platform/default/build_config.bzl
+index 196af9fb686..9302861c6aa 100644
+--- a/xla/tsl/platform/default/build_config.bzl
++++ b/xla/tsl/platform/default/build_config.bzl
+@@ -432,7 +432,7 @@ def py_proto_library(
+
+     genproto_deps = []
+     for dep in deps:
+-        if dep != "@com_google_protobuf//:protobuf_python":
++        if dep != "@com_google_protobuf//:protobuf_python" and not dep.endswith(":protobuf_python"):
+             genproto_deps.append(dep + "_genproto")
+         else:
+             genproto_deps.append("@com_google_protobuf//:well_known_types_py_pb2_genproto")