From 4c3aa84761903f85629b8a57ef7768032b332743 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Thu, 5 Dec 2024 05:05:17 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 703074301 --- third_party/xla/xla/hlo/transforms/BUILD | 58 +++++++++++++++++++ .../simplifiers/batch_dot_simplification.cc | 4 ++ .../batch_dot_simplification_test.cc | 3 + .../simplifiers/broadcast_canonicalizer.h | 3 + .../broadcast_canonicalizer_test.cc | 2 - .../simplifiers/conditional_canonicalizer.cc | 5 ++ .../simplifiers/conditional_canonicalizer.h | 3 + .../conditional_canonicalizer_test.cc | 2 + .../transforms/simplifiers/convert_mover.cc | 3 + .../simplifiers/convert_mover_test.cc | 2 + .../convert_operand_folder_test.cc | 2 + .../convolution_group_converter.cc | 7 ++- .../simplifiers/convolution_group_converter.h | 2 + .../simplifiers/dot_dimension_merger.cc | 2 + .../hlo/transforms/simplifiers/dot_merger.cc | 2 + .../hlo/transforms/simplifiers/dot_merger.h | 1 + .../transforms/simplifiers/dot_merger_test.cc | 4 ++ .../dynamic_dimension_simplifier.cc | 6 ++ .../dynamic_dimension_simplifier.h | 3 + .../dynamic_dimension_simplifier_test.cc | 3 - .../simplifiers/flatten_call_graph.h | 2 + .../simplifiers/float_normalization.h | 5 ++ .../simplifiers/fusion_constant_sinking.h | 3 + .../simplifiers/gather_simplifier.cc | 2 + .../simplifiers/gather_simplifier.h | 2 + .../simplifiers/gather_simplifier_test.cc | 2 + .../simplifiers/hlo_constant_splitter.cc | 2 +- .../transforms/simplifiers/hlo_dce_test.cc | 2 +- .../simplifiers/hlo_element_type_converter.cc | 9 ++- .../simplifiers/hlo_element_type_converter.h | 4 ++ .../hlo_element_type_converter_test.cc | 5 ++ .../simplifiers/hlo_rematerialization.cc | 2 + .../simplifiers/hlo_rematerialization_test.cc | 3 + .../host_memory_transfer_asyncifier.cc | 1 + .../host_memory_transfer_asyncifier.h | 1 + ...optimize_input_output_buffer_alias_test.cc | 2 +- .../simplifiers/reduce_window_rewriter.cc | 1 - .../transforms/simplifiers/reshape_mover.cc | 12 +++- .../transforms/simplifiers/reshape_mover.h | 4 ++ .../simplifiers/reshape_mover_test.cc | 3 + .../simplifiers/result_caster_test.cc | 3 + .../simplifiers/root_instruction_sinker.cc | 4 ++ .../simplifiers/root_instruction_sinker.h | 3 + .../root_instruction_sinker_test.cc | 3 + .../transforms/simplifiers/slice_sinker.cc | 4 +- .../simplifiers/slice_sinker_test.cc | 1 - .../transforms/simplifiers/sort_simplifier.cc | 2 + .../simplifiers/sort_simplifier_test.cc | 2 + .../simplifiers/sub_byte_normalization.cc | 5 ++ .../simplifiers/tree_reduction_rewriter.cc | 6 +- .../simplifiers/tree_reduction_rewriter.h | 2 + 51 files changed, 201 insertions(+), 18 deletions(-) diff --git a/third_party/xla/xla/hlo/transforms/BUILD b/third_party/xla/xla/hlo/transforms/BUILD index 58ac67cc258a8a..bc1c0c2424bdb9 100644 --- a/third_party/xla/xla/hlo/transforms/BUILD +++ b/third_party/xla/xla/hlo/transforms/BUILD @@ -183,6 +183,7 @@ cc_library( "@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/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@local_tsl//tsl/platform:errors", @@ -684,6 +685,9 @@ cc_library( "//xla/hlo/pass:hlo_pass", "//xla/service:shape_inference", "@com_google_absl//absl/algorithm:container", + "@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", "@local_tsl//tsl/platform:errors", @@ -895,6 +899,9 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/service:hlo_creation_utils", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@local_tsl//tsl/platform:errors", "@local_tsl//tsl/platform:logging", @@ -1052,6 +1059,7 @@ cc_library( "//xla/service:hlo_creation_utils", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -1081,6 +1089,7 @@ cc_library( "//xla:protobuf_util", "//xla:shape_util", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/service:shape_inference", @@ -1105,12 +1114,14 @@ xla_cc_test( ":algebraic_simplifier", ":dot_merger", "//xla:shape_util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:statusor", "@local_tsl//tsl/platform:test_main", # fixdeps: keep ], @@ -1200,6 +1211,14 @@ cc_library( "//xla/hlo/pass:hlo_pass", "//xla/service:hlo_creation_utils", "@com_google_absl//absl/algorithm:container", + "@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:statusor", + "@com_google_absl//absl/strings", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", "@local_tsl//tsl/platform:errors", ], ) @@ -1262,6 +1281,9 @@ cc_library( "//xla:status_macros", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", ], ) @@ -1303,6 +1325,8 @@ xla_cc_test( "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/status", + "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:test_main", ], ) @@ -1359,6 +1383,7 @@ cc_library( "//xla:shape_util", "//xla:status_macros", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/analysis:hlo_dataflow_analysis", "//xla/hlo/analysis:tuple_points_to_analysis", "//xla/hlo/ir:hlo", @@ -1419,11 +1444,14 @@ xla_cc_test( ":hlo_rematerialization_test_utils", "//xla:shape_util", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/utils:hlo_matchers", "//xla/service:hlo_cost_analysis", "//xla/tsl/lib/core:status_test_util", + "@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", "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:statusor", @@ -1447,6 +1475,7 @@ xla_cc_test( "//xla/service:pattern_matcher_gmock", "//xla/tests:literal_test_util", "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest_main", "@local_tsl//tsl/platform:test_main", @@ -1525,10 +1554,15 @@ cc_library( "//xla:literal", "//xla:shape_util", "//xla:types", + "//xla:xla_data_proto_cc", "//xla/hlo/evaluator:hlo_evaluator", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/hlo/utils:hlo_query", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", "@local_tsl//tsl/platform:errors", ], ) @@ -1538,8 +1572,10 @@ xla_cc_test( srcs = ["simplifiers/hlo_element_type_converter_test.cc"], deps = [ ":hlo_element_type_converter", + "//xla:xla_data_proto_cc", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", + "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:test_main", ], ) @@ -1552,6 +1588,11 @@ cc_library( "//xla:status_macros", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/status", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", + "@com_google_absl//absl/types:span", ], ) @@ -1569,6 +1610,7 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "//xla/tests:literal_test_util", "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:test_main", ], ) @@ -1684,6 +1726,10 @@ cc_library( "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "//xla/service:tuple_util", + "@com_google_absl//absl/container:flat_hash_set", + "@com_google_absl//absl/log:check", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", ], ) @@ -1694,6 +1740,8 @@ xla_cc_test( ":root_instruction_sinker", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:test_main", ], ) @@ -1741,6 +1789,7 @@ cc_library( deps = [ "//xla:shape_util", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", "@com_google_absl//absl/container:flat_hash_set", @@ -2010,6 +2059,7 @@ xla_cc_test( "//xla:shape_util", "//xla:test", "//xla:test_helpers", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_absl//absl/status", @@ -2225,10 +2275,12 @@ xla_cc_test( deps = [ ":result_caster", "//xla:shape_util", + "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:statusor", "@local_tsl//tsl/platform:test_main", ], @@ -2293,6 +2345,8 @@ cc_library( "//xla/service:gather_scatter_utils", "//xla/service:hlo_creation_utils", "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/status:statusor", + "@com_google_absl//absl/strings:string_view", "@local_tsl//tsl/platform:statusor", ], ) @@ -2375,9 +2429,11 @@ cc_library( deps = [ "//xla:shape_layout", "//xla:shape_util", + "//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/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@local_tsl//tsl/platform:errors", @@ -2408,6 +2464,8 @@ xla_cc_test( deps = [ ":gather_simplifier", "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", "@local_tsl//tsl/platform:test_main", ], ) diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification.cc b/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification.cc index 52fc16518a0302..8bc868ff2a3497 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification.cc @@ -15,6 +15,10 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/batch_dot_simplification.h" +#include +#include +#include + #include "absl/algorithm/container.h" #include "absl/container/flat_hash_set.h" #include "absl/log/log.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification_test.cc index ef2c10ac0b36dd..0cc0fca1e703cc 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/batch_dot_simplification_test.cc @@ -15,6 +15,9 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/batch_dot_simplification.h" +#include +#include + #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer.h b/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer.h index 08cc85b814933b..fe21b1b997579c 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer.h @@ -18,6 +18,9 @@ limitations under the License. #include +#include "absl/container/flat_hash_set.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer_test.cc index a227b1324d28f3..10e83a21695406 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/broadcast_canonicalizer_test.cc @@ -15,8 +15,6 @@ limitations under the License. ==============================================================================*/ #include "xla/hlo/transforms/simplifiers/broadcast_canonicalizer.h" -#include -#include #include #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.cc b/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.cc index a511652e8eb5b5..a7214f59fa38bc 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.cc @@ -15,6 +15,11 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/conditional_canonicalizer.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 "absl/types/span.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/status_macros.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.h b/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.h index 36de42ce428f51..c36c443b884a26 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer.h @@ -17,6 +17,9 @@ limitations under the License. #include +#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/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer_test.cc index 54d9b3c163809f..73fbdbb6b0279c 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/conditional_canonicalizer_test.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/conditional_canonicalizer.h" +#include +#include #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover.cc b/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover.cc index adc4c6d62e7c76..40c91f1c5e9f81 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover.cc @@ -15,6 +15,9 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/convert_mover.h" +#include +#include + #include "absl/algorithm/container.h" #include "absl/container/flat_hash_set.h" #include "absl/container/inlined_vector.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover_test.cc index 1cbe5c3415a881..3afca6c27985a8 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/convert_mover_test.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/convert_mover.h" +#include + #include #include #include "absl/strings/string_view.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/convert_operand_folder_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/convert_operand_folder_test.cc index 063b2d4130555f..7c359ab610bbd1 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/convert_operand_folder_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/convert_operand_folder_test.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/convert_operand_folder.h" +#include + #include #include #include "absl/strings/string_view.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.cc b/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.cc index d8f9fcd50d7e09..7e64cc06dfe2d8 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.cc @@ -15,10 +15,15 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/convolution_group_converter.h" -#include +#include +#include #include +#include #include +#include "absl/container/flat_hash_set.h" +#include "absl/status/status.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.h b/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.h index 16d39ad72382d5..8e7059d6109aed 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/convolution_group_converter.h @@ -18,6 +18,8 @@ limitations under the License. #include +#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/third_party/xla/xla/hlo/transforms/simplifiers/dot_dimension_merger.cc b/third_party/xla/xla/hlo/transforms/simplifiers/dot_dimension_merger.cc index c05ef071f2e592..8039d5ea13150e 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/dot_dimension_merger.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/dot_dimension_merger.cc @@ -22,7 +22,9 @@ limitations under the License. #include "absl/algorithm/container.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/strings/string_view.h" #include "absl/types/span.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.cc b/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.cc index 812c46112c62a1..9f070497caf794 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.cc @@ -16,6 +16,7 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/dot_merger.h" #include +#include #include #include #include @@ -39,6 +40,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/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.h b/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.h index bde8d3f9df6140..c303f45df4824d 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger.h @@ -17,6 +17,7 @@ limitations under the License. #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_DOT_MERGER_H_ #include +#include #include "absl/container/flat_hash_set.h" #include "absl/status/statusor.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger_test.cc index e352f74636beca..e8e90e57bf5759 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/dot_merger_test.cc @@ -16,9 +16,12 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/dot_merger.h" #include +#include #include #include +#include +#include #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/hlo/ir/hlo_module.h" @@ -30,6 +33,7 @@ limitations under the License. #include "xla/shape.h" #include "xla/shape_util.h" #include "xla/tsl/lib/core/status_test_util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/statusor.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.cc b/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.cc index c220ae1df608a1..18eb7f0eca5682 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.cc @@ -15,6 +15,12 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.h" +#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_instruction.h" #include "xla/hlo/ir/hlo_opcode.h" #include "xla/status_macros.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.h b/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.h index 171e9c02f59b93..eb52ffaa82c08f 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.h @@ -17,6 +17,9 @@ limitations under the License. #include +#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/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier_test.cc index 6112503842fdec..aecfaf483cc5c2 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier_test.cc @@ -16,10 +16,7 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/dynamic_dimension_simplifier.h" #include -#include -#include "absl/strings/str_cat.h" -#include "absl/strings/str_join.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_computation.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/flatten_call_graph.h b/third_party/xla/xla/hlo/transforms/simplifiers/flatten_call_graph.h index 6d35a483e6e4aa..25a3abb97b009c 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/flatten_call_graph.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/flatten_call_graph.h @@ -18,7 +18,9 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_FLATTEN_CALL_GRAPH_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_FLATTEN_CALL_GRAPH_H_ +#include "absl/container/flat_hash_set.h" #include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/pass/hlo_pass_interface.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/float_normalization.h b/third_party/xla/xla/hlo/transforms/simplifiers/float_normalization.h index 2f3807cb5815b3..84a19601ca289d 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/float_normalization.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/float_normalization.h @@ -16,11 +16,16 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_FLOAT_NORMALIZATION_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_FLOAT_NORMALIZATION_H_ +#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_module.h" #include "xla/hlo/pass/hlo_pass_interface.h" #include "xla/service/float_support.h" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/fusion_constant_sinking.h b/third_party/xla/xla/hlo/transforms/simplifiers/fusion_constant_sinking.h index 6e1c7d9813d239..0fe609f8582f11 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/fusion_constant_sinking.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/fusion_constant_sinking.h @@ -16,6 +16,9 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_FUSION_CONSTANT_SINKING_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_FUSION_CONSTANT_SINKING_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/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.cc b/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.cc index 64b7aa8efba0de..799120e5be0f61 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.cc @@ -15,10 +15,12 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/gather_simplifier.h" +#include #include #include #include "absl/algorithm/container.h" +#include "absl/status/statusor.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_instructions.h" #include "xla/literal_util.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.h b/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.h index da7b17c847158b..15edb691c661de 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier.h @@ -16,6 +16,8 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_GATHER_SIMPLIFIER_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_GATHER_SIMPLIFIER_H_ +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_instructions.h" #include "xla/hlo/transforms/expanders/op_expander_pass.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier_test.cc index 22f4e2e9205793..6cfa33dfaf45c4 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/gather_simplifier_test.cc @@ -17,6 +17,8 @@ limitations under the License. #include +#include +#include "absl/strings/string_view.h" #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_constant_splitter.cc b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_constant_splitter.cc index d804e24cc985ed..61d269bfbf3bdd 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_constant_splitter.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_constant_splitter.cc @@ -14,7 +14,7 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/hlo_constant_splitter.h" -#include +#include #include #include diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_dce_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_dce_test.cc index a1b18d1a5a9c0d..cd73bb80904bd0 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_dce_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_dce_test.cc @@ -15,11 +15,11 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/hlo_dce.h" -#include #include #include #include +#include "absl/algorithm/container.h" #include "absl/types/span.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.cc b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.cc index 7639055d7b4fe0..98a9122da5fc35 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.cc @@ -15,11 +15,13 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/hlo_element_type_converter.h" -#include -#include -#include +#include #include +#include "absl/container/flat_hash_set.h" +#include "absl/log/check.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/evaluator/hlo_evaluator.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" #include "xla/hlo/ir/hlo_computation.h" @@ -30,6 +32,7 @@ limitations under the License. #include "xla/literal.h" #include "xla/shape_util.h" #include "xla/types.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.h b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.h index 382c09b685e245..52a2f6cd924142 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter.h @@ -16,8 +16,12 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_HLO_ELEMENT_TYPE_CONVERTER_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_HLO_ELEMENT_TYPE_CONVERTER_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" +#include "xla/xla_data.pb.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter_test.cc index 4adf215fdc5791..4515038a2bba08 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_element_type_converter_test.cc @@ -15,8 +15,13 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/hlo_element_type_converter.h" +#include + +#include +#include #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" #include "xla/hlo/utils/hlo_matchers.h" +#include "xla/xla_data.pb.h" namespace xla { namespace { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization.cc b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization.cc index 7ad655f9a888a1..e40bb1d872ced4 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization.cc @@ -18,6 +18,7 @@ limitations under the License. #include #include #include +#include #include #include #include @@ -63,6 +64,7 @@ limitations under the License. #include "xla/shape_util.h" #include "xla/status_macros.h" #include "xla/util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" #include "tsl/platform/numbers.h" #include "tsl/platform/statusor.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization_test.cc index 65beab9a155bb2..e9742e2e28e874 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/hlo_rematerialization_test.cc @@ -22,7 +22,9 @@ 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/match.h" #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_computation.h" @@ -37,6 +39,7 @@ limitations under the License. #include "xla/shape_util.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" #include "tsl/platform/test.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.cc b/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.cc index 278d9f8e704424..794df7c96073dd 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.cc @@ -29,6 +29,7 @@ limitations under the License. #include "xla/hlo/ir/hlo_opcode.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/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.h b/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.h index 632faeb9e81730..7ec27b9d85f332 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/host_memory_transfer_asyncifier.h @@ -17,6 +17,7 @@ limitations under the License. #include +#include "absl/container/flat_hash_set.h" #include "absl/status/statusor.h" #include "absl/strings/string_view.h" #include "xla/hlo/pass/hlo_pass_interface.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/optimize_input_output_buffer_alias_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/optimize_input_output_buffer_alias_test.cc index ec8954b58d7866..04d546073ce9c9 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/optimize_input_output_buffer_alias_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/optimize_input_output_buffer_alias_test.cc @@ -19,7 +19,6 @@ limitations under the License. #include #include -#include #include #include "absl/status/status.h" #include "xla/hlo/ir/hlo_input_output_alias_config.h" @@ -28,6 +27,7 @@ limitations under the License. #include "xla/shape_util.h" #include "xla/test.h" #include "xla/test_helpers.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/test.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/reduce_window_rewriter.cc b/third_party/xla/xla/hlo/transforms/simplifiers/reduce_window_rewriter.cc index ddae0b21286be7..86151cdb7c3236 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/reduce_window_rewriter.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/reduce_window_rewriter.cc @@ -18,7 +18,6 @@ limitations under the License. #include #include #include -#include #include #include "absl/algorithm/container.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.cc b/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.cc index de6c2494274a91..b8a3c7600ae566 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.cc @@ -15,11 +15,21 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/reshape_mover.h" -#include +#include +#include #include +#include #include #include "absl/algorithm/container.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/statusor.h" +#include "absl/strings/str_cat.h" +#include "absl/strings/string_view.h" +#include "absl/types/span.h" #include "xla/permutation_util.h" #include "xla/service/hlo_creation_utils.h" #include "xla/shape_util.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.h b/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.h index 21d67fd8ed5a31..20a16b292185b3 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover.h @@ -16,6 +16,10 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_RESHAPE_MOVER_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_RESHAPE_MOVER_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" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover_test.cc index c72ae7ce392220..db95b48535b326 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/reshape_mover_test.cc @@ -18,6 +18,9 @@ limitations under the License. #include #include +#include +#include +#include "absl/status/status.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/third_party/xla/xla/hlo/transforms/simplifiers/result_caster_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/result_caster_test.cc index 58c550147ab214..1d2fb85a7e4657 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/result_caster_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/result_caster_test.cc @@ -18,12 +18,15 @@ limitations under the License. #include #include +#include +#include #include "absl/strings/string_view.h" #include "absl/strings/substitute.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" #include "xla/hlo/utils/hlo_matchers.h" #include "xla/primitive_util.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/statusor.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.cc b/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.cc index 9fcbab4222c26d..179742831b9de0 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.cc @@ -15,6 +15,10 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/root_instruction_sinker.h" +#include "absl/container/flat_hash_set.h" +#include "absl/log/check.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/service/tuple_util.h" namespace xla { diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.h b/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.h index 0e692943da18d4..4ead2aa5c501f3 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker.h @@ -16,6 +16,9 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_ROOT_INSTRUCTION_SINKER_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_ROOT_INSTRUCTION_SINKER_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/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker_test.cc index bbc472061f8032..8b5089fe1ccf4a 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/root_instruction_sinker_test.cc @@ -15,6 +15,9 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/root_instruction_sinker.h" +#include +#include +#include "absl/strings/string_view.h" #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" #include "xla/hlo/utils/hlo_matchers.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker.cc b/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker.cc index e1304eddc9ab57..d34ba3de5f4059 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker.cc @@ -15,9 +15,9 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/slice_sinker.h" -#include +#include +#include #include -#include #include #include "absl/algorithm/container.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker_test.cc index 1a983aa5335f4a..a2b3e8273145e8 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/slice_sinker_test.cc @@ -16,7 +16,6 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/slice_sinker.h" #include -#include #include #include diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier.cc b/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier.cc index 16335f0d1d21af..cf9b37f695613e 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/sort_simplifier.h" +#include +#include #include #include diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier_test.cc b/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier_test.cc index f9cfd9f7f99150..adc2c735c3ad36 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier_test.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/sort_simplifier_test.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/sort_simplifier.h" +#include + #include "xla/hlo/parser/hlo_parser.h" #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" #include "xla/service/pattern_matcher.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/sub_byte_normalization.cc b/third_party/xla/xla/hlo/transforms/simplifiers/sub_byte_normalization.cc index 9e8521bbf83623..10a94a981e59b1 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/sub_byte_normalization.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/sub_byte_normalization.cc @@ -17,12 +17,17 @@ 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/dfs_hlo_visitor_with_default.h" #include "xla/hlo/ir/hlo_instruction.h" #include "xla/layout.h" #include "xla/primitive_util.h" #include "xla/shape.h" #include "xla/shape_layout.h" +#include "xla/xla_data.pb.h" #include "tsl/platform/errors.h" #include "tsl/platform/status.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.cc b/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.cc index 15c930d52e1fb1..bc81e69684e775 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.cc +++ b/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.cc @@ -16,13 +16,17 @@ limitations under the License. #include "xla/hlo/transforms/simplifiers/tree_reduction_rewriter.h" #include +#include #include #include #include #include "absl/algorithm/container.h" +#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/str_join.h" +#include "absl/strings/string_view.h" #include "xla/hlo/builder/padding.h" #include "xla/hlo/ir/dfs_hlo_visitor_with_default.h" #include "xla/hlo/ir/hlo_instruction.h" diff --git a/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.h b/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.h index c332642ba82005..b5b6f91d3ca334 100644 --- a/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.h +++ b/third_party/xla/xla/hlo/transforms/simplifiers/tree_reduction_rewriter.h @@ -16,8 +16,10 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_SIMPLIFIERS_TREE_REDUCTION_REWRITER_H_ #define XLA_HLO_TRANSFORMS_SIMPLIFIERS_TREE_REDUCTION_REWRITER_H_ +#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_module.h"