aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorJames Ward <james.ward@arm.com>2022-11-23 11:55:32 +0000
committerEric Kunze <eric.kunze@arm.com>2022-12-13 21:22:31 +0000
commit635bc99c8fd012e4e7b3dcc1dc72a9a0db6661fb (patch)
treef41b67c1c25e72549a59df0bc0fb03209afbd557
parentc8774a1de2feff92f462e02116e64eca3ea10cbd (diff)
downloadreference_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.py2
-rw-r--r--verif/conformance/tosa_main_profile_ops_info.json72
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