aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-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": [