From bfe2431e936a8bd601ec32134aa14f2fdaae1c66 Mon Sep 17 00:00:00 2001 From: l30036321 Date: Thu, 7 Mar 2024 10:15:58 +0800 Subject: [PATCH] fix checkpoint bug --- .../ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) 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 a6b769ff2a..9129e0e1bd 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 @@ -151,14 +151,15 @@ def dump_tensor(x, prefix, dump_step): if x.is_meta: print_info_log(f"Meta tensor {prefix} is skipped.") return - if x.numel() == 0 or len(x.shape) == 0 or not x.is_floating_point(): + x_clone = x.clone().detach() + if x_clone.numel() == 0 or len(x_clone.shape) == 0 or not x_clone.is_floating_point(): if DumpUtil.dump_filter_switch == Const.OFF: - data_info = get_not_float_tensor_info(x) + data_info = get_not_float_tensor_info(x_clone) dump_data_by_rank_count(dump_step, prefix, data_info) else: return else: - data_info = get_float_tensor_info(x) + data_info = get_float_tensor_info(x_clone) dump_data_by_rank_count(dump_step, prefix, data_info) elif DumpUtil.dump_filter_switch == Const.OFF: -- Gitee