From 67e9fc539dd014745f8e2559b967489b8479a8f8 Mon Sep 17 00:00:00 2001
From: Luke Hutton <luke.hutton@arm.com>
Date: Thu, 12 Jan 2023 11:30:33 +0000
Subject: Support translation of FFT2d

Signed-off-by: Luke Hutton <luke.hutton@arm.com>
Change-Id: I4aae94438380d394b9c13015aa69ac52f9b73f74
---
 src/TosaSerialize.cpp | 22 ++++++++++++++++++++++
 1 file changed, 22 insertions(+)

(limited to 'src/TosaSerialize.cpp')

diff --git a/src/TosaSerialize.cpp b/src/TosaSerialize.cpp
index e69fcba..8a95b68 100644
--- a/src/TosaSerialize.cpp
+++ b/src/TosaSerialize.cpp
@@ -1367,6 +1367,28 @@ TosaSerializationOperatorBuilder::build<mlir::tosa::RFFT2dOp>(
   return tyop;
 }
 
+template<>
+TosaSerializationOperator *
+TosaSerializationOperatorBuilder::build<mlir::tosa::FFT2dOp>(
+    mlir::Operation &op) const {
+
+  bool inverse = op.getAttr("inverse").dyn_cast<mlir::BoolAttr>().getValue();
+
+  std::string input_real_name = GetTensorName(op.getOperand(0));
+  std::string input_imag_name = GetTensorName(op.getOperand(1));
+  std::string output_real_name = GetTensorName(op.getResult(0));
+  std::string output_imag_name = GetTensorName(op.getResult(1));
+
+  TosaFFTAttribute attribute(inverse);
+
+  TosaSerializationOperator *tyop = new TosaSerializationOperator(
+    Op_FFT2D, Attribute_FFTAttribute, &attribute,
+    std::vector<std::string>{input_real_name, input_imag_name},
+    std::vector<std::string>{output_real_name, output_imag_name});
+
+  return tyop;
+}
+
 /* End translating TOSA operator */
 mlir::LogicalResult TosaSerializationRegionBuilder::BuildAllBlocksInRegion(std::vector<mlir::Value>& return_values) {
   std::string region_name = ser_region->GetName();
-- 
cgit v1.2.1