[Transform][Fusion] fix getClosestExtractSliceOfOperand when operand is block argument of function (#304)

Yun-Fly · web-flow · commit 686c34c2da17 · 2024-08-30T12:36:11.000+08:00
diff --git a/lib/gc/Transforms/IterativeTilingAndFusion.cpp b/lib/gc/Transforms/IterativeTilingAndFusion.cpp
@@ -46,6 +46,8 @@ getClosestExtractSliceOfOperand(OpOperand &operand) {
     if (auto loop =
             dyn_cast<LoopLikeOpInterface>(iterArg.getOwner()->getParentOp()))
       return getClosestExtractSliceOfOperand(*loop.getTiedLoopInit(iterArg));
+    // If operand is not using loop init.
+    return failure();
   }
 
   Operation *defineOp = operand.get().getDefiningOp();

Original file line number	Diff line number	Diff line change
`@@ -46,6 +46,8 @@ getClosestExtractSliceOfOperand(OpOperand &operand) {`
`46`	`46`	`if (auto loop =`
`47`	`47`	`dyn_cast<LoopLikeOpInterface>(iterArg.getOwner()->getParentOp()))`
`48`	`48`	`return getClosestExtractSliceOfOperand(*loop.getTiedLoopInit(iterArg));`
	`49`	`+ // If operand is not using loop init.`
	`50`	`+ return failure();`
`49`	`51`	`}`
`50`	`52`
`51`	`53`	`Operation *defineOp = operand.get().getDefiningOp();`