commit b7f5eed148f515bc07e23ec8903d05dfe35dc8d7
author evacha01 <evan.chandler@arm.com> Thu Apr 18 14:49:35 2024 +0100
committer Eric Kunze <eric.kunze@arm.com> Thu Apr 18 16:14:15 2024 +0000
tree 8891d5ed05300f468dca2def32877dae013ccc82
parent 1eb1455568e2a23971f2c1b7be1077a8c1494685

Non-finite data tag for FP_SPECIAL data gen tests

Signed-off-by: evacha01 <evan.chandler@arm.com>
Change-Id: I21e305dbe4f406e85c6ed3d5b30b5b85dd208825

verif/generator/tosa_arg_gen.py

diff --git a/verif/generator/tosa_arg_gen.py b/verif/generator/tosa_arg_gen.py
index 22b23fe..41b0936 100644
--- a/verif/generator/tosa_arg_gen.py
+++ b/verif/generator/tosa_arg_gen.py
@@ -1935,6 +1935,9 @@
                             shapeList
                         )
                     arg_str = f"{arg_str}_fs" if arg_str else "fs"
+                    gen_args_dict["tags"] = args_dict.get("tags", []) + [
+                        "non_finite_fp_data"
+                    ]
 
                 gen_args_dict["dg_type"] = dg_type
                 if num_test_sets > 0: