diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2017-08-18 11:18:09 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-08-18 11:22:08 -0700 |
| commit | e6c60fb368aa1261663cef1204e515179c36db8d (patch) | |
| tree | d879d83fc9fa726d7c924a8cac6d3db7e1ef9401 /tensorflow/python/profiler | |
| parent | 360bff8ae51f2341e3586e247195cc80a4a3933f (diff) | |
Fix flakyness, sometimes the op takes ms to run.
PiperOrigin-RevId: 165728705
Diffstat (limited to 'tensorflow/python/profiler')
| -rw-r--r-- | tensorflow/python/profiler/model_analyzer_test.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/python/profiler/model_analyzer_test.py b/tensorflow/python/profiler/model_analyzer_test.py index 7c0080c206..9492cadb8b 100644 --- a/tensorflow/python/profiler/model_analyzer_test.py +++ b/tensorflow/python/profiler/model_analyzer_test.py @@ -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. |