diff --git a/xla/stream_executor/tpu/BUILD b/xla/stream_executor/tpu/BUILD index 777fc83b68c43a..fcd72ef1796cff 100644 --- a/xla/stream_executor/tpu/BUILD +++ b/xla/stream_executor/tpu/BUILD @@ -77,6 +77,7 @@ cc_library( "//xla/stream_executor:device_memory", "//xla/stream_executor:device_memory_allocator", "@com_google_absl//absl/container:inlined_vector", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:span", "@tsl//tsl/platform:logging", @@ -154,6 +155,7 @@ cc_library( hdrs = ["proto_helper.h"], deps = [ ":c_api_decl", + "@com_google_absl//absl/log:check", "@tsl//tsl/platform:logging", ], ) @@ -212,6 +214,7 @@ cc_library( "//xla/stream_executor/platform:initialize", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/functional:any_invocable", + "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -244,6 +247,7 @@ cc_library( "//xla/stream_executor:stream", "//xla/stream_executor:stream_executor_h", "@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/strings:string_view", @@ -282,6 +286,7 @@ cc_library( "//xla/stream_executor:stream_executor_h", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/functional:any_invocable", + "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", @@ -338,6 +343,8 @@ cc_library( "@com_google_absl//absl/cleanup", "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/functional:any_invocable", + "@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:string_view", @@ -458,6 +465,7 @@ cc_library( "//xla/stream_executor:stream", "//xla/stream_executor:stream_executor_h", "@com_google_absl//absl/cleanup", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/types:span", @@ -506,6 +514,8 @@ cc_library( "//xla/stream_executor:platform", "//xla/stream_executor:platform_manager", "//xla/tsl/protobuf:error_codes_proto_impl_cc", + "@com_google_absl//absl/log", + "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/synchronization", @@ -537,6 +547,7 @@ cc_library( ], visibility = ["//visibility:public"], deps = [ + "@com_google_absl//absl/log", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", @@ -608,6 +619,7 @@ cc_library( "//xla/stream_executor:stream_executor_h", "@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/status:statusor", "@com_google_absl//absl/strings", @@ -641,6 +653,7 @@ cc_library( "@com_google_absl//absl/cleanup", "@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", diff --git a/xla/stream_executor/tpu/c_api_conversions.cc b/xla/stream_executor/tpu/c_api_conversions.cc index cc8e0274da063d..9cdee13ba01a08 100644 --- a/xla/stream_executor/tpu/c_api_conversions.cc +++ b/xla/stream_executor/tpu/c_api_conversions.cc @@ -23,6 +23,8 @@ limitations under the License. #include #include "absl/container/inlined_vector.h" +#include "absl/log/check.h" +#include "absl/status/statusor.h" #include "absl/types/span.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/layout.h" diff --git a/xla/stream_executor/tpu/proto_helper.cc b/xla/stream_executor/tpu/proto_helper.cc index 0e852880f542aa..01b2df7250770f 100644 --- a/xla/stream_executor/tpu/proto_helper.cc +++ b/xla/stream_executor/tpu/proto_helper.cc @@ -15,6 +15,7 @@ limitations under the License. #include "xla/stream_executor/tpu/proto_helper.h" +#include "absl/log/check.h" #include "xla/stream_executor/tpu/c_api_decl.h" #include "tsl/platform/logging.h" // IWYU pragma: keep diff --git a/xla/stream_executor/tpu/proto_helper.h b/xla/stream_executor/tpu/proto_helper.h index cda8b489102668..e1c9cab24b5406 100644 --- a/xla/stream_executor/tpu/proto_helper.h +++ b/xla/stream_executor/tpu/proto_helper.h @@ -18,6 +18,7 @@ limitations under the License. #include +#include "absl/log/check.h" #include "xla/stream_executor/tpu/c_api_decl.h" #include "tsl/platform/logging.h" // IWYU pragma: keep diff --git a/xla/stream_executor/tpu/tpu_executable.cc b/xla/stream_executor/tpu/tpu_executable.cc index e91f1d279138a1..a59ab5d38f15ee 100644 --- a/xla/stream_executor/tpu/tpu_executable.cc +++ b/xla/stream_executor/tpu/tpu_executable.cc @@ -24,6 +24,8 @@ limitations under the License. #include "absl/cleanup/cleanup.h" #include "absl/container/inlined_vector.h" +#include "absl/log/check.h" +#include "absl/status/statusor.h" #include "absl/strings/string_view.h" #include "xla/hlo/ir/hlo_module.h" #include "xla/service/executable.h" diff --git a/xla/stream_executor/tpu/tpu_executable_interface.cc b/xla/stream_executor/tpu/tpu_executable_interface.cc index 5fef42035f58df..7cdc0b7a2b9236 100644 --- a/xla/stream_executor/tpu/tpu_executable_interface.cc +++ b/xla/stream_executor/tpu/tpu_executable_interface.cc @@ -23,6 +23,7 @@ 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" diff --git a/xla/stream_executor/tpu/tpu_executor.cc b/xla/stream_executor/tpu/tpu_executor.cc index d52b07d1d9ca2e..4201c667869aca 100644 --- a/xla/stream_executor/tpu/tpu_executor.cc +++ b/xla/stream_executor/tpu/tpu_executor.cc @@ -22,7 +22,9 @@ limitations under the License. #include #include "absl/cleanup/cleanup.h" +#include "absl/log/check.h" #include "absl/status/status.h" +#include "absl/status/statusor.h" #include "absl/types/span.h" #include "xla/stream_executor/allocator_stats.h" #include "xla/stream_executor/device_description.h" diff --git a/xla/stream_executor/tpu/tpu_executor.h b/xla/stream_executor/tpu/tpu_executor.h index 65ca29a0b7317f..31cad08ef2fdba 100644 --- a/xla/stream_executor/tpu/tpu_executor.h +++ b/xla/stream_executor/tpu/tpu_executor.h @@ -24,6 +24,7 @@ limitations under the License. #include "absl/container/flat_hash_map.h" #include "absl/functional/any_invocable.h" +#include "absl/log/log.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/synchronization/mutex.h" diff --git a/xla/stream_executor/tpu/tpu_initialize_util.cc b/xla/stream_executor/tpu/tpu_initialize_util.cc index 0868ba5ead48a3..cb52adbf7534e0 100644 --- a/xla/stream_executor/tpu/tpu_initialize_util.cc +++ b/xla/stream_executor/tpu/tpu_initialize_util.cc @@ -31,6 +31,7 @@ limitations under the License. #include #include +#include "absl/log/log.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/ascii.h" diff --git a/xla/stream_executor/tpu/tpu_platform.cc b/xla/stream_executor/tpu/tpu_platform.cc index f263e0973ad6ca..8a3c5997e76f2e 100644 --- a/xla/stream_executor/tpu/tpu_platform.cc +++ b/xla/stream_executor/tpu/tpu_platform.cc @@ -20,6 +20,7 @@ limitations under the License. #include #include +#include "absl/log/check.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/synchronization/mutex.h" diff --git a/xla/stream_executor/tpu/tpu_platform.h b/xla/stream_executor/tpu/tpu_platform.h index 8eb6f19b7cdd6b..65cdd0e294acbc 100644 --- a/xla/stream_executor/tpu/tpu_platform.h +++ b/xla/stream_executor/tpu/tpu_platform.h @@ -22,6 +22,7 @@ limitations under the License. #include #include "absl/container/flat_hash_map.h" +#include "absl/log/log.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/strings/string_view.h" diff --git a/xla/stream_executor/tpu/tpu_platform_interface.cc b/xla/stream_executor/tpu/tpu_platform_interface.cc index d1248356d914a4..4f4a45b67aefea 100644 --- a/xla/stream_executor/tpu/tpu_platform_interface.cc +++ b/xla/stream_executor/tpu/tpu_platform_interface.cc @@ -15,6 +15,8 @@ limitations under the License. #include "xla/stream_executor/tpu/tpu_platform_interface.h" +#include "absl/log/check.h" +#include "absl/log/log.h" #include "absl/synchronization/mutex.h" #include "xla/stream_executor/platform.h" #include "xla/stream_executor/platform_manager.h" diff --git a/xla/stream_executor/tpu/tpu_transfer_manager.cc b/xla/stream_executor/tpu/tpu_transfer_manager.cc index 45fe7aa8eab607..a369813a5fe366 100644 --- a/xla/stream_executor/tpu/tpu_transfer_manager.cc +++ b/xla/stream_executor/tpu/tpu_transfer_manager.cc @@ -24,6 +24,7 @@ limitations under the License. #include #include "absl/cleanup/cleanup.h" +#include "absl/log/check.h" #include "absl/status/status.h" #include "absl/status/statusor.h" #include "absl/types/span.h"