diff --git a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py index 3a931f487929c1bae1d59bc997b9fd7c8b66a5e5..da2d5c7e6904ae2a8f35a4e10008d69e1dea6b12 100644 --- a/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py +++ b/debug/accuracy_tools/msprobe/core/data_dump/data_processor/mindspore_processor.py @@ -76,10 +76,7 @@ class MindsporeDataProcessor(BaseDataProcessor): get_max_value = api_register.mint_ops_ori_attr.get("max", mint.max) get_min_value = api_register.mint_ops_ori_attr.get("min", mint.min) get_mean_value = api_register.mint_ops_ori_attr.get("mean", mint.mean) - if hasattr(mint, "norm"): - get_norm_value = api_register.mint_ops_ori_attr.get("norm", mint.norm) - else: - get_norm_value = api_register.functional_ori_attr.get("norm", ops.norm) + get_norm_value = api_register.functional_ori_attr.get("norm", ops.norm) tensor_stat.max = get_max_value(data).item() tensor_stat.min = get_min_value(data).item() tensor_stat.mean = get_mean_value(data).item()