Merge pull request #35677 from atrick/add-phioperand

Add PhiOperand and PhiValue types.

include/swift/SIL/SILBasicBlock.h

@@ -488,4 +488,141 @@ private:
 
 } // end llvm namespace
 
+//===----------------------------------------------------------------------===//
+//                           PhiOperand & PhiValue
+//===----------------------------------------------------------------------===//
+
+namespace swift {
+
+/// Represent a phi argument without storing pointers to branches or their
+/// operands which are invalidated by adding new, unrelated phi values. Because
+/// this only stores a block pointer, it remains valid as long as the CFG is
+/// immutable and the index of the phi value does not change.
+///
+/// Note: this should not be confused with SILPhiArgument which should be
+/// renamed to SILPhiValue and only used for actual phis.
+///
+/// Warning: This is invalid for CondBranchInst arguments. Clients assume that
+/// any instructions inserted at the phi argument is post-dominated by that phi
+/// argument. This warning can be removed once the SILVerifier fully prohibits
+/// CondBranchInst arguments at all SIL stages.
+struct PhiOperand {
+  SILBasicBlock *predBlock = nullptr;
+  unsigned argIndex = 0;
+
+  PhiOperand() = default;
+
+  PhiOperand(Operand *operand) {
+    auto *branch = dyn_cast<BranchInst>(operand->getUser());
+    if (!branch)
+      return;
+
+    predBlock = branch->getParent();
+    argIndex = operand->getOperandNumber();
+  }
+
+  explicit operator bool() const { return predBlock != nullptr; }
+
+  bool operator==(PhiOperand other) const {
+    return predBlock == other.predBlock && argIndex == other.argIndex;
+  }
+
+  bool operator!=(PhiOperand other) const { return !(*this == other); }
+
+  BranchInst *getBranch() const {
+    return cast<BranchInst>(predBlock->getTerminator());
+  }
+
+  Operand *getOperand() const {
+    return &getBranch()->getAllOperands()[argIndex];
+  }
+
+  SILPhiArgument *getValue() const {
+    auto *branch = cast<BranchInst>(predBlock->getTerminator());
+    return cast<SILPhiArgument>(branch->getDestBB()->getArgument(argIndex));
+  }
+
+  SILValue getSource() const {
+    return getOperand()->get();
+  }
+
+  operator Operand *() const { return getOperand(); }
+  Operand *operator*() const { return getOperand(); }
+  Operand *operator->() const { return getOperand(); }
+};
+
+/// Represent a phi value without referencing the SILValue, which is invalidated
+/// by adding new, unrelated phi values. Because this only stores a block
+/// pointer, it remains valid as long as the CFG is immutable and the index of
+/// the phi value does not change.
+struct PhiValue {
+  SILBasicBlock *phiBlock = nullptr;
+  unsigned argIndex = 0;
+
+  PhiValue() = default;
+
+  PhiValue(SILValue value) {
+    auto *blockArg = dyn_cast<SILPhiArgument>(value);
+    if (!blockArg || !blockArg->isPhiArgument())
+      return;
+
+    phiBlock = blockArg->getParent();
+    argIndex = blockArg->getIndex();
+  }
+
+  explicit operator bool() const { return phiBlock != nullptr; }
+
+  bool operator==(PhiValue other) const {
+    return phiBlock == other.phiBlock && argIndex == other.argIndex;
+  }
+
+  bool operator!=(PhiValue other) const { return !(*this == other); }
+
+  SILPhiArgument *getValue() const {
+    return cast<SILPhiArgument>(phiBlock->getArgument(argIndex));
+  }
+
+  operator SILValue() const { return getValue(); }
+  SILValue operator*() const { return getValue(); }
+  SILValue operator->() const { return getValue(); }
+};
+
+} // namespace swift
+
+namespace llvm {
+
+template <> struct DenseMapInfo<swift::PhiOperand> {
+  static swift::PhiOperand getEmptyKey() { return swift::PhiOperand(); }
+  static swift::PhiOperand getTombstoneKey() {
+    swift::PhiOperand phiOper;
+    phiOper.predBlock =
+        llvm::DenseMapInfo<swift::SILBasicBlock *>::getTombstoneKey();
+    return phiOper;
+  }
+  static unsigned getHashValue(swift::PhiOperand phiOper) {
+    return llvm::hash_combine(phiOper.predBlock, phiOper.argIndex);
+  }
+  static bool isEqual(swift::PhiOperand lhs, swift::PhiOperand rhs) {
+    return lhs == rhs;
+  }
+};
+
+template <> struct DenseMapInfo<swift::PhiValue> {
+  static swift::PhiValue getEmptyKey() { return swift::PhiValue(); }
+  static swift::PhiValue getTombstoneKey() {
+    swift::PhiValue phiValue;
+    phiValue.phiBlock =
+        llvm::DenseMapInfo<swift::SILBasicBlock *>::getTombstoneKey();
+    return phiValue;
+  }
+  static unsigned getHashValue(swift::PhiValue phiValue) {
+    return llvm::hash_combine(phiValue.phiBlock, phiValue.argIndex);
+  }
+  static bool isEqual(swift::PhiValue lhs, swift::PhiValue rhs) {
+    return lhs == rhs;
+  }
+};
+
+} // end namespace llvm
+
 #endif

include/swift/SILOptimizer/Utils/OwnershipOptUtils.h

@@ -40,7 +40,7 @@ struct OwnershipFixupContext {
   JointPostDominanceSetComputer &jointPostDomSetComputer;
 
   SmallVector<Operand *, 8> transitiveBorrowedUses;
-  SmallVector<std::pair<SILBasicBlock *, unsigned>, 8> recursiveReborrows;
+  SmallVector<PhiOperand, 8> recursiveReborrows;
 
   /// Extra state initialized by OwnershipRAUWFixupHelper::get() that we use
   /// when RAUWing addresses. This ensures we do not need to recompute this

lib/SILOptimizer/Utils/OwnershipOptUtils.cpp

@@ -180,9 +180,7 @@ static bool canFixUpOwnershipForRAUW(SILValue oldValue, SILValue newValue,
     // TODO: if non-phi reborrows even exist, handle them using a separate
     // SILValue list since we don't want to refer directly to phi SILValues.
     reborrows.insert(borrowingOper.getBorrowIntroducingUserResult().value);
-    auto *branch = endScope->getUser();
-    context.recursiveReborrows.push_back(
-      {branch->getParent(), endScope->getOperandNumber()});
+    context.recursiveReborrows.push_back(endScope);
   };
   if (!findTransitiveGuaranteedUses(oldValue, context.transitiveBorrowedUses,
                                     visitReborrow))
@@ -452,7 +450,7 @@ OwnershipLifetimeExtender::createPlusZeroBorrow(SILValue newValue,
 //===----------------------------------------------------------------------===//
 
 static void eliminateReborrowsOfRecursiveBorrows(
-    ArrayRef<std::pair<SILBasicBlock *, unsigned>> transitiveReborrows,
+    ArrayRef<PhiOperand> transitiveReborrows,
     SmallVectorImpl<Operand *> &usePoints, InstModCallbacks &callbacks) {
   SmallVector<std::pair<SILPhiArgument *, SILPhiArgument *>, 8>
       baseBorrowedValuePair;
@@ -462,8 +460,8 @@ static void eliminateReborrowsOfRecursiveBorrows(
     // edge from the base value and using that for the reborrow instead of the
     // actual value. We of course insert an end_borrow for our original incoming
     // value.
-    auto *bi = cast<BranchInst>(it.first->getTerminator());
-    auto &op = bi->getOperandRef(it.second);
+    auto *bi = cast<BranchInst>(it.predBlock->getTerminator());
+    auto &op = bi->getOperandRef(it.argIndex);
     BorrowingOperand borrowingOperand(&op);
     SILValue value = borrowingOperand->get();
     SILBuilderWithScope reborrowBuilder(bi);
@@ -1288,9 +1286,7 @@ OwnershipReplaceSingleUseHelper::OwnershipReplaceSingleUseHelper(
     if (reborrowOperand.isReborrow()) {
       // Check that the old lifetime can be extended and record the necessary
       // book-keeping in the OwnershipFixupContext.
-      auto *branch = reborrowOperand->getUser();
-      ctx->recursiveReborrows.push_back(
-          {branch->getParent(), reborrowOperand->getOperandNumber()});
+      ctx->recursiveReborrows.push_back(use);
     }
   }
 }