Update MoveOnlyAddressCheckerUtils for mutate accessors

Author: meg-gupta

include/swift/SIL/SILInstruction.h

@@ -3162,6 +3162,8 @@ class ApplyInst final
   bool hasAddressResult() const {
     return getSubstCalleeConv().hasAddressResult();
   }
+
+  bool hasInoutResult() const { return getSubstCalleeConv().hasInoutResult(); }
 };
 
 /// PartialApplyInst - Represents the creation of a closure object by partial

lib/SILOptimizer/Mandatory/MoveOnlyAddressCheckerUtils.cpp

@@ -472,6 +472,18 @@ static bool visitScopeEndsRequiringInit(
     return true;
   }
 
+  // Check for mutate accessor.
+  if (auto ai =
+          dyn_cast_or_null<ApplyInst>(operand->getDefiningInstruction())) {
+    if (ai->hasInoutResult()) {
+      SmallVector<SILBasicBlock *, 8> exitBlocks;
+      markedAddr->getFunction()->findExitingBlocks(exitBlocks);
+      for (auto *block : exitBlocks) {
+        visit(block->getTerminator(), ScopeRequiringFinalInit::InoutArgument);
+      }
+      return true;
+    }
+  }
   return false;
 }