Merge pull request #73270 from aschwaighofer/large_types_reg2mem_yield_begin_apply

LargeTypesReg2Mem: Fix yield and begin_apply instructions

Author: aschwaighofer

lib/IRGen/LoadableByAddress.cpp

@@ -3612,6 +3612,23 @@ class AssignAddressToDef : SILInstructionVisitor<AssignAddressToDef> {
     singleValueInstructionFallback(kp);
   }
 
+  void visitBeginApplyInst(BeginApplyInst *apply) {
+    auto builder = assignment.getBuilder(++apply->getIterator());
+    auto addr = assignment.createAllocStack(origValue->getType());
+    assignment.mapValueToAddress(origValue, addr);
+    for (auto &opd : apply->getAllOperands()) {
+      if (assignment.contains(opd.get())) {
+        auto builder = assignment.getBuilder(apply->getIterator());
+        auto loaded = builder.createLoad(
+            apply->getLoc(), assignment.getAddressForValue(opd.get()),
+            LoadOwnershipQualifier::Unqualified);
+        opd.set(loaded);
+      }
+    }
+    builder.createStore(apply->getLoc(), origValue, addr,
+                        StoreOwnershipQualifier::Unqualified);
+  }
+
   void visitApplyInst(ApplyInst *apply) {
     // The loadable by address transformation ignores large tuple return types.
     auto builder = assignment.getBuilder(++apply->getIterator());
@@ -3865,6 +3882,8 @@ class RewriteUser : SILInstructionVisitor<RewriteUser> {
     userInstructionFallback(kp);
   }
 
+  void visitYieldInst(YieldInst *yield) { userInstructionFallback(yield); }
+
   void visitFixLifetimeInst(FixLifetimeInst *f) {
     auto addr = assignment.getAddressForValue(f->getOperand());
     auto builder = assignment.getBuilder(f->getIterator());