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 57f9258feb50a8f609ae2ee207824afcf02a4cdb..82fc275da3b53fd4796a2e7b9cd74d5f5dd9df1a 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