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 9bb4c3c8aafeb2984f8d4acfbdafd4b2fb929489..cda89fdd8d094233835a85195b1ea827d2219d5f 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]: