[sil-generic-specializer] Provide a possibility to disable the indirect-to-direct conversions of parameters and results

This is required by the capture propagation pass. Indirect-to-direct conversions are still performed by default.

Author: swiftix

include/swift/SILOptimizer/Utils/Generics.h

@@ -50,6 +50,10 @@ class ReabstractionInfo {
   /// to direct.
   llvm::SmallBitVector Conversions;
 
+  /// If set, indirect to direct conversions should be performned by the generic
+  /// specializer.
+  bool ConvertIndirectToDirect;
+
   /// The first NumResults bits in Conversions refer to formal indirect
   /// out-parameters.
   unsigned NumFormalIndirectResults;
@@ -130,7 +134,8 @@ class ReabstractionInfo {
   /// If specialization is not possible getSpecializedType() will return an
   /// invalid type.
   ReabstractionInfo(ApplySite Apply, SILFunction *Callee,
-                    SubstitutionList ParamSubs);
+                    SubstitutionList ParamSubs,
+                    bool ConvertIndirectToDirect = true);
 
   /// Constructs the ReabstractionInfo for generic function \p Orig with
   /// additional requirements. Requirements may contain new layout,
@@ -140,14 +145,15 @@ class ReabstractionInfo {
   /// Returns true if the \p ParamIdx'th (non-result) formal parameter is
   /// converted from indirect to direct.
   bool isParamConverted(unsigned ParamIdx) const {
-    return Conversions.test(ParamIdx + NumFormalIndirectResults);
+    return ConvertIndirectToDirect &&
+           Conversions.test(ParamIdx + NumFormalIndirectResults);
   }
 
   /// Returns true if the \p ResultIdx'th formal result is converted from
   /// indirect to direct.
   bool isFormalResultConverted(unsigned ResultIdx) const {
     assert(ResultIdx < NumFormalIndirectResults);
-    return Conversions.test(ResultIdx);
+    return ConvertIndirectToDirect && Conversions.test(ResultIdx);
   }
 
   /// Gets the total number of original function arguments.

lib/SILOptimizer/Utils/Generics.cpp

@@ -290,10 +290,13 @@ bool ReabstractionInfo::canBeSpecialized(ApplySite Apply, SILFunction *Callee,
 }
 
 ReabstractionInfo::ReabstractionInfo(ApplySite Apply, SILFunction *Callee,
-                                     ArrayRef<Substitution> ParamSubs) {
+                                     ArrayRef<Substitution> ParamSubs,
+                                     bool ConvertIndirectToDirect) {
   if (!prepareAndCheck(Apply, Callee, ParamSubs))
     return;
 
+  this->ConvertIndirectToDirect = ConvertIndirectToDirect;
+
   if (SpecializeGenericSubstitutions) {
     specializeConcreteAndGenericSubstitutions(Apply, Callee, ParamSubs);
   } else {