commit 714aa6039a7e3585bf81ac90ce301767c08295af
author Luke Hutton <luke.hutton@arm.com> Wed Feb 08 19:45:26 2023 +0000
committer Luke Hutton <luke.hutton@arm.com> Thu Apr 13 17:32:14 2023 +0100
tree b50dd8a4a8faa70df8e91397dec08530afa1d770
parent abf8718d3bce6c76fb281d3911f566cd90c44f28

Add framework tests for tfl.real and tfl.imag

Change-Id: I665acac9b5171efd0c5a2b68b516609048f6e187
Signed-off-by: Luke Hutton <luke.hutton@arm.com>

verif/frameworks/tosa_verif_framework_compiler_runner.py

diff --git a/verif/frameworks/tosa_verif_framework_compiler_runner.py b/verif/frameworks/tosa_verif_framework_compiler_runner.py
index c55864a..71723ae 100755
--- a/verif/frameworks/tosa_verif_framework_compiler_runner.py
+++ b/verif/frameworks/tosa_verif_framework_compiler_runner.py
@@ -384,7 +384,13 @@
             while len(list(ifm_np.shape)) < len(test_desc["ifm_shape"][i]):
                 ifm_np = np.expand_dims(ifm_np, axis=0)
 
-            assert list(ifm_np.shape) == test_desc["ifm_shape"][i]
+            # After legalization, complex tensors are expected to be represented
+            # as a single floating point tensor of shape [?, ..., ?, 2].
+            expected_shape = test_desc["ifm_shape"][i]
+            if test.endswith("c64"):
+                expected_shape.append(2)
+
+            assert list(ifm_np.shape) == expected_shape
 
             reference_runner_ifm_name.append(ifm_tensor_name)