Move tsl/concurrency to xla/tsl/concurrency and update users

PiperOrigin-RevId: 625096733
tensorflow · Apr 15, 2024 · f14af49 · f14af49
1 parent 02562bf
commit f14af49
Show file tree

Hide file tree

Showing 6 changed files with 8 additions and 8 deletions.
diff --git 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,15 +221,15 @@ 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(
     name = "ref_count",
     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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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_