aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/tfprof
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2017-06-29 14:03:26 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2017-06-29 14:07:08 -0700
commit62c1104e766c15c71f1fa53145503899eb470d62 (patch)
treeea01dc781dde14f3d2feb8a4b5b510a0b31cb1e5 /tensorflow/contrib/tfprof
parent8be70f540bf57688c31eb807ff0d50e2186eb02f (diff)
Renable a subset of test cases in run_metadata_test.py
The flaky test case is due to RunMetadata randomly missing gpu kernel execution time, the reason of which is still being investigated. Enable the test to avoid other behaviors got broken as well. PiperOrigin-RevId: 160570056
Diffstat (limited to 'tensorflow/contrib/tfprof')
-rw-r--r--tensorflow/contrib/tfprof/python/tools/tfprof/internal/BUILD3
-rw-r--r--tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py6
2 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/internal/BUILD b/tensorflow/contrib/tfprof/python/tools/tfprof/internal/BUILD
index ca60e504ed..a498574a95 100644
--- a/tensorflow/contrib/tfprof/python/tools/tfprof/internal/BUILD
+++ b/tensorflow/contrib/tfprof/python/tools/tfprof/internal/BUILD
@@ -60,10 +60,7 @@ cuda_py_test(
"//tensorflow/tools/tfprof:protos_all_py",
],
tags = [
- "manual",
"no_pip",
- "noguitar",
- "notap",
],
)
diff --git a/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py b/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py
index a472ade362..8f3351999f 100644
--- a/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py
+++ b/tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py
@@ -146,6 +146,10 @@ class RunMetadataTest(test.TestCase):
self.assertEqual(mm_node.cpu_exec_micros, total_cpu_execs)
self.assertEqual(mm_node.exec_micros, total_cpu_execs)
+ # pylint: disable=pointless-string-statement
+ """
+ TODO(xpan): This test is flaky because RunMetadata returned from TensorFlow
+ is random. Still being investigated.
def testLoopGPU(self):
if not test.is_gpu_available():
return
@@ -180,6 +184,8 @@ class RunMetadataTest(test.TestCase):
self.assertEqual(mm_node.cpu_exec_micros, total_cpu_execs)
self.assertEqual(mm_node.exec_micros,
total_cpu_execs + total_accelerator_execs)
+ """
+
if __name__ == '__main__':
test.main()