From 8b4fbe4ec760f0396505e772135311e07cef481e Mon Sep 17 00:00:00 2001 From: l30044004 Date: Tue, 12 Mar 2024 15:07:15 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0acl=20dump=E7=9A=84=E5=90=8C?= =?UTF-8?q?=E6=AD=A5=E6=93=8D=E4=BD=9C?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- .../ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py index 9129e0e1b..f23e59960 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py @@ -339,6 +339,7 @@ def forward_acl_dump(module, module_name): global backward_init_status if not forward_init_status and not backward_init_status: forward_init_status = True + torch_npu.npu.synchronize() torch_npu.npu.init_dump() torch_npu.npu.set_dump(DumpUtil.dump_config) torch_npu.npu.synchronize() @@ -348,6 +349,7 @@ def forward_acl_dump(module, module_name): module.forward(*module.input_args, **module.input_kwargs) torch_npu.npu.synchronize() torch_npu.npu.finalize_dump() + torch_npu.npu.synchronize() del module.input_args del module.input_kwargs forward_init_status = False -- Gitee