Disable loop fusion for now. To revise later if we need to drop it

Xilinx · Jan 24, 2025 · d92fbc6 · d92fbc6
1 parent 386675b
commit d92fbc6
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/mlir/lib/Transform/AIRDependencyScheduleOpt.cpp b/mlir/lib/Transform/AIRDependencyScheduleOpt.cpp
@@ -5603,7 +5603,6 @@ class AIROptimizeShimDMABDs
 
   void runOnOperation() override {
     auto func = getOperation();
-    MLIRContext *ctx = &getContext();
     auto device = AIE::symbolizeAIEDevice(clDevice);
     if (!device) {
       func.emitOpError("Invalid aie.device option");
@@ -5618,9 +5617,11 @@ class AIROptimizeShimDMABDs
     air::applyAIRSpecializeChannelWrapAndStridePattern(
         &func.getRegion(),
         /*maxNumDims*/ maxNumDims, /*enableForLoopUnrolling*/ false);
-    RewritePatternSet patterns(ctx);
-    populateAIRLoopFusionPattern(patterns);
-    (void)applyPatternsGreedily(func, std::move(patterns));
+
+    // TODO: Loop fusion is commented out below. To revise later if we should
+    // drop loop fusion. MLIRContext *ctx = &getContext(); RewritePatternSet
+    // patterns(ctx); populateAIRLoopFusionPattern(patterns);
+    // (void)applyPatternsGreedily(func, std::move(patterns));
   }
 
 private: