diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-02-07 16:09:09 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-02-08 21:06:36 +0000 |
commit | 01e1c1c7f965ceb07e78a3b1ad063161c0f47b94 (patch) | |
tree | 10235a2a231f5366572577462ca4d1902f9a098c /verif/generator/tosa_test_gen.py | |
parent | 59d8f50f5b3399a6255643aad0e5857e30370761 (diff) | |
download | reference_model-01e1c1c7f965ceb07e78a3b1ad063161c0f47b94.tar.gz |
Improve Avg_Pool2D ErrorIf Testing
* Add test for invalid accumulator dtype
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I506e2047623372670b82db6e9c0010fa89802851
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index f5eca18..2d471c0 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -835,6 +835,7 @@ class TosaTestGen: input_dtype=input.dtype, output_shape=result_tensor.shape, output_dtype=result_tensor.dtype, + accum_dtype=accum_dtype, kernel=kernel, stride=stride, pad=pad, @@ -3218,6 +3219,7 @@ class TosaTestGen: TosaErrorValidator.evPadLargerEqualKernel, TosaErrorValidator.evPoolingOutputShapeMismatch, TosaErrorValidator.evPoolingOutputShapeNonInteger, + TosaErrorValidator.evWrongAccumulatorType, ), "data_gen": { "fp": (gtu.DataGenType.DOT_PRODUCT,), |