Automated Code Change

PiperOrigin-RevId: 609315877
tensorflow · Feb 22, 2024 · ee56b5f · ee56b5f
1 parent b51db3c
commit ee56b5f
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/third_party/xla/xla/ffi/ffi_api.cc b/third_party/xla/xla/ffi/ffi_api.cc
@@ -108,8 +108,8 @@ static Status RegisterHandler(std::string_view name, std::string_view platform,
   return OkStatus();
 }
 
-StatusOr<XLA_FFI_Handler*> FindHandler(std::string_view name,
-                                       std::string_view platform) {
+absl::StatusOr<XLA_FFI_Handler*> FindHandler(std::string_view name,
+                                             std::string_view platform) {
   auto it = GetHandlerRegistry().find(MakeHandlerKey(name, platform));
   if (it == GetHandlerRegistry().end())
     return absl::NotFoundError(absl::StrCat("No FFI handler registered for ",

diff --git a/third_party/xla/xla/ffi/ffi_api.h b/third_party/xla/xla/ffi/ffi_api.h
@@ -62,8 +62,8 @@ Status Call(XLA_FFI_Handler* handler, CallFrame& call_frame,
 
 // Returns registered FFI handler for a given name and platform, or an error if
 // it's not found in the static registry.
-StatusOr<XLA_FFI_Handler*> FindHandler(std::string_view name,
-                                       std::string_view platform);
+absl::StatusOr<XLA_FFI_Handler*> FindHandler(std::string_view name,
+                                             std::string_view platform);
 
 //===----------------------------------------------------------------------===//
 // XLA FFI Api Implementation