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

Author: meg-gupta

lib/SIL/Utils/OwnershipUtils.cpp

@@ -56,6 +56,11 @@ bool swift::findPointerEscape(SILValue original) {
         worklist.pushIfNotVisited(phi);
         break;
       }
+      case OperandOwnership::Borrow: {
+        auto *borrow = cast<SingleValueInstruction>(use->getUser());
+        worklist.pushIfNotVisited(borrow);
+        break;
+      }
       case OperandOwnership::Reborrow: {
         SILArgument *phi = PhiOperand(use).getValue();
         worklist.pushIfNotVisited(phi);
@@ -72,6 +77,13 @@ bool swift::findPointerEscape(SILValue original) {
         });
         break;
       }
+      case OperandOwnership::InteriorPointer: {
+        if (InteriorPointerOperand(use).findTransitiveUses() !=
+            AddressUseKind::NonEscaping) {
+          return true;
+        }
+        break;
+      }
       default:
         break;
       }