Merge pull request #71784 from jckarter/move-only-wrapped-yield

SILGen: Unwrap `@moveOnly` wrapped values before yielding them.

Author: jckarter

lib/SILGen/SILGenApply.cpp

@@ -5992,8 +5992,18 @@ void SILGenFunction::emitRawYield(SILLocation loc,
                                   JumpDest unwindDest,
                                   bool isUniqueYield) {
   SmallVector<SILValue, 4> yieldValues;
-  for (auto arg : yieldArgs)
-    yieldValues.push_back(arg.getValue());
+  for (auto arg : yieldArgs) {
+    auto value = arg.getValue();
+    if (value->getType().isMoveOnlyWrapped()) {
+      if (value->getType().isAddress()) {
+        value = B.createMoveOnlyWrapperToCopyableAddr(loc, value);
+      } else {
+        value = B.createGuaranteedMoveOnlyWrapperToCopyableValue(loc, value);
+      }
+    }
+    
+    yieldValues.push_back(value);
+  }
 
   // The normal continuation block.
   auto resumeBB = createBasicBlock();

test/SILGen/yield_borrowing.swift

@@ -0,0 +1,20 @@
+// RUN: %target-swift-emit-silgen -verify %s
+struct Foo {
+    var x: String
+
+    var y: Foo {
+        borrowing _read {
+            yield self
+        }
+    }
+}
+
+struct BigFoo {
+    var x: Any
+
+    var y: BigFoo {
+        borrowing _read {
+            yield self
+        }
+    }
+}

test/SILOptimizer/moveonly_borrowing_switch.swift

@@ -300,10 +300,6 @@ extension List {
         }
     }
 
-/*
-    TODO: type mismatch because of `@moveOnly` wrapper. yield needs to peel it
-    off
-
     var head: Element {
         _read {
             switch self {
@@ -314,7 +310,6 @@ extension List {
             }
         }
     }
-*/
 }
 
 extension ChonkyList {
@@ -336,10 +331,6 @@ extension ChonkyList {
         }
     }
 
-/*
-    TODO: type mismatch because of `@moveOnly` wrapper. yield needs to peel it
-    off
-
     var head: Element {
         _read {
             switch self {
@@ -350,5 +341,4 @@ extension ChonkyList {
             }
         }
     }
-*/
 }