commit b227ae5ff22411c3f7c82e2b4b5a4618008e897a
author Kevin Cheng <kevin.cheng@arm.com> Thu Sep 02 13:43:17 2021 -0700
committer Kevin Cheng <kevin.cheng@arm.com> Tue Sep 14 14:54:51 2021 -0700
tree 71072b49d2b1d102634f454e536b9aa7144a899d
parent 9034747c223cd5ab53148ec40aaf64765fcd6531

Fix while_loop test.

- add output tensor to main block
- fix bug where tensor cannot serialize list of const data

Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I40b96d634642e24e6fae6fa3ad6846f5911005f7

verif/tosa_serializer.py

diff --git a/verif/tosa_serializer.py b/verif/tosa_serializer.py
index b7f5565..e9ec027 100644
--- a/verif/tosa_serializer.py
+++ b/verif/tosa_serializer.py
@@ -358,6 +358,9 @@
             data = data.flatten().astype(int).tolist()
             data = list(map(int, data))
             self.data = data
+        elif isinstance(data, list):
+            data = list(map(int, data))
+            self.data = data
         else:
             self.data = None
 

verif/tosa_test_gen.py

diff --git a/verif/tosa_test_gen.py b/verif/tosa_test_gen.py
index 760ed06..942f75c 100644
--- a/verif/tosa_test_gen.py
+++ b/verif/tosa_test_gen.py
@@ -1798,6 +1798,7 @@
             [iter_out.name, a_out.name, acc_out.name],
             attr,
         )
+        self.ser.addOutputTensor(acc_out)
 
         # COND block (input: iter, output: cond_tens )
         self.ser.startBasicBlock(cond_block)