[mlir][IR][tests] Fix incorrect API usage in RewritePatterns

[springerm]

Summary

Incorrect API usage was detected by D144552.

mlir/test/lib/Dialect/Test/TestPatterns.cpp

struct HoistEligibleOps : public OpRewritePattern<test::OneRegionOp> {
   LogicalResult matchAndRewrite(test::OneRegionOp op,
                                 PatternRewriter &rewriter) const override {
     Operation *terminator = op.getRegion().front().getTerminator();
     Operation *toBeHoisted = terminator->getOperands()[0].getDefiningOp();
     if (toBeHoisted->getParentOp() != op)
       return failure();
     if (!toBeHoisted->hasAttr("eligible"))
       return failure();
-    toBeHoisted->moveBefore(op);
+    // Hoisting means removing an op from the enclosing op. I.e., the enclosing
+    // op is modified.
+    rewriter.updateRootInPlace(op, [&]() { toBeHoisted->moveBefore(op); });
     return success();
   }
 };
 
 struct TestPatternDriver
     : public PassWrapper<TestPatternDriver, OperationPass<func::FuncOp>> {
   MLIR_DEFINE_EXPLICIT_INTERNAL_INLINE_TYPE_ID(TestPatternDriver)
 
public:
     LogicalResult matchAndRewrite(Operation *op,
                                   PatternRewriter &rewriter) const override {
       if (op->hasAttr("skip"))
         return failure();
 
       Operation *newOp =
           rewriter.create(op->getLoc(), op->getName().getIdentifier(),
                           op->getOperands(), op->getResultTypes());
-      op->setAttr("skip", rewriter.getBoolAttr(true));
+      rewriter.updateRootInPlace(
+          op, [&]() { op->setAttr("skip", rewriter.getBoolAttr(true)); });
       newOp->setAttr("skip", rewriter.getBoolAttr(true));
 
       return success();
     }
   };
 
   // Replace an operation may introduce the re-visiting of its users.
   class ReplaceWithNewOp : public RewritePattern {