diff options
| author | James Ward <james.ward@arm.com> | 2022-08-12 20:48:56 +0100 |
|---|---|---|
| committer | James Ward <james.ward@arm.com> | 2022-10-11 11:56:02 +0100 |
| commit | 8b39043c70332e1e4c95ee6a9616aec40dd3baf1 (patch) | |
| tree | fea519246b698eb944b9d58537fc90bc30481d11 /verif/checker/tosa_result_checker.py | |
| parent | ba5fad356a926d5e1c6e0fe6b546a310230cc5a8 (diff) | |
| download | reference_model-8b39043c70332e1e4c95ee6a9616aec40dd3baf1.tar.gz | |
Reference model changes for fp16 support
Change-Id: I72f21fcfa153046274969d327313e3349981dbe6
Signed-off-by: James Ward <james.ward@arm.com>
Diffstat (limited to 'verif/checker/tosa_result_checker.py')
| -rw-r--r-- | verif/checker/tosa_result_checker.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/verif/checker/tosa_result_checker.py b/verif/checker/tosa_result_checker.py index 66864c2..8ae3218 100644 --- a/verif/checker/tosa_result_checker.py +++ b/verif/checker/tosa_result_checker.py @@ -147,14 +147,14 @@ def test_check( tolerance = 0.0 # Fall-through to below to add failure values - elif reference_result.dtype == np.float32: + # TODO: update for fp16 tolerance + elif reference_result.dtype == np.float32 or reference_result.dtype == np.float16: tolerance = float_tolerance if np.allclose(reference_result, test_result, atol=tolerance, equal_nan=True): print_color(LogColors.GREEN, "Results PASS {}".format(test_name)) return (TestResult.PASS, tolerance, "") msg = "Float result does not match within tolerance of {}".format(tolerance) # Fall-through to below to add failure values - else: print_color(LogColors.RED, "Results UNSUPPORTED TYPE {}".format(test_name)) msg = "Unsupported results type: {}".format(reference_result.dtype) |