From b7f5eed148f515bc07e23ec8903d05dfe35dc8d7 Mon Sep 17 00:00:00 2001
From: evacha01 <evan.chandler@arm.com>
Date: Thu, 18 Apr 2024 14:49:35 +0100
Subject: 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 | 3 +++
 1 file changed, 3 insertions(+)

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 @@ class TosaArgGen:
                             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:
-- 
cgit v1.2.1