commit fc4bde92120567a98189f95cfe90bb1699d25809
author Jeremy Johnson <jeremy.johnson@arm.com> Thu Jan 25 12:53:21 2024 +0000
committer Eric Kunze <eric.kunze@arm.com> Tue Jan 30 16:11:26 2024 +0000
tree 48ab320c288440816734fc771a817257f3a229f9
parent 95a6710ffb8cadcb8658a967ab29cac1bffad930

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

verif/generator/tosa_test_gen.py

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 @@
         "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 @@
         "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 @@
         "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 @@
         "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 @@
         "concat_shape": {
             "op": Op.CONCAT_SHAPE,
             "operands": (2, 0),
+            "rank": (1, 1),
             "build_fcn": (
                 build_concat,
                 TosaTensorGen.tgConcat,
@@ -4774,6 +4779,7 @@
         "const_shape": {
             "op": Op.CONST_SHAPE,
             "operands": (0, 1),
+            "rank": (1, 1),
             "build_fcn": (
                 build_const,
                 TosaTensorGen.tgBasic,