aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJeremy Johnson <jeremy.johnson@arm.com>2022-12-05 16:59:28 +0000
committerJeremy Johnson <jeremy.johnson@arm.com>2023-01-04 17:14:12 +0000
commit6ffb7c8fb88d3a5d0e94f3e604dc3548deea95d7 (patch)
tree2e520284ce4527c5bdb4b40466df33591dfc079d
parent98c281fb9bd5fb91d9fce09a979debd045345ea8 (diff)
downloadreference_model-6ffb7c8fb88d3a5d0e94f3e604dc3548deea95d7.tar.gz
Create MI tests for EW Binary: POW
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com> Change-Id: I8d478b7bd13559ea37e27f77b05ee5db56c8f9e3
-rw-r--r--verif/conformance/test_select.py6
-rw-r--r--verif/conformance/tosa_main_profile_ops_info.json53
2 files changed, 59 insertions, 0 deletions
diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py
index 696b466..59feae1 100644
--- a/verif/conformance/test_select.py
+++ b/verif/conformance/test_select.py
@@ -594,6 +594,12 @@ class PadOperator(Operator):
param_names = ["shape", "type", "pad"]
+class PowOperator(Operator):
+ """Test selector for the POW operator."""
+
+ name = "pow"
+
+
class ReduceAllOperator(Operator):
"""Test selector for the REDUCE_ALL operator."""
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index 0cbe044..da04d85 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -731,6 +731,59 @@
"tosa-mi"
]
},
+ "pow": {
+ "group": "ew_binary",
+ "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",
+ "fp16",
+ "--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",
+ "65534,3,1,1",
+ "--target-shape",
+ "2,65541,1,1,1"
+ ]
+ ],
+ "params": {},
+ "permutes": [
+ "shape",
+ "type"
+ ],
+ "profile": [
+ "tosa-mi"
+ ]
+ },
"reshape": {
"group": "data_layout",
"generator_args": [