Guard rewriter insertion point during signature conversion.
authorChristian Sigg <csigg@google.com>
Wed, 9 Oct 2019 18:32:54 +0000 (11:32 -0700)
committerA. Unique TensorFlower <gardener@tensorflow.org>
Wed, 9 Oct 2019 18:33:28 +0000 (11:33 -0700)
Avoid unexpected side effect in rewriter insertion point.

PiperOrigin-RevId: 273785794

mlir/lib/Transforms/DialectConversion.cpp

index a47f17b..d65eae7 100644 (file)
@@ -284,6 +284,7 @@ void ArgConverter::applySignatureConversion(
   // Remap each of the original arguments as determined by the signature
   // conversion.
   auto &newArgMapping = argMapping[block];
+  OpBuilder::InsertionGuard guard(rewriter);
   rewriter.setInsertionPointToStart(block);
   for (unsigned i = 0; i != origArgCount; ++i) {
     ArrayRef<Value *> remappedValues;