From 98d2e02db91ffe6276657e52ed20aace95f6842b Mon Sep 17 00:00:00 2001 From: s30048155 Date: Sun, 14 Jan 2024 19:18:02 -1200 Subject: [PATCH] bugfix --- debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) 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 cf9bfe29b..c784cab47 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 @@ -236,7 +236,8 @@ def run_backward(api_full_name, args, backward_content, grad_index, device_args, if grad_index is not None: out[grad_index].backward(cpu_grad) elif isinstance(out, (list, tuple)): - raise NotImplementedError("Multiple backward is not supported.") + print_warn_log("backward result is list or tuple, skip backward.") + grad_out = None else: out.backward(cpu_grad) args_grad = [] -- Gitee