Swift SIL: add a few ownership APIs to ForwardingInstruction

* `canForwardGuaranteedValues`
* `canForwardOwnedValues`
* `setForwardingOwnership`
* `Operand.canAccept(ownership:)`

Author: eeckstein

SwiftCompilerSources/Sources/Optimizer/PassManager/Context.swift

@@ -526,6 +526,13 @@ extension TermInst {
   }
 }
 
+extension ForwardingInstruction {
+  func setForwardingOwnership(to ownership: Ownership, _ context: some MutatingContext) {
+    context.notifyInstructionsChanged()
+    bridged.ForwardingInst_setForwardingOwnership(ownership._bridged)
+  }
+}
+
 extension Function {
   func set(needStackProtection: Bool, _ context: FunctionPassContext) {
     context.notifyEffectsChanged()

SwiftCompilerSources/Sources/SIL/ForwardingInstruction.swift

@@ -18,6 +18,10 @@ public protocol ForwardingInstruction : Instruction {
 
   /// Return true if the forwarded value has the same representation. If true, then the result can be mapped to the same storage without a move or copy.
   var preservesRepresentation: Bool { get }
+
+  var canForwardGuaranteedValues: Bool { get }
+
+  var canForwardOwnedValues: Bool { get }
 }
 
 extension ForwardingInstruction {
@@ -47,6 +51,14 @@ extension ForwardingInstruction {
   public var preservesReferenceCounts: Bool {
     bridged.ForwardingInst_preservesOwnership()
   }
+
+  // Most forwarding instructions can forward owned and guaranteed values.
+  // Therefore define a default implementation of `true`.
+  public var canForwardGuaranteedValues: Bool { true }
+
+  // Most forwarding instructions can forward owned and guaranteed values.
+  // Therefore define a default implementation of `true`.
+  public var canForwardOwnedValues: Bool { true }
 }
 
 // An instruction that forwards a single value to a single result.
@@ -242,12 +254,15 @@ extension MarkUninitializedInst {
 
 extension StructExtractInst {
   public var preservesRepresentation: Bool { true }
+  public var canForwardOwnedValues: Bool { false }
 }
 
 extension TupleExtractInst {
   public var preservesRepresentation: Bool { true }
+  public var canForwardOwnedValues: Bool { false }
 }
 
 extension TuplePackExtractInst {
   public var preservesRepresentation: Bool { true }
+  public var canForwardOwnedValues: Bool { false }
 }

SwiftCompilerSources/Sources/SIL/Operand.swift

@@ -42,7 +42,9 @@ public struct Operand : CustomStringConvertible, NoReflectionChildren {
   public var isTypeDependent: Bool { bridged.isTypeDependent() }
 
   public var endsLifetime: Bool { bridged.isLifetimeEnding() }
-  
+
+  public func canAccept(ownership: Ownership) -> Bool { bridged.canAcceptOwnership(ownership._bridged) }
+
   public var description: String { "operand #\(index) of \(instruction)" }
 }
 

include/swift/SIL/SILBridging.h

@@ -215,6 +215,7 @@ struct BridgedOperand {
 
   BRIDGED_INLINE bool isTypeDependent() const;
   BRIDGED_INLINE bool isLifetimeEnding() const;
+  BRIDGED_INLINE bool canAcceptOwnership(BridgedValue::Ownership ownership) const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE OptionalBridgedOperand getNextUse() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedValue getValue() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedInstruction getUser() const;
@@ -537,6 +538,7 @@ struct BridgedInstruction {
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE OptionalBridgedOperand ForwardingInst_singleForwardedOperand() const;
   SWIFT_IMPORT_UNSAFE BRIDGED_INLINE BridgedOperandArray ForwardingInst_forwardedOperands() const;
   BRIDGED_INLINE BridgedValue::Ownership ForwardingInst_forwardingOwnership() const;
+  BRIDGED_INLINE void ForwardingInst_setForwardingOwnership(BridgedValue::Ownership ownership) const;
   BRIDGED_INLINE bool ForwardingInst_preservesOwnership() const;
 
   // =========================================================================//

include/swift/SIL/SILBridgingImpl.h

@@ -269,6 +269,10 @@ bool BridgedOperand::isTypeDependent() const { return op->isTypeDependent(); }
 
 bool BridgedOperand::isLifetimeEnding() const { return op->isLifetimeEnding(); }
 
+bool BridgedOperand::canAcceptOwnership(BridgedValue::Ownership ownership) const {
+  return op->canAcceptKind(BridgedValue::castToOwnership(ownership));
+}
+
 OptionalBridgedOperand BridgedOperand::getNextUse() const {
   return {op->getNextUse()};
 }
@@ -683,6 +687,11 @@ BridgedValue::Ownership BridgedInstruction::ForwardingInst_forwardingOwnership()
   return castOwnership(forwardingInst->getForwardingOwnershipKind());
 }
 
+void BridgedInstruction::ForwardingInst_setForwardingOwnership(BridgedValue::Ownership ownership) const {
+  auto *forwardingInst = swift::ForwardingInstruction::get(unbridged());
+  return forwardingInst->setForwardingOwnershipKind(BridgedValue::castToOwnership(ownership));
+}
+
 bool BridgedInstruction::ForwardingInst_preservesOwnership() const {
   return swift::ForwardingInstruction::get(unbridged())->preservesOwnership();
 }