From fab148c925ec5fa2aa5f65b4c82c681efe919fe2 Mon Sep 17 00:00:00 2001 From: xla authors Date: Thu, 24 Oct 2024 19:28:47 -0700 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 689601446 --- xla/tools/BUILD | 7 +++++++ xla/tools/dumped_computation_to_operation_list.cc | 2 ++ xla/tools/dumped_computation_to_text.cc | 1 + xla/tools/hlo_control_flow_flattening.cc | 1 + xla/tools/hlo_control_flow_flattening_test.cc | 7 +++++++ xla/tools/hlo_decomposer.cc | 1 - xla/tools/hlo_module_loader.cc | 5 +++++ xla/tools/hlo_module_loader.h | 2 ++ xla/tools/hlo_module_loader_test.cc | 1 + 9 files changed, 26 insertions(+), 1 deletion(-) diff --git a/xla/tools/BUILD b/xla/tools/BUILD index 1b28e8fb92cfe..89946f3008dc7 100644 --- 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 index 071ec6192b414..b6be3188ffa02 100644 --- 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 #include #include #include #include +#include #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 index b053ef88ba448..72e1710e19450 100644 --- 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 #include #include +#include #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 index 55ea578d3460e..61ef1175457e1 100644 --- 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 index 9c8520803a0b4..df23c3c9527e8 100644 --- 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 #include #include +#include +#include +#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 index 30741733af213..577456c24787a 100644 --- 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 index 81f8167e56f59..3ab573dfa2ac4 100644 --- a/xla/tools/hlo_module_loader.cc +++ b/xla/tools/hlo_module_loader.cc @@ -22,7 +22,9 @@ limitations under the License. #include #include #include +#include +#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 index da44e71428332..4dc0653cd9729 100644 --- 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 #include #include #include @@ -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 index 16fbe45e4ae45..b3b4a8fb0b9c3 100644 --- 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 #include #include "xla/tests/hlo_test_base.h"