diff options
-rw-r--r-- | verif/conformance/test_select.py | 6 | ||||
-rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 53 |
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": [ |