gpu_timeline(python code)

update mindspore/profiler/parser/integrator.py.

update mindspore/profiler/profiling.py.
This commit is contained in:
gzhcv 2020-09-09 21:56:28 +08:00
parent 6240189190
commit 693c6da09c
1 changed files with 2 additions and 2 deletions

View File

@ -490,8 +490,6 @@ class BaseTimelineGenerator:
__col_names__ = ['op_name', 'stream_id', 'start_time', 'duration']
_output_timeline_data_file_path = 'output_timeline_data_{}.txt'
_min_cycle_counter_file_path = 'min_cycle_counter_{}.txt'
_display_filename = 'timeline_display_{}.json'
_timeline_summary_filename = 'timeline_summary_{}.json'
_timeline_meta = []
_timeline_summary = {
'total_time': 0,
@ -770,6 +768,8 @@ class GpuTimelineGenerator(BaseTimelineGenerator):
class AscendTimelineGenerator(BaseTimelineGenerator):
"""Generate ascend Timeline data from file."""
_display_filename = 'ascend_timeline_display_{}.json'
_timeline_summary_filename = 'ascend_timeline_summary_{}.json'
def _load_timeline_data(self):
"""Load timeline data from file."""