Automated Code Change

PiperOrigin-RevId: 716880401
openxla · Jan 18, 2025 · 518c014 · 518c014
1 parent e141532
commit 518c014
Show file tree

Hide file tree

Showing 8 changed files with 24 additions and 1 deletion.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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 {