From 24a84e5eadfa7caa09b7eef9e3b3a989e896f8ff Mon Sep 17 00:00:00 2001 From: yang-minghai22 Date: Sat, 11 May 2024 14:51:56 +0800 Subject: [PATCH] support npu_apply_adam_w api dump --- .../src/python/ptdbg_ascend/hook_module/support_wrap_ops.yaml | 1 + 1 file changed, 1 insertion(+) 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 92096fc4b..2dbf5b7b2 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 -- Gitee