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 453a4c82477fb60d9baf30e3e9d35ad855d7fa07..33e2e5adf5e46e9c82a7ed334c29a72814b3bc0e 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 @@ -207,21 +207,21 @@ def initialize_save_error_data(): def _run_ut_parser(parser): - parser.add_argument("-forward", "--forward_input_file", dest="forward_input_file", default="", + parser.add_argument("-forward", "--forward_input_file", dest="forward_input_file", default="", type=str, help=" The api param tool forward result file: generate from api param tool, " "a json file.", required=True) - parser.add_argument("-backward", "--backward_input_file", dest="backward_input_file", default="", + parser.add_argument("-backward", "--backward_input_file", dest="backward_input_file", default="", type=str, help=" The api param tool backward result file: generate from api param tool, " "a json file.", required=True) - parser.add_argument("-o", "--out_path", dest="out_path", default="", + parser.add_argument("-o", "--out_path", dest="out_path", default="", type=str, help=" The ut task result out path.", required=False) parser.add_argument('-save_error_data', dest="save_error_data", action="store_true", help=" Save compare failed api output.", required=False) - parser.add_argument("-j", "--jit_compile", dest="jit_compile", help=" whether to turn on jit compile", - default=False, required=False) + parser.add_argument("-j", "--jit_compile", dest="jit_compile", action="store_true", + help=" whether to turn on jit compile", required=False) parser.add_argument("-d", "--device", dest="device_id", type=int, help=" set NPU device id to run ut", default=0, required=False)