diff --git a/mindinsight/backend/profiler/profile_api.py b/mindinsight/backend/profiler/profile_api.py index a20e351e254cd5610fe6ecfca60ff09a6787b80e..1fd0849672cd601773f5477ce7b20dd54155f1b3 100644 --- a/mindinsight/backend/profiler/profile_api.py +++ b/mindinsight/backend/profiler/profile_api.py @@ -134,7 +134,7 @@ def get_training_trace_graph(): 'mode': 'step', 'step_id': graph_type }}) - + graph_info['summary'] = analyser.summary return jsonify(graph_info) diff --git a/mindinsight/profiler/analyser/step_trace_analyser.py b/mindinsight/profiler/analyser/step_trace_analyser.py index 865c36e1063f22687d0506277f64f41257a793a3..e611e3697c56a55613501948abf809bc605b019c 100644 --- a/mindinsight/profiler/analyser/step_trace_analyser.py +++ b/mindinsight/profiler/analyser/step_trace_analyser.py @@ -192,10 +192,11 @@ class StepTraceAnalyser(BaseAnalyser): fp_point = row_info_dict.get('fp_point', 0) bp_point = row_info_dict.get('bp_point', 0) points = [ - self._construct_time_point('', 0, row_info_dict.get('iteration_interval', 0)), - self._construct_time_point('fp_and_bp', fp_point - start_point, - row_info_dict.get('fp_and_bp', 0)), - self._construct_time_point('', bp_point - start_point, row_info_dict.get('tail', 0)), + self._construct_time_point( + 'iteration_interval', 0, row_info_dict.get('iteration_interval', 0)), + self._construct_time_point( + 'fp_and_bp', fp_point - start_point, row_info_dict.get('fp_and_bp', 0)), + self._construct_time_point('tail', bp_point - start_point, row_info_dict.get('tail', 0)) ] return points diff --git a/tests/st/func/profiler/test_analyse.py b/tests/st/func/profiler/test_analyse.py index 6e91b45ac1651e3e83c11efa61c9a73e8837f075..96867842b2eafa6ee58d625de5f03a600e55c4da 100644 --- a/tests/st/func/profiler/test_analyse.py +++ b/tests/st/func/profiler/test_analyse.py @@ -98,7 +98,6 @@ class TestProfilerAnalyse(TestCase): {'name': 'stream_540_parallel_0', 'start': 118.3705, 'duration': 49.281}, {'name': '', 'start': 167.6515, 'duration': 37.7294}] - @pytest.mark.level0 @pytest.mark.env_single @pytest.mark.platform_x86_cpu