Merge branch 'main' into bump-swift-version-to-6

Author: shahmishal

SwiftCompilerSources/Sources/Optimizer/FunctionPasses/LifetimeDependenceDiagnostics.swift

@@ -375,6 +375,10 @@ extension DiagnoseDependenceWalker : LifetimeDependenceDefUseWalker {
     return diagnostics.checkFunctionResult(operand: operand)
   }
 
+  mutating func yieldedDependence(result: Operand) -> WalkResult {
+    return diagnostics.checkFunctionResult(operand: result)
+  }
+
   // Override AddressUseVisitor here because LifetimeDependenceDefUseWalker
   // returns .abortWalk, and we want a more useful crash report.
   mutating func unknownAddressUse(of operand: Operand) -> WalkResult {

SwiftCompilerSources/Sources/Optimizer/FunctionPasses/LifetimeDependenceScopeFixup.swift

@@ -168,5 +168,9 @@ private struct LifetimeDependenceScopeFixupWalker : LifetimeDependenceDefUseWalk
                                    using operand: Operand) -> WalkResult {
     return .continueWalk
   }
+
+  mutating func yieldedDependence(result: Operand) -> WalkResult {
+    return .continueWalk
+  }
 }
 

SwiftCompilerSources/Sources/Optimizer/Utilities/LifetimeDependenceUtils.swift

@@ -780,7 +780,7 @@ extension LifetimeDependenceUseDefWalker {
 ///   escapingDependence(on operand: Operand) -> WalkResult
 ///   returnedDependence(result: Operand) -> WalkResult
 ///   returnedDependence(address: FunctionArgument, using: Operand) -> WalkResult
-///
+///   yieldedDependence(result: Operand) -> WalkResult
 /// Start walking:
 ///   walkDown(root: Value)
 protocol LifetimeDependenceDefUseWalker : ForwardingDefUseWalker,
@@ -796,6 +796,8 @@ protocol LifetimeDependenceDefUseWalker : ForwardingDefUseWalker,
 
   mutating func returnedDependence(address: FunctionArgument, using: Operand)
     -> WalkResult
+
+  mutating func yieldedDependence(result: Operand) -> WalkResult
 }
 
 extension LifetimeDependenceDefUseWalker {
@@ -841,6 +843,9 @@ extension LifetimeDependenceDefUseWalker {
     if operand.instruction is ReturnInst, !operand.value.type.isEscapable {
       return returnedDependence(result: operand)
     }
+    if operand.instruction is YieldInst, !operand.value.type.isEscapable {
+      return yieldedDependence(result: operand)
+    }
     return escapingDependence(on: operand)
   }
 }
@@ -1009,6 +1014,9 @@ extension LifetimeDependenceDefUseWalker {
     if operand.instruction is ReturnInst, !operand.value.type.isEscapable {
       return returnedDependence(result: operand)
     }
+    if operand.instruction is YieldInst, !operand.value.type.isEscapable {
+      return yieldedDependence(result: operand)
+    }
     return escapingDependence(on: operand)
   }
 
@@ -1208,6 +1216,11 @@ private struct LifetimeDependenceUsePrinter : LifetimeDependenceDefUseWalker {
     print("Returned use: \(operand) in: \(address)")
     return .continueWalk
   }
+
+  mutating func yieldedDependence(result: Operand) -> WalkResult {
+    print("Yielded use: \(result)")
+    return .continueWalk
+  }
 }
 
 let lifetimeDependenceUseTest = FunctionTest("lifetime_dependence_use") {