diff --git a/mindinsight/backend/profiler/profile_api.py b/mindinsight/backend/profiler/profile_api.py index e0ffdbd5..5253d84b 100644 --- a/mindinsight/backend/profiler/profile_api.py +++ b/mindinsight/backend/profiler/profile_api.py @@ -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) diff --git a/mindinsight/profiler/analyser/cpu_analyser.py b/mindinsight/profiler/analyser/cpu_analyser.py index f582565a..0793b06c 100644 --- a/mindinsight/profiler/analyser/cpu_analyser.py +++ b/mindinsight/profiler/analyser/cpu_analyser.py @@ -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.')