forked from mindspore-Ecosystem/mindspore
gpu_timeline(python code)
update mindspore/profiler/parser/integrator.py. update mindspore/profiler/profiling.py.
This commit is contained in:
parent
6240189190
commit
693c6da09c
|
@ -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…
Reference in New Issue