Browse Source

gpu_timeline(python code)

update mindspore/profiler/parser/integrator.py.

update mindspore/profiler/profiling.py.
tags/v1.0.0
gzhcv 5 years ago
parent
commit
693c6da09c
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      mindspore/profiler/parser/integrator.py

+ 2
- 2
mindspore/profiler/parser/integrator.py 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."""


Loading…
Cancel
Save