From e4fa3ac727cbaf5f0513ac8466391c9e02ee0c8a Mon Sep 17 00:00:00 2001 From: yang-minghai22 Date: Tue, 30 Apr 2024 15:25:00 +0800 Subject: [PATCH] support complex data summary calculation --- .../ptdbg_ascend/src/python/ptdbg_ascend/dump/dump.py | 6 ++++++ 1 file changed, 6 insertions(+) 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 2e49a9743..b8a8e026a 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 @@ -106,6 +106,12 @@ def get_not_float_tensor_info(data): tensor_max = item tensor_min = item tensor_mean = item + elif torch.is_complex(data): + data_np = data.detach().cpu().numpy() + data_abs = np.abs(data_np) + tensor_max = np.max(data_abs).item() + tensor_min = np.min(data_abs).item() + tensor_mean = np.mean(data_abs).item() else: tensor_max = torch._C._VariableFunctionsClass.max(data).float().item() tensor_min = torch._C._VariableFunctionsClass.min(data).float().item() -- Gitee