From 03d02c57ce0d8de4e43bf5cd9506748cfb753d6f Mon Sep 17 00:00:00 2001 From: xla authors Date: Thu, 19 Dec 2024 05:20:20 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 707878810 --- xla/hlo/transforms/BUILD | 11 +++++++++++ xla/hlo/transforms/host_offload_legalize.cc | 1 - xla/hlo/transforms/host_offload_legalize.h | 2 ++ xla/hlo/transforms/host_offload_legalize_test.cc | 3 --- xla/hlo/transforms/host_offloader.cc | 8 ++------ xla/hlo/transforms/host_offloader.h | 3 +++ xla/hlo/transforms/host_offloader_test.cc | 1 - xla/hlo/transforms/memory_space_propagation.cc | 4 ++++ xla/hlo/transforms/memory_space_propagation.h | 6 ++++++ xla/hlo/transforms/memory_space_propagation_test.cc | 4 ++++ xla/hlo/transforms/operand_upcaster_test.cc | 3 +++ 11 files changed, 35 insertions(+), 11 deletions(-) diff --git a/xla/hlo/transforms/BUILD b/xla/hlo/transforms/BUILD index d895b3cc91730..84cf00b65702b 100644 --- a/xla/hlo/transforms/BUILD +++ b/xla/hlo/transforms/BUILD @@ -1335,6 +1335,9 @@ cc_library( "//xla/hlo/analysis:hlo_dataflow_analysis", "//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", ], ) @@ -1346,6 +1349,10 @@ xla_cc_test( "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/tsl/lib/core:status_test_util", + "@com_google_absl//absl/hash", + "@com_google_absl//absl/status", + "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:test_main", ], ) @@ -1838,6 +1845,7 @@ cc_library( "@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/strings:str_format", "@tsl//tsl/platform:errors", @@ -1879,6 +1887,7 @@ cc_library( "//xla:side_effect_util", "//xla:status_macros", "//xla:util", + "//xla:xla_data_proto_cc", "//xla/hlo/analysis:hlo_alias_analysis", "//xla/hlo/ir:hlo", "//xla/hlo/pass:hlo_pass", @@ -2274,10 +2283,12 @@ xla_cc_test( deps = [ ":operand_upcaster", "//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", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test_main", ], diff --git a/xla/hlo/transforms/host_offload_legalize.cc b/xla/hlo/transforms/host_offload_legalize.cc index 639e37874ceb4..5e70dbb26c7d2 100644 --- a/xla/hlo/transforms/host_offload_legalize.cc +++ b/xla/hlo/transforms/host_offload_legalize.cc @@ -20,7 +20,6 @@ limitations under the License. #include #include #include -#include #include #include "absl/algorithm/container.h" diff --git a/xla/hlo/transforms/host_offload_legalize.h b/xla/hlo/transforms/host_offload_legalize.h index a5d85fa40a8a5..e08c842ee0bc6 100644 --- a/xla/hlo/transforms/host_offload_legalize.h +++ b/xla/hlo/transforms/host_offload_legalize.h @@ -17,8 +17,10 @@ #include #include +#include #include "absl/container/flat_hash_set.h" +#include "absl/status/statusor.h" #include "absl/strings/string_view.h" #include "xla/hlo/analysis/hlo_alias_analysis.h" #include "xla/hlo/pass/hlo_pass_interface.h" diff --git a/xla/hlo/transforms/host_offload_legalize_test.cc b/xla/hlo/transforms/host_offload_legalize_test.cc index 4aedc40b8ca2b..a37a73fc149f9 100644 --- a/xla/hlo/transforms/host_offload_legalize_test.cc +++ b/xla/hlo/transforms/host_offload_legalize_test.cc @@ -16,12 +16,9 @@ limitations under the License. #include "xla/hlo/transforms/host_offload_legalize.h" #include -#include #include -#include #include -#include "absl/container/flat_hash_set.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "xla/hlo/ir/hlo_computation.h" diff --git a/xla/hlo/transforms/host_offloader.cc b/xla/hlo/transforms/host_offloader.cc index 7b798fe38eef7..833fa176b78b0 100644 --- a/xla/hlo/transforms/host_offloader.cc +++ b/xla/hlo/transforms/host_offloader.cc @@ -15,15 +15,10 @@ limitations under the License. #include "xla/hlo/transforms/host_offloader.h" -#include -#include #include #include #include -#include #include -#include -#include #include #include "absl/algorithm/container.h" @@ -35,7 +30,7 @@ limitations under the License. #include "absl/status/statusor.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/analysis/hlo_alias_analysis.h" #include "xla/hlo/ir/hlo_casting_utils.h" #include "xla/hlo/ir/hlo_computation.h" @@ -56,6 +51,7 @@ limitations under the License. #include "xla/side_effect_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/status.h" #include "tsl/platform/statusor.h" diff --git a/xla/hlo/transforms/host_offloader.h b/xla/hlo/transforms/host_offloader.h index 765b3c2709856..8e79a44926178 100644 --- a/xla/hlo/transforms/host_offloader.h +++ b/xla/hlo/transforms/host_offloader.h @@ -18,8 +18,11 @@ #include #include #include +#include +#include "absl/container/flat_hash_map.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/analysis/hlo_alias_analysis.h" diff --git a/xla/hlo/transforms/host_offloader_test.cc b/xla/hlo/transforms/host_offloader_test.cc index 1452815127f1a..d38526e93178a 100644 --- a/xla/hlo/transforms/host_offloader_test.cc +++ b/xla/hlo/transforms/host_offloader_test.cc @@ -17,7 +17,6 @@ limitations under the License. #include #include -#include #include #include diff --git a/xla/hlo/transforms/memory_space_propagation.cc b/xla/hlo/transforms/memory_space_propagation.cc index d0704df0e88af..3dc14572dc408 100644 --- a/xla/hlo/transforms/memory_space_propagation.cc +++ b/xla/hlo/transforms/memory_space_propagation.cc @@ -16,7 +16,11 @@ limitations under the License. #include "xla/hlo/transforms/memory_space_propagation.h" #include +#include +#include "absl/container/flat_hash_set.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/shape.h" #include "xla/shape_util.h" diff --git a/xla/hlo/transforms/memory_space_propagation.h b/xla/hlo/transforms/memory_space_propagation.h index bb0da70bf1a7f..b3998f542d39f 100644 --- a/xla/hlo/transforms/memory_space_propagation.h +++ b/xla/hlo/transforms/memory_space_propagation.h @@ -16,6 +16,12 @@ limitations under the License. #ifndef XLA_HLO_TRANSFORMS_MEMORY_SPACE_PROPAGATION_H_ #define XLA_HLO_TRANSFORMS_MEMORY_SPACE_PROPAGATION_H_ +#include +#include + +#include "absl/container/flat_hash_set.h" +#include "absl/status/statusor.h" +#include "absl/strings/string_view.h" #include "xla/hlo/analysis/hlo_dataflow_analysis.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/hlo/pass/hlo_pass_interface.h" diff --git a/xla/hlo/transforms/memory_space_propagation_test.cc b/xla/hlo/transforms/memory_space_propagation_test.cc index 15cd6c4cd4cbf..a1252d596ee28 100644 --- a/xla/hlo/transforms/memory_space_propagation_test.cc +++ b/xla/hlo/transforms/memory_space_propagation_test.cc @@ -15,6 +15,10 @@ limitations under the License. #include "xla/hlo/transforms/memory_space_propagation.h" +#include +#include "absl/hash/hash.h" +#include "absl/status/status.h" +#include "absl/strings/string_view.h" #include "xla/hlo/parser/hlo_parser.h" #include "xla/hlo/testlib/hlo_hardware_independent_test_base.h" #include "xla/tsl/lib/core/status_test_util.h" diff --git a/xla/hlo/transforms/operand_upcaster_test.cc b/xla/hlo/transforms/operand_upcaster_test.cc index 8a143b365af61..ed61bb63d2dad 100644 --- a/xla/hlo/transforms/operand_upcaster_test.cc +++ b/xla/hlo/transforms/operand_upcaster_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 {