[go: nahoru, domu]
Skip to content

Commit

Permalink
Restore non-CUDA dependencies for :autotuner_util_test
Browse files Browse the repository at this point in the history 
The dependency `:xla_internal_test_main` was recently
accidentally moved into the `if_cuda_is_configured` macro
which makes the test not compiled anymore in a non-CUDA
build config.

This change restores the dependency and also sorts
the CUDA specific dependencies.

PiperOrigin-RevId: 646480891
  • Loading branch information
@beckerhe @tensorflower-gardener
beckerhe authored and tensorflower-gardener committed Jun 25, 2024
1 parent f14adf7 commit 9873a6a
Showing 1 changed file with 13 additions and 11 deletions.
24 changes: 13 additions & 11 deletions third_party/xla/xla/service/gpu/BUILD
View file
Original file line number Diff line number Diff line change
Expand Up @@ -5840,35 +5840,37 @@ xla_cc_test(
name = "autotuner_util_test",
srcs = if_cuda_is_configured(["autotuner_util_test.cc"]),
deps = if_cuda_is_configured([
# keep sorted
":autotuner_util",
"@com_google_googletest//:gtest",
"@com_google_absl//absl/base:log_severity",
"@com_google_absl//absl/container:flat_hash_set",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/log:scoped_mock_log",
"@com_google_absl//absl/status",
"@com_google_absl//absl/strings",
"//xla:autotune_results_proto_cc",
"//xla:autotuning_proto_cc",
"//xla:xla_proto_cc",
"//xla/hlo/ir:hlo",
"//xla/hlo/utils:hlo_query",
"//xla/stream_executor:platform_manager",
"//xla/stream_executor:platform",
"//xla/stream_executor:platform_manager",
"//xla/stream_executor/host:host_platform",
"//xla/tests:hlo_test_base",
"//xla/tests:verified_hlo_module",
"//xla/tests:xla_internal_test_main", # Keep outside GPU guard
"@com_google_absl//absl/base:log_severity",
"@com_google_absl//absl/container:flat_hash_set",
"@com_google_absl//absl/log:check",
"@com_google_absl//absl/log:scoped_mock_log",
"@com_google_absl//absl/status",
"@com_google_absl//absl/strings",
"@com_google_googletest//:gtest",
"@local_tsl//tsl/lib/core:status_test_util",
"@local_tsl//tsl/platform:env",
"@local_tsl//tsl/platform:errors",
"@local_tsl//tsl/platform:logging",
"@local_tsl//tsl/platform:path",
"@local_tsl//tsl/platform:protobuf",
"@local_tsl//tsl/platform:status_matchers",
"@local_tsl//tsl/platform:status",
"@local_tsl//tsl/platform:status_matchers",
"@local_tsl//tsl/platform:statusor",
]),
]) + [
"//xla/tests:xla_internal_test_main", # Keep outside GPU guard
],
)

cc_library(
Expand Down

0 comments on commit 9873a6a

Please sign in to comment.