[Result Builders] Correct type witness substitution when computing inferred

result builder attributes from protocol requirements.

Author: hborla

lib/Sema/TypeCheckRequestFunctions.cpp

@@ -297,9 +297,11 @@ static Type inferResultBuilderType(ValueDecl *decl)  {
         if (witness != lookupDecl)
           continue;
 
-        // Substitute into the result builder type.
-        auto subs =
-            conformance->getSubstitutions(lookupDecl->getModuleContext());
+        // Substitute Self and associated type witnesses into the
+        // result builder type.
+        auto subs = SubstitutionMap::getProtocolSubstitutions(
+            protocol, dc->getSelfTypeInContext(),
+            ProtocolConformanceRef(conformance));
         Type subResultBuilderType = resultBuilderType.subst(subs);
 
         matches.push_back(

test/Constraints/result_builder_generic_infer.swift

@@ -0,0 +1,31 @@
+// RUN: %target-swift-frontend -dump-ast %s | %FileCheck %s
+
+protocol P {
+  associatedtype A
+
+  @Builder<A>
+  var x1: [S] { get }
+    
+  @Builder<Self>
+  var x2: [S] { get }
+}
+
+@resultBuilder
+enum Builder<T> {
+  static func buildBlock(_ args: S...) -> [S] { args }
+}
+
+struct S {}
+
+// CHECK: struct_decl{{.*}}TestGenericBuilderInference
+struct TestGenericBuilderInference: P {
+  typealias A = Int
+
+  // CHECK: var_decl{{.*}}x1
+  // CHECK: Builder.buildBlock{{.*}}(substitution_map generic_signature=<T> (substitution T -> Int))
+  var x1: [S] { S() }
+
+  // CHECK: var_decl{{.*}}x2
+  // CHECK: Builder.buildBlock{{.*}}(substitution_map generic_signature=<T> (substitution T -> TestGenericBuilderInference))
+  var x2: [S] { S() }
+}