diff --git a/debug/accuracy_tools/api_accuracy_checker/run_ut/data_generate.py b/debug/accuracy_tools/api_accuracy_checker/run_ut/data_generate.py index 9b34e51c4fe674b5e4ef02df2f1130489f367d2b..a46e819b8758c97af1db7bd19ca12a0866bd5f73 100644 --- a/debug/accuracy_tools/api_accuracy_checker/run_ut/data_generate.py +++ b/debug/accuracy_tools/api_accuracy_checker/run_ut/data_generate.py @@ -19,7 +19,7 @@ import os import torch import numpy as np -from api_accuracy_checker.common.utils import check_file_or_directory_path, check_object_type, print_warn_log, print_error_log, \ +from api_accuracy_checker.common.utils import Const, check_file_or_directory_path, check_object_type, print_warn_log, print_error_log, \ CompareException TENSOR_DATA_LIST = ["torch.Tensor", "torch.nn.parameter.Parameter"] diff --git a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py index c71b32cefe02418bc4148545d862921c18a7c773..09bc3bf6d4310d1a031ac4c701236f5c50c68b65 100644 --- a/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py +++ b/debug/accuracy_tools/api_accuracy_checker/run_ut/run_ut.py @@ -5,7 +5,7 @@ import torch_npu import yaml import torch from api_accuracy_checker.run_ut.data_generate import gen_api_params, gen_args -from api_accuracy_checker.common.utils import print_info_log, print_warn_log, get_json_contents +from api_accuracy_checker.common.utils import print_info_log, print_warn_log, get_json_contents, check_need_convert from api_accuracy_checker.compare.compare import Comparator cur_path = os.path.dirname(os.path.realpath(__file__))