diff --git a/xla/service/BUILD b/xla/service/BUILD index f32d241ce05fd..6719153139ad2 100644 --- a/xla/service/BUILD +++ b/xla/service/BUILD @@ -924,6 +924,7 @@ xla_cc_test( ":hlo_proto_cc", ":pattern_matcher", ":pattern_matcher_gmock", + ":remote_mtest_lib_dynamic_padder_test_gpu_b200", "//xla:comparison_util", "//xla:literal_util", "//xla:protobuf_util", @@ -941,6 +942,7 @@ xla_cc_test( "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/status", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:protobuf", "@tsl//tsl/platform:statusor", ], @@ -1866,6 +1868,10 @@ xla_cc_test( "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/strings:str_format", + "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", ], ) @@ -1880,10 +1886,15 @@ cc_library( "//xla/hlo/analysis:hlo_alias_analysis", "//xla/hlo/analysis:tuple_points_to_analysis", "//xla/hlo/ir:hlo", + "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:status", @@ -1898,11 +1909,13 @@ cc_library( ":compiler", ":computation_layout", ":hlo_module_config", + ":hlo_proto_cc", ":hlo_verifier", "//xla:debug_options_flags", "//xla:shape_layout", "//xla:shape_util", "//xla:util", + "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/tsl/platform:env", @@ -1931,6 +1944,7 @@ cc_library( "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/functional:function_ref", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -2132,6 +2146,7 @@ xla_cc_test( ":hlo_creation_utils", ":pattern_matcher", ":pattern_matcher_gmock", + ":remote_mtest_lib_dynamic_padder_test_gpu_b200", "//xla:array2d", "//xla:literal", "//xla:literal_util", @@ -3572,6 +3587,8 @@ cc_library( "//xla/tsl/lib/gtl:map_util", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -3591,6 +3608,7 @@ xla_cc_test( "//xla:shape_util", "//xla:test_helpers", "//xla:xla_data_proto_cc", + "//xla:xla_proto_cc", "//xla/client", "//xla/client:client_library", "//xla/client:local_client", @@ -3601,7 +3619,11 @@ xla_cc_test( "//xla/hlo/parser:hlo_parser", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", + "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", ], ) @@ -3620,6 +3642,9 @@ cc_library( "//xla/hlo/ir:hlo", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", ], ) @@ -3631,9 +3656,12 @@ xla_cc_test( ":cpu_plugin", ":hlo_cost_analysis", ":hlo_execution_profile", + ":hlo_profile_printer_data_cc", + "//xla:xla_data_proto_cc", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", ], ) @@ -3643,12 +3671,14 @@ xla_cc_test( deps = [ ":pattern_matcher", ":pattern_matcher_gmock", + ":remote_mtest_lib_dynamic_padder_test_gpu_b200", "//xla:comparison_util", "//xla:literal_util", "//xla:shape_tree", "//xla:shape_util", "//xla:test", "//xla:test_helpers", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/hlo/utils:hlo_matchers", @@ -3658,6 +3688,7 @@ xla_cc_test( "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/status", + "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", @@ -3704,10 +3735,12 @@ xla_cc_test( name = "hlo_module_metadata_test", srcs = ["hlo_module_metadata_test.cc"], deps = [ + ":hlo_proto_cc", "//xla:test", "//xla:test_helpers", "//xla/hlo/ir:hlo", "//xla/tests:xla_internal_test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -3791,8 +3824,13 @@ cc_library( deps = [ ":hlo_value", "//xla/hlo/ir:hlo", + "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/types:span", ], ) @@ -3804,6 +3842,7 @@ xla_cc_test( ":hlo_phi_graph", "//xla:literal_util", "//xla/tests:xla_internal_test_main", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", ], ) @@ -3841,7 +3880,9 @@ cc_library( "//xla:util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", @@ -4076,8 +4117,11 @@ cc_library( "//xla/hlo/pass:hlo_pass", "//xla/hlo/transforms/simplifiers:hlo_dce", "//xla/hlo/transforms/simplifiers:tuple_simplifier", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", ], @@ -4204,6 +4248,8 @@ xla_cc_test( "//xla/tests:test_utils", "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/algorithm:container", + "@com_google_googletest//:gtest_main", ], ) @@ -4282,11 +4328,17 @@ cc_library( ":hlo_domain_map", "//xla:literal", "//xla:shape_util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/hash", + "@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", "@tsl//tsl/platform:errors", ], ) @@ -4298,6 +4350,7 @@ xla_cc_test( ":hlo_cse", ":pattern_matcher", ":pattern_matcher_gmock", + ":remote_mtest_lib_dynamic_padder_test_gpu_b200", "//xla:literal", "//xla:shape_util", "//xla:util", @@ -4310,6 +4363,7 @@ xla_cc_test( "//xla/tests:xla_internal_test_main", "//xla/tsl/platform:status_matchers", "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/log", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", ], @@ -4351,6 +4405,11 @@ cc_library( "//xla:types", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:status", ], ) @@ -4363,6 +4422,10 @@ cc_library( ":hlo_domain_remover", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:statusor", ], ) @@ -4378,6 +4441,11 @@ cc_library( "//xla:types", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", "@tsl//tsl/platform:status", ], ) @@ -4393,11 +4461,17 @@ xla_cc_test( ":sharding_propagation", "//xla:debug_options_flags", "//xla:test", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", ], ) @@ -4530,6 +4604,7 @@ cc_library( ":computation_placer", ":hlo_proto_cc", ":sharding_config", + "//third_party/protobuf:protobuf_lite", "//xla:debug_options_flags", "//xla:shape_layout", "//xla:shape_util", @@ -4588,6 +4663,7 @@ cc_library( "//xla:types", "//xla:util", "//xla:window_util", + "//xla:xla_data_proto_cc", "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/service/gpu:backend_configs_cc", @@ -4628,6 +4704,7 @@ xla_cc_test( ":hlo_graph_dumper", "//xla:literal_util", "//xla:test", + "//xla:xla_data_proto_cc", "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/tests:hlo_test_base", @@ -4636,6 +4713,7 @@ xla_cc_test( "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", + "@com_google_googletest//:gtest_main", ], ) @@ -4720,8 +4798,10 @@ cc_library( ":buffer_assignment", ":hlo_proto_cc", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", ], ) @@ -4738,6 +4818,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -4776,6 +4857,7 @@ cc_library( ":computation_placer", ":executable", ":hlo_module_util", + ":hlo_proto_cc", ":hlo_runner_interface", ":maybe_owning_device_memory", ":shaped_buffer", @@ -5454,8 +5536,10 @@ xla_cc_test( name = "hlo_casting_utils_test", srcs = ["hlo_casting_utils_test.cc"], deps = [ + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/tests:xla_internal_test_main", # fixdeps: keep + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", ], ) diff --git a/xla/service/hlo_buffer.cc b/xla/service/hlo_buffer.cc index 4c7569d10d9b2..5535ba267ff1d 100644 --- a/xla/service/hlo_buffer.cc +++ b/xla/service/hlo_buffer.cc @@ -17,10 +17,11 @@ limitations under the License. #include #include -#include +#include #include -#include "absl/container/flat_hash_set.h" +#include "absl/algorithm/container.h" +#include "absl/log/check.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_join.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/xla/service/hlo_buffer.h b/xla/service/hlo_buffer.h index 398b08d8cfdee..a163401baa418 100644 --- a/xla/service/hlo_buffer.h +++ b/xla/service/hlo_buffer.h @@ -21,6 +21,7 @@ limitations under the License. #include #include +#include "absl/log/check.h" #include "xla/service/hlo_value.h" #include "xla/shape_tree.h" #include "xla/types.h" diff --git a/xla/service/hlo_casting_utils_test.cc b/xla/service/hlo_casting_utils_test.cc index 010559bc71e9f..36a59424fed43 100644 --- a/xla/service/hlo_casting_utils_test.cc +++ b/xla/service/hlo_casting_utils_test.cc @@ -15,7 +15,9 @@ limitations under the License. #include "xla/hlo/ir/hlo_casting_utils.h" +#include #include "xla/hlo/ir/hlo_instruction.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/test.h" namespace xla { diff --git a/xla/service/hlo_computation_test.cc b/xla/service/hlo_computation_test.cc index 4103dcac62d8d..fd400f6f38dce 100644 --- a/xla/service/hlo_computation_test.cc +++ b/xla/service/hlo_computation_test.cc @@ -21,9 +21,11 @@ limitations under the License. #include #include +#include #include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_set.h" #include "absl/status/status.h" +#include "absl/strings/string_view.h" #include "xla/comparison_util.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" #include "xla/hlo/ir/hlo_clone_context.h" @@ -41,6 +43,7 @@ limitations under the License. #include "xla/test_helpers.h" #include "xla/tests/hlo_test_base.h" #include "xla/tsl/lib/core/status_test_util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/status.h" #include "tsl/platform/statusor.h" diff --git a/xla/service/hlo_cost_analysis.cc b/xla/service/hlo_cost_analysis.cc index d2ad0cf847c0e..97fccb52c9d56 100644 --- a/xla/service/hlo_cost_analysis.cc +++ b/xla/service/hlo_cost_analysis.cc @@ -18,16 +18,19 @@ limitations under the License. #include #include #include -#include #include #include #include #include #include "absl/algorithm/container.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/status/status.h" +#include "absl/status/statusor.h" #include "absl/strings/match.h" #include "absl/strings/str_cat.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" @@ -39,6 +42,7 @@ limitations under the License. #include "xla/tsl/lib/gtl/map_util.h" #include "xla/util.h" #include "xla/window_util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" namespace xla { diff --git a/xla/service/hlo_cost_analysis.h b/xla/service/hlo_cost_analysis.h index 2bedab36928ae..f38f59cec4efd 100644 --- a/xla/service/hlo_cost_analysis.h +++ b/xla/service/hlo_cost_analysis.h @@ -23,8 +23,11 @@ limitations under the License. #include #include "absl/container/flat_hash_map.h" +#include "absl/log/check.h" +#include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/str_format.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/dfs_hlo_visitor.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/xla/service/hlo_cost_analysis_test.cc b/xla/service/hlo_cost_analysis_test.cc index cb38f56e7884b..f520be44d4142 100644 --- a/xla/service/hlo_cost_analysis_test.cc +++ b/xla/service/hlo_cost_analysis_test.cc @@ -15,12 +15,18 @@ limitations under the License. #include "xla/service/hlo_cost_analysis.h" +#include #include #include #include #include +#include +#include +#include "absl/container/inlined_vector.h" #include "absl/status/statusor.h" +#include "absl/strings/string_view.h" +#include "absl/types/span.h" #include "xla/client/client.h" #include "xla/client/client_library.h" #include "xla/client/local_client.h" @@ -35,6 +41,7 @@ limitations under the License. #include "xla/shape_util.h" #include "xla/test_helpers.h" #include "xla/tests/hlo_test_base.h" +#include "xla/xla.pb.h" #include "xla/xla_data.pb.h" #include "tsl/platform/logging.h" diff --git a/xla/service/hlo_creation_utils.h b/xla/service/hlo_creation_utils.h index 0fe2da44434a2..0f731a1455019 100644 --- a/xla/service/hlo_creation_utils.h +++ b/xla/service/hlo_creation_utils.h @@ -21,6 +21,9 @@ limitations under the License. #include #include +#include "absl/log/log.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "absl/types/span.h" #include "xla/hlo/builder/xla_computation.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/xla/service/hlo_creation_utils_test.cc b/xla/service/hlo_creation_utils_test.cc index debabe09c3c51..844c4de58b570 100644 --- a/xla/service/hlo_creation_utils_test.cc +++ b/xla/service/hlo_creation_utils_test.cc @@ -18,6 +18,7 @@ limitations under the License. #include #include +#include #include #include "absl/log/check.h" #include "absl/types/span.h" diff --git a/xla/service/hlo_cse.cc b/xla/service/hlo_cse.cc index 2b98151580efc..9a730e1417aaf 100644 --- a/xla/service/hlo_cse.cc +++ b/xla/service/hlo_cse.cc @@ -15,6 +15,7 @@ limitations under the License. #include "xla/service/hlo_cse.h" +#include #include #include #include @@ -22,7 +23,12 @@ limitations under the License. #include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_set.h" +#include "absl/hash/hash.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/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" @@ -31,6 +37,7 @@ limitations under the License. #include "xla/literal.h" #include "xla/service/hlo_domain_map.h" #include "xla/shape_util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" namespace xla { diff --git a/xla/service/hlo_cse.h b/xla/service/hlo_cse.h index 423a99b92935a..9357aaafa938e 100644 --- a/xla/service/hlo_cse.h +++ b/xla/service/hlo_cse.h @@ -16,6 +16,9 @@ limitations under the License. #ifndef XLA_SERVICE_HLO_CSE_H_ #define XLA_SERVICE_HLO_CSE_H_ +#include "absl/container/flat_hash_set.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/pass/hlo_pass_interface.h" diff --git a/xla/service/hlo_cse_test.cc b/xla/service/hlo_cse_test.cc index a0de5d978a6dd..3f28c49f1e7bc 100644 --- a/xla/service/hlo_cse_test.cc +++ b/xla/service/hlo_cse_test.cc @@ -15,13 +15,18 @@ limitations under the License. #include "xla/service/hlo_cse.h" +#include #include #include +#include #include #include #include #include "absl/algorithm/container.h" +#include "absl/log/log.h" +#include "absl/strings/str_cat.h" +#include "absl/strings/string_view.h" #include "absl/strings/substitute.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/xla/service/hlo_domain_isolator.cc b/xla/service/hlo_domain_isolator.cc index c5355ef98d677..aea6ce1993409 100644 --- a/xla/service/hlo_domain_isolator.cc +++ b/xla/service/hlo_domain_isolator.cc @@ -16,7 +16,13 @@ limitations under the License. #include "xla/service/hlo_domain_isolator.h" #include +#include +#include +#include "absl/container/flat_hash_set.h" +#include "absl/log/log.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" diff --git a/xla/service/hlo_domain_isolator.h b/xla/service/hlo_domain_isolator.h index d10dc501babc3..0667ba9512c2d 100644 --- a/xla/service/hlo_domain_isolator.h +++ b/xla/service/hlo_domain_isolator.h @@ -16,6 +16,9 @@ limitations under the License. #ifndef XLA_SERVICE_HLO_DOMAIN_ISOLATOR_H_ #define XLA_SERVICE_HLO_DOMAIN_ISOLATOR_H_ +#include "absl/container/flat_hash_set.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/pass/hlo_pass_interface.h" diff --git a/xla/service/hlo_domain_remover.cc b/xla/service/hlo_domain_remover.cc index 4f2d2efc6afc3..7394bc17c9318 100644 --- a/xla/service/hlo_domain_remover.cc +++ b/xla/service/hlo_domain_remover.cc @@ -15,6 +15,15 @@ limitations under the License. #include "xla/service/hlo_domain_remover.h" +#include +#include +#include + +#include "absl/container/flat_hash_set.h" +#include "absl/log/log.h" +#include "absl/status/status.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" diff --git a/xla/service/hlo_domain_remover.h b/xla/service/hlo_domain_remover.h index 970f4f164a5e2..db4f38b2dd3a0 100644 --- a/xla/service/hlo_domain_remover.h +++ b/xla/service/hlo_domain_remover.h @@ -16,6 +16,10 @@ limitations under the License. #ifndef XLA_SERVICE_HLO_DOMAIN_REMOVER_H_ #define XLA_SERVICE_HLO_DOMAIN_REMOVER_H_ +#include "absl/container/flat_hash_set.h" +#include "absl/status/status.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_domain_metadata.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" diff --git a/xla/service/hlo_domain_test.cc b/xla/service/hlo_domain_test.cc index 11acf73bf6cff..2b5adad9c9772 100644 --- a/xla/service/hlo_domain_test.cc +++ b/xla/service/hlo_domain_test.cc @@ -13,10 +13,18 @@ See the License for the specific language governing permissions and limitations under the License. ==============================================================================*/ +#include +#include +#include #include #include #include +#include +#include "absl/log/check.h" +#include "absl/log/log.h" +#include "absl/status/status.h" +#include "absl/strings/string_view.h" #include "xla/debug_options_flags.h" #include "xla/hlo/ir/hlo_domain_metadata.h" #include "xla/hlo/ir/hlo_sharding_metadata.h" @@ -29,6 +37,7 @@ limitations under the License. #include "xla/test.h" #include "xla/tests/hlo_test_base.h" #include "xla/tsl/lib/core/status_test_util.h" +#include "xla/xla_data.pb.h" namespace xla { namespace { diff --git a/xla/service/hlo_domain_verifier.cc b/xla/service/hlo_domain_verifier.cc index 519f572c3dfbb..a7345384a3aa0 100644 --- a/xla/service/hlo_domain_verifier.cc +++ b/xla/service/hlo_domain_verifier.cc @@ -15,8 +15,15 @@ limitations under the License. #include "xla/service/hlo_domain_verifier.h" +#include #include +#include +#include "absl/container/flat_hash_set.h" +#include "absl/log/log.h" +#include "absl/status/status.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" diff --git a/xla/service/hlo_domain_verifier.h b/xla/service/hlo_domain_verifier.h index 32dce237cb03d..b4f4ceaa2738f 100644 --- a/xla/service/hlo_domain_verifier.h +++ b/xla/service/hlo_domain_verifier.h @@ -19,6 +19,9 @@ limitations under the License. #include #include +#include "absl/container/flat_hash_set.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_domain_metadata.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/pass/hlo_pass_interface.h" diff --git a/xla/service/hlo_execution_profile.cc b/xla/service/hlo_execution_profile.cc index 17de9cffdbe38..b19b588c51608 100644 --- a/xla/service/hlo_execution_profile.cc +++ b/xla/service/hlo_execution_profile.cc @@ -15,12 +15,17 @@ limitations under the License. #include "xla/service/hlo_execution_profile.h" -#include +#include +#include #include +#include #include #include #include "absl/algorithm/container.h" +#include "absl/log/check.h" +#include "absl/strings/string_view.h" +#include "absl/types/span.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/service/hlo_execution_profile_data.pb.h" diff --git a/xla/service/hlo_execution_profile.h b/xla/service/hlo_execution_profile.h index 0540a981a67fb..a9f76277c85e1 100644 --- a/xla/service/hlo_execution_profile.h +++ b/xla/service/hlo_execution_profile.h @@ -19,6 +19,8 @@ limitations under the License. #include #include "absl/container/flat_hash_map.h" +#include "absl/strings/string_view.h" +#include "absl/types/span.h" #include "xla/map_util.h" #include "xla/service/hlo_cost_analysis.h" #include "xla/service/hlo_execution_profile_data.pb.h" diff --git a/xla/service/hlo_execution_profile_test.cc b/xla/service/hlo_execution_profile_test.cc index 5a27be1d80028..dc32be217345a 100644 --- a/xla/service/hlo_execution_profile_test.cc +++ b/xla/service/hlo_execution_profile_test.cc @@ -14,9 +14,17 @@ limitations under the License. ==============================================================================*/ #include "xla/service/hlo_execution_profile.h" + +#include +#include + +#include +#include #include "absl/strings/str_cat.h" #include "xla/service/hlo_cost_analysis.h" +#include "xla/service/hlo_profile_printer_data.pb.h" #include "xla/tests/hlo_test_base.h" +#include "xla/xla_data.pb.h" namespace xla { namespace { diff --git a/xla/service/hlo_graph_dumper.cc b/xla/service/hlo_graph_dumper.cc index 7c5a5869990f0..bde2b10bdcef3 100644 --- a/xla/service/hlo_graph_dumper.cc +++ b/xla/service/hlo_graph_dumper.cc @@ -31,6 +31,8 @@ limitations under the License. #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_sharding.h" #include "xla/shape.h" +#include "xla/xla.pb.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" #include "tsl/platform/file_system.h" #include "tsl/platform/statusor.h" @@ -40,14 +42,9 @@ limitations under the License. #include #endif -#include -#include #include #include -#include -#include #include -#include #include #include #include diff --git a/xla/service/hlo_graph_dumper.h b/xla/service/hlo_graph_dumper.h index 8c5d5ade70960..ddfcd5435b940 100644 --- a/xla/service/hlo_graph_dumper.h +++ b/xla/service/hlo_graph_dumper.h @@ -20,6 +20,7 @@ limitations under the License. #include #include "absl/container/flat_hash_map.h" +#include "absl/container/flat_hash_set.h" #include "absl/status/statusor.h" #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/xla/service/hlo_graph_dumper_test.cc b/xla/service/hlo_graph_dumper_test.cc index d76e734f33ec7..a21c040d01932 100644 --- a/xla/service/hlo_graph_dumper_test.cc +++ b/xla/service/hlo_graph_dumper_test.cc @@ -15,6 +15,11 @@ limitations under the License. #include "xla/service/hlo_graph_dumper.h" +#include +#include + +#include +#include #include "absl/container/flat_hash_map.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_format.h" @@ -27,6 +32,7 @@ limitations under the License. #include "xla/tests/hlo_test_base.h" #include "xla/tests/test_utils.h" #include "xla/xla.pb.h" +#include "xla/xla_data.pb.h" namespace xla { namespace { diff --git a/xla/service/hlo_input_output_alias_config_test.cc b/xla/service/hlo_input_output_alias_config_test.cc index 7c2cc1d945c0d..a0734f8e4c566 100644 --- a/xla/service/hlo_input_output_alias_config_test.cc +++ b/xla/service/hlo_input_output_alias_config_test.cc @@ -15,11 +15,12 @@ limitations under the License. #include "xla/hlo/ir/hlo_input_output_alias_config.h" +#include #include +#include #include #include -#include "absl/algorithm/container.h" #include "xla/hlo/analysis/hlo_ordering.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/xla/service/hlo_instruction_test.cc b/xla/service/hlo_instruction_test.cc index 7b9e1f762d30b..bcb5bd66ba49d 100644 --- a/xla/service/hlo_instruction_test.cc +++ b/xla/service/hlo_instruction_test.cc @@ -26,6 +26,8 @@ limitations under the License. #include #include +#include +#include #include "absl/container/flat_hash_map.h" #include "absl/status/status.h" #include "absl/strings/string_view.h" diff --git a/xla/service/hlo_module_config.cc b/xla/service/hlo_module_config.cc index 9e30e5525cfcb..a961a6a9752b0 100644 --- a/xla/service/hlo_module_config.cc +++ b/xla/service/hlo_module_config.cc @@ -29,6 +29,7 @@ limitations under the License. #include "absl/strings/escaping.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_join.h" +#include "third_party/protobuf/repeated_ptr_field.h" #include "xla/service/computation_layout.h" #include "xla/service/computation_placer.h" #include "xla/service/hlo.pb.h" diff --git a/xla/service/hlo_module_config_test.cc b/xla/service/hlo_module_config_test.cc index 3f3703ff8fe46..98671fd0f00de 100644 --- a/xla/service/hlo_module_config_test.cc +++ b/xla/service/hlo_module_config_test.cc @@ -17,6 +17,7 @@ limitations under the License. #include +#include #include "xla/tests/test_utils.h" #include "xla/xla.pb.h" #include "tsl/platform/statusor.h" diff --git a/xla/service/hlo_module_dce.cc b/xla/service/hlo_module_dce.cc index fa4da84979250..37ae3cea8339a 100644 --- a/xla/service/hlo_module_dce.cc +++ b/xla/service/hlo_module_dce.cc @@ -15,10 +15,14 @@ limitations under the License. #include "xla/service/hlo_module_dce.h" -#include +#include +#include +#include -#include "absl/status/status.h" +#include "absl/container/flat_hash_set.h" +#include "absl/log/log.h" #include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/analysis/hlo_liveness_analysis.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/xla/service/hlo_module_dce.h b/xla/service/hlo_module_dce.h index 453deb26bb847..e46e0ffcf7bf6 100644 --- a/xla/service/hlo_module_dce.h +++ b/xla/service/hlo_module_dce.h @@ -16,7 +16,9 @@ limitations under the License. #ifndef XLA_SERVICE_HLO_MODULE_DCE_H_ #define XLA_SERVICE_HLO_MODULE_DCE_H_ +#include "absl/container/flat_hash_set.h" #include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/pass/hlo_pass_interface.h" diff --git a/xla/service/hlo_module_dce_test.cc b/xla/service/hlo_module_dce_test.cc index 4b1a7b7e2e440..b20d27a320ed1 100644 --- a/xla/service/hlo_module_dce_test.cc +++ b/xla/service/hlo_module_dce_test.cc @@ -15,6 +15,12 @@ limitations under the License. #include "xla/service/hlo_module_dce.h" +#include +#include +#include + +#include +#include "absl/algorithm/container.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" diff --git a/xla/service/hlo_module_group_metadata.cc b/xla/service/hlo_module_group_metadata.cc index 22a833d4870e7..4da79733c7157 100644 --- a/xla/service/hlo_module_group_metadata.cc +++ b/xla/service/hlo_module_group_metadata.cc @@ -15,12 +15,26 @@ limitations under the License. #include "xla/service/hlo_module_group_metadata.h" +#include +#include +#include +#include #include +#include +#include #include #include #include +#include +#include "absl/algorithm/container.h" #include "absl/container/flat_hash_set.h" +#include "absl/log/check.h" +#include "absl/log/log.h" +#include "absl/status/status.h" +#include "absl/status/statusor.h" +#include "absl/strings/str_cat.h" +#include "absl/types/span.h" #include "xla/hlo/analysis/hlo_alias_analysis.h" #include "xla/hlo/analysis/tuple_points_to_analysis.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" diff --git a/xla/service/hlo_module_group_metadata.h b/xla/service/hlo_module_group_metadata.h index a53fac2206acb..f74e6c5ba0310 100644 --- a/xla/service/hlo_module_group_metadata.h +++ b/xla/service/hlo_module_group_metadata.h @@ -23,8 +23,10 @@ limitations under the License. #include #include "absl/container/flat_hash_map.h" +#include "absl/log/check.h" #include "absl/status/status.h" #include "absl/status/statusor.h" +#include "absl/types/span.h" #include "xla/hlo/analysis/hlo_alias_analysis.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/xla/service/hlo_module_group_test.cc b/xla/service/hlo_module_group_test.cc index 007df88bdcc9d..26d2718211c35 100644 --- a/xla/service/hlo_module_group_test.cc +++ b/xla/service/hlo_module_group_test.cc @@ -15,6 +15,17 @@ limitations under the License. #include "xla/hlo/ir/hlo_module_group.h" +#include +#include +#include +#include +#include + +#include +#include +#include "absl/algorithm/container.h" +#include "absl/strings/str_format.h" +#include "absl/types/span.h" #include "xla/hlo/utils/hlo_matchers.h" #include "xla/service/hlo.pb.h" #include "xla/service/hlo_module_group_metadata.h" diff --git a/xla/service/hlo_module_group_util.cc b/xla/service/hlo_module_group_util.cc index de4b61736e747..e04becd8a0aa2 100644 --- a/xla/service/hlo_module_group_util.cc +++ b/xla/service/hlo_module_group_util.cc @@ -16,15 +16,22 @@ limitations under the License. #include "xla/service/hlo_module_group_util.h" #include -#include +#include +#include #include #include #include #include #include +#include #include "absl/container/flat_hash_set.h" -#include "absl/strings/str_cat.h" +#include "absl/log/check.h" +#include "absl/status/status.h" +#include "absl/status/statusor.h" +#include "absl/strings/str_join.h" +#include "absl/strings/string_view.h" +#include "absl/types/span.h" #include "xla/hlo/analysis/hlo_reachability.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_instructions.h" diff --git a/xla/service/hlo_module_metadata_test.cc b/xla/service/hlo_module_metadata_test.cc index fc64b31d019ab..1574c801c25b2 100644 --- a/xla/service/hlo_module_metadata_test.cc +++ b/xla/service/hlo_module_metadata_test.cc @@ -15,6 +15,11 @@ limitations under the License. #include "xla/hlo/ir/hlo_module_metadata.h" +#include + +#include +#include +#include "xla/service/hlo.pb.h" #include "xla/test.h" #include "xla/test_helpers.h" diff --git a/xla/service/hlo_module_test.cc b/xla/service/hlo_module_test.cc index 5f6e25acdf616..c96ba7a12e193 100644 --- a/xla/service/hlo_module_test.cc +++ b/xla/service/hlo_module_test.cc @@ -23,12 +23,14 @@ limitations under the License. #include #include +#include #include #include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_set.h" #include "absl/status/statusor.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_format.h" +#include "absl/strings/string_view.h" #include "absl/types/span.h" #include "xla/comparison_util.h" #include "xla/debug_options_flags.h" diff --git a/xla/service/hlo_module_util.cc b/xla/service/hlo_module_util.cc index 7ad2350dd1550..3c183fe9e2aa2 100644 --- a/xla/service/hlo_module_util.cc +++ b/xla/service/hlo_module_util.cc @@ -33,6 +33,7 @@ limitations under the License. #include "xla/hlo/parser/hlo_parser.h" #include "xla/service/compiler.h" #include "xla/service/computation_layout.h" +#include "xla/service/hlo.pb.h" #include "xla/service/hlo_module_config.h" #include "xla/service/hlo_verifier.h" #include "xla/shape.h" @@ -42,6 +43,7 @@ limitations under the License. #include "xla/tsl/platform/errors.h" #include "xla/tsl/platform/statusor.h" #include "xla/util.h" +#include "xla/xla.pb.h" namespace xla { diff --git a/xla/service/hlo_module_util.h b/xla/service/hlo_module_util.h index 4651a40c19ea3..1bcc50b681c87 100644 --- a/xla/service/hlo_module_util.h +++ b/xla/service/hlo_module_util.h @@ -29,9 +29,11 @@ limitations under the License. #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/parser/hlo_parser.h" #include "xla/service/compiler.h" +#include "xla/service/hlo.pb.h" #include "xla/service/hlo_module_config.h" #include "xla/shape.h" #include "xla/util.h" +#include "xla/xla.pb.h" namespace xla { diff --git a/xla/service/hlo_phi_graph.cc b/xla/service/hlo_phi_graph.cc index d272907ee9b0b..bc412fcb6c33f 100644 --- a/xla/service/hlo_phi_graph.cc +++ b/xla/service/hlo_phi_graph.cc @@ -15,7 +15,17 @@ limitations under the License. #include "xla/service/hlo_phi_graph.h" +#include +#include #include +#include + +#include "absl/algorithm/container.h" +#include "absl/container/flat_hash_set.h" +#include "absl/log/check.h" +#include "absl/log/log.h" +#include "absl/strings/str_cat.h" +#include "absl/types/span.h" namespace xla { HloValue::Id PhiGraph::GetOptimizedId(const HloValue& value) { diff --git a/xla/service/hlo_phi_graph.h b/xla/service/hlo_phi_graph.h index b8c56454f07ca..b09d54fb388e8 100644 --- a/xla/service/hlo_phi_graph.h +++ b/xla/service/hlo_phi_graph.h @@ -23,6 +23,7 @@ limitations under the License. #include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_set.h" +#include "absl/types/span.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/service/hlo_value.h" diff --git a/xla/service/hlo_phi_graph_test.cc b/xla/service/hlo_phi_graph_test.cc index b5184aa4d41ed..72f500a4a8420 100644 --- a/xla/service/hlo_phi_graph_test.cc +++ b/xla/service/hlo_phi_graph_test.cc @@ -15,6 +15,10 @@ limitations under the License. #include "xla/service/hlo_phi_graph.h" +#include +#include + +#include #include "xla/literal_util.h" #include "tsl/platform/test.h" diff --git a/xla/service/hlo_profile_printer.cc b/xla/service/hlo_profile_printer.cc index f26e430323fb5..e797a289bf552 100644 --- a/xla/service/hlo_profile_printer.cc +++ b/xla/service/hlo_profile_printer.cc @@ -15,8 +15,12 @@ limitations under the License. #include "xla/service/hlo_profile_printer.h" +#include +#include + #include "absl/algorithm/container.h" #include "absl/strings/str_cat.h" +#include "xla/service/hlo_profile_printer_data.pb.h" #include "xla/service/human_readable_profile_builder.h" namespace xla { diff --git a/xla/service/hlo_proto_util.cc b/xla/service/hlo_proto_util.cc index 04fcff317548f..197d21ea8b42b 100644 --- a/xla/service/hlo_proto_util.cc +++ b/xla/service/hlo_proto_util.cc @@ -15,10 +15,12 @@ limitations under the License. #include "xla/service/hlo_proto_util.h" -#include #include +#include "absl/status/statusor.h" +#include "xla/service/hlo.pb.h" #include "xla/util.h" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/hlo_proto_util.h b/xla/service/hlo_proto_util.h index 9a86bfcb0d503..d495b27974897 100644 --- a/xla/service/hlo_proto_util.h +++ b/xla/service/hlo_proto_util.h @@ -19,6 +19,7 @@ limitations under the License. #define XLA_SERVICE_HLO_PROTO_UTIL_H_ #include "absl/status/status.h" +#include "absl/status/statusor.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/service/buffer_assignment.h" #include "xla/service/hlo.pb.h" diff --git a/xla/service/hlo_proto_util_test.cc b/xla/service/hlo_proto_util_test.cc index d5ef461e58878..36fa024606bf2 100644 --- a/xla/service/hlo_proto_util_test.cc +++ b/xla/service/hlo_proto_util_test.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/service/hlo_proto_util.h" +#include +#include #include "xla/hlo/ir/hlo_opcode.h" #include "xla/service/hlo.pb.h" #include "xla/shape_util.h" diff --git a/xla/service/hlo_runner.cc b/xla/service/hlo_runner.cc index 3885cc2194be2..96c3573ddccf6 100644 --- a/xla/service/hlo_runner.cc +++ b/xla/service/hlo_runner.cc @@ -40,6 +40,7 @@ limitations under the License. #include "xla/service/computation_placer.h" #include "xla/service/executable.h" #include "xla/service/gpu/gpu_executable_run_options.h" +#include "xla/service/hlo.pb.h" #include "xla/service/hlo_module_util.h" #include "xla/service/hlo_runner_interface.h" #include "xla/service/maybe_owning_device_memory.h" diff --git a/xla/service/hlo_runner.h b/xla/service/hlo_runner.h index 3ee0234c98da0..563e4ce0295e2 100644 --- a/xla/service/hlo_runner.h +++ b/xla/service/hlo_runner.h @@ -37,6 +37,7 @@ limitations under the License. #include "xla/service/computation_layout.h" #include "xla/service/computation_placer.h" #include "xla/service/executable.h" +#include "xla/service/hlo.pb.h" #include "xla/service/hlo_runner_interface.h" #include "xla/service/service_executable_run_options.h" #include "xla/service/shaped_buffer.h" diff --git a/xla/service/hlo_runner_interface.cc b/xla/service/hlo_runner_interface.cc index 71ab84bfb0c26..16e0adeff48a4 100644 --- a/xla/service/hlo_runner_interface.cc +++ b/xla/service/hlo_runner_interface.cc @@ -23,6 +23,7 @@ limitations under the License. #include "absl/types/span.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/literal.h" +#include "xla/xla_data.pb.h" namespace xla {