From 518c01484c8e6cee4e3bf87be88ddee7e56428d2 Mon Sep 17 00:00:00 2001
From: xla authors <google-ml-automation@google.com>
Date: Fri, 17 Jan 2025 19:43:25 -0800
Subject: [PATCH] Automated Code Change

PiperOrigin-RevId: 716880401
---
 xla/BUILD                        | 6 ++++++
 xla/packed_literal_reader.cc     | 6 +++++-
 xla/packed_literal_reader.h      | 1 +
 xla/parse_flags_from_env.cc      | 2 ++
 xla/parse_flags_from_env_test.cc | 4 ++++
 xla/permutation_util.cc          | 2 ++
 xla/permutation_util.h           | 3 +++
 xla/permutation_util_test.cc     | 1 +
 8 files changed, 24 insertions(+), 1 deletion(-)

diff --git a/xla/BUILD b/xla/BUILD
index 860c690de3ab6..0f0cf7be09e5b 100644
--- a/xla/BUILD
+++ b/xla/BUILD
@@ -889,6 +889,8 @@ cc_library(
         ":util",
         ":xla_data_proto_cc",
         "@com_google_absl//absl/base",
+        "@com_google_absl//absl/log",
+        "@com_google_absl//absl/log:check",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings:string_view",
         "@com_google_absl//absl/types:span",
@@ -1124,6 +1126,8 @@ cc_library(
             "//xla/tsl/util:command_line_flags",
             "@com_google_absl//absl/base:core_headers",
             "@com_google_absl//absl/container:flat_hash_map",
+            "@com_google_absl//absl/log",
+            "@com_google_absl//absl/log:check",
             "@com_google_absl//absl/strings",
             "@com_google_absl//absl/strings:str_format",
             "@com_google_absl//absl/synchronization",
@@ -1140,7 +1144,9 @@ xla_cc_test(
             ":parse_flags_from_env",
             "//xla/tsl/platform:subprocess",
             "//xla/tsl/util:command_line_flags",
+            "@com_google_absl//absl/log:check",
             "@com_google_absl//absl/strings:str_format",
+            "@com_google_googletest//:gtest",
             "@tsl//tsl/platform:env",
             "@tsl//tsl/platform:logging",
             "@tsl//tsl/platform:macros",
diff --git a/xla/packed_literal_reader.cc b/xla/packed_literal_reader.cc
index e707778a55669..609d190401808 100644
--- a/xla/packed_literal_reader.cc
+++ b/xla/packed_literal_reader.cc
@@ -15,12 +15,16 @@ limitations under the License.
 
 #include "xla/packed_literal_reader.h"
 
+#include <cstdint>
+#include <cstring>
 #include <limits>
-#include <memory>
 #include <string>
 #include <utility>
 
 #include "absl/base/casts.h"
+#include "absl/log/check.h"
+#include "absl/log/log.h"
+#include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "absl/types/span.h"
 #include "xla/layout.h"
diff --git a/xla/packed_literal_reader.h b/xla/packed_literal_reader.h
index 4a5d5fdb9bdc9..e23d526ddfa54 100644
--- a/xla/packed_literal_reader.h
+++ b/xla/packed_literal_reader.h
@@ -16,6 +16,7 @@ limitations under the License.
 #ifndef XLA_PACKED_LITERAL_READER_H_
 #define XLA_PACKED_LITERAL_READER_H_
 
+#include <cstdint>
 #include <memory>
 
 #include "absl/status/statusor.h"
diff --git a/xla/parse_flags_from_env.cc b/xla/parse_flags_from_env.cc
index 7e5e7e686b04b..331cccb536be8 100644
--- a/xla/parse_flags_from_env.cc
+++ b/xla/parse_flags_from_env.cc
@@ -29,6 +29,8 @@ limitations under the License.
 
 #include "absl/base/const_init.h"
 #include "absl/container/flat_hash_map.h"
+#include "absl/log/check.h"
+#include "absl/log/log.h"
 #include "absl/strings/ascii.h"
 #include "absl/strings/str_join.h"
 #include "absl/strings/string_view.h"
diff --git a/xla/parse_flags_from_env_test.cc b/xla/parse_flags_from_env_test.cc
index fc72ce20ccbcf..a29c72e564591 100644
--- a/xla/parse_flags_from_env_test.cc
+++ b/xla/parse_flags_from_env_test.cc
@@ -20,9 +20,13 @@ limitations under the License.
 #include <stdio.h>
 #include <stdlib.h>
 
+#include <algorithm>
+#include <cstdint>
 #include <string>
 #include <vector>
 
+#include <gtest/gtest.h>
+#include "absl/log/check.h"
 #include "absl/strings/str_format.h"
 #include "xla/tsl/platform/subprocess.h"
 #include "xla/tsl/util/command_line_flags.h"
diff --git a/xla/permutation_util.cc b/xla/permutation_util.cc
index 66f2fb134bbd2..4df3e57fc234b 100644
--- a/xla/permutation_util.cc
+++ b/xla/permutation_util.cc
@@ -15,6 +15,8 @@ limitations under the License.
 
 #include "xla/permutation_util.h"
 
+#include <cstddef>
+#include <cstdint>
 #include <vector>
 
 #include "absl/container/inlined_vector.h"
diff --git a/xla/permutation_util.h b/xla/permutation_util.h
index 61dde9ca117c4..cec6421353354 100644
--- a/xla/permutation_util.h
+++ b/xla/permutation_util.h
@@ -18,8 +18,11 @@ limitations under the License.
 #ifndef XLA_PERMUTATION_UTIL_H_
 #define XLA_PERMUTATION_UTIL_H_
 
+#include <cstddef>
+#include <cstdint>
 #include <vector>
 
+#include "absl/log/check.h"
 #include "absl/types/span.h"
 #include "xla/types.h"
 #include "tsl/platform/logging.h"
diff --git a/xla/permutation_util_test.cc b/xla/permutation_util_test.cc
index 9926650940476..5754965bbbdcf 100644
--- a/xla/permutation_util_test.cc
+++ b/xla/permutation_util_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "xla/permutation_util.h"
 
+#include <gtest/gtest.h>
 #include "xla/hlo/testlib/test.h"
 
 namespace xla {