From b45db9a696f5df7b233f374248f329c16ee7ae64 Mon Sep 17 00:00:00 2001
From: James Ward <james.ward@arm.com>
Date: Mon, 12 Dec 2022 13:02:44 +0000
Subject: Create MI tests for Activation: SIGMOID, TANH

Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I75d6280ecb5219d50c91d14029e90ff3a2f6eb06
---
 verif/conformance/test_select.py                  |  12 +++
 verif/conformance/tosa_main_profile_ops_info.json | 114 ++++++++++++++++++++++
 2 files changed, 126 insertions(+)

diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py
index 99588a9..696b466 100644
--- a/verif/conformance/test_select.py
+++ b/verif/conformance/test_select.py
@@ -684,6 +684,12 @@ class SelectOperator(Operator):
     name = "select"
 
 
+class SigmoidOperator(Operator):
+    """Test selector for the SIGMOID operator."""
+
+    name = "sigmoid"
+
+
 class SliceOperator(Operator):
     """Test selector for the SLICE operator."""
 
@@ -703,6 +709,12 @@ class TableOperator(Operator):
     name = "table"
 
 
+class TanhOperator(Operator):
+    """Test selector for the TANH operator."""
+
+    name = "tanh"
+
+
 class TileOperator(Operator):
     """Test selector for the TILE operator."""
 
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index ce15b2f..4328b9f 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -1183,5 +1183,119 @@
         "profile": [
             "tosa-mi"
         ]
+    },
+    "sigmoid": {
+        "group": "activation",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "16,64",
+                "--target-rank",
+                "1",
+                "--target-rank",
+                "2",
+                "--target-rank",
+                "3"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "1,19",
+                "--target-rank",
+                "4",
+                "--target-rank",
+                "5"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,2,65538,2,1",
+                "--target-shape",
+                "1,3,1,65533"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
+    "tanh": {
+        "group": "activation",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "18,60",
+                "--target-rank",
+                "1",
+                "--target-rank",
+                "2",
+                "--target-rank",
+                "3"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "1,24",
+                "--target-rank",
+                "4",
+                "--target-rank",
+                "5"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,65535,2,1,1",
+                "--target-shape",
+                "1,65540,1,2"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
     }
 }
\ No newline at end of file
-- 
cgit v1.2.1