diff --git a/.bazelrc b/.bazelrc index c5ff8a4f50e8f..a0e4028ac05b4 100644 --- a/.bazelrc +++ b/.bazelrc @@ -127,6 +127,10 @@ build --incompatible_enforce_config_setting_visibility # TODO: also enable this flag after fixing the visibility violations # build --incompatible_config_setting_private_default_visibility +# Print a stacktrace when a test is killed +test --test_env="GTEST_INSTALL_FAILURE_SIGNAL_HANDLER=1" + + # Default options should come above this line. # Android configs. Bazel needs to have --cpu and --fat_apk_cpu both set to the diff --git a/third_party/tsl/.bazelrc b/third_party/tsl/.bazelrc index 7a8d56b3af396..56ade15da0390 100644 --- a/third_party/tsl/.bazelrc +++ b/third_party/tsl/.bazelrc @@ -127,6 +127,10 @@ build --incompatible_enforce_config_setting_visibility # TODO: also enable this flag after fixing the visibility violations # build --incompatible_config_setting_private_default_visibility +# Print a stacktrace when a test is killed +test --test_env="GTEST_INSTALL_FAILURE_SIGNAL_HANDLER=1" + + # Default options should come above this line. # Android configs. Bazel needs to have --cpu and --fat_apk_cpu both set to the diff --git a/xla/BUILD b/xla/BUILD index 2662ac834c4f2..9bb90d84a97ac 100644 --- a/xla/BUILD +++ b/xla/BUILD @@ -91,9 +91,9 @@ xla_cc_test( deps = [ ":bit_cast", ":test", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@tsl//tsl/platform:bfloat16", - "@tsl//tsl/platform:test_main", ], ) @@ -130,7 +130,7 @@ xla_cc_test( ":test", ":types", ":xla_data_proto_cc", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -162,7 +162,7 @@ xla_cc_test( "@com_google_absl//absl/log:log_streamer", "@com_google_absl//absl/random", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -226,7 +226,7 @@ xla_cc_test( ":test", ":types", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -262,9 +262,9 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -291,7 +291,6 @@ xla_cc_test( "@com_google_absl//absl/numeric:bits", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:ml_dtypes", - "@tsl//tsl/platform:test_main", ], ) @@ -354,10 +353,10 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@ml_dtypes//:float8", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:ml_dtypes", - "@tsl//tsl/platform:test_main", ], ) @@ -380,7 +379,7 @@ xla_cc_test( deps = [ ":permutation_util", ":test", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -408,7 +407,7 @@ xla_cc_test( deps = [ ":test", ":util", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -486,8 +485,8 @@ xla_cc_test( ":xla_data_proto_cc", "@com_google_absl//absl/hash:hash_testing", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -504,10 +503,10 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:protobuf", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -519,8 +518,8 @@ xla_cc_test( ":test", ":test_helpers", ":xla_data_proto_cc", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -534,10 +533,10 @@ xla_cc_test( ":xla_data_proto_cc", "@com_google_absl//absl/log", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:status_matchers", - "@tsl//tsl/platform:test_main", ], ) @@ -548,7 +547,7 @@ xla_cc_test( ":shape_util", ":test", ":xla_data_proto_cc", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -560,7 +559,7 @@ xla_cc_test( ":test", ":xla_data_proto_cc", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -633,7 +632,6 @@ xla_cc_test( "@tsl//tsl/platform:ml_dtypes", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -659,8 +657,8 @@ xla_cc_test( ":literal", ":literal_pool", ":literal_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -711,7 +709,6 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:ml_dtypes", - "@tsl//tsl/platform:test_main", ], ) @@ -787,8 +784,8 @@ xla_cc_test( ":test", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", - "@tsl//tsl/platform:test_main", ], ) @@ -811,9 +808,9 @@ xla_cc_test( deps = [ ":array2d", ":test", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@tsl//tsl/platform:ml_dtypes", - "@tsl//tsl/platform:test_main", ], ) @@ -835,7 +832,7 @@ xla_cc_test( ":array3d", ":test", ":types", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -862,8 +859,8 @@ xla_cc_test( ":test", "@com_google_absl//absl/log", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", - "@tsl//tsl/platform:test_main", ], ) @@ -947,8 +944,8 @@ xla_cc_test( ":types", ":xla_data_proto_cc", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", - "@tsl//tsl/platform:test_main", ], ) @@ -983,8 +980,8 @@ xla_cc_test( ":types", "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", - "@tsl//tsl/platform:test_main", ], ) @@ -1017,8 +1014,8 @@ xla_cc_test( ":test", ":xla_data_proto_cc", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -1063,7 +1060,7 @@ xla_cc_test( ":window_util", ":xla_data_proto_cc", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1113,7 +1110,7 @@ xla_cc_test( "//xla/hlo/builder:padding", "//xla/tests:literal_test_util", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1290,8 +1287,8 @@ xla_cc_test( ":autotuning_proto_cc", ":test", ":test_helpers", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1348,7 +1345,6 @@ xla_cc_test( "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/backends/cpu/codegen/BUILD b/xla/backends/cpu/codegen/BUILD index 9f442ae560686..0e29de1751af9 100644 --- a/xla/backends/cpu/codegen/BUILD +++ b/xla/backends/cpu/codegen/BUILD @@ -138,6 +138,7 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:AsmParser", "@llvm-project//llvm:Core", "@llvm-project//llvm:OrcJIT", @@ -148,7 +149,6 @@ xla_cc_test( "@tsl//tsl/platform:errors", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/backends/cpu/nanort/BUILD b/xla/backends/cpu/nanort/BUILD index 55e7bfc2687ce..003bfb03fc843 100644 --- a/xla/backends/cpu/nanort/BUILD +++ b/xla/backends/cpu/nanort/BUILD @@ -60,11 +60,11 @@ xla_cc_test( "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/backends/cpu/runtime/BUILD b/xla/backends/cpu/runtime/BUILD index 3d20852890007..6bfed764c5896 100644 --- a/xla/backends/cpu/runtime/BUILD +++ b/xla/backends/cpu/runtime/BUILD @@ -113,11 +113,11 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:platform_port", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -370,11 +370,11 @@ xla_cc_test( "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status", "@com_google_absl//absl/synchronization", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@tsl//tsl/platform:env", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -816,9 +816,9 @@ xla_cc_test( "//xla/tsl/concurrency:async_value", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -904,9 +904,9 @@ xla_cc_test( "//xla/tsl/concurrency:async_value", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -988,8 +988,8 @@ xla_cc_test( deps = [ ":resource_use", "//xla/service:buffer_assignment", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/backends/cpu/runtime/xnnpack/BUILD b/xla/backends/cpu/runtime/xnnpack/BUILD index 4b01b7614131f..f1be295687976 100644 --- a/xla/backends/cpu/runtime/xnnpack/BUILD +++ b/xla/backends/cpu/runtime/xnnpack/BUILD @@ -118,12 +118,12 @@ xla_cc_test( "@XNNPACK", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/synchronization", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@pthreadpool", "@tsl//tsl/platform:env", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/backends/cpu/testlib/BUILD b/xla/backends/cpu/testlib/BUILD index 1248e5f3d56ee..c90589c09bf4f 100644 --- a/xla/backends/cpu/testlib/BUILD +++ b/xla/backends/cpu/testlib/BUILD @@ -58,9 +58,9 @@ xla_cc_test( "//xla/runtime:buffer_use", "//xla/stream_executor:launch_dim", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -137,10 +137,10 @@ xla_cc_test( "//xla/codegen:llvm_ir_kernel_source", "//xla/runtime:buffer_use", "//xla/stream_executor:launch_dim", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:casts", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/backends/gpu/collectives/BUILD b/xla/backends/gpu/collectives/BUILD index 122a49bbbdddb..cc5818f6a8fbd 100644 --- a/xla/backends/gpu/collectives/BUILD +++ b/xla/backends/gpu/collectives/BUILD @@ -79,9 +79,9 @@ xla_cc_test( "@com_google_absl//absl/container:btree", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/backends/profiler/gpu/BUILD b/xla/backends/profiler/gpu/BUILD index ee591b48deb8d..8c195230c8aec 100644 --- a/xla/backends/profiler/gpu/BUILD +++ b/xla/backends/profiler/gpu/BUILD @@ -109,8 +109,8 @@ xla_test( ], deps = [ ":cupti_interface", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_cuda_is_configured([ ":cuda_test", ":cupti_error_manager", @@ -397,8 +397,8 @@ xla_test( ":cupti_collector", ":cupti_tracer", ":cupti_utils", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -418,9 +418,9 @@ xla_test( ]) + [ "//xla/tsl/profiler/utils:xplane_builder", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", "@tsl//tsl/profiler/protobuf:xplane_proto_cc", ], ) @@ -449,6 +449,5 @@ xla_test( deps = [ ":nvtx_with_cuda_kernels", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/client/BUILD b/xla/client/BUILD index 7ae6ae2f6c97b..de2e23dfa657e 100644 --- a/xla/client/BUILD +++ b/xla/client/BUILD @@ -116,11 +116,11 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:protobuf", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/ffi/BUILD b/xla/ffi/BUILD index d752db481938a..1746bb949bccb 100644 --- a/xla/ffi/BUILD +++ b/xla/ffi/BUILD @@ -49,10 +49,10 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -80,9 +80,9 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -109,9 +109,9 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -225,13 +225,13 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@tsl//tsl/platform:env", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -257,8 +257,8 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/ffi/api/BUILD b/xla/ffi/api/BUILD index 29262de8e70cf..a3207605591f1 100644 --- a/xla/ffi/api/BUILD +++ b/xla/ffi/api/BUILD @@ -92,12 +92,12 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/synchronization", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@tsl//tsl/platform:env", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/analysis/BUILD b/xla/hlo/analysis/BUILD index 141088fd9c7fc..36a90a8638585 100644 --- a/xla/hlo/analysis/BUILD +++ b/xla/hlo/analysis/BUILD @@ -44,9 +44,9 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:computation_placer_hdr", "//xla/service:hlo_module_config", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -77,9 +77,9 @@ xla_cc_test( "//xla/service:computation_placer", "//xla/service:hlo_module_config", "@com_google_absl//absl/random", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -126,7 +126,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -176,8 +175,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -234,9 +233,9 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -277,8 +276,8 @@ xla_cc_test( "@com_google_absl//absl/log", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -311,7 +310,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/tsl/lib/core:status_test_util", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -351,9 +350,9 @@ xla_cc_test( "//xla:test_helpers", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -405,10 +404,10 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -472,10 +471,10 @@ xla_cc_test( "//xla/service:logical_buffer", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -517,7 +516,6 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/builder/BUILD b/xla/hlo/builder/BUILD index 55c339c3b85d1..d325e39be6aed 100644 --- a/xla/hlo/builder/BUILD +++ b/xla/hlo/builder/BUILD @@ -46,8 +46,8 @@ xla_cc_test( srcs = ["padding_test.cc"], deps = [ ":padding", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -183,8 +183,8 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/builder/lib/BUILD b/xla/hlo/builder/lib/BUILD index f0022ca15116c..e22418d995401 100644 --- a/xla/hlo/builder/lib/BUILD +++ b/xla/hlo/builder/lib/BUILD @@ -50,7 +50,7 @@ xla_test( "//xla/tests:client_library_test_base", "//xla/tests:test_macros_header", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -91,7 +91,6 @@ xla_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:protobuf", - "@tsl//tsl/platform:test_main", ], ) @@ -143,7 +142,6 @@ xla_test( "//xla/tests:client_library_test_base", "//xla/tests:test_macros_header", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -251,7 +249,6 @@ xla_test( "//xla/tests:test_macros_header", "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -302,7 +299,6 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -341,7 +337,7 @@ xla_test( "//xla/tests:test_macros_header", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -378,7 +374,6 @@ xla_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -420,7 +415,6 @@ xla_test( "//xla/tests:test_macros_header", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -498,8 +492,8 @@ xla_test( "//xla/hlo/builder:xla_builder", "//xla/tests:client_library_test_base", "//xla/tests:test_macros_header", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -537,7 +531,7 @@ xla_test( "//xla/tests:client_library_test_base", "//xla/tests:test_macros_header", "@com_google_absl//absl/algorithm:container", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -573,7 +567,6 @@ xla_test( "//xla/tests:test_macros_header", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:bfloat16", - "@tsl//tsl/platform:test_main", ], ) @@ -619,7 +612,7 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -668,7 +661,7 @@ xla_test( "//xla/tests:client_library_test_base", "//xla/tests:test_macros_header", "@com_google_absl//absl/status:statusor", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -714,7 +707,6 @@ xla_test( "@com_google_absl//absl/status", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -753,7 +745,7 @@ xla_test( "//xla/hlo/builder:xla_builder", "//xla/tests:client_library_test_base", "//xla/tests:test_macros_header", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -785,7 +777,7 @@ xla_test( "//xla/hlo/builder:xla_builder", "//xla/service", "//xla/tests:client_library_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/evaluator/BUILD b/xla/hlo/evaluator/BUILD index b9a3c348ef30b..99948b3e7abe8 100644 --- a/xla/hlo/evaluator/BUILD +++ b/xla/hlo/evaluator/BUILD @@ -113,7 +113,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -156,10 +155,10 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:errors", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) diff --git a/xla/hlo/experimental/auto_sharding/BUILD b/xla/hlo/experimental/auto_sharding/BUILD index 8e9d0cf51234e..3aeac130503ff 100644 --- a/xla/hlo/experimental/auto_sharding/BUILD +++ b/xla/hlo/experimental/auto_sharding/BUILD @@ -409,8 +409,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -434,8 +434,8 @@ xla_cc_test( "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ] + if_google(["@com_google_ortools//ortools/linear_solver:linear_solver_scip"]), ) diff --git a/xla/hlo/ir/BUILD b/xla/hlo/ir/BUILD index 560f46517e07e..a78c7a0be3f8a 100644 --- a/xla/hlo/ir/BUILD +++ b/xla/hlo/ir/BUILD @@ -199,8 +199,8 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/synchronization", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -240,8 +240,8 @@ xla_cc_test( ":hlo_instruction_utils", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_query", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -275,10 +275,10 @@ cc_test( ":ptrvec", "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -307,6 +307,5 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/service:hlo_proto_cc", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/parser/BUILD b/xla/hlo/parser/BUILD index 5f52322750cee..c224ea4d1ee1d 100644 --- a/xla/hlo/parser/BUILD +++ b/xla/hlo/parser/BUILD @@ -87,11 +87,11 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/pass/BUILD b/xla/hlo/pass/BUILD index 7ff8149a96b9e..5ed56a7e3b3f7 100644 --- a/xla/hlo/pass/BUILD +++ b/xla/hlo/pass/BUILD @@ -98,6 +98,5 @@ xla_cc_test( "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/transforms/BUILD b/xla/hlo/transforms/BUILD index 2c8f45317a59e..d2a4481f7da11 100644 --- a/xla/hlo/transforms/BUILD +++ b/xla/hlo/transforms/BUILD @@ -66,8 +66,8 @@ xla_cc_test( "//xla/tests:literal_test_util", "@com_google_absl//absl/log", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -212,7 +212,7 @@ xla_cc_test( "//xla/service:pattern_matcher_gmock", "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -272,8 +272,8 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -308,7 +308,6 @@ xla_cc_test( "//xla/service:pattern_matcher_gmock", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -345,8 +344,8 @@ xla_cc_test( "//xla/service:pattern_matcher_gmock", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -392,7 +391,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -421,8 +419,8 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -462,7 +460,6 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -478,7 +475,6 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "@com_google_absl//absl/log", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -517,9 +513,9 @@ xla_cc_test( "//xla/service:pattern_matcher_gmock", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -565,7 +561,7 @@ xla_cc_test( "//xla/hlo/parser:hlo_parser", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -581,7 +577,7 @@ xla_cc_test( "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -612,7 +608,7 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -647,7 +643,7 @@ xla_cc_test( "//xla/service:pattern_matcher_gmock", "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -687,7 +683,7 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -737,8 +733,8 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -803,8 +799,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -841,8 +837,8 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -897,9 +893,9 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -932,7 +928,7 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1010,8 +1006,8 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1049,7 +1045,7 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1104,6 +1100,5 @@ xla_cc_test( "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/transforms/collectives/BUILD b/xla/hlo/transforms/collectives/BUILD index b7f8aba244bdf..ea67ab7c14934 100644 --- a/xla/hlo/transforms/collectives/BUILD +++ b/xla/hlo/transforms/collectives/BUILD @@ -45,8 +45,8 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -89,7 +89,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -130,7 +129,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -164,8 +162,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -197,7 +195,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:pattern_matcher", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -230,7 +228,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -271,7 +268,6 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -317,7 +313,6 @@ xla_cc_test( "@com_google_absl//absl/log:check", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -352,7 +347,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -389,8 +383,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -462,7 +456,7 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) diff --git a/xla/hlo/transforms/simplifiers/BUILD b/xla/hlo/transforms/simplifiers/BUILD index e19f32f2cbbb7..23c6f447f447b 100644 --- a/xla/hlo/transforms/simplifiers/BUILD +++ b/xla/hlo/transforms/simplifiers/BUILD @@ -54,8 +54,8 @@ 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_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -92,8 +92,8 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -122,7 +122,7 @@ xla_cc_test( "//xla:test", "//xla:test_helpers", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@tsl//tsl/platform:test_main", # fixdeps: keep + "@com_google_googletest//:gtest_main", # fixdeps: keep ], ) @@ -162,7 +162,7 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:float_support", "@com_google_absl//absl/status:statusor", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -211,8 +211,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -249,7 +249,6 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -287,8 +286,8 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:call_graph", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -347,11 +346,11 @@ xla_cc_test( "@com_google_absl//absl/container:flat_hash_map", "@com_google_absl//absl/status", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -433,7 +432,6 @@ xla_cc_test( "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -468,7 +466,6 @@ xla_cc_test( "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -502,8 +499,8 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -543,7 +540,7 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", - "@tsl//tsl/platform:test_main", # fixdeps: keep + "@com_google_googletest//:gtest_main", # fixdeps: keep ], ) @@ -578,7 +575,6 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -623,8 +619,8 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -663,7 +659,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", # fixdeps: keep ], ) @@ -695,8 +690,8 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -735,7 +730,7 @@ xla_cc_test( "//xla/service:shape_inference", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings", - "@tsl//tsl/platform:test_main", # fixdeps: keep + "@com_google_googletest//:gtest_main", # fixdeps: keep ], ) @@ -778,7 +773,7 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -819,7 +814,6 @@ xla_cc_test( "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -833,7 +827,7 @@ cc_library( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -844,7 +838,6 @@ xla_cc_test( ":hlo_rematerialization_test_utils", "//xla/hlo/ir:hlo", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -903,9 +896,9 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -952,8 +945,8 @@ xla_cc_test( "//xla/service:pattern_matcher_gmock", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -964,11 +957,11 @@ xla_cc_test( # deps = [ # ":hlo_constant_folding", # "@com_google_googletest//:gtest", +# "@com_google_googletest//:gtest_main", # "@com_google_absl//absl/strings:str_format", # "//xla:test", # "//xla/hlo/testlib:hlo_hardware_independent_test_base", # "@tsl//tsl/platform:statusor", -# "@tsl//tsl/platform:test_main", # ], # ) # copybara:uncomment_end @@ -1003,7 +996,7 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/hlo/utils:hlo_matchers", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1038,7 +1031,7 @@ xla_cc_test( "//xla/tests:literal_test_util", "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1070,9 +1063,9 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:hlo_hardware_independent_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1110,8 +1103,8 @@ xla_cc_test( "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1139,7 +1132,7 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1179,8 +1172,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1213,8 +1206,8 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1251,8 +1244,8 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -1298,7 +1291,6 @@ xla_cc_test( "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1340,7 +1332,6 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1371,8 +1362,8 @@ xla_cc_test( "//xla/hlo/utils:hlo_matchers", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1402,7 +1393,6 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1441,7 +1431,7 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_absl//absl/strings", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1472,7 +1462,7 @@ xla_cc_test( "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) diff --git a/xla/hlo/translate/hlo_to_mhlo/BUILD b/xla/hlo/translate/hlo_to_mhlo/BUILD index 6c503918aa63c..3e1589fe77072 100644 --- a/xla/hlo/translate/hlo_to_mhlo/BUILD +++ b/xla/hlo/translate/hlo_to_mhlo/BUILD @@ -201,10 +201,10 @@ xla_cc_test( "//xla/mlir_hlo", "//xla/tsl/lib/core:status_test_util", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@llvm-project//mlir:IR", "@llvm-project//mlir:Support", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/translate/mhlo_to_hlo/BUILD b/xla/hlo/translate/mhlo_to_hlo/BUILD index 196174941a6e6..e9c60512470b4 100644 --- a/xla/hlo/translate/mhlo_to_hlo/BUILD +++ b/xla/hlo/translate/mhlo_to_hlo/BUILD @@ -219,6 +219,7 @@ xla_cc_test( ":mlir_hlo_to_hlo", "//xla/mlir/utils:error_util", "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest_main", "@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:IR", "@llvm-project//mlir:Parser", @@ -226,7 +227,6 @@ xla_cc_test( "@stablehlo//:register", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -310,9 +310,9 @@ xla_cc_test( "//xla/hlo/translate/hlo_to_mhlo:hlo_utils", "//xla/mlir_hlo", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:Support", "@llvm-project//mlir:IR", "@tsl//tsl/platform:protobuf", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/hlo/utils/BUILD b/xla/hlo/utils/BUILD index d1d75b6460e4c..1f1b2b887a772 100644 --- a/xla/hlo/utils/BUILD +++ b/xla/hlo/utils/BUILD @@ -68,7 +68,6 @@ xla_cc_test( "@com_google_absl//absl/container:flat_hash_map", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -96,7 +95,6 @@ xla_cc_test( "//xla:xla_data_proto_cc", "//xla/hlo/testlib:hlo_hardware_independent_test_base", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -158,7 +156,6 @@ xla_cc_test( "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/mlir/utils/BUILD b/xla/mlir/utils/BUILD index f80d1fe160077..4b6baae25b278 100644 --- a/xla/mlir/utils/BUILD +++ b/xla/mlir/utils/BUILD @@ -41,7 +41,6 @@ cc_test( "@llvm-project//llvm:Support", "@llvm-project//mlir:IR", "@tsl//tsl/platform:status", - "@tsl//tsl/platform:test_main", ], ) @@ -70,9 +69,9 @@ xla_cc_test( "//xla:xla_data_proto_cc", "@com_google_absl//absl/functional:function_ref", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:Support", "@llvm-project//mlir:IR", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/pjrt/BUILD b/xla/pjrt/BUILD index abc88ca806cd5..827d5a4d01814 100644 --- a/xla/pjrt/BUILD +++ b/xla/pjrt/BUILD @@ -77,7 +77,6 @@ xla_cc_test( "@com_google_absl//absl/synchronization", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", - "@tsl//tsl/platform:test_main", ], ) @@ -134,8 +133,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -678,8 +677,8 @@ xla_cc_test( ":pjrt_future", "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -724,7 +723,7 @@ xla_cc_test( deps = [ ":lru_cache", "//xla:test", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -772,11 +771,11 @@ xla_cc_test( "//xla/tsl/protobuf:error_codes_proto_impl_cc", "@com_google_absl//absl/container:inlined_vector", "@com_google_absl//absl/numeric:int128", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@tsl//tsl/platform:env", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/pjrt/cpu/BUILD b/xla/pjrt/cpu/BUILD index 6a7d18cdbaabf..419049911ea2b 100644 --- a/xla/pjrt/cpu/BUILD +++ b/xla/pjrt/cpu/BUILD @@ -18,7 +18,7 @@ xla_cc_test( deps = [ ":cpu_client", "//xla/pjrt:pjrt_client_test_common", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -251,6 +251,7 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/synchronization", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:logging", @@ -258,6 +259,5 @@ xla_cc_test( "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/pjrt/distributed/BUILD b/xla/pjrt/distributed/BUILD index 963d5c697c58c..368da962a107e 100644 --- a/xla/pjrt/distributed/BUILD +++ b/xla/pjrt/distributed/BUILD @@ -53,10 +53,10 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/time", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -152,11 +152,11 @@ xla_cc_test( "@com_google_absl//absl/synchronization", "@com_google_absl//absl/time", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + tsl_grpc_cc_dependencies(), ) diff --git a/xla/pjrt/gpu/BUILD b/xla/pjrt/gpu/BUILD index b1d392ce0b4b3..9f8972c0e4db2 100644 --- a/xla/pjrt/gpu/BUILD +++ b/xla/pjrt/gpu/BUILD @@ -252,7 +252,7 @@ xla_test( deps = [ ":se_gpu_pjrt_client", "//xla/pjrt:pjrt_client_test_common", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -404,12 +404,12 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:IR", "@llvm-project//mlir:Parser", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -436,12 +436,12 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@llvm-project//mlir:FuncDialect", "@llvm-project//mlir:IR", "@llvm-project//mlir:Parser", "@tsl//tsl/platform:casts", "@tsl//tsl/platform:protobuf", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/pjrt/plugin/xla_cpu/BUILD b/xla/pjrt/plugin/xla_cpu/BUILD index a7c085e290080..67ac6550fe39c 100644 --- a/xla/pjrt/plugin/xla_cpu/BUILD +++ b/xla/pjrt/plugin/xla_cpu/BUILD @@ -101,8 +101,8 @@ xla_cc_test( deps = [ ":cpu_topology", ":cpu_topology_proto_cc", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:protobuf", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/python/BUILD b/xla/python/BUILD index 49fc21ee54019..e848931e3f42b 100644 --- a/xla/python/BUILD +++ b/xla/python/BUILD @@ -1403,8 +1403,8 @@ xla_cc_test( "//xla/tsl/profiler/utils:file_system_utils", "//xla/tsl/profiler/utils:xplane_builder", "//xla/tsl/profiler/utils:xplane_schema", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", "@tsl//tsl/profiler/protobuf:profiled_instructions_proto_cc_impl", "@tsl//tsl/profiler/protobuf:xplane_proto_cc", ], @@ -1514,8 +1514,8 @@ xla_cc_test( deps = [ ":aggregate_profile", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", "@tsl//tsl/profiler/protobuf:profiled_instructions_proto_cc", "@tsl//tsl/profiler/protobuf:profiled_instructions_proto_cc_impl", ], diff --git a/xla/runtime/BUILD b/xla/runtime/BUILD index 03fc0d112dc32..3e8c90862a93c 100644 --- a/xla/runtime/BUILD +++ b/xla/runtime/BUILD @@ -33,7 +33,7 @@ xla_cc_test( ":buffer_use", "//xla/service:buffer_assignment", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/BUILD b/xla/service/BUILD index 7b0dd97c91226..996515e1e6d8c 100644 --- a/xla/service/BUILD +++ b/xla/service/BUILD @@ -162,7 +162,6 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1270,8 +1269,8 @@ xla_cc_test( "@com_google_absl//absl/log", "@com_google_absl//absl/strings:str_format", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -2496,7 +2495,7 @@ xla_cc_test( "//xla/tests:hlo_test_base", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/log", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -5361,7 +5360,7 @@ cc_test( ":custom_call_status", ":custom_call_target_registry", "//xla:test", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -5430,8 +5429,8 @@ xla_cc_test( deps = [ ":custom_call_status_internal", ":custom_call_status_test_c_caller", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -5734,9 +5733,9 @@ xla_cc_test( ":lockable", "@com_google_absl//absl/synchronization", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -5771,10 +5770,10 @@ xla_cc_test( "@com_google_absl//absl/synchronization", "@com_google_absl//absl/time", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -5938,9 +5937,9 @@ xla_cc_test( "//xla/hlo/ir:hlo", "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/cpu/BUILD b/xla/service/cpu/BUILD index f82a470855e31..ad1ffc81f5881 100644 --- a/xla/service/cpu/BUILD +++ b/xla/service/cpu/BUILD @@ -713,6 +713,7 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:Core", "@llvm-project//llvm:Support", "@llvm-project//llvm:Target", @@ -721,7 +722,6 @@ xla_cc_test( "@tsl//tsl/platform:errors", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -1703,7 +1703,6 @@ xla_cc_test( "@com_google_googletest//:gtest_main", "@llvm-project//llvm:Core", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/cpu/benchmarks/BUILD b/xla/service/cpu/benchmarks/BUILD index 13a9e23535396..f39025e170cd2 100644 --- a/xla/service/cpu/benchmarks/BUILD +++ b/xla/service/cpu/benchmarks/BUILD @@ -55,7 +55,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -73,7 +73,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -91,7 +91,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -109,7 +109,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -128,7 +128,7 @@ xla_cc_test( "@com_google_absl//absl/strings:str_format", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -146,7 +146,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -164,7 +164,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -180,7 +180,7 @@ xla_cc_test( "//xla/tsl/platform:logging", "//xla/tsl/platform:test_benchmark", "@com_google_absl//absl/strings", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -201,7 +201,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -220,7 +220,7 @@ xla_cc_test( "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -238,7 +238,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -256,7 +256,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -272,7 +272,7 @@ xla_cc_test( "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -290,7 +290,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -308,7 +308,7 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -325,6 +325,6 @@ xla_cc_test( "//xla/tsl/platform:test_benchmark", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) diff --git a/xla/service/cpu/tests/BUILD b/xla/service/cpu/tests/BUILD index 14be483c11dae..2fd154879cd7c 100644 --- a/xla/service/cpu/tests/BUILD +++ b/xla/service/cpu/tests/BUILD @@ -34,7 +34,7 @@ cc_library( deps = [ "//xla/service:cpu_plugin", "//xla/tests:llvm_irgen_test_base", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -57,10 +57,10 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:ARMCodeGen", # fixdeps: keep "@llvm-project//llvm:X86CodeGen", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -75,7 +75,6 @@ xla_cc_test( "//xla/service/cpu:cpu_compiler", "//xla/service/cpu:test_header_helper", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -94,8 +93,8 @@ xla_cc_test( "//xla/service/cpu:cpu_instruction_fusion", "//xla/tests:hlo_test_base", "//xla/tests:literal_test_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -106,9 +105,9 @@ xla_cc_test( "//xla:shape_util", "//xla:xla_data_proto_cc", "//xla/service/llvm_ir:llvm_util", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:ir_headers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -144,11 +143,11 @@ xla_cc_test( "//xla/service/llvm_ir:ir_array", "//xla/service/llvm_ir:llvm_util", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:Core", "@llvm-project//llvm:Support", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -169,7 +168,6 @@ xla_cc_test( "@llvm-project//llvm:Target", "@llvm-project//llvm:X86CodeGen", # fixdeps: keep "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -188,7 +186,6 @@ xla_cc_test( "//xla/tests:test_utils", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -205,7 +202,6 @@ xla_cc_test( "@llvm-project//llvm:Target", "@llvm-project//llvm:X86CodeGen", # fixdeps: keep "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -215,8 +211,8 @@ xla_cc_test( deps = [ ":cpu_codegen_test", "//xla/service/cpu:cpu_compiler", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -238,9 +234,9 @@ xla_cc_test( "//xla/service:cpu_plugin", "//xla/tests:client_library_test_base", "//xla/tests:literal_test_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -254,7 +250,6 @@ xla_cc_test( "//xla/service/cpu:test_header_helper", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -267,7 +262,6 @@ xla_cc_test( "//xla/service/cpu:test_header_helper", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -280,7 +274,6 @@ xla_cc_test( "//xla/service/cpu:test_header_helper", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -296,8 +289,8 @@ xla_cc_test( "//xla/service/cpu:test_header_helper", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -317,7 +310,6 @@ xla_cc_test( "//xla/service/cpu:test_header_helper", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -341,7 +333,6 @@ xla_cc_test( "@llvm-project//llvm:X86CodeGen", # fixdeps: keep "@tsl//tsl/platform:platform_port", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -356,7 +347,6 @@ xla_cc_test( "@llvm-project//llvm:ARMCodeGen", # fixdeps: keep "@llvm-project//llvm:X86CodeGen", # fixdeps: keep "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/gpu/BUILD b/xla/service/gpu/BUILD index e70562a6afc12..416f7337e7395 100644 --- a/xla/service/gpu/BUILD +++ b/xla/service/gpu/BUILD @@ -2515,10 +2515,10 @@ xla_test( "//xla/stream_executor:platform_manager", "//xla/stream_executor:stream", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:ml_dtypes", "@tsl//tsl/platform:status", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_gpu_is_configured([ ":buffer_comparator", "//xla/stream_executor:device_memory", @@ -2629,8 +2629,8 @@ xla_test( "//xla/stream_executor:device_description", "//xla/tests:hlo_test_base", "//xla/tests:test_macros_header", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -2663,10 +2663,10 @@ xla_cc_test( "//xla/stream_executor:dnn", "//xla/tests:test_utils", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:path", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/gpu/autotuning/BUILD b/xla/service/gpu/autotuning/BUILD index 7bde131d059d7..e3997a2e526c7 100644 --- a/xla/service/gpu/autotuning/BUILD +++ b/xla/service/gpu/autotuning/BUILD @@ -427,9 +427,9 @@ xla_test( "//xla/tsl/protobuf:dnn_proto_cc", "@com_google_absl//absl/log", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -531,9 +531,9 @@ xla_test( "//xla/tests:hlo_test_base", "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/gpu/fusions/BUILD b/xla/service/gpu/fusions/BUILD index aaaf8728d1100..d4193482ef95f 100644 --- a/xla/service/gpu/fusions/BUILD +++ b/xla/service/gpu/fusions/BUILD @@ -186,10 +186,10 @@ xla_test( "//xla/tests:test_utils", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_cuda_is_configured([ "@local_config_cuda//cuda:cuda_headers", ]) + if_rocm_is_configured([ @@ -487,7 +487,6 @@ xla_test( "@tsl//tsl/platform:path", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/gpu/kernels/BUILD b/xla/service/gpu/kernels/BUILD index 2cbc547a85edd..3a11d8bae997c 100644 --- a/xla/service/gpu/kernels/BUILD +++ b/xla/service/gpu/kernels/BUILD @@ -138,7 +138,6 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -211,7 +210,6 @@ xla_test( "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -256,10 +254,10 @@ xla_test( "@com_google_absl//absl/random", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:path", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -314,7 +312,6 @@ xla_test( "@tsl//tsl/platform:path", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -338,11 +335,11 @@ xla_cc_binary( "//xla/stream_executor:stream", "//xla/stream_executor:stream_executor_h", "//xla/stream_executor/cuda:cuda_platform", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -549,6 +546,5 @@ xla_test( "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/gpu/model/BUILD b/xla/service/gpu/model/BUILD index 65246c3cac534..6b32e49517129 100644 --- a/xla/service/gpu/model/BUILD +++ b/xla/service/gpu/model/BUILD @@ -938,9 +938,9 @@ xla_test( "//xla/hlo/ir:hlo", "//xla/tests:hlo_test_base", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:status_matchers", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/gpu/runtime/BUILD b/xla/service/gpu/runtime/BUILD index fab5653b0f748..c7d7a0bff5294 100644 --- a/xla/service/gpu/runtime/BUILD +++ b/xla/service/gpu/runtime/BUILD @@ -172,11 +172,11 @@ xla_test( "@com_google_absl//absl/status", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -260,9 +260,9 @@ xla_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_cuda_is_configured([ "@local_config_cuda//cuda:cuda_headers", ]), @@ -374,9 +374,9 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", "@tsl//tsl/profiler/lib:profiler_lock", ] + if_cuda_is_configured([ "@local_config_cuda//cuda:cuda_headers", diff --git a/xla/service/gpu/tests/BUILD b/xla/service/gpu/tests/BUILD index 3140bdd3bbe03..a2dd9c1eecb70 100644 --- a/xla/service/gpu/tests/BUILD +++ b/xla/service/gpu/tests/BUILD @@ -110,8 +110,8 @@ xla_test( deps = [ ":gpu_codegen_test", "//xla:literal_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -129,7 +129,6 @@ xla_cc_test( "//xla/tests:hlo_test_base", "//xla/tests:literal_test_util", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -146,7 +145,6 @@ xla_cc_test( "//xla/tests:hlo_test_base", "//xla/tests:literal_test_util", "@com_google_googletest//:gtest_main", - "@tsl//tsl/platform:test_main", ], ) @@ -158,8 +156,8 @@ xla_test( ":gpu_codegen_test", "//xla:error_spec", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -178,8 +176,8 @@ xla_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/algorithm:container", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -196,9 +194,9 @@ xla_test( "//xla/service:executable", "//xla/stream_executor:stream_executor_memory_allocator", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -214,9 +212,9 @@ xla_test( "//xla:error_spec", "//xla/stream_executor:device_description", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -232,9 +230,9 @@ xla_test( "//xla/hlo/parser:hlo_parser", "//xla/stream_executor:device_description", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -254,7 +252,6 @@ xla_test( "//xla/tests:hlo_test_base", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -268,8 +265,8 @@ xla_test( ":gpu_codegen_test", "//xla:error_spec", "//xla/hlo/testlib:verified_hlo_module", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -284,9 +281,9 @@ xla_test( "//xla:literal_util", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:verified_hlo_module", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -300,8 +297,8 @@ xla_test( ":gpu_codegen_test", "//xla:error_spec", "//xla/hlo/testlib:verified_hlo_module", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -313,7 +310,7 @@ xla_test( ":gpu_codegen_test", "//xla:shape_util", "//xla/hlo/ir:hlo", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -339,7 +336,6 @@ xla_test( "@llvm-project//mlir:IR", "@llvm-project//mlir:Support", "@tsl//tsl/platform:status_matchers", - "@tsl//tsl/platform:test_main", ], ) @@ -352,7 +348,7 @@ xla_test( "//xla:shape_util", "//xla/hlo/ir:hlo", "//xla/hlo/testlib:verified_hlo_module", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -369,8 +365,8 @@ xla_test( "//xla/hlo/ir:hlo", "//xla/service:hlo_module_config", "//xla/tests:hlo_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -389,8 +385,8 @@ xla_test( "//xla/client:local_client", "//xla/hlo/builder:xla_builder", "//xla/tests:client_library_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", - "@tsl//tsl/platform:test_main", ], ) @@ -407,8 +403,8 @@ xla_test( "//xla/service:hlo_module_config", "//xla/tests:hlo_test_base", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -422,8 +418,8 @@ xla_test( "//xla:shape_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -436,8 +432,8 @@ xla_test( "//xla:shape_util", "//xla:xla_data_proto_cc", "//xla/hlo/ir:hlo", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -450,8 +446,8 @@ xla_test( "//xla:debug_options_flags", "//xla/service:hlo_module_config", "//xla/tests:hlo_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -461,8 +457,8 @@ xla_test( backends = ["gpu"], deps = [ ":gpu_codegen_test", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -472,8 +468,8 @@ xla_test( backends = ["gpu"], deps = [ ":gpu_codegen_test", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -486,8 +482,8 @@ xla_test( "//xla:error_spec", "//xla/service:hlo_module_config", "//xla/tests:hlo_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -521,7 +517,7 @@ xla_test( deps = [ ":gpu_codegen_test", "//xla:error_spec", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -542,9 +538,9 @@ xla_test( "//xla:xla_data_proto_cc", "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@eigen_archive//:eigen3", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -632,7 +628,7 @@ xla_test( deps = [ ":gpu_codegen_test", "//xla:error_spec", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -642,7 +638,7 @@ xla_test( backends = ["gpu"], deps = [ ":gpu_codegen_test", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -653,7 +649,7 @@ xla_test( deps = [ "//xla:debug_options_flags", "//xla/tests:hlo_test_base", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -672,10 +668,10 @@ xla_test( "//xla/stream_executor:stream_executor_h", "@com_google_absl//absl/log", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_cuda_is_configured([ "//xla/service/gpu:stream_executor_util", "//xla/stream_executor:device_memory", @@ -695,8 +691,8 @@ xla_test( deps = [ "//xla:error_spec", "//xla/tests:hlo_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:tensor_float_32_utils", - "@tsl//tsl/platform:test_main", ], ) @@ -720,7 +716,7 @@ xla_test( "//xla/tests:xla_internal_test_main", "//xla/tsl/lib/core:status_test_util", ], - ["@tsl//tsl/platform:test_main"], # b/317293391 + ["@com_google_googletest//:gtest_main"], # b/317293391 ), ) @@ -741,7 +737,6 @@ xla_test( "@com_google_absl//absl/strings", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -867,8 +862,8 @@ xla_test( deps = [ ":gpu_codegen_test", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -880,7 +875,7 @@ xla_test( "//xla:xla_proto_cc", "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings:string_view", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -894,6 +889,6 @@ xla_test( "//xla/tests:hlo_test_base", "//xla/tests:literal_test_util", "//xla/tsl/platform:test", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) diff --git a/xla/service/gpu/transforms/BUILD b/xla/service/gpu/transforms/BUILD index f8f0e93a387f5..8af157511f1d5 100644 --- a/xla/service/gpu/transforms/BUILD +++ b/xla/service/gpu/transforms/BUILD @@ -238,9 +238,9 @@ xla_cc_test( "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -361,7 +361,6 @@ xla_cc_test( "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -392,8 +391,8 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -587,7 +586,7 @@ xla_cc_test( "//xla/hlo/transforms:while_loop_trip_count_annotator", "//xla/service:collective_ops_utils", "//xla/tests:hlo_test_base", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -969,7 +968,6 @@ xla_test( "@com_google_absl//absl/strings:str_format", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1050,7 +1048,6 @@ xla_test( "@com_google_absl//absl/strings", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ] + if_cuda_is_configured([ "@local_config_cuda//cuda:cuda_headers", "@local_config_cuda//cuda:cudnn_header", @@ -1400,8 +1397,8 @@ xla_cc_test( "//xla/service/gpu/kernels:custom_kernel_fusion_pattern", "//xla/stream_executor:device_description", "//xla/tests:hlo_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -1466,10 +1463,10 @@ xla_cc_test( "//xla/service:pattern_matcher", "//xla/service:pattern_matcher_gmock", "//xla/tests:hlo_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -1515,7 +1512,7 @@ xla_test( ], ) + [ # b/317293391 - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -1660,9 +1657,9 @@ xla_cc_test( "//xla/stream_executor:stream", "//xla/tests:hlo_test_base", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -1725,9 +1722,9 @@ xla_test( "//xla/hlo/ir:hlo", "//xla/service/gpu/tests:gpu_codegen_test", "//xla/stream_executor:semantic_version", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -1914,8 +1911,8 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1938,8 +1935,8 @@ xla_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -1968,8 +1965,8 @@ xla_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -2228,8 +2225,8 @@ xla_cc_test( "//xla/service:layout_assignment", "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -2385,8 +2382,8 @@ xla_cc_test( "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings:string_view", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -2535,8 +2532,8 @@ xla_cc_test( ":reduction_degenerate_dim_remover", "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -2569,8 +2566,8 @@ xla_cc_test( ":reduction_dimension_grouper", "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -2612,7 +2609,6 @@ xla_test( "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -3075,7 +3071,6 @@ xla_test( "@com_google_absl//absl/strings", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -3149,11 +3144,11 @@ xla_cc_test( ":transpose_dimension_grouper", "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -3195,8 +3190,8 @@ xla_cc_test( "//xla/stream_executor:device_description", "//xla/tests:hlo_test_base", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/service/graphcycles/BUILD b/xla/service/graphcycles/BUILD index c965a37ab6855..94d1ec6ff6396 100644 --- a/xla/service/graphcycles/BUILD +++ b/xla/service/graphcycles/BUILD @@ -42,10 +42,10 @@ xla_cc_test( ":graphcycles", "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/random", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -54,7 +54,7 @@ xla_cc_test( srcs = ["ordered_set_test.cc"], deps = [ ":ordered_set", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/stream_executor/BUILD b/xla/stream_executor/BUILD index 7a87a9a835474..20ba49422c041 100644 --- a/xla/stream_executor/BUILD +++ b/xla/stream_executor/BUILD @@ -706,7 +706,6 @@ cc_test( "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -722,9 +721,9 @@ xla_cc_test( ":stream_executor_h", ":typed_kernel_factory", "//xla/stream_executor/host:host_platform", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -737,9 +736,9 @@ xla_cc_test( ":stream_executor_h", "//xla/stream_executor/host:host_platform", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -754,9 +753,9 @@ xla_cc_test( ":stream_executor_h", "//xla/stream_executor/host:host_platform", "@com_google_absl//absl/log:check", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -766,8 +765,8 @@ xla_cc_test( srcs = ["dnn_test.cc"], deps = [ ":dnn", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -780,8 +779,8 @@ xla_cc_test( ":module_spec", ":scoped_module_handle", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -792,8 +791,8 @@ xla_cc_test( ":device_memory", ":device_memory_handle", ":mock_stream_executor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -803,8 +802,8 @@ xla_cc_test( deps = [ ":device_description", ":semantic_version", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -816,9 +815,9 @@ xla_cc_test( ":mock_stream_executor", ":stream", "@com_google_absl//absl/log", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -832,9 +831,9 @@ xla_cc_test( ":stream_finder", "//xla:test", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -845,7 +844,6 @@ cc_test( ":bit_pattern", "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/stream_executor/cuda/BUILD b/xla/stream_executor/cuda/BUILD index 76d686a82e1a4..6ff1197a00feb 100644 --- a/xla/stream_executor/cuda/BUILD +++ b/xla/stream_executor/cuda/BUILD @@ -275,7 +275,6 @@ xla_test( "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -298,7 +297,6 @@ xla_test( "@tsl//tsl/platform:status", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -787,7 +785,6 @@ xla_test( "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -807,9 +804,9 @@ xla_cc_test( "@com_google_absl//absl/status", "@com_google_absl//absl/status:statusor", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -927,9 +924,9 @@ xla_cc_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -952,8 +949,8 @@ xla_cc_test( ":nvjitlink_known_issues", ":nvjitlink_support", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/stream_executor/gpu/BUILD b/xla/stream_executor/gpu/BUILD index 61d7d787321f0..9b900b163d6f2 100644 --- a/xla/stream_executor/gpu/BUILD +++ b/xla/stream_executor/gpu/BUILD @@ -83,8 +83,8 @@ xla_cc_test( srcs = ["context_map_test.cc"], deps = [ ":context_map", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -131,7 +131,6 @@ xla_cc_test( "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -459,9 +458,9 @@ xla_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/strings:string_view", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -507,9 +506,9 @@ xla_test( "//xla/tsl/framework:device_id", "@com_google_absl//absl/log:check", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -616,9 +615,9 @@ xla_test( backends = ["gpu"], deps = [ ":gpu_test_kernels_fatbin", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -645,9 +644,9 @@ xla_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -680,13 +679,13 @@ xla_test( "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:status", "@tsl//tsl/platform:status_matchers", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ] + if_cuda([ "//xla/stream_executor/cuda:cuda_platform", ]) + if_rocm([ @@ -705,9 +704,9 @@ xla_test( "//xla/stream_executor:stream", "//xla/stream_executor:stream_executor_h", "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_cuda([ "//xla/stream_executor/cuda:cuda_platform", ]) + if_rocm([ @@ -728,9 +727,9 @@ xla_test( "//xla/stream_executor:stream_finder", "//xla/stream_executor/host:host_platform", "@com_google_absl//absl/status:statusor", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_cuda([ "//xla/stream_executor/cuda:cuda_platform", ]) + if_rocm([ @@ -790,11 +789,11 @@ xla_test( "@com_google_absl//absl/log", "@com_google_absl//absl/strings", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:path", "@tsl//tsl/platform:protobuf", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/stream_executor/host/BUILD b/xla/stream_executor/host/BUILD index fc895e3b002a7..efde87d7d0541 100644 --- a/xla/stream_executor/host/BUILD +++ b/xla/stream_executor/host/BUILD @@ -154,12 +154,12 @@ xla_cc_test( "@com_google_absl//absl/status:statusor", "@com_google_absl//absl/strings", "@com_google_absl//absl/types:span", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:platform_port", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", "@tsl//tsl/platform:test_benchmark", - "@tsl//tsl/platform:test_main", ], ) @@ -210,10 +210,10 @@ xla_cc_test( "//xla/tsl/lib/core:status_test_util", "@com_google_absl//absl/status", "@com_google_absl//absl/synchronization", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:errors", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/stream_executor/integrations/BUILD b/xla/stream_executor/integrations/BUILD index 6d7b017aacca7..298c29ea30edf 100644 --- a/xla/stream_executor/integrations/BUILD +++ b/xla/stream_executor/integrations/BUILD @@ -102,10 +102,10 @@ xla_cc_test( "@com_google_absl//absl/log:check", "@com_google_absl//absl/status", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:status", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ] + if_google([ "//xla/tsl/framework:allocator", ]), diff --git a/xla/stream_executor/sycl/BUILD b/xla/stream_executor/sycl/BUILD index 08695853034e8..781a76a9e2fd7 100644 --- a/xla/stream_executor/sycl/BUILD +++ b/xla/stream_executor/sycl/BUILD @@ -57,7 +57,7 @@ xla_cc_test( "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", ]) + [ - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) diff --git a/xla/tests/BUILD b/xla/tests/BUILD index 25045815a5a03..b65683bb3ca49 100644 --- a/xla/tests/BUILD +++ b/xla/tests/BUILD @@ -34,7 +34,7 @@ xla_tests_package_groups() generate_backend_suites() # Target to add main for tests. Do not link this target and -# //third_party/tensorflow/tsl/platform:test_main into the same target. +# @com_google_googletest//:gtest_main into the same target. cc_library( name = "xla_internal_test_main", testonly = True, @@ -2828,11 +2828,11 @@ xla_test( "//xla/stream_executor:device_description", "//xla/stream_executor:stream_executor_h", "@com_google_absl//absl/status", + "@com_google_googletest//:gtest_main", "@llvm-project//llvm:Core", "@tsl//tsl/platform:casts", "@tsl//tsl/platform:env", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -2931,9 +2931,9 @@ xla_cc_test( deps = [ "//xla:executable_run_options", "@com_google_absl//absl/base:dynamic_annotations", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -3025,7 +3025,7 @@ xla_cc_test( "//xla/hlo/builder:xla_builder", "//xla/service:cpu_plugin", "//xla/service:local_service", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], ) @@ -3092,11 +3092,11 @@ xla_cc_test( "//xla:literal", "//xla:test_helpers", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:logging", "@tsl//tsl/platform:path", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) diff --git a/xla/tests/build_defs.bzl b/xla/tests/build_defs.bzl index 1da9050a079af..05504df531df6 100644 --- a/xla/tests/build_defs.bzl +++ b/xla/tests/build_defs.bzl @@ -364,7 +364,7 @@ def xla_test( if test_names: native.test_suite(name = name, tags = tags + ["manual"], tests = test_names) else: - native.cc_test(name = name, deps = ["@tsl//tsl/platform:test_main"]) + native.cc_test(name = name, deps = ["@com_google_googletest//:gtest_main"]) def xla_test_library( name, diff --git a/xla/tests/fuzz/build_defs.bzl b/xla/tests/fuzz/build_defs.bzl index f690523035647..698e501ecf9f6 100644 --- a/xla/tests/fuzz/build_defs.bzl +++ b/xla/tests/fuzz/build_defs.bzl @@ -42,7 +42,7 @@ def hlo_test(name, hlo, **kwargs): real_hardware_only = True, deps = [ "//xla/tests/fuzz:hlo_test_lib", - "@tsl//tsl/platform:test_main", + "@com_google_googletest//:gtest_main", ], **kwargs ) diff --git a/xla/tools/BUILD b/xla/tools/BUILD index 7ebb2f4130677..ed52155e99a59 100644 --- a/xla/tools/BUILD +++ b/xla/tools/BUILD @@ -256,9 +256,9 @@ xla_cc_test( deps = [ "//xla/tsl/platform:subprocess", "@com_google_googletest//:gtest", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:path", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -339,10 +339,10 @@ xla_cc_test( deps = [ "//xla/tsl/platform:subprocess", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform", "@tsl//tsl/platform:path", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -411,8 +411,8 @@ xla_cc_test( "//xla:test", "//xla/hlo/ir:hlo", "//xla/tests:hlo_test_base", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:statusor", - "@tsl//tsl/platform:test_main", ], ) @@ -494,9 +494,9 @@ xla_cc_test( "//xla:literal_util", "//xla:xla_data_proto_cc", "//xla/tsl/lib/core:status_test_util", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], ) @@ -553,11 +553,11 @@ xla_cc_test( "//xla/tsl/platform:subprocess", "@com_google_absl//absl/log", "@com_google_absl//absl/strings", + "@com_google_googletest//:gtest_main", "@tsl//tsl/platform:env", "@tsl//tsl/platform:path", "@tsl//tsl/platform:statusor", "@tsl//tsl/platform:test", - "@tsl//tsl/platform:test_main", ], )