From c5d759328fbfe9772fba56e63c81d196b3a7620f Mon Sep 17 00:00:00 2001
From: Jeremy Johnson <jeremy.johnson@arm.com>
Date: Tue, 14 Feb 2023 11:47:46 +0000
Subject: Create MI tests for Tensor: FFT2D & RFFT2D

Signed-off-by: Jeremy Johnson <jeremy.johnson@arm.com>
Change-Id: I44ad6513fad831ff50fcc5ab801b46a5bf25d0a8
---
 verif/conformance/test_select.py                  | 13 +++++
 verif/conformance/tosa_main_profile_ops_info.json | 59 +++++++++++++++++++++++
 2 files changed, 72 insertions(+)

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": [
-- 
cgit v1.2.1