From d75c70bc2d6b9f7ae6d6b58f65cfe1b7aa21e84f Mon Sep 17 00:00:00 2001 From: Guangda Lai Date: Mon, 14 May 2018 14:15:14 -0700 Subject: [PATCH] Reenable virtual gpu test, and decrease the number of testing rounds. PiperOrigin-RevId: 196565153 --- tensorflow/python/BUILD | 1 - tensorflow/python/client/virtual_gpu_test.py | 2 +- 2 files changed, 1 insertion(+), 2 deletions(-) diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index ea11b70..d804578 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3969,7 +3969,6 @@ cuda_py_test( ":math_ops", "//tensorflow/core:protos_all_py", ], - tags = ["noguitar"], ) py_test( diff --git a/tensorflow/python/client/virtual_gpu_test.py b/tensorflow/python/client/virtual_gpu_test.py index addf634..ae653e0 100644 --- a/tensorflow/python/client/virtual_gpu_test.py +++ b/tensorflow/python/client/virtual_gpu_test.py @@ -236,7 +236,7 @@ class VirtualGpuTest(test_util.TensorFlowTestCase): with self.test_session(config=self._util.config) as sess: if not test.is_gpu_available(cuda_only=True): self.skipTest('No GPU available') - for _ in range(10): + for _ in range(5): if not self._util.TestRandomGraph(sess): return -- 2.7.4