From 71616fe89e17bf5a18ed156dc6f69f4241c8432d Mon Sep 17 00:00:00 2001
From: James Ward <james.ward@arm.com>
Date: Wed, 23 Nov 2022 11:00:47 +0000
Subject: Create MI tests for EW Unary: ABS, CEIL, FLOOR, NEGATE

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

diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py
index 2b8e7d2..99588a9 100644
--- a/verif/conformance/test_select.py
+++ b/verif/conformance/test_select.py
@@ -399,6 +399,12 @@ class CastOperator(Operator):
     param_names = ["shape", "type", "output_type"]
 
 
+class CeilOperator(Operator):
+    """Test selector for the CEIL operator."""
+
+    name = "ceil"
+
+
 class ClampOperator(Operator):
     """Test selector for the CLAMP operator."""
 
@@ -458,6 +464,12 @@ class EqualOperator(Operator):
     name = "equal"
 
 
+class FloorOperator(Operator):
+    """Test selector for the FLOOR operator."""
+
+    name = "floor"
+
+
 class FullyConnectedOperator(Operator):
     """Test selector for the FULLY_CONNECTED operator."""
 
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index ce9b71f..0bd9b1f 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -1,4 +1,62 @@
+
 {
+    "abs": {
+        "group": "ew_unary",
+        "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,16",
+                "--target-rank",
+                "4",
+                "--target-rank",
+                "5"
+            ],
+            [
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,65531,2,1",
+                "--target-shape",
+                "65538,1,1,1"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
     "argmax": {
         "group": "tensor",
         "generator_args": [
@@ -105,6 +163,63 @@
             "tosa-mi"
         ]
     },
+    "ceil": {
+        "group": "ew_unary",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "4,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,16",
+                "--target-rank",
+                "4",
+                "--target-rank",
+                "5"
+            ],
+            [
+                "--target-dtype",
+                "fp16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "2,1,65530,1",
+                "--target-shape",
+                "1,1,65539,1"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
     "concat": {
         "group": "data_layout",
         "generator_args": [
@@ -274,6 +389,63 @@
             "tosa-mi"
         ]
     },
+    "floor": {
+        "group": "ew_unary",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "4,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,16",
+                "--target-rank",
+                "4",
+                "--target-rank",
+                "5"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,1,65533,2",
+                "--target-shape",
+                "1,65537,1,1"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
     "gather": {
         "group": "scatter_gather",
         "generator_args": [
@@ -423,6 +595,63 @@
             "tosa-mi"
         ]
     },
+    "negate": {
+        "group": "ew_unary",
+        "generator_args": [
+            [
+                "--target-dtype",
+                "fp32",
+                "--target-dtype",
+                "fp16",
+                "--target-dtype",
+                "bf16",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--tensor-dim-range",
+                "17,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,22",
+                "--target-rank",
+                "4",
+                "--target-rank",
+                "5"
+            ],
+            [
+                "--target-dtype",
+                "fp32",
+                "--fp-values-range",
+                "-2.0,2.0",
+                "--target-shape",
+                "1,65541,3,1",
+                "--target-shape",
+                "1,2,1,65532,1"
+            ]
+        ],
+        "params": {},
+        "permutes": [
+            "shape",
+            "type"
+        ],
+        "profile": [
+            "tosa-mi"
+        ]
+    },
     "pad": {
         "group": "data_layout",
         "generator_args": [
-- 
cgit v1.2.1