auto

Author: gottesmm

lib/SILOptimizer/Mandatory/TransferNonSendable.cpp

@@ -695,6 +695,7 @@ struct InferredCallerArgumentTypeInfo {
   /// we may represent all of the actor isolation information at the SIL level,
   /// but we are not there yet today.
   void initForApply(ApplyIsolationCrossing isolationCrossing);
+  void initForPartialApply(const Operand *op, AbstractClosureExpr *expr);
 
   void initForApply(const Operand *op, ApplyExpr *expr);
   void initForAutoclosure(const Operand *op, AutoClosureExpr *expr);
@@ -727,6 +728,16 @@ void InferredCallerArgumentTypeInfo::initForApply(
   applyUses.emplace_back(baseInferredType, isolationCrossing);
 }
 
+void InferredCallerArgumentTypeInfo::initForPartialApply(const Operand *op, AbstractClosureExpr *ace) {
+  SmallVector<std::tuple<unsigned, const CapturedValue *, ApplyIsolationCrossing>, 8> capturedIsolationCrossings;
+  ace->getCaptureIsolationCrossings(capturedIsolationCrossings);
+  assert(capturedIsolationCrossings.size());
+
+  auto &data = capturedIsolationCrossings.front();
+
+  applyUses.emplace_back(baseInferredType, std::get<2>(capturedIsolationCrossings.front()));
+}
+
 void InferredCallerArgumentTypeInfo::initForApply(const Operand *op,
                                                   ApplyExpr *sourceApply) {
   auto isolationCrossing = *sourceApply->getIsolationCrossing();
@@ -846,6 +857,12 @@ void InferredCallerArgumentTypeInfo::init(const Operand *op) {
     }
   }
 
+  if (auto *pai = dyn_cast<PartialApplyInst>(nonConstOp->getUser())) {
+    if (auto *ace = pai->getLoc().getAsASTNode<AbstractClosureExpr>()) {
+      return initForPartialApply(op, ace);
+    }
+  }
+
   auto *autoClosureExpr = loc.getAsASTNode<AutoClosureExpr>();
   if (!autoClosureExpr) {
     llvm::report_fatal_error("Unknown node");