diff options
author | 2018-09-19 03:14:28 -0700 | |
---|---|---|
committer | 2018-09-19 03:18:32 -0700 | |
commit | 227b819d55c3b24103026cdaf1897892422c5cd3 (patch) | |
tree | 99122ed9aaf747446410c5fe6884386607882643 /tensorflow/compiler/tests | |
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/tests')
-rw-r--r-- | tensorflow/compiler/tests/build_defs.bzl | 6 |
1 files changed, 5 insertions, 1 deletions
diff --git a/tensorflow/compiler/tests/build_defs.bzl b/tensorflow/compiler/tests/build_defs.bzl index 114793352e..b8583c9bdb 100644 --- a/tensorflow/compiler/tests/build_defs.bzl +++ b/tensorflow/compiler/tests/build_defs.bzl @@ -2,6 +2,10 @@ load("@local_config_cuda//cuda:build_defs.bzl", "cuda_is_configured") load("//tensorflow/compiler/tests:plugin.bzl", "plugins") +load( + "//tensorflow/core:platform/default/build_config_root.bzl", + "tf_cuda_tests_tags", +) def all_backends(): b = ["cpu"] + plugins.keys() @@ -65,7 +69,7 @@ def tf_xla_py_test( "--test_device=XLA_GPU", "--types=DT_HALF,DT_FLOAT,DT_DOUBLE,DT_UINT8,DT_INT8,DT_INT32,DT_INT64,DT_BOOL,DT_COMPLEX64,DT_BFLOAT16", ] - backend_tags += ["requires-gpu-sm35"] + backend_tags += tf_cuda_tests_tags() elif backend in plugins: backend_args += [ "--test_device=" + plugins[backend]["device"], |