From ee2bd1ce8b7cfb21a6122a76a956f956ecf11e0f Mon Sep 17 00:00:00 2001 From: l30044004 Date: Sat, 9 Sep 2023 17:48:32 +0800 Subject: [PATCH] =?UTF-8?q?csv=E6=96=87=E4=BB=B6api=E5=91=BD=E5=90=8D?= =?UTF-8?q?=E5=92=8Cdump=E6=95=B0=E6=8D=AE=E5=91=BD=E5=90=8D=E4=B8=8D?= =?UTF-8?q?=E4=B8=80=E8=87=B4=E9=97=AE=E9=A2=98?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../api_accuracy_checker/common/base_api.py | 2 ++ .../api_accuracy_checker/run_ut/run_ut.py | 13 +++++-------- 2 files changed, 7 insertions(+), 8 deletions(-) diff --git a/debug/accuracy_tools/api_accuracy_checker/common/base_api.py b/debug/accuracy_tools/api_accuracy_checker/common/base_api.py index 0db84b45bf..627bd76976 100644 --- a/debug/accuracy_tools/api_accuracy_checker/common/base_api.py +++ b/debug/accuracy_tools/api_accuracy_checker/common/base_api.py @@ -71,6 +71,8 @@ class BaseAPIInfo: def analyze_builtin(self, arg): single_arg = {} + if self.is_save_data: + self.args_num += 1 if isinstance(arg, slice): single_arg.update({'type' : "slice"}) single_arg.update({'value' : [arg.start, arg.stop, arg.step]}) diff --git a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py index 707d6cbed9..b70720f589 100644 --- a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py +++ b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py @@ -123,14 +123,11 @@ def do_save_error_data(api_full_name, data_info, is_fwd_success, is_bwd_success) api_full_name = api_full_name.replace("*", ".") for element in data_info.in_fwd_data_list: UtAPIInfo(api_full_name + '.forward.input', element) - if data_info.bench_out is not None: - UtAPIInfo(api_full_name + '.forward.output.bench', data_info.bench_out) - UtAPIInfo(api_full_name + '.forward.output.npu', data_info.npu_out) - if data_info.grad_in is not None: - UtAPIInfo(api_full_name + '.backward.input', data_info.grad_in) - if data_info.bench_grad_out is not None: - UtAPIInfo(api_full_name + '.backward.output.bench', data_info.bench_grad_out) - UtAPIInfo(api_full_name + '.backward.output.npu', data_info.npu_grad_out) + UtAPIInfo(api_full_name + '.forward.output.bench', data_info.bench_out) + UtAPIInfo(api_full_name + '.forward.output.npu', data_info.npu_out) + UtAPIInfo(api_full_name + '.backward.input', data_info.grad_in) + UtAPIInfo(api_full_name + '.backward.output.bench', data_info.bench_grad_out) + UtAPIInfo(api_full_name + '.backward.output.npu', data_info.npu_grad_out) -- Gitee