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 cf9bfe29bdfd054db96ae878813c099892430bbb..c784cab471394cf78a89cb6477453ac2d8007e54 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 = []