diff options
author | James Ward <james.ward@arm.com> | 2022-11-23 11:55:32 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2022-12-13 21:22:31 +0000 |
commit | 635bc99c8fd012e4e7b3dcc1dc72a9a0db6661fb (patch) | |
tree | f41b67c1c25e72549a59df0bc0fb03209afbd557 | |
parent | c8774a1de2feff92f462e02116e64eca3ea10cbd (diff) | |
download | reference_model-635bc99c8fd012e4e7b3dcc1dc72a9a0db6661fb.tar.gz |
Create MI tests for Reduction: REDUCE_MIN, REDUCE_MAX
* Also include test_select improvements
Signed-off-by: James Ward <james.ward@arm.com>
Change-Id: I306a8ed57975bc429d6cd5a8a24777b889593ddf
-rw-r--r-- | verif/conformance/test_select.py | 2 | ||||
-rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 72 |
2 files changed, 73 insertions, 1 deletions
diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py index c04b7ef..2b8e7d2 100644 --- a/verif/conformance/test_select.py +++ b/verif/conformance/test_select.py @@ -793,7 +793,7 @@ def main(): args = parse_args() loglevels = (logging.ERROR, logging.WARNING, logging.INFO, logging.DEBUG) - logger.basicConfig(level=loglevels[min(args.verbosity, len(loglevels) - 1)]) + logger.setLevel(loglevels[min(args.verbosity, len(loglevels) - 1)]) logger.info(f"{__file__}: args: {args}") try: diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json index ee2abb7..ce9b71f 100644 --- a/verif/conformance/tosa_main_profile_ops_info.json +++ b/verif/conformance/tosa_main_profile_ops_info.json @@ -823,5 +823,77 @@ "profile": [ "tosa-mi" ] + }, + "reduce_max": { + "group": "reduction", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,32" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,1,1,65531", + "--target-shape", + "1,1,65539,2" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ], + "profile": [ + "tosa-mi" + ] + }, + "reduce_min": { + "group": "reduction", + "generator_args": [ + [ + "--target-dtype", + "fp32", + "--target-dtype", + "fp16", + "--target-dtype", + "bf16", + "--fp-values-range", + "-2.0,2.0", + "--tensor-dim-range", + "1,32" + ], + [ + "--target-dtype", + "fp32", + "--fp-values-range", + "-2.0,2.0", + "--target-shape", + "1,1,65531,1", + "--target-shape", + "1,65539,2,1" + ] + ], + "params": {}, + "permutes": [ + "shape", + "type", + "axis" + ], + "profile": [ + "tosa-mi" + ] } }
\ No newline at end of file |