[embedded] Use load_borrow/end_borrow in KeyPathProjector when in OSSA

Author: kubamracek

lib/SILOptimizer/Utils/KeyPathProjector.cpp

@@ -120,8 +120,13 @@ class StoredPropertyProjector : public ComponentProjector {
     } else {
       // Accessing a class member -> reading the class
       parent->project(AccessType::Get, [&](SILValue parentValue) {
-        SingleValueInstruction *Ref = builder.createLoad(loc, parentValue,
-                                                         LoadOwnershipQualifier::Unqualified);
+        SingleValueInstruction *Borrow = nullptr;
+        SingleValueInstruction *Ref;
+        if (builder.hasOwnership()) {
+          Ref = Borrow = builder.createLoadBorrow(loc, parentValue);
+        } else {
+          Ref = builder.createLoad(loc, parentValue, LoadOwnershipQualifier::Unqualified);
+        }
 
         // If we were previously accessing a class member, we're done now.
         insertEndAccess(beginAccess, builder);
@@ -135,6 +140,9 @@ class StoredPropertyProjector : public ComponentProjector {
             // decl or in a superclass of it. Just handle this to be on the safe
             // side.
             callback(SILValue());
+            if (Borrow) {
+              builder.createEndBorrow(loc, Borrow);
+            }
             return;
           }
           Ref = builder.createUpcast(loc, Ref, superCl);
@@ -163,6 +171,10 @@ class StoredPropertyProjector : public ComponentProjector {
         if (beginAccess == addr) {
           insertEndAccess(beginAccess, builder);
         }
+        
+        if (Borrow) {
+          builder.createEndBorrow(loc, Borrow);
+        }
       });
     }
   }