Add insertDestroyOfCapturedArguments to Local.h

Author: aschwaighofer

include/swift/SILOptimizer/Utils/Local.h

@@ -212,6 +212,12 @@ void releasePartialApplyCapturedArg(
     SILBuilder &Builder, SILLocation Loc, SILValue Arg, SILParameterInfo PInfo,
     InstModCallbacks Callbacks = InstModCallbacks());
 
+/// Insert destroys of captured arguments of partial_apply [stack].
+void insertDestroyOfCapturedArguments(
+    PartialApplyInst *PAI, SILBuilder &B,
+    llvm::function_ref<bool(SILValue)> shouldInsertDestroy =
+        [](SILValue arg) -> bool { return true; });
+
 /// This computes the lifetime of a single SILValue.
 ///
 /// This does not compute a set of jointly postdominating use points. Instead it

lib/SILOptimizer/Utils/Local.cpp

@@ -1793,3 +1793,22 @@ swift::findLocalApplySites(FunctionRefBaseInst *FRI) {
     return None;
   return f;
 }
+
+/// Insert destroys of captured arguments of partial_apply [stack].
+void swift::insertDestroyOfCapturedArguments(
+    PartialApplyInst *PAI, SILBuilder &B,
+    llvm::function_ref<bool(SILValue)> shouldInsertDestroy) {
+  assert(PAI->isOnStack());
+
+  ApplySite site(PAI);
+  SILFunctionConventions calleeConv(site.getSubstCalleeType(),
+                                    PAI->getModule());
+  auto loc = RegularLocation::getAutoGeneratedLocation();
+  for (auto &arg : PAI->getArgumentOperands()) {
+    if (!shouldInsertDestroy(arg.get())) continue;
+    unsigned calleeArgumentIndex = site.getCalleeArgIndex(arg);
+    assert(calleeArgumentIndex >= calleeConv.getSILArgIndexOfFirstParam());
+    auto paramInfo = calleeConv.getParamInfoForSILArg(calleeArgumentIndex);
+    releasePartialApplyCapturedArg(B, loc, arg.get(), paramInfo);
+  }
+}