Make memref.assume_alignment op illegal in AIRToAIE (Xilinx#854)

nod-ai · Jan 11, 2025 · 646b67b · 646b67b
1 parent ad90c73
commit 646b67b
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/mlir/lib/Conversion/AIRToAIEPass.cpp b/mlir/lib/Conversion/AIRToAIEPass.cpp
@@ -3466,9 +3466,11 @@ class AIRToAIEPass : public air::impl::AIRToAIEBase<AIRToAIEPass> {
       RewritePatternSet removepatterns(ctx);
       removepatterns.add<OpRemovalPattern<memref::DeallocOp>,
                          OpRemovalPattern<air::WaitAllOp>,
-                         OpRemovalPattern<memref::CopyOp>>(ctx);
+                         OpRemovalPattern<memref::CopyOp>,
+                         OpRemovalPattern<memref::AssumeAlignmentOp>>(ctx);
       ConversionTarget target(*ctx);
-      target.addIllegalOp<memref::DeallocOp, air::WaitAllOp, memref::CopyOp>();
+      target.addIllegalOp<memref::DeallocOp, air::WaitAllOp, memref::CopyOp,
+                          memref::AssumeAlignmentOp>();
       if (failed(applyPartialConversion(device, target,
                                         std::move(removepatterns))))
         signalPassFailure();