[XLA:GPU] Migrate from tsl::errors to absl::Status error creation functions.

This change replaces calls to `tsl::errors::FailedPrecondition`, `tsl::errors::Unavailable`, `tsl::errors::Internal`, and `tsl::errors::Aborted` with their equivalent `absl::FailedPreconditionError`, `absl::UnavailableError`, `absl::InternalError`, and `absl::AbortedError` functions.

PiperOrigin-RevId: 839243361

Author: loislo

xla/hlo/evaluator/hlo_evaluator.cc

@@ -3675,7 +3675,7 @@ absl::StatusOr<Literal> TryParseAndEvaluateWhileInductionVar(
   std::optional<ParsedWhileLoop> parsed_while_loop =
       PatternMatchParseWhileLoop(while_hlo, /*precomputed_analyses=*/{});
   if (!parsed_while_loop.has_value() || parsed_while_loop->is_dynamic()) {
-    return FailedPrecondition(
+    return absl::FailedPreconditionError(
         "Cannot evaluate a while loop's induction variable since the loop "
         "does not match a known loop pattern or the loop is not static.");
   }

xla/tsl/platform/cloud/BUILD

@@ -357,6 +357,7 @@ cc_library(
         "//xla/tsl/platform:errors",
         "//xla/tsl/platform:status",
         "@boringssl//:crypto",
+        "@com_google_absl//absl/status",
         "@jsoncpp_git//:jsoncpp",
         "@tsl//tsl/platform:base64",
     ],

xla/tsl/platform/cloud/gcs_file_system_test.cc

@@ -181,7 +181,7 @@ TEST(GcsFileSystemTest, NewRandomAccessFile_Buffered_Errors) {
           "Auth Token: fake_token\n"
           "Range: 0-9\n"
           "Timeouts: 5 1 20\n",
-          "Server Not", errors::Unavailable("important HTTP error 308"),
+          "Server Not", absl::UnavailableError("important HTTP error 308"),
           nullptr, {}, 308),
       new FakeHttpRequest(
           "Uri: https://storage.googleapis.com/bucket/random_access.txt\n"
@@ -1127,7 +1127,7 @@ TEST(GcsFileSystemTest, NewWritableFile_ResumeUploadSucceedsOnGetStatus) {
                            "Header Content-Range: bytes 0-16/17\n"
                            "Timeouts: 5 1 30\n"
                            "Put body: content1,content2\n",
-                           "", errors::Unavailable("503"), 503),
+                           "", absl::UnavailableError("503"), 503),
        new FakeHttpRequest("Uri: https://custom/upload/location\n"
                            "Auth Token: fake_token\n"
                            "Timeouts: 5 1 10\n"

xla/tsl/platform/cloud/google_auth_provider.cc

@@ -120,7 +120,7 @@ absl::Status GetWellKnownFileName(string* filename) {
     // Determine the home dir path.
     const char* home_dir = std::getenv("HOME");
     if (!home_dir) {
-      return errors::FailedPrecondition("Could not read $HOME.");
+      return absl::FailedPreconditionError("Could not read $HOME.");
     }
     config_dir = io::JoinPath(home_dir, kGCloudConfigFolder);
   }
@@ -240,7 +240,7 @@ absl::Status GoogleAuthProvider::GetTokenFromFiles() {
         json, kOAuthV4Url, kOAuthScope, &current_token_,
         &expiration_timestamp_sec_));
   } else {
-    return errors::FailedPrecondition(
+    return absl::FailedPreconditionError(
         "Unexpected content of the JSON credentials file.");
   }
   return absl::OkStatus();

xla/tsl/platform/cloud/oauth_client.cc

@@ -14,6 +14,8 @@ limitations under the License.
 ==============================================================================*/
 
 #include "xla/tsl/platform/cloud/oauth_client.h"
+
+#include "absl/status/status.h"
 #ifndef _WIN32
 #include <pwd.h>
 #include <sys/types.h>
@@ -94,7 +96,7 @@ absl::Status CreateSignature(RSA* private_key, absl::string_view to_sign,
 
   const auto md = EVP_sha256();
   if (!md) {
-    return errors::Internal("Could not get a sha256 encryptor.");
+    return absl::InternalError("Could not get a sha256 encryptor.");
   }
 
   // EVP_MD_CTX_destroy is renamed to EVP_MD_CTX_free in OpenSSL 1.1.0 but
@@ -104,26 +106,26 @@ absl::Status CreateSignature(RSA* private_key, absl::string_view to_sign,
   std::unique_ptr<EVP_MD_CTX, std::function<void(EVP_MD_CTX*)>> md_ctx(
       EVP_MD_CTX_create(), [](EVP_MD_CTX* ptr) { EVP_MD_CTX_destroy(ptr); });
   if (!md_ctx) {
-    return errors::Internal("Could not create MD_CTX.");
+    return absl::InternalError("Could not create MD_CTX.");
   }
 
   std::unique_ptr<EVP_PKEY, std::function<void(EVP_PKEY*)>> key(
       EVP_PKEY_new(), [](EVP_PKEY* ptr) { EVP_PKEY_free(ptr); });
   EVP_PKEY_set1_RSA(key.get(), private_key);
 
   if (EVP_DigestSignInit(md_ctx.get(), nullptr, md, nullptr, key.get()) != 1) {
-    return errors::Internal("DigestInit failed.");
+    return absl::InternalError("DigestInit failed.");
   }
   if (EVP_DigestSignUpdate(md_ctx.get(), to_sign.data(), to_sign.size()) != 1) {
-    return errors::Internal("DigestUpdate failed.");
+    return absl::InternalError("DigestUpdate failed.");
   }
   size_t sig_len = 0;
   if (EVP_DigestSignFinal(md_ctx.get(), nullptr, &sig_len) != 1) {
-    return errors::Internal("DigestFinal (get signature length) failed.");
+    return absl::InternalError("DigestFinal (get signature length) failed.");
   }
   std::unique_ptr<unsigned char[]> sig(new unsigned char[sig_len]);
   if (EVP_DigestSignFinal(md_ctx.get(), sig.get(), &sig_len) != 1) {
-    return errors::Internal("DigestFinal (signature compute) failed.");
+    return absl::InternalError("DigestFinal (signature compute) failed.");
   }
   return Base64Encode(
       absl::string_view(reinterpret_cast<char*>(sig.get()), sig_len),
@@ -198,13 +200,13 @@ absl::Status OAuthClient::GetTokenFromServiceAccountJson(
       BIO_new(BIO_s_mem()), [](BIO* ptr) { BIO_free_all(ptr); });
   if (BIO_puts(bio.get(), private_key_serialized.c_str()) !=
       static_cast<int>(private_key_serialized.size())) {
-    return errors::Internal("Could not load the private key.");
+    return absl::InternalError("Could not load the private key.");
   }
   std::unique_ptr<RSA, std::function<void(RSA*)>> private_key(
       PEM_read_bio_RSAPrivateKey(bio.get(), nullptr, nullptr, nullptr),
       [](RSA* ptr) { RSA_free(ptr); });
   if (!private_key) {
-    return errors::Internal("Could not deserialize the private key.");
+    return absl::InternalError("Could not deserialize the private key.");
   }
 
   const uint64 request_timestamp_sec = env_->NowSeconds();
@@ -278,14 +280,15 @@ absl::Status OAuthClient::ParseOAuthResponse(absl::string_view response,
   Json::Value root;
   Json::Reader reader;
   if (!reader.parse(response.data(), response.data() + response.size(), root)) {
-    return errors::Internal("Couldn't parse JSON response from OAuth server.");
+    return absl::InternalError(
+        "Couldn't parse JSON response from OAuth server.");
   }
 
   string token_type;
   TF_RETURN_IF_ERROR(ReadJsonString(root, "token_type", &token_type));
   if (token_type != "Bearer") {
-    return errors::FailedPrecondition("Unexpected Oauth token type: " +
-                                      token_type);
+    return absl::FailedPreconditionError("Unexpected Oauth token type: " +
+                                         token_type);
   }
   int64_t expires_in = 0;
   TF_RETURN_IF_ERROR(ReadJsonInt(root, "expires_in", &expires_in));

xla/tsl/platform/default/human_readable_json.cc

@@ -40,7 +40,7 @@ absl::StatusOr<std::string> ProtoToHumanReadableJson(
     // Convert error_msg google::protobuf::StringPiece to
     // tsl::StringPiece.
     auto error_msg = status.message();
-    return errors::Internal(
+    return absl::InternalError(
         absl::StrCat("Could not convert proto to JSON string: ", error_msg));
   }
   return std::move(result);
@@ -59,15 +59,15 @@ absl::Status HumanReadableJsonToProto(const std::string& str,
     // Convert error_msg google::protobuf::StringPiece to
     // tsl::StringPiece.
     auto error_msg = status.message();
-    return errors::Internal(
+    return absl::InternalError(
         absl::StrCat("Could not convert JSON string to proto: ", error_msg));
   }
   return absl::OkStatus();
 }
 
 absl::Status HumanReadableJsonToProto(const std::string& str,
                                       protobuf::MessageLite* proto) {
-  return errors::Internal("Cannot parse JSON protos on Android");
+  return absl::InternalError("Cannot parse JSON protos on Android");
 }
 
 }  // namespace tsl

xla/tsl/platform/env.cc

@@ -504,8 +504,9 @@ absl::Status ReadFileToString(Env* env, const std::string& fname,
   if (!s.ok()) {
     data->clear();
   } else if (result.size() != file_size) {
-    s = errors::Aborted("File ", fname, " changed while reading: ", file_size,
-                        " vs. ", result.size());
+    s = absl::AbortedError(absl::StrCat("File ", fname,
+                                        " changed while reading: ", file_size,
+                                        " vs. ", result.size()));
     data->clear();
   } else if (result.data() == p) {
     // Data is already in the correct location

xla/tsl/platform/errors_test.cc

@@ -21,7 +21,7 @@ limitations under the License.
 namespace tsl {
 
 TEST(AppendToMessageTest, PayloadsAreCopied) {
-  absl::Status status = errors::Aborted("Aborted Error Message");
+  absl::Status status = absl::AbortedError("Aborted Error Message");
   status.SetPayload("payload_key", absl::Cord("payload_value"));
   errors::AppendToMessage(&status, "Appended Message");
 

xla/tsl/platform/status_test.cc

@@ -1,9 +1,11 @@
-
 /* Copyright 2021 The TensorFlow Authors. All Rights Reserved.
+
 Licensed under the Apache License, Version 2.0 (the "License");
 you may not use this file except in compliance with the License.
 You may obtain a copy of the License at
+
     http://www.apache.org/licenses/LICENSE-2.0
+
 Unless required by applicable law or agreed to in writing, software
 distributed under the License is distributed on an "AS IS" BASIS,
 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
@@ -37,7 +39,7 @@ using ::testing::IsEmpty;
 using ::testing::Pair;
 
 TEST(ToStringTest, PayloadsArePrinted) {
-  absl::Status status = errors::Aborted("Aborted Error Message");
+  absl::Status status = absl::AbortedError("Aborted Error Message");
   status.SetPayload("payload_key", absl::Cord(absl::StrFormat(
                                        "payload_value %c%c%c", 1, 2, 3)));
 
@@ -47,7 +49,7 @@ TEST(ToStringTest, PayloadsArePrinted) {
 }
 
 TEST(ToStringTest, MatchesAbslStatus) {
-  absl::Status status = errors::Aborted("Aborted Error Message");
+  absl::Status status = absl::AbortedError("Aborted Error Message");
   status.SetPayload("payload_key", absl::Cord(absl::StrFormat(
                                        "payload_value %c%c%c", 1, 2, 3)));
 
@@ -60,7 +62,7 @@ TEST(ToStringTest, MatchesAbslStatus) {
 }
 
 TEST(StackTrace, SerializeAndDeserializeCorrectly) {
-  absl::Status status = errors::Aborted("Aborted Error Message");
+  absl::Status status = absl::AbortedError("Aborted Error Message");
   std::vector<StackFrame> stack_trace;
   stack_trace.push_back(StackFrame("filename_1", 33, "func_name_1"));
   stack_trace.push_back(StackFrame("filename_2", 66, "func_name_2"));
@@ -75,9 +77,9 @@ TEST(StackTrace, SerializeAndDeserializeCorrectly) {
 }
 
 TEST(StatusGroupTest, DeterministicOrderWithoutPayloads) {
-  absl::Status status_a = errors::Aborted("Status A");
-  absl::Status status_b = errors::Aborted("Status B");
-  absl::Status status_c = errors::Aborted("Status C");
+  absl::Status status_a = absl::AbortedError("Status A");
+  absl::Status status_b = absl::AbortedError("Status B");
+  absl::Status status_c = absl::AbortedError("Status C");
 
   absl::Status combined =
       StatusGroup({status_a, status_b, status_c}).as_summary_status();
@@ -97,11 +99,11 @@ TEST(StatusGroupTest, DeterministicOrderWithoutPayloads) {
 }
 
 TEST(StatusGroupTest, DeterministicOrderWithPayloads) {
-  absl::Status status_a = errors::Aborted("Status A");
+  absl::Status status_a = absl::AbortedError("Status A");
   status_a.SetPayload("payload_key", absl::Cord("payload_value_a"));
-  absl::Status status_b = errors::Aborted("Status B");
+  absl::Status status_b = absl::AbortedError("Status B");
   status_b.SetPayload("payload_key", absl::Cord("payload_value_b"));
-  absl::Status status_c = errors::Aborted("Status C");
+  absl::Status status_c = absl::AbortedError("Status C");
   status_c.SetPayload("payload_key", absl::Cord("payload_value_c"));
 
   absl::Status combined =
@@ -130,17 +132,17 @@ TEST(StatusGroupTest, DeterministicOrderWithPayloads) {
 }
 
 TEST(StatusGroupTest, PayloadsMergedProperly) {
-  absl::Status status_a = errors::Aborted("Status A");
+  absl::Status status_a = absl::AbortedError("Status A");
   status_a.SetPayload("payload_key_a",
                       absl::Cord(std::string("payload_value_a")));
-  absl::Status status_b = errors::Aborted("Status B");
+  absl::Status status_b = absl::AbortedError("Status B");
   status_b.SetPayload("payload_key_b",
                       absl::Cord(std::string("payload_value_b")));
-  absl::Status status_c = errors::Aborted("Status C");
+  absl::Status status_c = absl::AbortedError("Status C");
   status_c.SetPayload("payload_key_c",
                       absl::Cord(std::string("payload_value_c")));
   absl::Status derived_status_c =
-      StatusGroup::MakeDerived(errors::Aborted("Status C"));
+      StatusGroup::MakeDerived(absl::AbortedError("Status C"));
   derived_status_c.SetPayload(
       "payload_key_c", absl::Cord(std::string("derived_payload_value_c")));