From edf405fbdabaeeef52fb6b5c4e983a5411b577f6 Mon Sep 17 00:00:00 2001 From: gitee Date: Wed, 25 Sep 2024 10:11:07 +0800 Subject: [PATCH] fix get min/max with inf or nan --- .../pytorch/api_accuracy_checker/run_ut/data_generate.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/debug/accuracy_tools/msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py b/debug/accuracy_tools/msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py index f67c849910..d64c2774a9 100644 --- a/debug/accuracy_tools/msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py +++ b/debug/accuracy_tools/msprobe/pytorch/api_accuracy_checker/run_ut/data_generate.py @@ -120,8 +120,12 @@ def gen_random_tensor(info, convert_type): convert_type: convert ori_type to dist_type flag. """ check_object_type(info, dict) - low, high = info.get('Min'), info.get('Max') - low_origin, high_origin = info.get('Min_origin'), info.get('Max_origin') + + low_origin = info.get('Min') + low = info.get('Min_except_inf_nan', low_origin) + high_origin = info.get('Max') + high = info.get('Max_except_inf_nan', high_origin) + low_info = [low, low_origin] high_info = [high, high_origin] data_dtype = info.get('dtype') -- Gitee