diff options
author | 2017-06-29 14:03:26 -0700 | |
---|---|---|
committer | 2017-06-29 14:07:08 -0700 | |
commit | 62c1104e766c15c71f1fa53145503899eb470d62 (patch) | |
tree | ea01dc781dde14f3d2feb8a4b5b510a0b31cb1e5 /tensorflow/contrib/tfprof | |
parent | 8be70f540bf57688c31eb807ff0d50e2186eb02f (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/BUILD | 3 | ||||
-rw-r--r-- | tensorflow/contrib/tfprof/python/tools/tfprof/internal/run_metadata_test.py | 6 |
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() |