diff options
| author | James Ward <james.ward@arm.com> | 2022-11-23 11:00:47 +0000 |
|---|---|---|
| committer | Eric Kunze <eric.kunze@arm.com> | 2022-12-13 21:22:43 +0000 |
| commit | 71616fe89e17bf5a18ed156dc6f69f4241c8432d (patch) | |
| tree | 4ccb93aaa472ec17becb9872de6754ad61bcae35 | |
| parent | 635bc99c8fd012e4e7b3dcc1dc72a9a0db6661fb (diff) | |
| download | reference_model-71616fe89e17bf5a18ed156dc6f69f4241c8432d.tar.gz | |
Create MI tests for EW Unary: ABS, CEIL, FLOOR, NEGATE
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I2ee6d4f706958b6261fd41dd245a3d01d3f8cecf
| -rw-r--r-- | verif/conformance/test_select.py | 12 | ||||
| -rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 229 |
2 files changed, 241 insertions, 0 deletions
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": [ |