Create MI tests for Tensor: FFT2D & RFFT2D
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I44ad6513fad831ff50fcc5ab801b46a5bf25d0a8
diff --git a/verif/conformance/tosa_main_profile_ops_info.json b/verif/conformance/tosa_main_profile_ops_info.json
index dc4e45b..b2cdf37 100644
--- a/verif/conformance/tosa_main_profile_ops_info.json
+++ b/verif/conformance/tosa_main_profile_ops_info.json
@@ -788,6 +788,36 @@
"tosa-mi"
]
},
+ "fft2d": {
+ "group": "tensor",
+ "generator_args": [
+ [
+ "--target-dtype",
+ "fp32",
+ "--fp-values-range",
+ "-2.0,2.0"
+ ],
+ [
+ "--target-dtype",
+ "fp32",
+ "--fp-values-range",
+ "-2.0,2.0",
+ "--target-shape",
+ "1,256,64",
+ "--target-shape",
+ "1,4,4096"
+ ]
+ ],
+ "params": {},
+ "permutes": [
+ "shape",
+ "type",
+ "inverse"
+ ],
+ "profile": [
+ "tosa-mi"
+ ]
+ },
"fully_connected": {
"group": "tensor",
"generator_negative_dim_range": "1,10",
@@ -2121,6 +2151,35 @@
"tosa-mi"
]
},
+ "rfft2d": {
+ "group": "tensor",
+ "generator_args": [
+ [
+ "--target-dtype",
+ "fp32",
+ "--fp-values-range",
+ "-2.0,2.0"
+ ],
+ [
+ "--target-dtype",
+ "fp32",
+ "--fp-values-range",
+ "-2.0,2.0",
+ "--target-shape",
+ "1,16,512",
+ "--target-shape",
+ "1,8192,2"
+ ]
+ ],
+ "params": {},
+ "permutes": [
+ "shape",
+ "type"
+ ],
+ "profile": [
+ "tosa-mi"
+ ]
+ },
"select": {
"group": "ew_ternary",
"generator_args": [