Skip to content

Commit

Permalink
PR tensorflow#14477: Allow to build multihost_hlo_runner for CPU backend
Browse files Browse the repository at this point in the history
Imported from GitHub PR openxla/xla#14477

Currently, Bazel skips building `multihost_hlo_runner:hlo_runner_main` on CPU because the target is tagged with `tf_gpu_tests_tags`.

This tag should only be applied to the dedicated `hlo_runner_main_gpu` target.

The regular `hlo_runner_main` works fine on CPU (when `--xla_force_host_platform_device_count=N` is used)

@bchetioui @ddunl @beckerhe Can you have a look?
Copybara import of the project:

--
a0d2129bfe74f2f9133f9e1c6b04e06cea6d0e5d by Alexander Pivovarov <pivovaa@amazon.com>:

Allow to build multihost_hlo_runner for CPU backend

Merging this change closes tensorflow#14477

PiperOrigin-RevId: 650160788
  • Loading branch information
apivovarov authored and tensorflower-gardener committed Jul 8, 2024
1 parent 104e27b commit 2cb20f3
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions third_party/xla/xla/tools/multihost_hlo_runner/BUILD
Original file line number Diff line number Diff line change
Expand Up @@ -31,7 +31,7 @@ cc_library(
compatible_with = None,
tags = [
"nomac",
] + tf_gpu_tests_tags(),
],
deps = [
":create_client",
":functional_hlo_runner",
Expand Down Expand Up @@ -73,7 +73,7 @@ xla_cc_binary(
testonly = True,
tags = [
"nomac",
] + tf_gpu_tests_tags(),
],
deps = [
":hlo_runner_main_lib",
],
Expand Down

0 comments on commit 2cb20f3

Please sign in to comment.