[mlir][linalg][bufferize] Fix bug in getInplaceableOpResult

matthias-springer · matthias-springer · commit 94213bc7decb · 2021-10-21T21:55:26.000+09:00
Differential Revision: https://reviews.llvm.org/D112123
diff --git a/mlir/lib/Dialect/Linalg/Transforms/ComprehensiveBufferize.cpp b/mlir/lib/Dialect/Linalg/Transforms/ComprehensiveBufferize.cpp
@@ -508,7 +508,7 @@ static OpResult getInplaceableOpResult(VectorTransferOpInterface op,
 /// when the op is bufferized inplace.
 /// Return null if no such result exists.
 static OpResult getInplaceableOpResult(InsertSliceOp op, OpOperand &opOperand) {
-  if (opOperand.get() != op.dest())
+  if (&opOperand != &op->getOpOperand(1) /*dest*/)
     return OpResult();
   return op->getResult(0);
 }

Original file line number	Diff line number	Diff line change
`@@ -508,7 +508,7 @@ static OpResult getInplaceableOpResult(VectorTransferOpInterface op,`
`508`	`508`	`/// when the op is bufferized inplace.`
`509`	`509`	`/// Return null if no such result exists.`
`510`	`510`	`static OpResult getInplaceableOpResult(InsertSliceOp op, OpOperand &opOperand) {`
`511`		`- if (opOperand.get() != op.dest())`
	`511`	`+ if (&opOperand != &op->getOpOperand(1) /dest/)`
`512`	`512`	`return OpResult();`
`513`	`513`	`return op->getResult(0);`
`514`	`514`	`}`