aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/python/data
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-09-29 00:59:02 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-09-29 01:03:16 -0700
commitd78595d333c9b5c8a0705ba6852c08b107d6c462 (patch)
tree61527521ca417bedd24243d019f5f5a0711c4143 /tensorflow/python/data
parent2e0e934e0b3c00863918c78bf55524eea3f0c0dc (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/python/data')
-rw-r--r--tensorflow/python/data/kernel_tests/BUILD1
1 files changed, 1 insertions, 0 deletions
diff --git a/tensorflow/python/data/kernel_tests/BUILD b/tensorflow/python/data/kernel_tests/BUILD
index cadfe7f9e0..99d7f70513 100644
--- a/tensorflow/python/data/kernel_tests/BUILD
+++ b/tensorflow/python/data/kernel_tests/BUILD
@@ -318,6 +318,7 @@ cuda_py_test(
"//tensorflow/python:framework_test_lib",
],
tags = [
+ "no_oss", # TODO(b/116813115): Investigate timeout and re-enable.
"no_windows_gpu",
],
)