Optimizer: add Context. mangle(withBoxToStackPromotedArguments)

Author: eeckstein

SwiftCompilerSources/Sources/Optimizer/PassManager/Context.swift

@@ -412,6 +412,12 @@ struct FunctionPassContext : MutatingContext {
     }
   }
 
+  func mangle(withBoxToStackPromotedArguments argIndices: [Int], from original: Function) -> String {
+    argIndices.withBridgedArrayRef { bridgedArgIndices in
+      String(taking: _bridged.mangleWithBoxToStackPromotedArgs(bridgedArgIndices, original.bridged))
+    }
+  }
+
   func createGlobalVariable(name: String, type: Type, linkage: Linkage, isLet: Bool) -> GlobalVariable {
     let gv = name._withBridgedStringRef {
       _bridged.createGlobalVariable($0, type.bridged, linkage.bridged, isLet)

include/swift/SILOptimizer/OptimizerBridging.h

@@ -254,6 +254,8 @@ struct BridgedPassContext {
   BridgedOwnedString mangleWithClosureArgs(BridgedValueArray closureArgs,
                                                                BridgedArrayRef closureArgIndices,
                                                                BridgedFunction applySiteCallee) const;
+  BridgedOwnedString mangleWithBoxToStackPromotedArgs(BridgedArrayRef bridgedPromotedArgIndices,
+                                                      BridgedFunction bridgedOriginalFunction) const;
 
   SWIFT_IMPORT_UNSAFE BridgedGlobalVar createGlobalVariable(BridgedStringRef name, BridgedType type,
                                                             BridgedLinkage linkage, bool isLet) const;

lib/SILOptimizer/Utils/OptimizerBridging.cpp

@@ -336,6 +336,20 @@ BridgedOwnedString BridgedPassContext::mangleWithClosureArgs(
   return BridgedOwnedString(mangler.mangle());
 }
 
+BridgedOwnedString BridgedPassContext::mangleWithBoxToStackPromotedArgs(
+  BridgedArrayRef bridgedPromotedArgIndices,
+  BridgedFunction bridgedOriginalFunction
+) const {
+  auto *original = bridgedOriginalFunction.getFunction();
+  Mangle::FunctionSignatureSpecializationMangler mangler(original->getASTContext(),
+                                                         Demangle::SpecializationPass::AllocBoxToStack,
+                                                         original->getSerializedKind(), original);
+  for (SwiftInt i : bridgedPromotedArgIndices.unbridged<SwiftInt>()) {
+    mangler.setArgumentBoxToStack((unsigned)i);
+  }
+  return BridgedOwnedString(mangler.mangle());
+}
+
 BridgedGlobalVar BridgedPassContext::createGlobalVariable(BridgedStringRef name, BridgedType type, BridgedLinkage linkage, bool isLet) const {
   auto *global = SILGlobalVariable::create(
       *invocation->getPassManager()->getModule(),