diff --git a/compiler/pota-quantization-value-test/compare_tensors_all.py b/compiler/pota-quantization-value-test/compare_tensors_all.py index 4a15be2c207..98fe1d05950 100644 --- a/compiler/pota-quantization-value-test/compare_tensors_all.py +++ b/compiler/pota-quantization-value-test/compare_tensors_all.py @@ -57,6 +57,9 @@ source_dir = args.source_dir mode = args.mode +if mode in ['mixed_fake_quantization', 'parallel_record_minmax', 'mixed_quantization']: + source_dir += '_config' + log_format = '%(levelname)s: %(message)s' formatter = logging.Formatter(log_format) streamer = StringIO() @@ -167,13 +170,13 @@ def chunks(lst, n): expect_dir = f'{source_dir}/expected_outputs/{model_name}/{granularity}/{dtype}/{modes_to_expected_folder[mode]}' if os.path.isfile(expect_dir + "/" + tensor_name + ".json"): test_result = False - if mode == "fake_quantization": + if mode in ["fake_quantization", "mixed_fake_quantization"]: test_result = compare_fake_quantization(input[tensor_name], tensor_name, expect_dir) - elif mode == "record_minmax": + elif mode in ["record_minmax", "parallel_record_minmax"]: test_result = compare_record_minmax(input[tensor_name], tensor_name, expect_dir) - elif mode == "quantization": + elif mode in ["quantization", "mixed_quantization"]: test_result = compare_quantization(input[tensor_name], tensor_name, expect_dir) elif mode == "weights_only_quantization": diff --git a/compiler/pota-quantization-value-test/test_fake_wquant_with_config.sh b/compiler/pota-quantization-value-test/test_fake_wquant_with_config.sh index 16e43ce5983..815ab4ca7a5 100755 --- a/compiler/pota-quantization-value-test/test_fake_wquant_with_config.sh +++ b/compiler/pota-quantization-value-test/test_fake_wquant_with_config.sh @@ -65,7 +65,7 @@ popd --test_param "${TEST_PARAMS}" \ --bin_dir ${BIN_PATH} \ --source_dir ${SOURCE_PATH} \ - --mode fake_quantization + --mode mixed_fake_quantization if [[ $? -ne 0 ]]; then exit 255 diff --git a/compiler/pota-quantization-value-test/test_parallel_record_minmax.sh b/compiler/pota-quantization-value-test/test_parallel_record_minmax.sh index 4e665879f32..612215dc29c 100755 --- a/compiler/pota-quantization-value-test/test_parallel_record_minmax.sh +++ b/compiler/pota-quantization-value-test/test_parallel_record_minmax.sh @@ -79,7 +79,7 @@ popd --test_param "${TEST_PARAMS}" \ --bin_dir ${BIN_PATH} \ --source_dir ${SOURCE_PATH} \ - --mode record_minmax + --mode parallel_record_minmax if [[ $? -ne 0 ]]; then exit 255 diff --git a/compiler/pota-quantization-value-test/test_quantization_with_config.sh b/compiler/pota-quantization-value-test/test_quantization_with_config.sh index f3f4fcf4979..e56f0fdc028 100755 --- a/compiler/pota-quantization-value-test/test_quantization_with_config.sh +++ b/compiler/pota-quantization-value-test/test_quantization_with_config.sh @@ -90,7 +90,7 @@ popd --test_param "${TEST_PARAMS}" \ --bin_dir ${BIN_PATH} \ --source_dir ${SOURCE_PATH} \ - --mode quantization + --mode mixed_quantization if [[ $? -ne 0 ]]; then exit 255