diff options
-rw-r--r-- | verif/checker/tosa_result_checker.py | 9 | ||||
-rw-r--r-- | verif/tests/test_tosa_result_checker.py | 8 |
2 files changed, 12 insertions, 5 deletions
diff --git a/verif/checker/tosa_result_checker.py b/verif/checker/tosa_result_checker.py index 38ed510..4ba5d4c 100644 --- a/verif/checker/tosa_result_checker.py +++ b/verif/checker/tosa_result_checker.py @@ -186,7 +186,14 @@ def test_check( return (TestResult.INCORRECT_FORMAT, 0.0, msg) # for quantized test, allow +-(quantize_tolerance) error - if reference_result.dtype == np.int32 or reference_result.dtype == np.int64: + if reference_result.dtype in ( + np.int8, + np.int16, + np.int32, + np.int64, + np.uint8, + np.uint16, + ): if np.all(np.absolute(reference_result - test_result) <= quantize_tolerance): _print_result(LogColors.GREEN, "Results PASS {}".format(test_name)) diff --git a/verif/tests/test_tosa_result_checker.py b/verif/tests/test_tosa_result_checker.py index d78d158..f9bda39 100644 --- a/verif/tests/test_tosa_result_checker.py +++ b/verif/tests/test_tosa_result_checker.py @@ -32,12 +32,12 @@ def _delete_data_file(file: Path): @pytest.mark.parametrize( "data_type,expected", [ - (np.int8, trc.TestResult.MISMATCH), - (np.int16, trc.TestResult.MISMATCH), + (np.int8, trc.TestResult.PASS), + (np.int16, trc.TestResult.PASS), (np.int32, trc.TestResult.PASS), (np.int64, trc.TestResult.PASS), - (np.uint8, trc.TestResult.MISMATCH), - (np.uint16, trc.TestResult.MISMATCH), + (np.uint8, trc.TestResult.PASS), + (np.uint16, trc.TestResult.PASS), (np.uint32, trc.TestResult.MISMATCH), (np.uint64, trc.TestResult.MISMATCH), (np.float16, trc.TestResult.PASS), |