From e34571397539dab6b09dbe6d4c98ff4690a2b963 Mon Sep 17 00:00:00 2001
From: yangminghai22 <yangminghai2@huawei.com>
Date: Thu, 14 Mar 2024 17:36:06 +0800
Subject: [PATCH] fix npu_to_cpu copy miss

---
 .../src/python/ptdbg_ascend/online_dispatch/utils.py            | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/online_dispatch/utils.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/online_dispatch/utils.py
index 57f9258fe..82fc275da 100644
--- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/online_dispatch/utils.py
+++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/online_dispatch/utils.py
@@ -113,6 +113,8 @@ def data_to_cpu(data, deep, data_cpu):
     elif isinstance(data, torch._C.device):
         return cpu_device
     else:
+        if deep == 0:
+            data_cpu.append(data)
         return data
 
 
-- 
Gitee