Skip to content

Commit

Permalink
Automated Code Change
Browse files Browse the repository at this point in the history 
PiperOrigin-RevId: 714433332
  • Loading branch information
@Google-ML-Automation
Google-ML-Automation committed Jan 11, 2025
1 parent 53af204 commit ff06db8
Show file tree
Hide file tree
Showing 13 changed files with 29 additions and 0 deletions.
13 changes: 13 additions & 0 deletions xla/stream_executor/tpu/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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",
Expand Down Expand Up @@ -154,6 +155,7 @@ cc_library(
hdrs = ["proto_helper.h"],
deps = [
":c_api_decl",
"@com_google_absl//absl/log:check",
"@tsl//tsl/platform:logging",
],
)
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down Expand Up @@ -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",
Expand Down
2 changes: 2 additions & 0 deletions xla/stream_executor/tpu/c_api_conversions.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,8 @@ limitations under the License.
#include <vector>

#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"
Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/proto_helper.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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

Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/proto_helper.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -18,6 +18,7 @@ limitations under the License.

#include <cstddef>

#include "absl/log/check.h"
#include "xla/stream_executor/tpu/c_api_decl.h"
#include "tsl/platform/logging.h" // IWYU pragma: keep

Expand Down
2 changes: 2 additions & 0 deletions xla/stream_executor/tpu/tpu_executable.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/tpu_executable_interface.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -23,6 +23,7 @@ limitations under the License.
#include <vector>

#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"
Expand Down
2 changes: 2 additions & 0 deletions xla/stream_executor/tpu/tpu_executor.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,7 +22,9 @@ limitations under the License.
#include <variant>

#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"
Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/tpu_executor.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/tpu_initialize_util.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -31,6 +31,7 @@ limitations under the License.
#include <utility>
#include <vector>

#include "absl/log/log.h"
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/strings/ascii.h"
Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/tpu_platform.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -20,6 +20,7 @@ limitations under the License.
#include <string>
#include <utility>

#include "absl/log/check.h"
#include "absl/status/status.h"
#include "absl/status/statusor.h"
#include "absl/synchronization/mutex.h"
Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/tpu_platform.h
View file
Original file line number Diff line number Diff line change
Expand Up @@ -22,6 +22,7 @@ limitations under the License.
#include <string>

#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"
Expand Down
2 changes: 2 additions & 0 deletions xla/stream_executor/tpu/tpu_platform_interface.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -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"
Expand Down
1 change: 1 addition & 0 deletions xla/stream_executor/tpu/tpu_transfer_manager.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -24,6 +24,7 @@ limitations under the License.
#include <vector>

#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"
Expand Down

0 comments on commit ff06db8

Please sign in to comment.