Fix REDUCE ops to support 8k MAX_RANK
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I365da379f599e5eca1841e6d299b4005a7b0f082
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index dc28bef..7792417 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -2450,7 +2450,15 @@
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
- "1,32"
+ "1,32",
+ "--target-rank",
+ "1",
+ "--target-rank",
+ "2",
+ "--target-rank",
+ "3",
+ "--target-rank",
+ "4"
],
[
"--target-dtype",
@@ -2463,6 +2471,28 @@
"1,1,65539,2"
]
]
+ },
+ "8k_level": {
+ "from_version": "v0.80.0",
+ "no_negative_tests": "true",
+ "generator_args": [
+ [
+ "--target-dtype",
+ "fp32",
+ "--target-dtype",
+ "fp16",
+ "--target-dtype",
+ "bf16",
+ "--fp-values-range",
+ "-max,max",
+ "--tensor-dim-range",
+ "1,10",
+ "--target-rank",
+ "5",
+ "--target-rank",
+ "6"
+ ]
+ ]
}
},
"selection": {
@@ -2496,7 +2526,15 @@
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
- "1,32"
+ "1,32",
+ "--target-rank",
+ "1",
+ "--target-rank",
+ "2",
+ "--target-rank",
+ "3",
+ "--target-rank",
+ "4"
],
[
"--target-dtype",
@@ -2509,6 +2547,28 @@
"1,65539,2,1"
]
]
+ },
+ "8k_level": {
+ "from_version": "v0.80.0",
+ "no_negative_tests": "true",
+ "generator_args": [
+ [
+ "--target-dtype",
+ "fp32",
+ "--target-dtype",
+ "fp16",
+ "--target-dtype",
+ "bf16",
+ "--fp-values-range",
+ "-max,max",
+ "--tensor-dim-range",
+ "1,9",
+ "--target-rank",
+ "5",
+ "--target-rank",
+ "6"
+ ]
+ ]
}
},
"selection": {
@@ -2541,7 +2601,15 @@
"--fp-values-range",
"-max,max",
"--tensor-dim-range",
- "1,34"
+ "1,34",
+ "--target-rank",
+ "1",
+ "--target-rank",
+ "2",
+ "--target-rank",
+ "3",
+ "--target-rank",
+ "4"
],
[
"--target-dtype",
@@ -2554,6 +2622,28 @@
"1,3,2,65541"
]
]
+ },
+ "8k_level": {
+ "from_version": "v0.80.0",
+ "no_negative_tests": "true",
+ "generator_args": [
+ [
+ "--target-dtype",
+ "fp32",
+ "--target-dtype",
+ "fp16",
+ "--target-dtype",
+ "bf16",
+ "--fp-values-range",
+ "-max,max",
+ "--tensor-dim-range",
+ "1,11",
+ "--target-rank",
+ "5",
+ "--target-rank",
+ "6"
+ ]
+ ]
}
},
"selection": {
@@ -2590,7 +2680,15 @@
"--target-rank",
"4",
"--tensor-dim-range",
- "10,37"
+ "10,37",
+ "--target-rank",
+ "1",
+ "--target-rank",
+ "2",
+ "--target-rank",
+ "3",
+ "--target-rank",
+ "4"
],
[
"--target-dtype",
@@ -2627,6 +2725,28 @@
"1,65537,1,1"
]
]
+ },
+ "8k_level": {
+ "from_version": "v0.80.0",
+ "no_negative_tests": "true",
+ "generator_args": [
+ [
+ "--target-dtype",
+ "fp32",
+ "--target-dtype",
+ "fp16",
+ "--target-dtype",
+ "bf16",
+ "--fp-values-range",
+ "-max,max",
+ "--tensor-dim-range",
+ "5,12",
+ "--target-rank",
+ "5",
+ "--target-rank",
+ "6"
+ ]
+ ]
}
},
"selection": {