diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/support_wrap_ops.yaml b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/support_wrap_ops.yaml index 92096fc4bb336928b2ddf9c3e8eba33dca71a12c..2dbf5b7b2fa94f379e93505705fe166700cc2a70 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/support_wrap_ops.yaml +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/hook_module/support_wrap_ops.yaml @@ -1131,6 +1131,7 @@ torch_npu: - npu_prompt_flash_attention - npu_lstm - npu_apply_adam + - npu_apply_adam_w aten: - signbit