diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2018-09-29 00:59:02 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-09-29 01:03:16 -0700 |
commit | d78595d333c9b5c8a0705ba6852c08b107d6c462 (patch) | |
tree | 61527521ca417bedd24243d019f5f5a0711c4143 /tensorflow/tools | |
parent | 2e0e934e0b3c00863918c78bf55524eea3f0c0dc (diff) |
Make cuda_py_test create a gpu and cpu target.
Currently, we run tests on machines with GPUs based on the "gpu" tag, and the
tests automatically adapt to whether a GPU is available. Creating two targets,
one tagged with "gpu" and one not, will make us run the tests in both modes.
PiperOrigin-RevId: 215045035
Diffstat (limited to 'tensorflow/tools')
-rw-r--r-- | tensorflow/tools/pip_package/pip_smoke_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/tools/pip_package/pip_smoke_test.py b/tensorflow/tools/pip_package/pip_smoke_test.py index c6ef82ccdc..e7f9628fa6 100644 --- a/tensorflow/tools/pip_package/pip_smoke_test.py +++ b/tensorflow/tools/pip_package/pip_smoke_test.py @@ -142,7 +142,7 @@ def main(): missing_dependencies = [] # File extensions and endings to ignore - ignore_extensions = ["_test", "_test.py"] + ignore_extensions = ["_test", "_test.py", "_test_gpu", "_test_gpu.py"] ignored_files = 0 blacklisted_files = len(BLACKLIST) |