From 635bc99c8fd012e4e7b3dcc1dc72a9a0db6661fb Mon Sep 17 00:00:00 2001 From: James Ward Date: Wed, 23 Nov 2022 11:55:32 +0000 Subject: Create MI tests for Reduction: REDUCE_MIN, REDUCE_MAX * Also include test_select improvements Signed-off-by: James Ward Change-Id: I306a8ed57975bc429d6cd5a8a24777b889593ddf --- verif/conformance/test_select.py | 2 +- verif/conformance/tosa_main_profile_ops_info.json | 72 +++++++++++++++++++++++ 2 files changed, 73 insertions(+), 1 deletion(-) 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 -- cgit v1.2.1