From fc4bde92120567a98189f95cfe90bb1699d25809 Mon Sep 17 00:00:00 2001
From: Jeremy Johnson <jeremy.johnson@arm.com>
Date: Thu, 25 Jan 2024 12:53:21 +0000
Subject: 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 | 14 ++++++++++----
 1 file changed, 10 insertions(+), 4 deletions(-)

(limited to '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 @@ 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,
-- 
cgit v1.2.1