diff options
author | Luke Hutton <luke.hutton@arm.com> | 2023-02-08 19:45:26 +0000 |
---|---|---|
committer | Luke Hutton <luke.hutton@arm.com> | 2023-04-13 17:32:14 +0100 |
commit | 714aa6039a7e3585bf81ac90ce301767c08295af (patch) | |
tree | b50dd8a4a8faa70df8e91397dec08530afa1d770 /verif/frameworks/tosa_verif_framework_compiler_runner.py | |
parent | abf8718d3bce6c76fb281d3911f566cd90c44f28 (diff) | |
download | reference_model-714aa6039a7e3585bf81ac90ce301767c08295af.tar.gz |
Add framework tests for tfl.real and tfl.imag
Change-Id: I665acac9b5171efd0c5a2b68b516609048f6e187
Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Diffstat (limited to 'verif/frameworks/tosa_verif_framework_compiler_runner.py')
-rwxr-xr-x | verif/frameworks/tosa_verif_framework_compiler_runner.py | 8 |
1 files changed, 7 insertions, 1 deletions
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 @@ def run_test(args, test, framework): 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) |