experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

TensorTracer: Add an explicit error message when trace_dir is not provided.

Browse Source 
PiperOrigin-RevId: 351278777
Change-Id: I78e922365f0c19a030a6d8ecbeb43e69107ea37d
This commit is contained in:
Mehmet Deveci 2021-01-11 18:31:21 -08:00 committed by TensorFlower Gardener
parent d68ead1718
commit c0d110128e
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/python/tpu/tensor_tracer_flags.py
View File

@ -141,6 +141,13 @@ class TTParameters(object):
self.collect_summary_per_core = self.is_flag_on(FLAG_NAME_SUMMARY_PER_CORE)
self.flush_summaries_with_outside_compile = self.is_flag_on(
FLAG_FLUSH_SUMMARY)
self._check_flag_errors()
def _check_flag_errors(self):
if self.trace_mode in (TRACE_MODE_SUMMARY, TRACE_MODE_FULL_TENSOR_SUMMARY):
if not self.trace_dir:
raise ValueError('trace_dir must be explicitly provided in '
'TENSOR_TRACER_FLAGS when summary mode is used.')
def _get_report_filepath(self):
"""Sets the path of the output report file."""