From 367cb98d0afbb9b4c9ca28ad52d572dc4ba7192e Mon Sep 17 00:00:00 2001 From: xla authors Date: Mon, 23 Dec 2024 22:13:25 -0800 Subject: [PATCH] Automated Code Change PiperOrigin-RevId: 709239699 --- xla/service/cpu/BUILD | 24 +++++++++++++++++++ xla/service/cpu/conv_canonicalization_test.cc | 1 + xla/service/cpu/cpu_compiler_test.cc | 1 + .../cpu/cpu_eigen_tensor_alignment_test.cc | 1 + xla/service/cpu/cpu_executable.cc | 4 ++++ xla/service/cpu/cpu_layout_assignment_test.cc | 1 + xla/service/cpu/cpu_runtime.cc | 3 +++ xla/service/cpu/cpu_runtime_test.cc | 3 +++ xla/service/cpu/cpu_transfer_manager.cc | 1 + xla/service/cpu/cpu_xfeed.cc | 1 + xla/service/cpu/dot_op_emitter.cc | 2 ++ xla/service/cpu/dot_op_emitter.h | 1 + xla/service/cpu/elemental_ir_emitter.cc | 1 + xla/service/cpu/elemental_ir_emitter.h | 1 + xla/service/cpu/ir_emission_utils_test.cc | 1 + xla/service/cpu/ir_emitter2_test.cc | 2 ++ xla/service/cpu/ir_emitter_test.cc | 1 + xla/service/cpu/metrics_test.cc | 1 + xla/service/cpu/parallel_task_assignment.cc | 1 + .../cpu/parallel_task_assignment_test.cc | 1 + xla/service/cpu/runtime_fork_join.cc | 2 ++ xla/service/cpu/runtime_handle_ffi_call.cc | 2 ++ xla/service/cpu/sample_harness.cc | 1 + 23 files changed, 57 insertions(+) diff --git a/xla/service/cpu/BUILD b/xla/service/cpu/BUILD index 6fbf6037cde06..98d4ef411707d 100644 --- a/xla/service/cpu/BUILD +++ b/xla/service/cpu/BUILD @@ -155,6 +155,7 @@ cc_library( "//xla/service:shaped_buffer", "@com_google_absl//absl/base", "@com_google_absl//absl/cleanup", + "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:span", @@ -608,6 +609,8 @@ cc_library( "//xla/tsl/concurrency:async_value", "@com_google_absl//absl/base:core_headers", "@com_google_absl//absl/base:dynamic_annotations", + "@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", @@ -695,6 +698,7 @@ xla_cc_test( ":runtime_symbol_generator", ":target_machine_features_stub", "//xla:cpu_function_runtime", + "//xla:xla_proto_cc", "//xla/backends/cpu/codegen:cpu_features", "//xla/backends/cpu/codegen:ir_compiler", "//xla/backends/cpu/codegen:jit_compiler", @@ -750,6 +754,7 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest", "@llvm-project//llvm:Core", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", @@ -973,6 +978,8 @@ cc_library( "//xla/service/llvm_ir:llvm_loop", "//xla/service/llvm_ir:llvm_util", "@com_google_absl//absl/algorithm:container", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", @@ -1005,6 +1012,7 @@ xla_cc_binary( "//xla/client:local_client", "//xla/hlo/builder:xla_builder", "//xla/hlo/builder:xla_computation", + "@com_google_absl//absl/log", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:str_format", "@tsl//tsl/platform:logging", @@ -1042,6 +1050,8 @@ cc_library( "@com_google_absl//absl/base:core_headers", "@com_google_absl//absl/base:dynamic_annotations", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -1339,6 +1349,8 @@ cc_library( "//xla:executable_run_options", "//xla/service:custom_call_status_internal", "@com_google_absl//absl/base:core_headers", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/synchronization", @@ -1364,7 +1376,9 @@ cc_library( "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/base:core_headers", "@com_google_absl//absl/base:dynamic_annotations", + "@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", "@llvm-project//mlir:AsmParser", @@ -1392,7 +1406,9 @@ xla_cc_test( "//xla/client:local_client", "//xla/service:custom_call_status_internal", "//xla/tests:xla_internal_test_main", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings:str_format", + "@com_google_googletest//:gtest", "@eigen_archive//:eigen3", "@tsl//tsl/platform:env", "@tsl//tsl/platform:logging", @@ -1480,6 +1496,7 @@ xla_cc_test( "//xla:test", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", + "@com_google_googletest//:gtest", ], ) @@ -1528,6 +1545,7 @@ xla_cc_test( "@com_google_absl//absl/log:check", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:status", ], ) @@ -1566,6 +1584,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", + "@com_google_googletest//:gtest", ], ) @@ -1612,6 +1631,7 @@ cc_library( "@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", @@ -1637,6 +1657,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:statusor", ], ) @@ -1674,6 +1695,7 @@ xla_cc_test( "//xla:test", "//xla/tests:hlo_test_base", "//xla/tests:xla_internal_test_main", + "@com_google_googletest//:gtest", ], ) @@ -2037,6 +2059,7 @@ cc_library( hdrs = ["elemental_ir_emitter.h"], deps = [ ":elemental_math_emitter", + "//xla:xla_data_proto_cc", "//xla/service:elemental_ir_emitter", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -2052,6 +2075,7 @@ xla_cc_test( "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/monitoring:collected_metrics", "//xla/tsl/lib/monitoring:collection_registry", + "@com_google_googletest//:gtest", "@tsl//tsl/platform:test", ], ) diff --git a/xla/service/cpu/conv_canonicalization_test.cc b/xla/service/cpu/conv_canonicalization_test.cc index 00c9ee256452c..11deedfa64d53 100644 --- a/xla/service/cpu/conv_canonicalization_test.cc +++ b/xla/service/cpu/conv_canonicalization_test.cc @@ -17,6 +17,7 @@ limitations under the License. #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/xla/service/cpu/cpu_compiler_test.cc b/xla/service/cpu/cpu_compiler_test.cc index a2afebce2e828..3b7aceca72519 100644 --- a/xla/service/cpu/cpu_compiler_test.cc +++ b/xla/service/cpu/cpu_compiler_test.cc @@ -15,6 +15,7 @@ limitations under the License. #include #include +#include #include "absl/strings/string_view.h" #include "xla/hlo/testlib/verified_hlo_module.h" #include "xla/tests/hlo_pjrt_test_base.h" diff --git a/xla/service/cpu/cpu_eigen_tensor_alignment_test.cc b/xla/service/cpu/cpu_eigen_tensor_alignment_test.cc index 1193443a806a3..ce7ca3bcb78b7 100644 --- a/xla/service/cpu/cpu_eigen_tensor_alignment_test.cc +++ b/xla/service/cpu/cpu_eigen_tensor_alignment_test.cc @@ -16,6 +16,7 @@ limitations under the License. #include #include +#include #include "xla/service/cpu/ir_emission_utils.h" #include "xla/service/cpu/target_machine_features_stub.h" #include "xla/test.h" diff --git a/xla/service/cpu/cpu_executable.cc b/xla/service/cpu/cpu_executable.cc index 52d1cea039e92..b6085a0ed9cc0 100644 --- a/xla/service/cpu/cpu_executable.cc +++ b/xla/service/cpu/cpu_executable.cc @@ -15,6 +15,10 @@ limitations under the License. #include "xla/service/cpu/cpu_executable.h" +#include "absl/log/check.h" +#include "absl/log/log.h" +#include "absl/status/statusor.h" + #define EIGEN_USE_THREADS #include diff --git a/xla/service/cpu/cpu_layout_assignment_test.cc b/xla/service/cpu/cpu_layout_assignment_test.cc index 252049664af8f..b3a2d87a2c8fd 100644 --- a/xla/service/cpu/cpu_layout_assignment_test.cc +++ b/xla/service/cpu/cpu_layout_assignment_test.cc @@ -20,6 +20,7 @@ limitations under the License. #include #include +#include #include "absl/log/check.h" #include "absl/status/statusor.h" #include "absl/types/span.h" diff --git a/xla/service/cpu/cpu_runtime.cc b/xla/service/cpu/cpu_runtime.cc index a99612e3c4447..f7d4d3e247c97 100644 --- a/xla/service/cpu/cpu_runtime.cc +++ b/xla/service/cpu/cpu_runtime.cc @@ -30,11 +30,14 @@ limitations under the License. #include "absl/base/attributes.h" #include "absl/base/dynamic_annotations.h" #include "absl/container/flat_hash_map.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/str_cat.h" #include "absl/strings/str_join.h" #include "absl/strings/str_split.h" +#include "absl/strings/string_view.h" #include "absl/synchronization/mutex.h" #include "absl/time/time.h" #include "absl/types/span.h" diff --git a/xla/service/cpu/cpu_runtime_test.cc b/xla/service/cpu/cpu_runtime_test.cc index 79df0e29e8ea9..680be270942e2 100644 --- a/xla/service/cpu/cpu_runtime_test.cc +++ b/xla/service/cpu/cpu_runtime_test.cc @@ -14,6 +14,9 @@ limitations under the License. ==============================================================================*/ #include #include + +#include +#include "absl/log/check.h" #define EIGEN_USE_THREADS #include "xla/service/cpu/cpu_runtime.h" diff --git a/xla/service/cpu/cpu_transfer_manager.cc b/xla/service/cpu/cpu_transfer_manager.cc index b87a7072bcdb9..11369f1341318 100644 --- a/xla/service/cpu/cpu_transfer_manager.cc +++ b/xla/service/cpu/cpu_transfer_manager.cc @@ -17,6 +17,7 @@ limitations under the License. #include +#include "absl/status/status.h" #include "xla/literal.h" #include "xla/literal_util.h" #include "xla/service/compiler.h" diff --git a/xla/service/cpu/cpu_xfeed.cc b/xla/service/cpu/cpu_xfeed.cc index 0c60155e7a3cc..a6f7c86915520 100644 --- a/xla/service/cpu/cpu_xfeed.cc +++ b/xla/service/cpu/cpu_xfeed.cc @@ -24,6 +24,7 @@ limitations under the License. #include "absl/base/casts.h" #include "absl/cleanup/cleanup.h" +#include "absl/log/log.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/types/span.h" diff --git a/xla/service/cpu/dot_op_emitter.cc b/xla/service/cpu/dot_op_emitter.cc index 4911cbcf235a0..eba3c300becde 100644 --- a/xla/service/cpu/dot_op_emitter.cc +++ b/xla/service/cpu/dot_op_emitter.cc @@ -28,6 +28,8 @@ limitations under the License. #include #include "absl/algorithm/container.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/status/status.h" #include "absl/types/span.h" #include "llvm/IR/BasicBlock.h" diff --git a/xla/service/cpu/dot_op_emitter.h b/xla/service/cpu/dot_op_emitter.h index f267cc48b0b1d..a632f5b5d20b1 100644 --- a/xla/service/cpu/dot_op_emitter.h +++ b/xla/service/cpu/dot_op_emitter.h @@ -19,6 +19,7 @@ limitations under the License. #include #include +#include "absl/log/check.h" #include "absl/status/status.h" #include "llvm/IR/IRBuilder.h" #include "llvm/IR/Value.h" diff --git a/xla/service/cpu/elemental_ir_emitter.cc b/xla/service/cpu/elemental_ir_emitter.cc index d5eb44f85ed7d..a200a43023bbd 100644 --- a/xla/service/cpu/elemental_ir_emitter.cc +++ b/xla/service/cpu/elemental_ir_emitter.cc @@ -19,6 +19,7 @@ limitations under the License. #include "absl/strings/string_view.h" #include "llvm/IR/Value.h" #include "xla/service/cpu/elemental_math_emitter.h" +#include "xla/xla_data.pb.h" namespace xla::cpu { diff --git a/xla/service/cpu/elemental_ir_emitter.h b/xla/service/cpu/elemental_ir_emitter.h index f5c0a719b1ec2..0dc4e6c3e8842 100644 --- a/xla/service/cpu/elemental_ir_emitter.h +++ b/xla/service/cpu/elemental_ir_emitter.h @@ -22,6 +22,7 @@ limitations under the License. #include "llvm/IR/Module.h" #include "llvm/IR/Value.h" #include "xla/service/elemental_ir_emitter.h" +#include "xla/xla_data.pb.h" namespace xla::cpu { diff --git a/xla/service/cpu/ir_emission_utils_test.cc b/xla/service/cpu/ir_emission_utils_test.cc index 6babf519fde9b..28d165548ff5c 100644 --- a/xla/service/cpu/ir_emission_utils_test.cc +++ b/xla/service/cpu/ir_emission_utils_test.cc @@ -18,6 +18,7 @@ limitations under the License. #include #include +#include #include "xla/service/cpu/target_machine_features_stub.h" #include "xla/test.h" #include "xla/tests/hlo_test_base.h" diff --git a/xla/service/cpu/ir_emitter2_test.cc b/xla/service/cpu/ir_emitter2_test.cc index 11031111f873e..399149dee7d17 100644 --- a/xla/service/cpu/ir_emitter2_test.cc +++ b/xla/service/cpu/ir_emitter2_test.cc @@ -19,10 +19,12 @@ limitations under the License. #include #include +#include #include "absl/memory/memory.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/str_cat.h" +#include "absl/strings/string_view.h" #include "llvm/IR/Constants.h" #include "llvm/IR/IRBuilder.h" #include "llvm/IR/LLVMContext.h" diff --git a/xla/service/cpu/ir_emitter_test.cc b/xla/service/cpu/ir_emitter_test.cc index d41cad880a38b..9313f5774b8b7 100644 --- a/xla/service/cpu/ir_emitter_test.cc +++ b/xla/service/cpu/ir_emitter_test.cc @@ -63,6 +63,7 @@ limitations under the License. #include "xla/service/logical_buffer.h" #include "xla/tests/hlo_test_base.h" #include "xla/tsl/lib/core/status_test_util.h" +#include "xla/xla.pb.h" #include "tsl/platform/env.h" #include "tsl/platform/errors.h" #include "tsl/platform/statusor.h" diff --git a/xla/service/cpu/metrics_test.cc b/xla/service/cpu/metrics_test.cc index 133f75664b71f..c1b575e859a1a 100644 --- a/xla/service/cpu/metrics_test.cc +++ b/xla/service/cpu/metrics_test.cc @@ -19,6 +19,7 @@ limitations under the License. #include #include +#include #include "xla/tsl/lib/monitoring/collected_metrics.h" #include "xla/tsl/lib/monitoring/collection_registry.h" #include "tsl/platform/test.h" diff --git a/xla/service/cpu/parallel_task_assignment.cc b/xla/service/cpu/parallel_task_assignment.cc index 8266696c1e9b8..6c1b6c1c63f04 100644 --- a/xla/service/cpu/parallel_task_assignment.cc +++ b/xla/service/cpu/parallel_task_assignment.cc @@ -24,6 +24,7 @@ limitations under the License. #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_cat.h" diff --git a/xla/service/cpu/parallel_task_assignment_test.cc b/xla/service/cpu/parallel_task_assignment_test.cc index 7c76e5f271ca9..0ccc58381c5e6 100644 --- a/xla/service/cpu/parallel_task_assignment_test.cc +++ b/xla/service/cpu/parallel_task_assignment_test.cc @@ -19,6 +19,7 @@ limitations under the License. #include #include +#include #include "absl/status/statusor.h" #include "xla/backends/cpu/codegen/target_machine_features.h" #include "xla/hlo/ir/hlo_module.h" diff --git a/xla/service/cpu/runtime_fork_join.cc b/xla/service/cpu/runtime_fork_join.cc index bf30ddfebd15f..f5d45d5ee147a 100644 --- a/xla/service/cpu/runtime_fork_join.cc +++ b/xla/service/cpu/runtime_fork_join.cc @@ -22,6 +22,8 @@ limitations under the License. #include #include "absl/base/attributes.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/strings/str_cat.h" #include "absl/strings/string_view.h" #include "absl/synchronization/blocking_counter.h" diff --git a/xla/service/cpu/runtime_handle_ffi_call.cc b/xla/service/cpu/runtime_handle_ffi_call.cc index 65a000c524a47..5d7596e774fb4 100644 --- a/xla/service/cpu/runtime_handle_ffi_call.cc +++ b/xla/service/cpu/runtime_handle_ffi_call.cc @@ -22,7 +22,9 @@ limitations under the License. #include "absl/algorithm/container.h" #include "absl/base/attributes.h" #include "absl/base/dynamic_annotations.h" +#include "absl/log/check.h" #include "absl/status/status.h" +#include "absl/status/statusor.h" #include "absl/strings/str_cat.h" #include "absl/strings/string_view.h" #include "absl/types/span.h" diff --git a/xla/service/cpu/sample_harness.cc b/xla/service/cpu/sample_harness.cc index 118d72578314e..b671b169f2488 100644 --- a/xla/service/cpu/sample_harness.cc +++ b/xla/service/cpu/sample_harness.cc @@ -17,6 +17,7 @@ limitations under the License. #include #include +#include "absl/log/log.h" #include "absl/status/statusor.h" #include "absl/strings/str_format.h" #include "xla/array4d.h"