OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Fix flakyness, sometimes the op takes ms to run.

Browse Source 
PiperOrigin-RevId: 165728705
This commit is contained in:
A. Unique TensorFlower
2017-08-18 11:18:09 -07:00
committed by TensorFlower Gardener
parent 360bff8ae5
commit e6c60fb368
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/python/profiler/model_analyzer_test.py
View File

@@ -95,7 +95,7 @@ class PrintModelAnalysisTest(test.TestCase):
# Make sure time is profiled.
gap = 1 if test.is_gpu_available() else 2
for i in range(3, 6, gap):
mat = re.search('(.*)us/(.*)us', metrics[i])
mat = re.search('(.*)[um]s/(.*)[um]s', metrics[i])
self.assertGreater(float(mat.group(1)), 0.0)
self.assertGreater(float(mat.group(2)), 0.0)
# Make sure device is profiled.