From fae0b637ded099af7f090c08f24866498052d772 Mon Sep 17 00:00:00 2001 From: lcw Date: Thu, 19 Jun 2025 19:30:35 +0800 Subject: [PATCH] =?UTF-8?q?=E3=80=90bugfix=E3=80=91mrtest?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../accuracy_tools/msprobe/pytorch/hook_module/api_register.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/debug/accuracy_tools/msprobe/pytorch/hook_module/api_register.py b/debug/accuracy_tools/msprobe/pytorch/hook_module/api_register.py index 9bb4c3c8aa..cda89fdd8d 100644 --- a/debug/accuracy_tools/msprobe/pytorch/hook_module/api_register.py +++ b/debug/accuracy_tools/msprobe/pytorch/hook_module/api_register.py @@ -107,7 +107,7 @@ def dist_module_forward(module, *args, **kwargs): def npu_module_forward(module, *args, **kwargs): if not module.need_hook: if module.api_name not in npu_custom_functions: - raise Exception(f'There is not bench function {module.api_name}') + raise Exception(f'There is not bench function test {module.api_name}') if module.device == Const.CUDA_LOWERCASE: module.api_name = _cuda_func_mapping.get(module.api_name, module.api_name) if module.device in [Const.CUDA_LOWERCASE, Const.CPU_LOWERCASE]: -- Gitee