Merge pull request #77715 from nate-chandler/general-coro/20241119/1

[CoroutineAccessors] Rewrite allocs in LoadableByAddress.

Author: nate-chandler

lib/IRGen/LoadableByAddress.cpp

@@ -2590,6 +2590,10 @@ void LoadableByAddress::recreateSingleApply(
     // Use the new token result.
     oldApply->getTokenResult()->replaceAllUsesWith(newApply->getTokenResult());
 
+    if (auto *allocation = oldApply->getCalleeAllocationResult()) {
+      allocation->replaceAllUsesWith(newApply->getCalleeAllocationResult());
+    }
+
     // Rewrite all the yields.
     auto oldYields = oldApply->getOrigCalleeType()->getYields();
     auto oldYieldedValues = oldApply->getYieldedValues();

test/IRGen/loadable_by_address.sil

@@ -0,0 +1,55 @@
+// RUN: %target-swift-frontend %s  -Xllvm -sil-print-after=loadable-address -import-objc-header %S/Inputs/large_c.h -c -o %t/t.o 2>&1 | %FileCheck %s
+
+sil_stage canonical
+
+import Builtin
+import Swift
+
+struct X {
+  var x1 : Int
+  var x2 : Int
+  var x3 : Int
+  var x4: Int
+  var x5: Int
+  var x6: Int
+  var x7: Int
+  var x8: Int
+  var x9: Int
+  var x10: Int
+  var x11: Int
+  var x12: Int
+  var x13: Int
+  var x14: Int
+  var x15: Int
+  var x16: Int
+}
+
+sil @yield_nothing_take_x : $@convention(thin) @yield_once_2 (@owned X) -> (@yields @in_guaranteed ()) {
+entry(%x : $X):
+  %void_addr = alloc_stack $()
+  %void = tuple ()
+  store %void to %void_addr : $*()
+  yield %void_addr : $*(), resume bb1, unwind bb2
+
+bb1:
+  dealloc_stack %void_addr : $*()
+  return %void : $()
+
+bb2:
+  dealloc_stack %void_addr : $*()
+  unwind
+}
+
+// CHECK-LABEL: sil @begin_apply_x_yielder : {{.*}} {
+// CHECK:         ({{%[^,]+}}, {{%[^,]+}}, [[ALLOCATION:%[^,]+]]) = begin_apply
+// CHECK:         dealloc_stack [[ALLOCATION]]
+// CHECK-LABEL: } // end sil function 'begin_apply_x_yielder'
+sil @begin_apply_x_yielder : $@convention(thin) (@owned X) -> () {
+entry(%x : $X):
+  %fn = function_ref @yield_nothing_take_x : $@convention(thin) @yield_once_2 (@owned X) -> (@yields @in_guaranteed ())
+  (%void, %token, %allocation) = begin_apply %fn(%x) : $@convention(thin) @yield_once_2 (@owned X) -> (@yields @in_guaranteed ())
+  end_apply %token as $()
+  dealloc_stack %allocation : $*Builtin.SILToken
+  %retval = tuple ()
+  return %retval : $()
+}