Fixup the location of creating constant op (Xilinx#915)

erwei-xilinx · web-flow · commit 42d516b06d75 · 2025-02-25T06:37:28.000Z
diff --git a/mlir/lib/Transform/AIRMiscPasses.cpp b/mlir/lib/Transform/AIRMiscPasses.cpp
@@ -1737,6 +1737,7 @@ void AIRSplitL2MemrefForBufferConstraintPass::runOnOperation() {
       SmallVector<Value> strides = theOtherChanOp[0].getStrides();
       if (wraps.empty()) {
         // Populate default wraps, if wraps is an empty vector.
+        rewriter.setInsertionPoint(theOtherChanOp[0]);
         air::populateDefaultWrapsAndStrides(
             rewriter, theOtherChanOp[0].getMemref(), offsets, wraps, strides);
       }

Original file line number	Diff line number	Diff line change
`@@ -1737,6 +1737,7 @@ void AIRSplitL2MemrefForBufferConstraintPass::runOnOperation() {`
`1737`	`1737`	`SmallVector<Value> strides = theOtherChanOp[0].getStrides();`
`1738`	`1738`	`if (wraps.empty()) {`
`1739`	`1739`	`// Populate default wraps, if wraps is an empty vector.`
	`1740`	`+ rewriter.setInsertionPoint(theOtherChanOp[0]);`
`1740`	`1741`	`air::populateDefaultWrapsAndStrides(`
`1741`	`1742`	`rewriter, theOtherChanOp[0].getMemref(), offsets, wraps, strides);`
`1742`	`1743`	`}`