From f14af49bfde6926709d37c75c370e3f56cf04516 Mon Sep 17 00:00:00 2001
From: David Dunleavy <ddunleavy@google.com>
Date: Mon, 15 Apr 2024 15:08:02 -0700
Subject: [PATCH] Move `tsl/concurrency` to `xla/tsl/concurrency` and update
 users

PiperOrigin-RevId: 625096733
---
 BUILD                                        | 6 +++---
 include/tfrt/concurrency/async_value.h       | 2 +-
 include/tfrt/concurrency/async_value_ref.h   | 2 +-
 include/tfrt/concurrency/chain.h             | 2 +-
 include/tfrt/concurrency/concurrent_vector.h | 2 +-
 include/tfrt/concurrency/ref_count.h         | 2 +-
 6 files changed, 8 insertions(+), 8 deletions(-)

diff --git a/BUILD b/BUILD
index 43d7bef5c04..9108d5d712c 100644
--- a/BUILD
+++ b/BUILD
@@ -212,7 +212,7 @@ tfrt_cc_library(
     visibility = ["//visibility:public"],
     deps = [
         ":support",
-        "@local_tsl//tsl/concurrency:async_value",
+        "//third_party/tensorflow/compiler/xla/tsl/concurrency:async_value",
     ],
 )
 
@@ -221,7 +221,7 @@ tfrt_cc_library(
     hdrs = ["include/tfrt/concurrency/concurrent_vector.h"],
     # copybara:uncomment compatible_with = ["//buildenv/target:non_prod"],
     visibility = ["//visibility:public"],
-    deps = ["@local_tsl//tsl/concurrency:concurrent_vector"],
+    deps = ["//third_party/tensorflow/compiler/xla/tsl/concurrency:concurrent_vector"],
 )
 
 tfrt_cc_library(
@@ -229,7 +229,7 @@ tfrt_cc_library(
     hdrs = ["include/tfrt/concurrency/ref_count.h"],
     # copybara:uncomment compatible_with = ["//buildenv/target:non_prod"],
     visibility = ["//visibility:public"],
-    deps = ["@local_tsl//tsl/concurrency:ref_count"],
+    deps = ["//third_party/tensorflow/compiler/xla/tsl/concurrency:ref_count"],
 )
 
 tfrt_cc_library(
diff --git a/include/tfrt/concurrency/async_value.h b/include/tfrt/concurrency/async_value.h
index 090ccc0b3db..6e6d5c9264f 100644
--- a/include/tfrt/concurrency/async_value.h
+++ b/include/tfrt/concurrency/async_value.h
@@ -16,7 +16,7 @@ limitations under the License.
 #ifndef TFRT_CONCURRENCY_ASYNC_VALUE_H_
 #define TFRT_CONCURRENCY_ASYNC_VALUE_H_
 
-#include "tsl/concurrency/async_value.h"  // from @local_tsl  // IWYU pragma: export
+#include "compiler/xla/tsl/concurrency/async_value.h"  // from @local_tsl  // IWYU pragma: export
 
 // TODO(ezhulenev): We have targets that depend on transitive includes. This
 // should be removed once we finally move to TSL concurrency package.
diff --git a/include/tfrt/concurrency/async_value_ref.h b/include/tfrt/concurrency/async_value_ref.h
index a35194faf92..e67e2472cd1 100644
--- a/include/tfrt/concurrency/async_value_ref.h
+++ b/include/tfrt/concurrency/async_value_ref.h
@@ -16,6 +16,6 @@ limitations under the License.
 #ifndef TFRT_CONCURRENCY_ASYNC_VALUE_REF_H_
 #define TFRT_CONCURRENCY_ASYNC_VALUE_REF_H_
 
-#include "tsl/concurrency/async_value_ref.h"  // from @local_tsl  // IWYU pragma: export
+#include "compiler/xla/tsl/concurrency/async_value_ref.h"  // from @local_tsl  // IWYU pragma: export
 
 #endif  // TFRT_CONCURRENCY_ASYNC_VALUE_REF_H_
diff --git a/include/tfrt/concurrency/chain.h b/include/tfrt/concurrency/chain.h
index 4bb5ad5504e..c0c5c95d060 100644
--- a/include/tfrt/concurrency/chain.h
+++ b/include/tfrt/concurrency/chain.h
@@ -16,6 +16,6 @@ limitations under the License.
 #ifndef TFRT_CONCURRENCY_CHAIN_H_
 #define TFRT_CONCURRENCY_CHAIN_H_
 
-#include "tsl/concurrency/chain.h"  // from @local_tsl  // IWYU pragma: export
+#include "compiler/xla/tsl/concurrency/chain.h"  // from @local_tsl  // IWYU pragma: export
 
 #endif  // TFRT_CONCURRENCY_CHAIN_H_
diff --git a/include/tfrt/concurrency/concurrent_vector.h b/include/tfrt/concurrency/concurrent_vector.h
index fa4a575f220..0b239570946 100644
--- a/include/tfrt/concurrency/concurrent_vector.h
+++ b/include/tfrt/concurrency/concurrent_vector.h
@@ -16,6 +16,6 @@ limitations under the License.
 #ifndef TFRT_CONCURRENCY_CONCURRENT_VECTOR_H_
 #define TFRT_CONCURRENCY_CONCURRENT_VECTOR_H_
 
-#include "tsl/concurrency/concurrent_vector.h"  // from @local_tsl  // IWYU pragma: export
+#include "compiler/xla/tsl/concurrency/concurrent_vector.h"  // from @local_tsl  // IWYU pragma: export
 
 #endif  // TFRT_CONCURRENCY_CONCURRENT_VECTOR_H_
diff --git a/include/tfrt/concurrency/ref_count.h b/include/tfrt/concurrency/ref_count.h
index f6d5812efa9..ecf109afce5 100644
--- a/include/tfrt/concurrency/ref_count.h
+++ b/include/tfrt/concurrency/ref_count.h
@@ -16,6 +16,6 @@ limitations under the License.
 #ifndef TFRT_CONCURRENCY_REF_COUNT_H_
 #define TFRT_CONCURRENCY_REF_COUNT_H_
 
-#include "tsl/concurrency/ref_count.h"  // from @local_tsl  // IWYU pragma: export
+#include "compiler/xla/tsl/concurrency/ref_count.h"  // from @local_tsl  // IWYU pragma: export
 
 #endif  // TFRT_CONCURRENCY_REF_COUNT_H_