aboutsummaryrefslogtreecommitdiff
path: root/verif/generator/tosa_test_gen.py
diff options
context:
space:
mode:
authorJeremy Johnson <jeremy.johnson@arm.com>2024-02-07 16:09:09 +0000
committerEric Kunze <eric.kunze@arm.com>2024-02-08 21:06:36 +0000
commit01e1c1c7f965ceb07e78a3b1ad063161c0f47b94 (patch)
tree10235a2a231f5366572577462ca4d1902f9a098c /verif/generator/tosa_test_gen.py
parent59d8f50f5b3399a6255643aad0e5857e30370761 (diff)
downloadreference_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.py2
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,),