aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/compiler/xrt
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-09-19 03:14:28 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-19 03:18:32 -0700
commit227b819d55c3b24103026cdaf1897892422c5cd3 (patch)
tree99122ed9aaf747446410c5fe6884386607882643 /tensorflow/compiler/xrt
parent0b274dcbe12a9f0ddd96f94cc80c2e66c195fd76 (diff)
Run CPU tests remotely.
Being able to run CPU tests remotely while running GPU tests locally required multiple changes: 1. Unify how we tag GPU tests in TF; we now always use tf_cuda_tests_tags(). 2. Tag tests using tf_cuda_tests_tags() with 'local' and 'gpu'; this makes them not run on non-gpu builds and always runs them locally. PiperOrigin-RevId: 213601626
Diffstat (limited to 'tensorflow/compiler/xrt')
-rw-r--r--tensorflow/compiler/xrt/tests/BUILD6
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/compiler/xrt/tests/BUILD b/tensorflow/compiler/xrt/tests/BUILD
index 09ab4ed95f..b6dcfc4eb9 100644
--- a/tensorflow/compiler/xrt/tests/BUILD
+++ b/tensorflow/compiler/xrt/tests/BUILD
@@ -8,6 +8,10 @@ package(
)
load("//tensorflow:tensorflow.bzl", "tf_cuda_cc_test", "tf_cc_test")
+load(
+ "//tensorflow/core:platform/default/build_config_root.bzl",
+ "tf_cuda_tests_tags",
+)
cc_library(
name = "raw_api_test_lib",
@@ -57,7 +61,7 @@ tf_cuda_cc_test(
size = "medium",
srcs = [],
args = ["--xla_test_device=XLA_GPU"],
- tags = ["requires-gpu-sm35"],
+ tags = tf_cuda_tests_tags(),
deps = [
":raw_api_test_lib",
"//tensorflow/compiler/jit:xla_gpu_device",