[Test] Add for PAI::visitOnStackLifetimeEnds.

Author: nate-chandler

lib/SIL/IR/SILInstruction.cpp

@@ -1904,6 +1904,22 @@ PartialApplyInst::visitOnStackLifetimeEnds(
   return !noUsers;
 }
 
+namespace swift::test {
+FunctionTest PartialApplyPrintOnStackLifetimeEnds(
+    "partial_apply_print_on_stack_lifetime_ends",
+    [](auto &function, auto &arguments, auto &test) {
+      auto *inst = arguments.takeInstruction();
+      auto *pai = cast<PartialApplyInst>(inst);
+      function.print(llvm::outs());
+      auto result = pai->visitOnStackLifetimeEnds([](auto *operand) {
+        operand->print(llvm::outs());
+        return true;
+      });
+      const char *resultString = result ? "true" : "false";
+      llvm::outs() << "returned: " << resultString << "\n";
+    });
+} // end namespace swift::test
+
 // FIXME: Rather than recursing through all results, this should only recurse
 // through ForwardingInstruction and OwnershipTransitionInstruction and the
 // client should prove that any other uses cannot be upstream from a consume of

test/SIL/Instruction/partial_apply_on_stack_lifetime_ends.sil

@@ -0,0 +1,40 @@
+// RUN: %target-sil-opt \
+// RUN:     -test-runner \
+// RUN:     %s \
+// RUN:     -o /dev/null \
+// RUN: 2>&1 | %FileCheck %s
+
+sil_stage canonical
+
+class C {}
+
+sil @borrowC : $@convention(thin) (@guaranteed C) -> ()
+
+// CHECK-LABEL: begin running test {{.*}} on copied_partial_apply
+// CHECK-LABEL: sil [ossa] @copied_partial_apply : {{.*}} {
+// CHECK:       bb0([[C:%[^,]+]] :
+// CHECK:         [[BORROW_C:%[^,]+]] = function_ref @borrowC
+// CHECK:         [[PA:%[^,]+]] = partial_apply [callee_guaranteed] [on_stack] [[BORROW_C]]([[C]])
+// CHECK:         [[PA2:%[^,]+]] = copy_value [[PA]]
+// CHECK:         destroy_value [[PA]]
+// CHECK:         destroy_value [[PA2]]
+// CHECK:         destroy_value [[C]]
+// CHECK-LABEL: } // end sil function 'copied_partial_apply'
+// CHECK:       Operand.
+// CHECK:       Owner:   destroy_value [[PA]]
+// CHECK:       returned: true
+// CHECK-LABEL: end running test {{.*}} on copied_partial_apply
+sil [ossa] @copied_partial_apply : $@convention(thin) (@owned C) -> () {
+entry(%c: @owned $C):
+  specify_test "partial_apply_print_on_stack_lifetime_ends %pa"
+
+  %callee = function_ref @borrowC : $@convention(thin) (@guaranteed C) -> ()
+  %pa = partial_apply [callee_guaranteed] [on_stack] %callee(%c) : $@convention(thin) (@guaranteed C) -> ()
+  %pa2 = copy_value %pa
+  destroy_value %pa
+
+  destroy_value %pa2
+  destroy_value %c
+  %retval = tuple ()
+  return %retval
+}