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 9129e0e1bd17438f70d49d6b55044299619e4597..f23e59960960e24df1648a6167b9c43b25abae78 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