Browse Source

!1274 StepTrace of UI shows not support hetergeneous training scene

From: @gzhcv
Reviewed-by: @ouwenchang,@lilongfei15
Signed-off-by: @lilongfei15
pull/1274/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
40680c7583
2 changed files with 4 additions and 7 deletions
  1. +2
    -1
      mindinsight/backend/profiler/profile_api.py
  2. +2
    -6
      mindinsight/profiler/analyser/cpu_analyser.py

+ 2
- 1
mindinsight/backend/profiler/profile_api.py View File

@@ -156,11 +156,12 @@ def get_training_trace_graph():
}})
graph_info['summary'] = analyser.summary
graph_info['point_info'] = analyser.point_info
graph_info['is_heterogeneous'] = False

# In heterogeneous training scene, do not display step trace data.
cpu_op_type_file_name = f"cpu_op_type_info_{device_id}.csv"
if cpu_op_type_file_name in os.listdir(profiler_dir_abs):
graph_info = {}
graph_info = {'is_heterogeneous': True}

return jsonify(graph_info)



+ 2
- 6
mindinsight/profiler/analyser/cpu_analyser.py View File

@@ -35,10 +35,8 @@ class CpuOpTypeAnalyser(GpuAnalyser):
Returns:
list, the converted data.
"""
factor_us_to_ms = 1e-3
try:
return [row[0], int(row[1]), int(row[2]), float(row[3]) * factor_us_to_ms,
float(row[4]) * factor_us_to_ms, float(row[5])*100]
return [row[0], int(row[1]), int(row[2]), float(row[3]), float(row[4]), float(row[5])*100]
except IndexError as err:
log.exception(err)
raise ProfilerRawFileException('failed to get HOST CPU operator type data.')
@@ -60,10 +58,8 @@ class CpuOpInfoAnalyser(GpuAnalyser):
Returns:
list, the converted data.
"""
factor_us_to_ms = 1e-3
try:
return [row[0], row[1], row[2], row[3], int(row[4]), float(row[5]) * factor_us_to_ms,
float(row[6]) * factor_us_to_ms, float(row[7]), row[8]]
return [row[0], row[1], row[2], row[3], int(row[4]), float(row[5]), float(row[6]), float(row[7]), row[8]]
except IndexError as err:
log.exception(err)
raise ProfilerRawFileException('failed to get HOST CPU operator detail data.')

Loading…
Cancel
Save