From bb8785670431b856d6d4168e1e6b1abd2b95621f Mon Sep 17 00:00:00 2001 From: Christian Sigg Date: Wed, 12 Jun 2024 10:41:31 -0700 Subject: [PATCH] [tsl:concurrency] NFC: Remove KeepAsyncValuePayloadOnError alias PiperOrigin-RevId: 642662861 --- cpp_tests/host_context/async_value_test.cc | 2 +- include/tfrt/host_context/async_value.h | 3 ++- 2 files changed, 3 insertions(+), 2 deletions(-) diff --git a/cpp_tests/host_context/async_value_test.cc b/cpp_tests/host_context/async_value_test.cc index 51f7aaf12d8..a837cea4c0c 100644 --- a/cpp_tests/host_context/async_value_test.cc +++ b/cpp_tests/host_context/async_value_test.cc @@ -104,7 +104,7 @@ TEST(AsyncValueTest, AddAndDropRef) { TEST(AsyncValueTest, KeepPayloadOnError) { int payload_value = 0; - struct Payload : tsl::AsyncPayload::KeepOnError { + struct Payload : internal::KeepAsyncValuePayloadOnError { explicit Payload(int* value) : value{value} { *value = 1; } ~Payload() { *value = 2; } diff --git a/include/tfrt/host_context/async_value.h b/include/tfrt/host_context/async_value.h index 24e19b44d4e..b03f145a404 100644 --- a/include/tfrt/host_context/async_value.h +++ b/include/tfrt/host_context/async_value.h @@ -32,7 +32,8 @@ namespace tfrt { using ::tsl::AsyncValue; // NOLINT namespace internal { -using ::tsl::internal::ConcreteAsyncValue; // NOLINT +using ::tsl::internal::ConcreteAsyncValue; // NOLINT +using ::tsl::internal::KeepAsyncValuePayloadOnError; // NOLINT } // namespace internal using ::tsl::DummyValueForErrorAsyncValue; // NOLINT