Minimize number of Copybara transforms that operate on `tensorflow/th…

…ird_party` PiperOrigin-RevId: 621679504
openxla · Apr 4, 2024 · 55cdde9 · 55cdde9
1 parent baf7544
commit 55cdde9
Show file tree

Hide file tree

Showing 8 changed files with 17 additions and 17 deletions.
diff --git a/third_party/compute_library/build_defs.bzl b/third_party/compute_library/build_defs.bzl
@@ -1,6 +1,6 @@
 def if_enable_acl(if_true, if_false = []):
     return select({
-        "@xla//third_party/compute_library:build_with_acl": if_true,
+        "@tsl//third_party/compute_library:build_with_acl": if_true,
         "//conditions:default": if_false,
     })
 
@@ -15,6 +15,6 @@ def acl_deps():
       inclusion in the deps attribute of rules.
     """
     return select({
-        "@xla//third_party/compute_library:build_with_acl": ["@compute_library//:arm_compute"],
+        "@tsl//third_party/compute_library:build_with_acl": ["@compute_library//:arm_compute"],
         "//conditions:default": [],
     })
diff --git a/third_party/llvm_openmp/BUILD b/third_party/llvm_openmp/BUILD
@@ -2,21 +2,21 @@
 
 load("@bazel_skylib//:bzl_library.bzl", "bzl_library")
 load(
-    "@tsl//tsl:tsl.bzl",
-    "if_linux_x86_64",
-    "if_macos",
-    "if_windows",
-)
-load(
-    "@xla//third_party/llvm_openmp:cmake_vars.bzl",
+    "@tsl//third_party/llvm_openmp:cmake_vars.bzl",
     "cmake_var_string",
     "expand_cmake_vars",
 )
 load(
-    "@xla//third_party/llvm_openmp:openmp.bzl",
+    "@tsl//third_party/llvm_openmp:openmp.bzl",
     "dict_add",
     "libiomp5_cc_binary",
 )
+load(
+    "@tsl//tsl:tsl.bzl",
+    "if_linux_x86_64",
+    "if_macos",
+    "if_windows",
+)
 
 package(
     default_visibility = [

diff --git a/third_party/llvm_openmp/cmake_vars.bzl b/third_party/llvm_openmp/cmake_vars.bzl
@@ -46,7 +46,7 @@ def expand_cmake_vars(name, src, dst, cmake_vars):
       cmake_vars: a string containing the CMake variables, as generated by
         cmake_var_string.
     """
-    expand_cmake_vars_tool = "@xla//third_party/llvm_openmp:expand_cmake_vars"
+    expand_cmake_vars_tool = "@tsl//third_party/llvm_openmp:expand_cmake_vars"
     native.genrule(
         name = name,
         srcs = [src],

diff --git a/third_party/nanobind/nanobind.BUILD b/third_party/nanobind/nanobind.BUILD
@@ -21,6 +21,6 @@ cc_library(
     ),
     deps = [
         "@robin_map",
-        "@xla//third_party/python_runtime:headers",
+        "@tsl//third_party/python_runtime:headers",
     ],
 )
diff --git a/third_party/py/ml_dtypes/ml_dtypes.BUILD b/third_party/py/ml_dtypes/ml_dtypes.BUILD
@@ -49,7 +49,7 @@ pybind_extension(
         ":float8",
         ":int4",
         "@eigen_archive//:eigen3",
-        "@xla//third_party/py/numpy:headers",
+        "@tsl//third_party/py/numpy:headers",
     ],
 )
 

diff --git a/third_party/py/ml_dtypes/ml_dtypes.tests.BUILD b/third_party/py/ml_dtypes/ml_dtypes.tests.BUILD
@@ -8,7 +8,7 @@ py_library(
         "//:ml_dtypes",
         "@absl_py//absl/testing:absltest",
         "@absl_py//absl/testing:parameterized",
-        "@xla//third_party/py/numpy",
+        "@tsl//third_party/py/numpy",
     ],
 )
 

diff --git a/third_party/tsl/third_party/git/git_configure.bzl b/third_party/tsl/third_party/git/git_configure.bzl
@@ -38,10 +38,10 @@ def _git_conf_impl(repository_ctx):
     )
 
     tensorflow_root_path = str(repository_ctx.path(
-        Label("@tsl//:BUILD"),
+        Label("@org_tensorflow//:BUILD"),
     ))[:-len("BUILD")]
     python_script_path = repository_ctx.path(
-        Label("@tsl//tensorflow/tools/git:gen_git_source.py"),
+        Label("@org_tensorflow//tensorflow/tools/git:gen_git_source.py"),
     )
     generated_files_path = repository_ctx.path("gen")
 

diff --git a/third_party/tsl/third_party/llvm_openmp/openmp.bzl b/third_party/tsl/third_party/llvm_openmp/openmp.bzl
@@ -4,7 +4,7 @@ after the TF 2.4 branch cut has passed.
 """
 
 load(
-    "//tsl/platform:rules_cc.bzl",
+    "@tsl//tsl/platform:rules_cc.bzl",
     "cc_binary",
 )
-Original file line number
+Diff line change
@@ Expand Up / @@ -4,7 +4,7 @@ after the TF 2.4 branch cut has passed. @@
     """
     load(
-        "//tsl/platform:rules_cc.bzl",
+        "@tsl//tsl/platform:rules_cc.bzl",
         "cc_binary",
     )
@@ Expand Down @@