[Mlir-commits] [mlir] Fix llvm-mlir-use-after-erase findings (PR #210733)

[llvmlistbot at llvm.org]

================
@@ -301,14 +301,15 @@ void NormalizeMemRefs::updateFunctionSignature(func::FuncOp funcOp,
         // If it failed (due to escapes for example), bail out.
         // It should never hit this part of the code because it is called by
         // only those functions which are normalizable.
-        newCallOp->erase();
         replacingMemRefUsesFailed = true;
         break;
       }
       returnTypeChanged = true;
     }
-    if (replacingMemRefUsesFailed)
+    if (replacingMemRefUsesFailed) {
+      newCallOp->erase();
       continue;
+    }
----------------
Harald-R wrote:

I have reverted this change.

https://github.com/llvm/llvm-project/pull/210733