commit 72dcab775c7a84037135bf365086ca976f3220ef
author Jeremy Johnson <jeremy.johnson@arm.com> Mon Oct 30 10:28:21 2023 +0000
committer Eric Kunze <eric.kunze@arm.com> Tue Oct 31 20:58:07 2023 +0000
tree b184d731582734795e77492290c1595ec8ed13d5
parent a7f5b995d618d26724812fc27011f87600e958dc

Allow more integer types in result check.

Added int8, int16, uint8 and uint16.

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I6b1174aa7369c34a613b322c4b05bf4fc586ee6e

verif/checker/tosa_result_checker.py

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 @@
             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))