Skip to content

Commit

Permalink
Automated Code Change
Browse files Browse the repository at this point in the history 
PiperOrigin-RevId: 724525028
  • Loading branch information
@Google-ML-Automation
Google-ML-Automation committed Feb 8, 2025
1 parent a3ede7c commit 9d24f39
Show file tree
Hide file tree
Showing 51 changed files with 267 additions and 20 deletions.
92 changes: 92 additions & 0 deletions xla/service/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -107,8 +107,14 @@ cc_library(
hdrs = ["collective_opt_utils.h"],
deps = [
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/container:inlined_vector",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/types:span",
],
)

Expand Down Expand Up @@ -156,6 +162,7 @@ xla_cc_test(
":all_reduce_promotion",
":pattern_matcher",
":pattern_matcher_gmock",
":remote_mtest_lib_dynamic_padder_test_gpu_b200",
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/tests:hlo_test_base",
Expand Down Expand Up @@ -330,6 +337,7 @@ cc_library(
":source_target_pairs",
"//xla:shape_util",
"//xla:xla_data_proto_cc",
"//xla:xla_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/parser:hlo_parser",
"//xla/hlo/pass:hlo_pass",
Expand All @@ -350,6 +358,8 @@ xla_cc_test(
deps = [
":collective_ops_utils",
":collective_permute_decomposer",
"//xla:xla_data_proto_cc",
"//xla:xla_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/testlib:hlo_hardware_independent_test_base",
"//xla/hlo/utils:hlo_matchers",
Expand All @@ -371,7 +381,10 @@ cc_library(
deps = [
"//xla:literal",
"//xla:util",
"@com_google_absl//absl/base",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
],
)

Expand Down Expand Up @@ -479,6 +492,7 @@ xla_cc_test(
"//xla:literal_util",
"//xla:test_helpers",
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/parser:hlo_parser",
"//xla/hlo/pass:hlo_pass_pipeline",
Expand Down Expand Up @@ -945,7 +959,9 @@ cc_library(
"@com_google_absl//absl/container:flat_hash_set",
"@com_google_absl//absl/container:inlined_vector",
"@com_google_absl//absl/functional:function_ref",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/memory",
"@com_google_absl//absl/status",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/strings:str_format",
Expand Down Expand Up @@ -1033,6 +1049,7 @@ xla_cc_test(
"//xla/tsl/lib/core:status_test_util",
"@com_google_absl//absl/log",
"@com_google_absl//absl/strings:string_view",
"@com_google_googletest//:gtest",
"@tsl//tsl/platform:statusor",
],
)
Expand Down Expand Up @@ -1363,16 +1380,21 @@ cc_library(
":compiler",
":computation_layout",
":dump",
":hlo_proto_cc",
":platform_util",
":service",
"//xla:debug_options_flags",
"//xla:status_macros",
"//xla:types",
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla:xla_proto_cc",
"//xla/stream_executor:stream_executor_h",
"@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",
"@tsl//tsl/platform:logging",
],
)
Expand Down Expand Up @@ -1552,16 +1574,25 @@ cc_library(
":computation_placer",
":executable",
":hlo_module_config",
":hlo_proto_cc",
":metrics_hook_interface",
"//third_party/protobuf",
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla:xla_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/ir:hlo_module_group",
"//xla/pjrt/distributed:key_value_store_interface",
"//xla/stream_executor:device_description_proto_cc",
"//xla/stream_executor:dnn",
"//xla/stream_executor:stream_executor_h",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/container:flat_hash_map",
"@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/synchronization",
"@tsl//tsl/platform:env",
"@tsl//tsl/platform:logging",
"@tsl//tsl/platform:protobuf",
Expand Down Expand Up @@ -1659,6 +1690,7 @@ cc_library(
"//xla/stream_executor:device_memory_allocator",
"@com_google_absl//absl/base:core_headers",
"@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/synchronization",
Expand Down Expand Up @@ -1691,7 +1723,9 @@ cc_library(
deps = [
"//xla:util",
"//xla:xla_data_proto_cc",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/synchronization",
],
)

Expand Down Expand Up @@ -1799,12 +1833,14 @@ xla_cc_test(
"//xla/tests:xla_internal_test_main",
"//xla/tsl/lib/core:status_test_util",
"@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",
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
"@tsl//tsl/platform:status",
"@tsl//tsl/platform:statusor",
],
Expand Down Expand Up @@ -2382,6 +2418,7 @@ xla_test(
"//xla/hlo/utils:hlo_matchers",
"//xla/tests:hlo_test_base",
"//xla/tests:xla_internal_test_main",
"@com_google_googletest//:gtest_main",
"@tsl//tsl/platform:statusor",
],
)
Expand Down Expand Up @@ -2463,7 +2500,10 @@ cc_library(
"//xla:literal_util",
"//xla:shape_util",
"//xla:status_macros",
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/types:span",
],
)

Expand Down Expand Up @@ -2575,6 +2615,7 @@ xla_cc_test(
":hlo_module_config",
":pattern_matcher",
":pattern_matcher_gmock",
":remote_mtest_lib_dynamic_padder_test_gpu_b200",
"//xla:shape_util",
"//xla:test",
"//xla:types",
Expand All @@ -2584,6 +2625,7 @@ xla_cc_test(
"//xla/tests:hlo_test_base",
"//xla/tests:xla_internal_test_main", # fixdeps: keep
"//xla/tsl/lib/core:status_test_util",
"@com_google_googletest//:gtest",
"@tsl//tsl/platform:statusor",
],
)
Expand Down Expand Up @@ -2683,11 +2725,14 @@ cc_library(
"//xla:status_macros",
"//xla:types",
"//xla:util",
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/pass:hlo_pass",
"@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/log:check",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
"@tsl//tsl/platform:errors",
Expand All @@ -2709,6 +2754,8 @@ xla_cc_test(
"//xla/tests:hlo_test_base",
"//xla/tests:xla_internal_test_main",
"//xla/tsl/lib/core:status_test_util",
"@com_google_absl//absl/strings:string_view",
"@com_google_googletest//:gtest",
"@tsl//tsl/platform:status",
],
)
Expand All @@ -2733,12 +2780,14 @@ cc_library(
"//xla/hlo/transforms/simplifiers:tuple_simplifier",
"@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/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",
"@com_google_absl//absl/types:span",
"@tsl//tsl/platform:errors",
],
)
Expand All @@ -2758,6 +2807,9 @@ xla_cc_test(
"//xla/tests:hlo_test_base",
"//xla/tests:xla_internal_test_main",
"//xla/tsl/lib/core:status_test_util",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/strings:string_view",
"@com_google_googletest//:gtest_main",
"@tsl//tsl/platform:status",
],
Expand Down Expand Up @@ -3365,11 +3417,15 @@ cc_library(
"//xla/stream_executor/host:host_platform_id",
"//xla/stream_executor/rocm:rocm_platform_id",
"//xla/stream_executor/sycl:sycl_platform_id",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/memory",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/synchronization",
"@tsl//tsl/platform:errors",
"@tsl//tsl/platform:logging",
"@tsl//tsl/platform:status",
Expand All @@ -3386,8 +3442,10 @@ cc_library(
"//xla:xla_data_proto_cc",
"//xla/stream_executor:platform",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/synchronization",
"@tsl//tsl/platform:status",
],
)
Expand Down Expand Up @@ -3643,10 +3701,13 @@ cc_library(
hdrs = ["buffer_value.h"],
deps = [
":hlo_proto_cc",
"//third_party/protobuf:protobuf_lite",
"//xla:shape_util",
"//xla:xla_data_proto_cc",
"//xla/hlo/ir:hlo",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/log:check",
"@tsl//tsl/platform:logging",
],
)
Expand Down Expand Up @@ -3805,7 +3866,12 @@ cc_library(
"//xla:types",
"//xla:util",
"//xla:xla_data_proto_cc",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/synchronization",
"@tsl//tsl/platform:logging",
"@tsl//tsl/platform:strcat",
],
Expand Down Expand Up @@ -4495,7 +4561,10 @@ cc_library(
"//xla:types",
"//xla:xla_data_proto_cc",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/strings:str_format",
],
)

Expand Down Expand Up @@ -5340,7 +5409,10 @@ cc_library(
hdrs = ["compilation_stats.h"],
deps = [
"//xla:types",
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/strings:str_format",
"@tsl//tsl/platform:env",
Expand Down Expand Up @@ -5393,8 +5465,14 @@ cc_library(
":hlo_creation_utils",
"//xla:status_macros",
"//xla:types",
"//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/log",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings:string_view",
"@tsl//tsl/platform:errors",
"@tsl//tsl/platform:logging",
"@tsl//tsl/platform:status",
Expand All @@ -5412,6 +5490,7 @@ xla_cc_test(
"//xla/hlo/utils:hlo_matchers",
"//xla/tests:hlo_test_base",
"//xla/tests:xla_internal_test_main", # fixdeps: keep
"@com_google_googletest//:gtest",
],
)

Expand Down Expand Up @@ -5867,7 +5946,10 @@ cc_library(
"@com_google_absl//absl/algorithm:container",
"@com_google_absl//absl/base:core_headers",
"@com_google_absl//absl/container:flat_hash_map",
"@com_google_absl//absl/log",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/memory",
"@com_google_absl//absl/status",
"@com_google_absl//absl/status:statusor",
"@com_google_absl//absl/strings",
"@com_google_absl//absl/synchronization",
Expand Down Expand Up @@ -5910,6 +5992,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:casts",
"@tsl//tsl/platform:protobuf",
],
Expand Down Expand Up @@ -6037,8 +6120,14 @@ cc_library(
copts = tsl_copts(),
deps = [
":hlo_creation_utils",
"//xla:xla_data_proto_cc",
"//xla/hlo/pass:hlo_pass",
"//xla/service/cpu:onednn_contraction_rewriter",
"@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:string_view",
"@com_google_absl//absl/types:span",
],
)

Expand Down Expand Up @@ -6148,9 +6237,12 @@ xla_cc_test(
":change_op_data_type",
":pattern_matcher",
":pattern_matcher_gmock",
":remote_mtest_lib_dynamic_padder_test_gpu_b200",
"//xla:xla_data_proto_cc",
"//xla/tests:hlo_test_base",
"//xla/tests:xla_internal_test_main", # fixdeps: keep
"@com_google_absl//absl/types:span",
"@com_google_googletest//:gtest",
],
)

Expand Down
1 change: 0 additions & 1 deletion xla/service/algebraic_simplifier_overflow_test.cc
View file
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,6 @@ limitations under the License.

#include <optional>
#include <string>
#include <vector>

#include <gtest/gtest.h>
#include "xla/error_spec.h"
Expand Down
Loading

0 comments on commit 9d24f39

Please sign in to comment.