Skip to content

Commit 2cb20f3

Browse files
apivovarovtensorflower-gardener
authored andcommitted
PR tensorflow#14477: Allow to build multihost_hlo_runner for CPU backend
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 <[email protected]>: Allow to build multihost_hlo_runner for CPU backend Merging this change closes tensorflow#14477 PiperOrigin-RevId: 650160788
1 parent 104e27b commit 2cb20f3

File tree

1 file changed

+2
-2
lines changed
  • third_party/xla/xla/tools/multihost_hlo_runner

1 file changed

+2
-2
lines changed

third_party/xla/xla/tools/multihost_hlo_runner/BUILD

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -31,7 +31,7 @@ cc_library(
3131
compatible_with = None,
3232
tags = [
3333
"nomac",
34-
] + tf_gpu_tests_tags(),
34+
],
3535
deps = [
3636
":create_client",
3737
":functional_hlo_runner",
@@ -73,7 +73,7 @@ xla_cc_binary(
7373
testonly = True,
7474
tags = [
7575
"nomac",
76-
] + tf_gpu_tests_tags(),
76+
],
7777
deps = [
7878
":hlo_runner_main_lib",
7979
],

0 commit comments

Comments
 (0)