Merge pull request #36141 from gottesmm/pr-fa4d423549d67732a05efc03383b0bcc8f15bd99

[sil-combine] Allow for storage casts to be cloned when rebasing an address def-use chain onto a new interior pointer

Author: gottesmm

lib/SILOptimizer/Utils/OwnershipOptUtils.cpp

@@ -848,8 +848,8 @@ class InteriorPointerAddressRebaseUseDefChainCloner
   }
 
   SILValue visitStorageCast(SingleValueInstruction *cast, Operand *sourceOper) {
-    assert(false && "unexpected storage cast on access path");
-    return SILValue();
+    // We are ok with cloning storage casts.
+    return cloneProjection(cast, sourceOper);
   }
 
   SILValue visitAccessProjection(SingleValueInstruction *projectedAddr,

test/SILOptimizer/sil_combine_ossa.sil

@@ -4953,3 +4953,35 @@ bb0(%0 :  $*Klass, %1 : @owned $Klass):
   return %9999 : $()
 }
 
+class KlassWithTailAllocatedElems {
+  var x : Builtin.NativeObject
+  init()
+}
+
+// CHECK-LABEL: sil [ossa] @unchecked_addr_cast_formation_handling_interior_pointer_rebase_of_cast : $@convention(thin) () -> Builtin.Word {
+// CHECK-NOT: pointer_to_address
+// CHECK-NOT: address_to_pointer
+// CHECK: [[TAIL_ADDR:%.*]] = ref_tail_addr
+// CHECK-NOT: pointer_to_address
+// CHECK-NOT: address_to_pointer
+// CHECK: [[CAST_RESULT:%.*]] = unchecked_addr_cast [[TAIL_ADDR]]
+// CHECK-NOT: pointer_to_address
+// CHECK-NOT: address_to_pointer
+// CHECK: load [trivial] [[CAST_RESULT]]
+// CHECK-NOT: pointer_to_address
+// CHECK-NOT: address_to_pointer
+// CHECK: } // end sil function 'unchecked_addr_cast_formation_handling_interior_pointer_rebase_of_cast'
+sil [ossa] @unchecked_addr_cast_formation_handling_interior_pointer_rebase_of_cast : $@convention(thin) () -> Builtin.Word {
+bb0:
+  %7 = integer_literal $Builtin.Word, 1
+  %8 = alloc_ref [tail_elems $(Builtin.NativeObject) * %7 : $Builtin.Word] $KlassWithTailAllocatedElems
+  %borrow = begin_borrow %8 : $KlassWithTailAllocatedElems
+  %addr = ref_tail_addr %borrow : $KlassWithTailAllocatedElems, $Builtin.NativeObject
+  %1 = address_to_pointer %addr : $*Builtin.NativeObject to $Builtin.RawPointer
+  end_borrow %borrow : $KlassWithTailAllocatedElems
+  %2 = pointer_to_address %1 : $Builtin.RawPointer to [strict] $*Builtin.Word
+  %3 = load [trivial] %2 : $*Builtin.Word
+  destroy_value %8 : $KlassWithTailAllocatedElems
+  return %3 : $Builtin.Word
+}
+