From e80afdc2d103cf54ccfb13ef0e1132afb793bf5f Mon Sep 17 00:00:00 2001 From: louyujing Date: Thu, 29 Feb 2024 03:31:30 +0000 Subject: [PATCH] update debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py. Signed-off-by: louyujing --- .../src/python/ptdbg_ascend/overflow_check/overflow_check.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py index df605848e..94d3b9a8d 100644 --- a/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py +++ b/debug/accuracy_tools/ptdbg_ascend/src/python/ptdbg_ascend/overflow_check/overflow_check.py @@ -50,6 +50,8 @@ def check_data_overflow(x): return False else: if isinstance(x, torch.Tensor) and x.numel() != 0 and x.dtype != torch.bool: + if x.is_meta: + return False if len(x.shape) == 0: tensor_max = x.cpu().detach().float().numpy().tolist() tensor_min = tensor_max -- Gitee