diff --git a/xla/service/BUILD b/xla/service/BUILD index 42ef16c97ec00..78542bae019fd 100644 --- a/xla/service/BUILD +++ b/xla/service/BUILD @@ -107,8 +107,14 @@ cc_library( hdrs = ["collective_opt_utils.h"], deps = [ "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/types:span", ], ) @@ -156,6 +162,7 @@ xla_cc_test( ":all_reduce_promotion", ":pattern_matcher", ":pattern_matcher_gmock", + ":remote_mtest_lib_dynamic_padder_test_gpu_b200", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/tests:hlo_test_base", @@ -330,6 +337,7 @@ cc_library( ":source_target_pairs", "//xla:shape_util", "//xla:xla_data_proto_cc", + "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass", @@ -350,6 +358,8 @@ xla_cc_test( deps = [ ":collective_ops_utils", ":collective_permute_decomposer", + "//xla:xla_data_proto_cc", + "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", @@ -371,7 +381,10 @@ cc_library( deps = [ "//xla:literal", "//xla:util", + "@com_google_absl//absl/base", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings", ], ) @@ -479,6 +492,7 @@ xla_cc_test( "//xla:literal_util", "//xla:test_helpers", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/parser:hlo_parser", "//xla/hlo/pass:hlo_pass_pipeline", @@ -945,7 +959,9 @@ cc_library( "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/functional:function_ref", + "@com_google_absl//absl/log", "@com_google_absl//absl/log:check", + "@com_google_absl//absl/memory", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", @@ -1033,6 +1049,7 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/log", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:statusor", ], ) @@ -1363,6 +1380,7 @@ cc_library( ":compiler", ":computation_layout", ":dump", + ":hlo_proto_cc", ":platform_util", ":service", "//xla:debug_options_flags", @@ -1370,9 +1388,13 @@ cc_library( "//xla:types", "//xla:util", "//xla:xla_data_proto_cc", + "//xla:xla_proto_cc", "//xla/stream_executor:stream_executor_h", + "@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:logging", ], ) @@ -1552,16 +1574,25 @@ cc_library( ":computation_placer", ":executable", ":hlo_module_config", + ":hlo_proto_cc", ":metrics_hook_interface", + "//third_party/protobuf", "//xla:util", + "//xla:xla_data_proto_cc", + "//xla:xla_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/ir:hlo_module_group", "//xla/pjrt/distributed:key_value_store_interface", + "//xla/stream_executor:device_description_proto_cc", "//xla/stream_executor:dnn", "//xla/stream_executor:stream_executor_h", + "@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/status:statusor", "@com_google_absl//absl/strings", + "@com_google_absl//absl/synchronization", "@tsl//tsl/platform:env", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:protobuf", @@ -1659,6 +1690,7 @@ cc_library( "//xla/stream_executor:device_memory_allocator", "@com_google_absl//absl/base:core_headers", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/synchronization", @@ -1691,7 +1723,9 @@ cc_library( deps = [ "//xla:util", "//xla:xla_data_proto_cc", + "@com_google_absl//absl/base:core_headers", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/synchronization", ], ) @@ -1799,12 +1833,14 @@ xla_cc_test( "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", "@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/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", ], @@ -2382,6 +2418,7 @@ xla_test( "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", ], ) @@ -2463,7 +2500,10 @@ cc_library( "//xla:literal_util", "//xla:shape_util", "//xla:status_macros", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/types:span", ], ) @@ -2575,6 +2615,7 @@ xla_cc_test( ":hlo_module_config", ":pattern_matcher", ":pattern_matcher_gmock", + ":remote_mtest_lib_dynamic_padder_test_gpu_b200", "//xla:shape_util", "//xla:test", "//xla:types", @@ -2584,6 +2625,7 @@ 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_googletest//:gtest", "@tsl//tsl/platform:statusor", ], ) @@ -2683,11 +2725,14 @@ cc_library( "//xla:status_macros", "//xla:types", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "@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:statusor", "@com_google_absl//absl/strings", "@tsl//tsl/platform:errors", @@ -2709,6 +2754,8 @@ 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/strings:string_view", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:status", ], ) @@ -2733,12 +2780,14 @@ cc_library( "//xla/hlo/transforms/simplifiers:tuple_simplifier", "@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/container:inlined_vector", "@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", ], ) @@ -2758,6 +2807,9 @@ 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/log", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", ], @@ -3365,11 +3417,15 @@ cc_library( "//xla/stream_executor/host:host_platform_id", "//xla/stream_executor/rocm:rocm_platform_id", "//xla/stream_executor/sycl:sycl_platform_id", + "@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/memory", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", + "@com_google_absl//absl/synchronization", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:status", @@ -3386,8 +3442,10 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/stream_executor:platform", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/synchronization", "@tsl//tsl/platform:status", ], ) @@ -3643,10 +3701,13 @@ cc_library( hdrs = ["buffer_value.h"], deps = [ ":hlo_proto_cc", + "//third_party/protobuf:protobuf_lite", "//xla:shape_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/base:core_headers", + "@com_google_absl//absl/log:check", "@tsl//tsl/platform:logging", ], ) @@ -3805,7 +3866,12 @@ cc_library( "//xla:types", "//xla:util", "//xla:xla_data_proto_cc", + "@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/status:statusor", + "@com_google_absl//absl/synchronization", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:strcat", ], @@ -4495,7 +4561,10 @@ cc_library( "//xla:types", "//xla:xla_data_proto_cc", "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:str_format", ], ) @@ -5340,7 +5409,10 @@ cc_library( hdrs = ["compilation_stats.h"], deps = [ "//xla:types", + "@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/strings", "@com_google_absl//absl/strings:str_format", "@tsl//tsl/platform:env", @@ -5393,8 +5465,14 @@ cc_library( ":hlo_creation_utils", "//xla:status_macros", "//xla:types", + "//xla:xla_data_proto_cc", "//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:errors", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:status", @@ -5412,6 +5490,7 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", # fixdeps: keep + "@com_google_googletest//:gtest", ], ) @@ -5867,7 +5946,10 @@ cc_library( "@com_google_absl//absl/algorithm:container", "@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/memory", + "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/synchronization", @@ -5910,6 +5992,7 @@ xla_cc_test( "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:casts", "@tsl//tsl/platform:protobuf", ], @@ -6037,8 +6120,14 @@ cc_library( copts = tsl_copts(), deps = [ ":hlo_creation_utils", + "//xla:xla_data_proto_cc", "//xla/hlo/pass:hlo_pass", "//xla/service/cpu:onednn_contraction_rewriter", + "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", ], ) @@ -6148,9 +6237,12 @@ xla_cc_test( ":change_op_data_type", ":pattern_matcher", ":pattern_matcher_gmock", + ":remote_mtest_lib_dynamic_padder_test_gpu_b200", + "//xla:xla_data_proto_cc", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", # fixdeps: keep "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest", ], ) diff --git a/xla/service/algebraic_simplifier_overflow_test.cc b/xla/service/algebraic_simplifier_overflow_test.cc index 071f9994b54a0..8ea4e36877894 100644 --- a/xla/service/algebraic_simplifier_overflow_test.cc +++ b/xla/service/algebraic_simplifier_overflow_test.cc @@ -15,7 +15,6 @@ limitations under the License. #include #include -#include #include #include "xla/error_spec.h" diff --git a/xla/service/all_gather_decomposer.cc b/xla/service/all_gather_decomposer.cc index ce3ed5f5f4402..83c259c9778eb 100644 --- a/xla/service/all_gather_decomposer.cc +++ b/xla/service/all_gather_decomposer.cc @@ -21,6 +21,7 @@ limitations under the License. #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_casting_utils.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/xla/service/all_reduce_promotion_test.cc b/xla/service/all_reduce_promotion_test.cc index 86d5fde6eb71c..5dd9c980d0349 100644 --- a/xla/service/all_reduce_promotion_test.cc +++ b/xla/service/all_reduce_promotion_test.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/service/all_reduce_promotion.h" +#include + #include #include #include "absl/strings/string_view.h" diff --git a/xla/service/all_reduce_simplifier_test.cc b/xla/service/all_reduce_simplifier_test.cc index 7048bf20a6163..7830e15738675 100644 --- a/xla/service/all_reduce_simplifier_test.cc +++ b/xla/service/all_reduce_simplifier_test.cc @@ -16,8 +16,9 @@ limitations under the License. #include "xla/service/all_reduce_simplifier.h" #include -#include +#include +#include #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/parser/hlo_parser.h" diff --git a/xla/service/allocation_tracker.cc b/xla/service/allocation_tracker.cc index 507107723093a..6f9d7f4d3ccc3 100644 --- a/xla/service/allocation_tracker.cc +++ b/xla/service/allocation_tracker.cc @@ -22,7 +22,9 @@ limitations under the License. #include #include +#include "absl/log/log.h" #include "absl/status/status.h" +#include "absl/status/statusor.h" #include "absl/synchronization/mutex.h" #include "xla/service/shaped_buffer.h" #include "xla/shape.h" diff --git a/xla/service/batchnorm_expander_test.cc b/xla/service/batchnorm_expander_test.cc index 25cfa87004be0..93d9e4fbea061 100644 --- a/xla/service/batchnorm_expander_test.cc +++ b/xla/service/batchnorm_expander_test.cc @@ -15,9 +15,10 @@ limitations under the License. #include "xla/service/batchnorm_expander.h" +#include #include -#include +#include #include "xla/error_spec.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/xla/service/buffer_assignment.h b/xla/service/buffer_assignment.h index 24093327cfd01..64c550661b391 100644 --- a/xla/service/buffer_assignment.h +++ b/xla/service/buffer_assignment.h @@ -30,6 +30,7 @@ limitations under the License. #include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_set.h" +#include "absl/log/check.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/types/span.h" diff --git a/xla/service/buffer_assignment_test.cc b/xla/service/buffer_assignment_test.cc index d91b13a7bff1d..5c52d82e615e9 100644 --- a/xla/service/buffer_assignment_test.cc +++ b/xla/service/buffer_assignment_test.cc @@ -22,7 +22,10 @@ limitations under the License. #include #include +#include +#include #include "absl/algorithm/container.h" +#include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_set.h" #include "absl/log/log.h" #include "absl/status/status.h" diff --git a/xla/service/buffer_value.cc b/xla/service/buffer_value.cc index 70b02a09dc002..78a328657fd70 100644 --- a/xla/service/buffer_value.cc +++ b/xla/service/buffer_value.cc @@ -18,8 +18,11 @@ limitations under the License. #include #include +#include "absl/algorithm/container.h" +#include "third_party/protobuf/repeated_field.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" +#include "xla/service/hlo.pb.h" namespace xla { diff --git a/xla/service/buffer_value.h b/xla/service/buffer_value.h index f75ec9b3998c8..e1cbb29f9b238 100644 --- a/xla/service/buffer_value.h +++ b/xla/service/buffer_value.h @@ -22,6 +22,7 @@ limitations under the License. #include #include "absl/base/attributes.h" +#include "absl/log/check.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/service/hlo.pb.h" #include "xla/shape.h" diff --git a/xla/service/call_graph.cc b/xla/service/call_graph.cc index 80515e13ea751..42c739c46f742 100644 --- a/xla/service/call_graph.cc +++ b/xla/service/call_graph.cc @@ -17,15 +17,22 @@ limitations under the License. #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/memory/memory.h" +#include "absl/status/status.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_format.h" #include "absl/strings/str_join.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/map_util.h" diff --git a/xla/service/call_inliner_test.cc b/xla/service/call_inliner_test.cc index 602a126344f28..d7bf298053473 100644 --- a/xla/service/call_inliner_test.cc +++ b/xla/service/call_inliner_test.cc @@ -16,8 +16,11 @@ limitations under the License. #include "xla/service/call_inliner.h" #include +#include #include +#include +#include #include "absl/log/log.h" #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/xla/service/change_op_data_type.h b/xla/service/change_op_data_type.h index 43e929d6e8e7c..54c03880f0ddd 100644 --- a/xla/service/change_op_data_type.h +++ b/xla/service/change_op_data_type.h @@ -20,7 +20,13 @@ limitations under the License. #include #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 "absl/types/span.h" #include "xla/hlo/pass/hlo_pass_interface.h" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/change_op_data_type_test.cc b/xla/service/change_op_data_type_test.cc index 2bd746b4bc6bd..c67b23de4f1c6 100644 --- a/xla/service/change_op_data_type_test.cc +++ b/xla/service/change_op_data_type_test.cc @@ -15,14 +15,15 @@ limitations under the License. #include "xla/service/change_op_data_type.h" -#include -#include -#include +#include +#include +#include #include "absl/types/span.h" #include "xla/service/pattern_matcher.h" #include "xla/service/pattern_matcher_gmock.h" #include "xla/tests/hlo_test_base.h" +#include "xla/xla_data.pb.h" namespace xla { namespace { diff --git a/xla/service/channel_tracker.cc b/xla/service/channel_tracker.cc index 8ad2445f082ef..adb55816be51c 100644 --- a/xla/service/channel_tracker.cc +++ b/xla/service/channel_tracker.cc @@ -15,7 +15,10 @@ limitations under the License. #include "xla/service/channel_tracker.h" +#include "absl/status/statusor.h" +#include "absl/synchronization/mutex.h" #include "xla/util.h" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/channel_tracker.h b/xla/service/channel_tracker.h index 0f1bc594ed621..28960d547cca0 100644 --- a/xla/service/channel_tracker.h +++ b/xla/service/channel_tracker.h @@ -16,7 +16,9 @@ limitations under the License. #ifndef XLA_SERVICE_CHANNEL_TRACKER_H_ #define XLA_SERVICE_CHANNEL_TRACKER_H_ +#include "absl/base/thread_annotations.h" #include "absl/status/statusor.h" +#include "absl/synchronization/mutex.h" #include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/collective_decomposer_utils.cc b/xla/service/collective_decomposer_utils.cc index d86c6b5ae4e91..32fc5b8fc699d 100644 --- a/xla/service/collective_decomposer_utils.cc +++ b/xla/service/collective_decomposer_utils.cc @@ -15,9 +15,14 @@ limitations under the License. #include "xla/service/collective_decomposer_utils.h" +#include +#include #include #include +#include +#include "absl/status/statusor.h" +#include "absl/types/span.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_instructions.h" @@ -26,6 +31,7 @@ limitations under the License. #include "xla/service/hlo_module_config.h" #include "xla/shape_util.h" #include "xla/status_macros.h" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/collective_decomposer_utils.h b/xla/service/collective_decomposer_utils.h index 905ab12c24069..facd46e9da4e0 100644 --- a/xla/service/collective_decomposer_utils.h +++ b/xla/service/collective_decomposer_utils.h @@ -15,8 +15,11 @@ limitations under the License. #include +#include "absl/status/statusor.h" +#include "absl/types/span.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/service/collective_ops_utils.h" +#include "xla/xla_data.pb.h" #ifndef XLA_SERVICE_COLLECTIVE_DECOMPOSER_UTILS_H_ #define XLA_SERVICE_COLLECTIVE_DECOMPOSER_UTILS_H_ diff --git a/xla/service/collective_ops_utils.h b/xla/service/collective_ops_utils.h index 0343eaa7f4635..1374cc8418a4e 100644 --- a/xla/service/collective_ops_utils.h +++ b/xla/service/collective_ops_utils.h @@ -42,6 +42,7 @@ limitations under the License. #include "xla/service/pattern_matcher.h" #include "xla/service/source_target_pairs.h" #include "xla/stream_executor/device_memory.h" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/collective_opt_utils.cc b/xla/service/collective_opt_utils.cc index 693b616c449ec..25f39634bf91f 100644 --- a/xla/service/collective_opt_utils.cc +++ b/xla/service/collective_opt_utils.cc @@ -15,18 +15,25 @@ limitations under the License. #include "xla/service/collective_opt_utils.h" +#include #include #include #include #include #include "absl/algorithm/container.h" +#include "absl/container/flat_hash_map.h" +#include "absl/container/inlined_vector.h" +#include "absl/log/check.h" +#include "absl/log/log.h" +#include "absl/types/span.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_instructions.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/util.h" +#include "xla/xla_data.pb.h" namespace xla { namespace { diff --git a/xla/service/collective_permute_decomposer.cc b/xla/service/collective_permute_decomposer.cc index 7e1fc08837bfc..c92feccc3f38d 100644 --- a/xla/service/collective_permute_decomposer.cc +++ b/xla/service/collective_permute_decomposer.cc @@ -46,6 +46,7 @@ limitations under the License. #include "xla/shape.h" #include "xla/shape_util.h" #include "xla/tsl/platform/errors.h" +#include "xla/xla.pb.h" #include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/collective_permute_decomposer.h b/xla/service/collective_permute_decomposer.h index 0c425a4acf8ed..6d530bf0fe630 100644 --- a/xla/service/collective_permute_decomposer.h +++ b/xla/service/collective_permute_decomposer.h @@ -23,6 +23,7 @@ limitations under the License. #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/pass/hlo_pass_interface.h" +#include "xla/xla.pb.h" namespace xla { diff --git a/xla/service/collective_permute_decomposer_test.cc b/xla/service/collective_permute_decomposer_test.cc index 9187c541d2b13..3dd8037df9627 100644 --- a/xla/service/collective_permute_decomposer_test.cc +++ b/xla/service/collective_permute_decomposer_test.cc @@ -33,6 +33,8 @@ limitations under the License. #include "xla/service/collective_ops_utils.h" #include "xla/tsl/lib/core/status_test_util.h" #include "xla/tsl/platform/statusor.h" +#include "xla/xla.pb.h" +#include "xla/xla_data.pb.h" namespace xla { namespace { diff --git a/xla/service/collective_pipeliner.h b/xla/service/collective_pipeliner.h index 0b5deb81d77df..674aa7beb9974 100644 --- a/xla/service/collective_pipeliner.h +++ b/xla/service/collective_pipeliner.h @@ -19,6 +19,7 @@ limitations under the License. #include #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_module.h" diff --git a/xla/service/collective_pipeliner_test.cc b/xla/service/collective_pipeliner_test.cc index c791b74e497b1..66c7c6c2e5992 100644 --- a/xla/service/collective_pipeliner_test.cc +++ b/xla/service/collective_pipeliner_test.cc @@ -48,6 +48,7 @@ limitations under the License. #include "xla/tests/hlo_test_base.h" #include "xla/tsl/lib/core/status_test_util.h" #include "xla/util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/statusor.h" namespace xla { diff --git a/xla/service/compilation_cache.cc b/xla/service/compilation_cache.cc index 9187924b6d6a9..59838513420aa 100644 --- a/xla/service/compilation_cache.cc +++ b/xla/service/compilation_cache.cc @@ -15,8 +15,15 @@ limitations under the License. #include "xla/service/compilation_cache.h" +#include +#include #include +#include "absl/base/const_init.h" +#include "absl/log/check.h" +#include "absl/log/log.h" +#include "absl/status/statusor.h" +#include "absl/synchronization/mutex.h" #include "xla/types.h" #include "xla/util.h" #include "xla/xla_data.pb.h" diff --git a/xla/service/compilation_cache.h b/xla/service/compilation_cache.h index 65384bf8340c4..d7585ba764c12 100644 --- a/xla/service/compilation_cache.h +++ b/xla/service/compilation_cache.h @@ -20,10 +20,14 @@ limitations under the License. #include #include +#include "absl/base/thread_annotations.h" #include "absl/container/flat_hash_map.h" +#include "absl/status/statusor.h" +#include "absl/synchronization/mutex.h" #include "xla/service/executable.h" #include "xla/service/hlo_module_config.h" #include "xla/types.h" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/xla/service/compilation_environments.cc b/xla/service/compilation_environments.cc index f2e0dff2b5c5b..786404eb9f097 100644 --- a/xla/service/compilation_environments.cc +++ b/xla/service/compilation_environments.cc @@ -15,7 +15,6 @@ limitations under the License. #include "xla/service/compilation_environments.h" -#include #include #include #include @@ -27,7 +26,10 @@ limitations under the License. #include "absl/base/const_init.h" #include "absl/base/thread_annotations.h" #include "absl/container/flat_hash_map.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/memory/memory.h" +#include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_join.h" diff --git a/xla/service/compilation_environments_test.cc b/xla/service/compilation_environments_test.cc index 35058aefd4599..a643f7dd35bb3 100644 --- a/xla/service/compilation_environments_test.cc +++ b/xla/service/compilation_environments_test.cc @@ -18,7 +18,7 @@ limitations under the License. #include #include -#include "absl/status/statusor.h" +#include #include "xla/service/test_compilation_environment.pb.h" #include "xla/test.h" #include "xla/tsl/lib/core/status_test_util.h" diff --git a/xla/service/compilation_stats.cc b/xla/service/compilation_stats.cc index d08994e4273b5..509d3b0e9fc2d 100644 --- a/xla/service/compilation_stats.cc +++ b/xla/service/compilation_stats.cc @@ -15,12 +15,17 @@ limitations under the License. #include "xla/service/compilation_stats.h" -#include +#include #include #include +#include +#include +#include "absl/algorithm/container.h" #include "absl/container/flat_hash_map.h" -#include "absl/strings/str_format.h" +#include "absl/log/check.h" +#include "absl/log/log.h" +#include "absl/strings/string_view.h" #include "xla/types.h" #include "tsl/platform/env.h" diff --git a/xla/service/compile_only_service.cc b/xla/service/compile_only_service.cc index a0e48bd8174d3..c5ffb803dbd78 100644 --- a/xla/service/compile_only_service.cc +++ b/xla/service/compile_only_service.cc @@ -15,11 +15,13 @@ limitations under the License. #include "xla/service/compile_only_service.h" -#include +#include #include #include -#include "absl/strings/str_cat.h" +#include "absl/log/check.h" +#include "absl/status/statusor.h" +#include "absl/types/span.h" #include "xla/debug_options_flags.h" #include "xla/service/backend.h" #include "xla/service/computation_layout.h" @@ -29,6 +31,7 @@ limitations under the License. #include "xla/stream_executor/stream_executor.h" #include "xla/types.h" #include "xla/util.h" +#include "xla/xla.pb.h" #include "tsl/platform/logging.h" namespace xla { diff --git a/xla/service/compile_only_service.h b/xla/service/compile_only_service.h index 0238a16f28294..0379fe6d9e422 100644 --- a/xla/service/compile_only_service.h +++ b/xla/service/compile_only_service.h @@ -16,9 +16,12 @@ limitations under the License. #ifndef XLA_SERVICE_COMPILE_ONLY_SERVICE_H_ #define XLA_SERVICE_COMPILE_ONLY_SERVICE_H_ +#include "absl/status/status.h" #include "absl/status/statusor.h" +#include "absl/types/span.h" #include "xla/service/backend.h" #include "xla/service/compiler.h" +#include "xla/service/hlo.pb.h" #include "xla/service/service.h" #include "xla/stream_executor/stream_executor.h" #include "xla/xla_data.pb.h" diff --git a/xla/service/compiler.cc b/xla/service/compiler.cc index 43f33e30ea418..cadf3f88aae96 100644 --- a/xla/service/compiler.cc +++ b/xla/service/compiler.cc @@ -21,6 +21,14 @@ limitations under the License. #include #include +#include "absl/base/const_init.h" +#include "absl/container/flat_hash_map.h" +#include "absl/log/check.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" +#include "absl/synchronization/mutex.h" +#include "third_party/protobuf/message.h" +#include "xla/stream_executor/device_description.pb.h" #include "xla/stream_executor/dnn.h" #include "xla/util.h" #include "tsl/platform/logging.h" diff --git a/xla/service/compiler.h b/xla/service/compiler.h index dd923a4ce4504..e69439e1dd101 100644 --- a/xla/service/compiler.h +++ b/xla/service/compiler.h @@ -28,8 +28,11 @@ limitations under the License. #include #include "absl/container/flat_hash_map.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/status/statusor.h" #include "absl/strings/string_view.h" +#include "absl/synchronization/mutex.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/ir/hlo_module_group.h" @@ -38,9 +41,13 @@ limitations under the License. #include "xla/service/buffer_value.h" #include "xla/service/computation_placer.h" #include "xla/service/executable.h" +#include "xla/service/hlo.pb.h" #include "xla/service/hlo_module_config.h" #include "xla/service/metrics_hook_interface.h" +#include "xla/stream_executor/device_description.pb.h" #include "xla/stream_executor/stream_executor.h" +#include "xla/xla.pb.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/protobuf.h" #include "tsl/platform/threadpool.h" diff --git a/xla/service/computation_layout.cc b/xla/service/computation_layout.cc index d62e1a19afeeb..53088df39c3ef 100644 --- a/xla/service/computation_layout.cc +++ b/xla/service/computation_layout.cc @@ -15,12 +15,16 @@ limitations under the License. #include "xla/service/computation_layout.h" +#include #include #include #include #include "absl/algorithm/container.h" +#include "absl/status/status.h" +#include "absl/status/statusor.h" #include "absl/strings/str_cat.h" +#include "absl/strings/str_format.h" #include "xla/layout.h" #include "xla/printer.h" #include "xla/shape.h" diff --git a/xla/service/computation_layout.h b/xla/service/computation_layout.h index b6c947b2b7c9e..bc22ae19e8fbd 100644 --- a/xla/service/computation_layout.h +++ b/xla/service/computation_layout.h @@ -21,6 +21,7 @@ limitations under the License. #include #include +#include "absl/status/statusor.h" #include "xla/printer.h" #include "xla/shape_layout.h" #include "xla/types.h" diff --git a/xla/service/computation_placer.cc b/xla/service/computation_placer.cc index 43f351a548959..c2bb4a315c102 100644 --- a/xla/service/computation_placer.cc +++ b/xla/service/computation_placer.cc @@ -16,14 +16,18 @@ limitations under the License. #include "xla/service/computation_placer.h" #include +#include #include #include #include #include -#include "absl/status/status.h" +#include "absl/base/const_init.h" +#include "absl/container/flat_hash_map.h" +#include "absl/log/log.h" #include "absl/status/statusor.h" #include "absl/strings/str_cat.h" +#include "absl/synchronization/mutex.h" #include "xla/literal.h" #include "xla/service/global_device_id.h" #include "xla/shape_util.h" @@ -34,6 +38,7 @@ limitations under the License. #include "xla/stream_executor/sycl/sycl_platform_id.h" #include "xla/types.h" #include "xla/util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" #include "tsl/platform/logging.h" #include "tsl/platform/status.h" diff --git a/xla/service/computation_placer.h b/xla/service/computation_placer.h index 552fce2d84ba9..1a0a5236e0805 100644 --- a/xla/service/computation_placer.h +++ b/xla/service/computation_placer.h @@ -22,8 +22,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/synchronization/mutex.h" #include "xla/array2d.h" #include "xla/service/global_device_id.h" #include "xla/stream_executor/platform.h" diff --git a/xla/service/conditional_code_motion.cc b/xla/service/conditional_code_motion.cc index f1ca8db8633fa..b8b05b56020cc 100644 --- a/xla/service/conditional_code_motion.cc +++ b/xla/service/conditional_code_motion.cc @@ -17,21 +17,24 @@ limitations under the License. #include #include -#include -#include -#include +#include #include #include #include #include "absl/algorithm/container.h" #include "absl/container/flat_hash_map.h" +#include "absl/container/flat_hash_set.h" #include "absl/container/inlined_vector.h" #include "absl/log/check.h" #include "absl/log/log.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/numbers.h" +#include "absl/strings/str_cat.h" +#include "absl/strings/str_split.h" +#include "absl/strings/string_view.h" +#include "absl/types/span.h" #include "xla/debug_options_flags.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/xla/service/conditional_code_motion.h b/xla/service/conditional_code_motion.h index 4f36e501ffc98..72b1633c52faa 100644 --- a/xla/service/conditional_code_motion.h +++ b/xla/service/conditional_code_motion.h @@ -18,6 +18,10 @@ limitations under the License. #include +#include "absl/algorithm/container.h" +#include "absl/container/flat_hash_map.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/ir/hlo_module.h" diff --git a/xla/service/conditional_code_motion_test.cc b/xla/service/conditional_code_motion_test.cc index 1398a9b1fdc8d..6db7333ce8a2e 100644 --- a/xla/service/conditional_code_motion_test.cc +++ b/xla/service/conditional_code_motion_test.cc @@ -15,12 +15,16 @@ limitations under the License. #include "xla/service/conditional_code_motion.h" +#include #include #include #include -#include #include +#include +#include "absl/log/check.h" +#include "absl/log/log.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/conditional_simplifier.cc b/xla/service/conditional_simplifier.cc index edd01e32da60c..615bc86afdcdc 100644 --- a/xla/service/conditional_simplifier.cc +++ b/xla/service/conditional_simplifier.cc @@ -15,17 +15,21 @@ limitations under the License. #include "xla/service/conditional_simplifier.h" +#include +#include +#include #include #include -#include #include #include #include "absl/algorithm/container.h" #include "absl/container/flat_hash_map.h" #include "absl/container/flat_hash_set.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/status/statusor.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" @@ -38,6 +42,7 @@ limitations under the License. #include "xla/status_macros.h" #include "xla/types.h" #include "xla/util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" namespace xla { diff --git a/xla/service/conditional_simplifier.h b/xla/service/conditional_simplifier.h index 01363f94580b6..d1f9986c4dd8b 100644 --- a/xla/service/conditional_simplifier.h +++ b/xla/service/conditional_simplifier.h @@ -16,6 +16,7 @@ limitations under the License. #ifndef XLA_SERVICE_CONDITIONAL_SIMPLIFIER_H_ #define XLA_SERVICE_CONDITIONAL_SIMPLIFIER_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" diff --git a/xla/service/conditional_simplifier_test.cc b/xla/service/conditional_simplifier_test.cc index 8394cdcdc7a00..1cf0231dfe95a 100644 --- a/xla/service/conditional_simplifier_test.cc +++ b/xla/service/conditional_simplifier_test.cc @@ -15,9 +15,14 @@ limitations under the License. #include "xla/service/conditional_simplifier.h" +#include +#include #include #include +#include +#include +#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/conditional_to_select.cc b/xla/service/conditional_to_select.cc index 11246fa29cf3f..bc33b18311334 100644 --- a/xla/service/conditional_to_select.cc +++ b/xla/service/conditional_to_select.cc @@ -15,6 +15,14 @@ limitations under the License. #include "xla/service/conditional_to_select.h" +#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" @@ -23,6 +31,7 @@ limitations under the License. #include "xla/service/hlo_creation_utils.h" #include "xla/status_macros.h" #include "xla/types.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" #include "tsl/platform/logging.h" #include "tsl/platform/status.h" diff --git a/xla/service/conditional_to_select.h b/xla/service/conditional_to_select.h index 4e3676468fd8a..8bba94a9329ff 100644 --- a/xla/service/conditional_to_select.h +++ b/xla/service/conditional_to_select.h @@ -16,6 +16,9 @@ limitations under the License. #ifndef XLA_SERVICE_CONDITIONAL_TO_SELECT_H_ #define XLA_SERVICE_CONDITIONAL_TO_SELECT_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/conditional_to_select_test.cc b/xla/service/conditional_to_select_test.cc index b32aeacf14e40..5321862915090 100644 --- a/xla/service/conditional_to_select_test.cc +++ b/xla/service/conditional_to_select_test.cc @@ -16,8 +16,10 @@ limitations under the License. #include "xla/service/conditional_to_select.h" #include -#include +#include +#include +#include #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/constant_value.cc b/xla/service/constant_value.cc index 14de2501c5c0e..0555a757b753f 100644 --- a/xla/service/constant_value.cc +++ b/xla/service/constant_value.cc @@ -15,8 +15,14 @@ limitations under the License. #include "xla/service/constant_value.h" +#include #include +#include "absl/base/casts.h" +#include "absl/log/check.h" +#include "absl/status/statusor.h" +#include "absl/strings/str_cat.h" + namespace xla { absl::StatusOr ConstantValue::FromLiteral( diff --git a/xla/service/constant_value.h b/xla/service/constant_value.h index 418fcf56ef8e2..3dd92b95f5ce1 100644 --- a/xla/service/constant_value.h +++ b/xla/service/constant_value.h @@ -18,6 +18,7 @@ limitations under the License. #include +#include "absl/base/casts.h" #include "absl/status/statusor.h" #include "xla/literal.h" #include "xla/util.h" diff --git a/xla/service/constant_value_test.cc b/xla/service/constant_value_test.cc index fd468fbe5f01f..574044b67dcc9 100644 --- a/xla/service/constant_value_test.cc +++ b/xla/service/constant_value_test.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/service/constant_value.h" +#include + #include #include "xla/literal_util.h"