|
|
|
@@ -205,6 +205,9 @@ class Profiler: |
|
|
|
except (ProfilerIOException, ProfilerFileNotFoundException, RuntimeError) as err: |
|
|
|
logger.warning('Fail to write timeline data: %s', err) |
|
|
|
|
|
|
|
os.environ['PROFILING_MODE'] = str("false") |
|
|
|
context.set_context(enable_profiling=False) |
|
|
|
|
|
|
|
def _analyse_step_trace(self, source_path, framework_parser): |
|
|
|
""" |
|
|
|
Analyse step trace data and save the result. |
|
|
|
@@ -275,11 +278,6 @@ class Profiler: |
|
|
|
timeline_analyser.write_timeline() |
|
|
|
timeline_analyser.write_timeline_summary() |
|
|
|
|
|
|
|
def __del__(self): |
|
|
|
"""Disable the profiling collection service, called after training.""" |
|
|
|
os.environ['PROFILING_MODE'] = str("false") |
|
|
|
context.set_context(enable_profiling=False) |
|
|
|
|
|
|
|
def _get_profiling_job_id(self): |
|
|
|
"""Get profiling job id, which was generated by ada service. |
|
|
|
|
|
|
|
|