diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2024-01-25 12:53:21 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2024-01-30 16:11:26 +0000 |
commit | fc4bde92120567a98189f95cfe90bb1699d25809 (patch) | |
tree | 48ab320c288440816734fc771a817257f3a229f9 /verif/generator/tosa_test_gen.py | |
parent | 95a6710ffb8cadcb8658a967ab29cac1bffad930 (diff) | |
download | reference_model-fc4bde92120567a98189f95cfe90bb1699d25809.tar.gz |
Fix up shape operator test errors
Update serialization_lib to store SHAPE as INT64.
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: Ie589cd6670dc79b77df981c81cd7c27b982f20fa
Diffstat (limited to 'verif/generator/tosa_test_gen.py')
-rw-r--r-- | verif/generator/tosa_test_gen.py | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/verif/generator/tosa_test_gen.py b/verif/generator/tosa_test_gen.py index 39b064d..a347b13 100644 --- a/verif/generator/tosa_test_gen.py +++ b/verif/generator/tosa_test_gen.py @@ -4714,9 +4714,10 @@ class TosaTestGen: "add_shape": { "op": Op.ADD_SHAPE, "operands": (2, 0), + "rank": (1, 1), "build_fcn": ( build_shape_op, - TosaTensorGen.tgShape, + TosaTensorGen.tgBasic, TosaTensorValuesGen.tvgAddSub, TosaArgGen.agNone, ), @@ -4726,9 +4727,10 @@ class TosaTestGen: "sub_shape": { "op": Op.SUB_SHAPE, "operands": (2, 0), + "rank": (1, 1), "build_fcn": ( build_shape_op, - TosaTensorGen.tgShape, + TosaTensorGen.tgBasic, TosaTensorValuesGen.tvgAddSub, TosaArgGen.agNone, ), @@ -4738,9 +4740,10 @@ class TosaTestGen: "mul_shape": { "op": Op.MUL_SHAPE, "operands": (2, 0), + "rank": (1, 1), "build_fcn": ( build_shape_op, - TosaTensorGen.tgShape, + TosaTensorGen.tgBasic, TosaTensorValuesGen.tvgMul, TosaArgGen.agNone, ), @@ -4750,9 +4753,10 @@ class TosaTestGen: "div_shape": { "op": Op.DIV_SHAPE, "operands": (2, 0), + "rank": (1, 1), "build_fcn": ( build_shape_op, - TosaTensorGen.tgShape, + TosaTensorGen.tgBasic, TosaTensorValuesGen.tvgIntDiv, TosaArgGen.agNone, ), @@ -4762,6 +4766,7 @@ class TosaTestGen: "concat_shape": { "op": Op.CONCAT_SHAPE, "operands": (2, 0), + "rank": (1, 1), "build_fcn": ( build_concat, TosaTensorGen.tgConcat, @@ -4774,6 +4779,7 @@ class TosaTestGen: "const_shape": { "op": Op.CONST_SHAPE, "operands": (0, 1), + "rank": (1, 1), "build_fcn": ( build_const, TosaTensorGen.tgBasic, |