Handle borrow uses and interior pointer uses in swift::findPointerEscape

Author: meg-gupta

lib/SIL/Utils/OwnershipUtils.cpp

@@ -56,6 +56,13 @@ bool swift::findPointerEscape(SILValue original) {
         worklist.pushIfNotVisited(phi);
         break;
       }
+      case OperandOwnership::Borrow: {
+        auto borrowOp = BorrowingOperand(use);
+        if (auto borrowValue = borrowOp.getBorrowIntroducingUserResult()) {
+          worklist.pushIfNotVisited(borrowValue.value);
+        }
+        break;
+      }
       case OperandOwnership::Reborrow: {
         SILArgument *phi = PhiOperand(use).getValue();
         worklist.pushIfNotVisited(phi);
@@ -72,6 +79,13 @@ bool swift::findPointerEscape(SILValue original) {
         });
         break;
       }
+      case OperandOwnership::InteriorPointer: {
+        if (InteriorPointerOperand(use).findTransitiveUses() !=
+            AddressUseKind::NonEscaping) {
+          return true;
+        }
+        break;
+      }
       default:
         break;
       }