From 0622b7cf715df3746b6ad34a84927c3328e76fb1 Mon Sep 17 00:00:00 2001 From: Linwei-Ying Date: Wed, 26 Mar 2025 10:19:25 +0800 Subject: [PATCH] api_mapping yaml outputs bugfix --- debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py b/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py index 7ef577c8ec..c0ef3bc66c 100644 --- a/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py +++ b/debug/accuracy_tools/msprobe/mindspore/compare/ms_compare.py @@ -315,7 +315,7 @@ class MSComparator(Comparator): for mapping_dict in self.api_mapping_dict: keys_to_compare = [ ('ms_args', 'pt_args'), - ('ms_output', 'pt_output'), + ('ms_outputs', 'pt_outputs'), ('ms_parameters', 'pt_parameters'), ('ms_parameters_grad', 'pt_parameters_grad'), ] -- Gitee