Automated Code Change

PiperOrigin-RevId: 689601446
openxla · Oct 25, 2024 · fab148c · fab148c
1 parent 31e7e36
commit fab148c
Show file tree

Hide file tree

Showing 9 changed files with 26 additions and 1 deletion.
diff --git a/xla/tools/BUILD b/xla/tools/BUILD
@@ -341,9 +341,12 @@ cc_library(
     deps = [
         ":run_hlo_module_proto_cc",
         "//xla:debug_options_flags",
+        "//xla:xla_proto_cc",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/parser:hlo_parser",
         "//xla/service:hlo_module_config",
+        "//xla/service:hlo_proto_cc",
+        "@com_google_absl//absl/status",
         "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
         "@com_googlesource_code_re2//:re2",
@@ -553,6 +556,7 @@ cc_library(
         "//xla:literal_util",
         "//xla:shape_util",
         "//xla:util",
+        "//xla:xla_data_proto_cc",
         "//xla/hlo/ir:hlo",
         "//xla/hlo/pass:hlo_pass",
         "//xla/hlo/transforms:hlo_dce",
@@ -583,7 +587,10 @@ xla_cc_test(
         "//xla/tests:hlo_test_base",
         "//xla/tests:xla_internal_test_main",  # fixdeps: keep
         "//xla/tsl/lib/core:status_test_util",
+        "@com_google_absl//absl/log",
+        "@com_google_absl//absl/status:statusor",
         "@com_google_absl//absl/strings",
+        "@com_google_googletest//:gtest",
     ],
 )
 

diff --git a/xla/tools/dumped_computation_to_operation_list.cc b/xla/tools/dumped_computation_to_operation_list.cc
@@ -15,10 +15,12 @@ limitations under the License.
 
 // Dumps out the operations that are present in a serialized computation.
 
+#include <cstdio>
 #include <iostream>
 #include <memory>
 #include <string>
 #include <utility>
+#include <vector>
 
 #include "absl/status/status.h"
 #include "absl/status/statusor.h"

diff --git a/xla/tools/dumped_computation_to_text.cc b/xla/tools/dumped_computation_to_text.cc
@@ -18,6 +18,7 @@ limitations under the License.
 #include <memory>
 #include <string>
 #include <utility>
+#include <vector>
 
 #include "absl/status/statusor.h"
 #include "absl/types/span.h"

diff --git a/xla/tools/hlo_control_flow_flattening.cc b/xla/tools/hlo_control_flow_flattening.cc
@@ -43,6 +43,7 @@ limitations under the License.
 #include "xla/shape.h"
 #include "xla/shape_util.h"
 #include "xla/util.h"
+#include "xla/xla_data.pb.h"
 #include "tsl/platform/errors.h"
 #include "tsl/platform/statusor.h"
 

diff --git a/xla/tools/hlo_control_flow_flattening_test.cc b/xla/tools/hlo_control_flow_flattening_test.cc
@@ -15,10 +15,17 @@ limitations under the License.
 
 #include "xla/tools/hlo_control_flow_flattening.h"
 
+#include <cstdint>
 #include <memory>
 #include <utility>
 
+#include <gmock/gmock.h>
+#include <gtest/gtest.h>
+#include "absl/log/log.h"
+#include "absl/status/statusor.h"
+#include "absl/strings/str_cat.h"
 #include "absl/strings/str_replace.h"
+#include "absl/strings/string_view.h"
 #include "xla/hlo/transforms/despecializer.h"
 #include "xla/hlo/utils/hlo_matchers.h"
 #include "xla/service/collective_ops_utils.h"

diff --git a/xla/tools/hlo_decomposer.cc b/xla/tools/hlo_decomposer.cc
@@ -24,7 +24,6 @@ limitations under the License.
 #include "absl/container/flat_hash_set.h"
 #include "absl/container/inlined_vector.h"
 #include "absl/log/check.h"
-#include "absl/status/status.h"
 #include "xla/hlo/ir/hlo_clone_context.h"
 #include "xla/hlo/ir/hlo_computation.h"
 #include "xla/hlo/ir/hlo_instruction.h"

diff --git a/xla/tools/hlo_module_loader.cc b/xla/tools/hlo_module_loader.cc
@@ -22,7 +22,9 @@ limitations under the License.
 #include <string>
 #include <string_view>
 #include <utility>
+#include <vector>
 
+#include "absl/status/status.h"
 #include "absl/strings/str_cat.h"
 #include "absl/strings/str_join.h"
 #include "absl/strings/str_split.h"
@@ -31,7 +33,10 @@ limitations under the License.
 #include "xla/hlo/ir/hlo_computation.h"
 #include "xla/hlo/ir/hlo_instruction.h"
 #include "xla/hlo/parser/hlo_parser.h"
+#include "xla/service/hlo.pb.h"
 #include "xla/service/hlo_module_config.h"
+#include "xla/tools/run_hlo_module.pb.h"
+#include "xla/xla.pb.h"
 #include "tsl/platform/env.h"
 #include "tsl/platform/logging.h"
 #include "tsl/platform/path.h"

diff --git a/xla/tools/hlo_module_loader.h b/xla/tools/hlo_module_loader.h
@@ -16,6 +16,7 @@ limitations under the License.
 #ifndef XLA_TOOLS_HLO_MODULE_LOADER_H_
 #define XLA_TOOLS_HLO_MODULE_LOADER_H_
 
+#include <cstdint>
 #include <functional>
 #include <memory>
 #include <string>
@@ -24,6 +25,7 @@ limitations under the License.
 #include "absl/status/statusor.h"
 #include "absl/strings/string_view.h"
 #include "xla/hlo/ir/hlo_module.h"
+#include "xla/service/hlo.pb.h"
 #include "xla/tools/run_hlo_module.pb.h"
 
 namespace xla {

diff --git a/xla/tools/hlo_module_loader_test.cc b/xla/tools/hlo_module_loader_test.cc
@@ -15,6 +15,7 @@ limitations under the License.
 
 #include "xla/tools/hlo_module_loader.h"
 
+#include <memory>
 #include <string>
 
 #include "xla/tests/hlo_test_base.h"