[Mlir-commits] [mlir] [TOSA] FFT2D operator (PR #77005)

Jakub Kuderski llvmlistbot at llvm.org
Tue Jan 9 19:50:37 PST 2024


================
@@ -2344,6 +2344,134 @@ struct RFFT2dConverter final : public OpRewritePattern<RFFT2dOp> {
   }
 };
 
+struct FFT2dConverter final : public OpRewritePattern<FFT2dOp> {
+  using OpRewritePattern<FFT2dOp>::OpRewritePattern;
+
+  LogicalResult matchAndRewrite(FFT2dOp fft2d,
+                                PatternRewriter &rewriter) const override {
+    if (!llvm::all_of(fft2d->getOperandTypes(),
+                      RFFT2dConverter::isRankedTensor) ||
+        !llvm::all_of(fft2d->getResultTypes(),
+                      RFFT2dConverter::isRankedTensor)) {
+      return rewriter.notifyMatchFailure(fft2d, "only supports ranked tensors");
+    }
+
+    auto loc = fft2d.getLoc();
----------------
kuhar wrote:

```suggestion
    Location loc = fft2d.getLoc();
```

https://github.com/llvm/llvm-project/pull/77005


More information about the Mlir-commits mailing list