diff options
author | Jeremy Johnson <jeremy.johnson@arm.com> | 2023-02-14 11:47:46 +0000 |
---|---|---|
committer | Eric Kunze <eric.kunze@arm.com> | 2023-02-22 16:27:57 +0000 |
commit | c5d759328fbfe9772fba56e63c81d196b3a7620f (patch) | |
tree | abd2e1941477605dbdc83c23e0f504242909edf8 | |
parent | ae7a81c1734a6853348afa2ae59298e5397fd769 (diff) | |
download | reference_model-c5d759328fbfe9772fba56e63c81d196b3a7620f.tar.gz |
Create MI tests for Tensor: FFT2D & RFFT2D
Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I44ad6513fad831ff50fcc5ab801b46a5bf25d0a8
-rw-r--r-- | verif/conformance/test_select.py | 13 | ||||
-rw-r--r-- | verif/conformance/tosa_main_profile_ops_info.json | 59 |
2 files changed, 72 insertions, 0 deletions
diff --git a/verif/conformance/test_select.py b/verif/conformance/test_select.py index 061a109..72a4e84 100644 --- a/verif/conformance/test_select.py +++ b/verif/conformance/test_select.py @@ -496,6 +496,13 @@ class ExpOperator(Operator): name = "exp" +class FFT2DOperator(Operator): + """Test selector for the FFT2D operator.""" + + name = "fft2d" + param_names = ["shape", "type", "inverse"] + + class FloorOperator(Operator): """Test selector for the FLOOR operator.""" @@ -729,6 +736,12 @@ class ReverseOperator(Operator): param_names = ["shape", "type", "axis"] +class RFFT2DOperator(Operator): + """Test selector for the RFFT2D operator.""" + + name = "rfft2d" + + class RsqrtOperator(Operator): """Test selector for the RSQRT operator.""" 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": [ |