diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-19 03:14:28 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-19 03:18:32 -0700 |
commit | 227b819d55c3b24103026cdaf1897892422c5cd3 (patch) | |
tree | 99122ed9aaf747446410c5fe6884386607882643 /tensorflow/compiler/xrt | |
parent | 0b274dcbe12a9f0ddd96f94cc80c2e66c195fd76 (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/BUILD | 6 |
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", |