From 2270153da141cd95a14a528b255815b94bfc5f40 Mon Sep 17 00:00:00 2001 From: yangminghai Date: Wed, 13 Mar 2024 19:05:38 +0800 Subject: [PATCH] fix copy to cpu error when output is not Tensor or list --- .../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 57f9258feb..82fc275da3 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