K1: introduce BUILDER_INFERENCE_STUB_PARAMETER_TYPE to prevent compiler crashes

This diagnostic is reported in rare situations when
StubTypeForBuilderInference is kept as a parameter type
of for loop or lambda. Before this commit, we had in K1
"Could not load module <error module>" from IrLinker instead.

Related to: KT-52757, KT-53109, KT-63841, KT-64066
#KT-53478 Fixed

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/DiagnosticCompilerTestFE10TestdataTestGenerated.java

@@ -4049,6 +4049,12 @@ public class DiagnosticCompilerTestFE10TestdataTestGenerated extends AbstractDia
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -4235,6 +4241,18 @@ public class DiagnosticCompilerTestFE10TestdataTestGenerated extends AbstractDia
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt63841.kt");
         }
 
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066.kt");
+        }
+
+        @Test
+        @TestMetadata("kt64066disabled.kt")
+        public void testKt64066disabled() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066disabled.kt");
+        }
+
         @Test
         @TestMetadata("kt64077.kt")
         public void testKt64077() {

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirBlackBoxCodegenBasedTestGenerated.java

@@ -3683,6 +3683,12 @@ public class LLFirBlackBoxCodegenBasedTestGenerated extends AbstractLLFirBlackBo
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3856,6 +3862,12 @@ public class LLFirBlackBoxCodegenBasedTestGenerated extends AbstractLLFirBlackBo
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated.java

@@ -4049,6 +4049,12 @@ public class LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -4235,6 +4241,18 @@ public class LLFirPreresolvedReversedDiagnosticCompilerFE10TestDataTestGenerated
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt63841.kt");
         }
 
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066.kt");
+        }
+
+        @Test
+        @TestMetadata("kt64066disabled.kt")
+        public void testKt64066disabled() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066disabled.kt");
+        }
+
         @Test
         @TestMetadata("kt64077.kt")
         public void testKt64077() {

analysis/low-level-api-fir/tests/org/jetbrains/kotlin/analysis/low/level/api/fir/diagnostic/compiler/based/LLFirReversedBlackBoxCodegenBasedTestGenerated.java

@@ -3683,6 +3683,12 @@ public class LLFirReversedBlackBoxCodegenBasedTestGenerated extends AbstractLLFi
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3856,6 +3862,12 @@ public class LLFirReversedBlackBoxCodegenBasedTestGenerated extends AbstractLLFi
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirLightTreeOldFrontendDiagnosticsTestGenerated.java

@@ -4043,6 +4043,12 @@ public class FirLightTreeOldFrontendDiagnosticsTestGenerated extends AbstractFir
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -4229,6 +4235,18 @@ public class FirLightTreeOldFrontendDiagnosticsTestGenerated extends AbstractFir
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt63841.kt");
         }
 
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066.kt");
+        }
+
+        @Test
+        @TestMetadata("kt64066disabled.kt")
+        public void testKt64066disabled() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066disabled.kt");
+        }
+
         @Test
         @TestMetadata("kt64077.kt")
         public void testKt64077() {

compiler/fir/analysis-tests/tests-gen/org/jetbrains/kotlin/test/runners/FirPsiOldFrontendDiagnosticsTestGenerated.java

@@ -4049,6 +4049,12 @@ public class FirPsiOldFrontendDiagnosticsTestGenerated extends AbstractFirPsiDia
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -4235,6 +4241,18 @@ public class FirPsiOldFrontendDiagnosticsTestGenerated extends AbstractFirPsiDia
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt63841.kt");
         }
 
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066.kt");
+        }
+
+        @Test
+        @TestMetadata("kt64066disabled.kt")
+        public void testKt64066disabled() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066disabled.kt");
+        }
+
         @Test
         @TestMetadata("kt64077.kt")
         public void testKt64077() {

compiler/fir/fir2ir/tests-gen/org/jetbrains/kotlin/test/runners/codegen/FirLightTreeBlackBoxCodegenTestGenerated.java

@@ -3666,6 +3666,12 @@ public class FirLightTreeBlackBoxCodegenTestGenerated extends AbstractFirLightTr
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3839,6 +3845,12 @@ public class FirLightTreeBlackBoxCodegenTestGenerated extends AbstractFirLightTr
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/fir/fir2ir/tests-gen/org/jetbrains/kotlin/test/runners/codegen/FirLightTreeBlackBoxCodegenWithIrFakeOverrideGeneratorTestGenerated.java

@@ -3666,6 +3666,12 @@ public class FirLightTreeBlackBoxCodegenWithIrFakeOverrideGeneratorTestGenerated
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3839,6 +3845,12 @@ public class FirLightTreeBlackBoxCodegenWithIrFakeOverrideGeneratorTestGenerated
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/fir/fir2ir/tests-gen/org/jetbrains/kotlin/test/runners/codegen/FirPsiBlackBoxCodegenTestGenerated.java

@@ -3666,6 +3666,12 @@ public class FirPsiBlackBoxCodegenTestGenerated extends AbstractFirPsiBlackBoxCo
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3839,6 +3845,12 @@ public class FirPsiBlackBoxCodegenTestGenerated extends AbstractFirPsiBlackBoxCo
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/frontend/src/org/jetbrains/kotlin/diagnostics/Errors.java

@@ -949,6 +949,7 @@ public interface Errors {
     DiagnosticFactory4<PsiElement, KotlinType, String, String, BuilderLambdaLabelingInfo> STUB_TYPE_IN_RECEIVER_CAUSES_AMBIGUITY = DiagnosticFactory4.create(ERROR);
     DiagnosticFactory2<PsiElement, Name, Name> BUILDER_INFERENCE_MULTI_LAMBDA_RESTRICTION = DiagnosticFactory2.create(ERROR);
     DiagnosticFactory2<PsiElement, Name, Name> BUILDER_INFERENCE_STUB_RECEIVER = DiagnosticFactory2.create(ERROR);
+    DiagnosticFactory1<KtDeclaration, Name> BUILDER_INFERENCE_STUB_PARAMETER_TYPE = DiagnosticFactory1.create(ERROR);
     DiagnosticFactory4<KtElement, Name, Name, KotlinType, KotlinType> UPPER_BOUND_VIOLATION_IN_CONSTRAINT = DiagnosticFactory4.create(ERROR);
 
     DiagnosticFactory1<PsiElement, Collection<? extends ResolvedCall<?>>> NONE_APPLICABLE = DiagnosticFactory1.create(ERROR);

compiler/frontend/src/org/jetbrains/kotlin/diagnostics/rendering/DefaultErrorMessages.java

@@ -1106,6 +1106,7 @@ public class DefaultErrorMessages {
         MAP.put(STUB_TYPE_IN_RECEIVER_CAUSES_AMBIGUITY, "The type of a receiver hasn''t been inferred yet. To disambiguate this call, explicitly cast it to `{0}` if you want the builder''s type parameter(s) `{1}` to be inferred to `{2}`.", RENDER_TYPE, STRING, STRING, null);
         MAP.put(BUILDER_INFERENCE_MULTI_LAMBDA_RESTRICTION, "Unstable inference behaviour with multiple lambdas. Please either specify the type argument for generic parameter `{0}` of `{1}` explicitly", TO_STRING, TO_STRING);
         MAP.put(BUILDER_INFERENCE_STUB_RECEIVER, "The type of a receiver hasn''t been inferred yet. Please specify type argument for generic parameter `{0}` of `{1}` explicitly", TO_STRING, TO_STRING);
+        MAP.put(BUILDER_INFERENCE_STUB_PARAMETER_TYPE, "The type of parameter `{0}` cannot be inferred by builder inference", TO_STRING);
         MAP.put(UPPER_BOUND_VIOLATION_IN_CONSTRAINT, "Upper bound violation for generic parameter `{0}` of `{1}`: {3} is not a subtype of {2}", TO_STRING, TO_STRING, RENDER_TYPE, RENDER_TYPE);
         MAP.put(NONE_APPLICABLE, "None of the following functions can be called with the arguments supplied: {0}", AMBIGUOUS_CALLS);
         MAP.put(CANNOT_COMPLETE_RESOLVE, "Cannot choose among the following candidates without completing type inference: {0}", AMBIGUOUS_CALLS);

compiler/frontend/src/org/jetbrains/kotlin/resolve/PlatformConfiguratorBase.kt

@@ -62,6 +62,7 @@ private val DEFAULT_DECLARATION_CHECKERS = listOf(
     EnumEntriesRedeclarationChecker,
     VolatileAnnotationChecker,
     ActualTypealiasToSpecialAnnotationChecker,
+    StubForBuilderInferenceParameterTypeChecker,
 )
 
 private val DEFAULT_CALL_CHECKERS = listOf(
@@ -80,7 +81,7 @@ private val DEFAULT_CALL_CHECKERS = listOf(
     NewSchemeOfIntegerOperatorResolutionChecker, EnumEntryVsCompanionPriorityCallChecker, CompanionInParenthesesLHSCallChecker,
     ResolutionToPrivateConstructorOfSealedClassChecker, EqualityCallChecker, UnsupportedUntilOperatorChecker,
     BuilderInferenceAssignmentChecker, IncorrectCapturedApproximationCallChecker, CompanionIncorrectlyUnboundedWhenUsedAsLHSCallChecker,
-    CustomEnumEntriesMigrationCallChecker, EnumEntriesUnsupportedChecker
+    CustomEnumEntriesMigrationCallChecker, EnumEntriesUnsupportedChecker,
 )
 private val DEFAULT_TYPE_CHECKERS = emptyList<AdditionalTypeChecker>()
 private val DEFAULT_CLASSIFIER_USAGE_CHECKERS = listOf(

compiler/frontend/src/org/jetbrains/kotlin/resolve/calls/checkers/BuilderInferenceAssignmentChecker.kt

@@ -11,8 +11,10 @@ import org.jetbrains.kotlin.descriptors.PropertyDescriptor
 import org.jetbrains.kotlin.diagnostics.Errors
 import org.jetbrains.kotlin.lexer.KtTokens
 import org.jetbrains.kotlin.psi.KtBinaryExpression
+import org.jetbrains.kotlin.psi.KtLambdaExpression
 import org.jetbrains.kotlin.psi.KtNameReferenceExpression
 import org.jetbrains.kotlin.psi.psiUtil.getParentOfType
+import org.jetbrains.kotlin.resolve.BindingContext
 import org.jetbrains.kotlin.resolve.calls.model.ResolvedCall
 import org.jetbrains.kotlin.resolve.calls.util.getType
 import org.jetbrains.kotlin.types.StubTypeForBuilderInference
@@ -24,18 +26,27 @@ object BuilderInferenceAssignmentChecker : CallChecker {
         val resultingDescriptor = resolvedCall.resultingDescriptor
         if (resultingDescriptor !is PropertyDescriptor) return
         if (context.languageVersionSettings.supportsFeature(LanguageFeature.NoBuilderInferenceWithoutAnnotationRestriction)) return
-        if (resolvedCall.candidateDescriptor.returnType !is StubTypeForBuilderInference) return
         val callElement = resolvedCall.call.callElement
         if (callElement !is KtNameReferenceExpression) return
         val binaryExpression = callElement.getParentOfType<KtBinaryExpression>(strict = true) ?: return
         if (binaryExpression.operationToken != KtTokens.EQ) return
         if (!BasicExpressionTypingVisitor.isLValue(callElement, binaryExpression)) return
-
-        val leftType = resultingDescriptor.returnType?.takeIf { !it.isError } ?: return
         val right = binaryExpression.right ?: return
-        val rightType = right.getType(context.trace.bindingContext) ?: return
 
-        if (isAssignmentCorrectWithDataFlowInfo(leftType, right, rightType, context)) return
-        context.trace.report(Errors.TYPE_MISMATCH.on(right, leftType, rightType))
+        if (resolvedCall.candidateDescriptor.returnType is StubTypeForBuilderInference) {
+            val leftType = resultingDescriptor.returnType?.takeIf { !it.isError } ?: return
+            val rightType = right.getType(context.trace.bindingContext) ?: return
+
+            if (isAssignmentCorrectWithDataFlowInfo(leftType, right, rightType, context)) return
+            context.trace.report(Errors.TYPE_MISMATCH.on(right, leftType, rightType))
+        } else if (right is KtLambdaExpression) {
+            val functionLiteral = right.functionLiteral
+            val functionDescriptor = context.trace.get(BindingContext.FUNCTION, right.functionLiteral) ?: return
+            for ((index, valueParameterDescriptor) in functionDescriptor.valueParameters.withIndex()) {
+                if (valueParameterDescriptor.type !is StubTypeForBuilderInference) continue
+                val target = functionLiteral.valueParameters.getOrNull(index) ?: functionLiteral
+                context.trace.report(Errors.BUILDER_INFERENCE_STUB_PARAMETER_TYPE.on(target, valueParameterDescriptor.name))
+            }
+        }
     }
 }

compiler/frontend/src/org/jetbrains/kotlin/resolve/checkers/StubForBuilderInferenceParameterTypeChecker.kt

@@ -0,0 +1,26 @@
+/*
+ * Copyright 2010-2024 JetBrains s.r.o. and Kotlin Programming Language contributors.
+ * Use of this source code is governed by the Apache 2.0 license that can be found in the license/LICENSE.txt file.
+ */
+
+package org.jetbrains.kotlin.resolve.checkers
+
+import org.jetbrains.kotlin.config.LanguageFeature
+import org.jetbrains.kotlin.descriptors.DeclarationDescriptor
+import org.jetbrains.kotlin.descriptors.ValueDescriptor
+import org.jetbrains.kotlin.diagnostics.Errors
+import org.jetbrains.kotlin.psi.KtDeclaration
+import org.jetbrains.kotlin.psi.KtParameter
+import org.jetbrains.kotlin.types.StubTypeForBuilderInference
+
+object StubForBuilderInferenceParameterTypeChecker : DeclarationChecker {
+    override fun check(declaration: KtDeclaration, descriptor: DeclarationDescriptor, context: DeclarationCheckerContext) {
+        if (declaration !is KtParameter ||
+            descriptor !is ValueDescriptor ||
+            descriptor.returnType !is StubTypeForBuilderInference
+        ) return
+        if (context.languageVersionSettings.supportsFeature(LanguageFeature.NoBuilderInferenceWithoutAnnotationRestriction)) return
+
+        context.trace.report(Errors.BUILDER_INFERENCE_STUB_PARAMETER_TYPE.on(declaration, declaration.nameAsSafeName))
+    }
+}

compiler/testData/codegen/box/builderInference/issues/kt52757.kt

@@ -2,8 +2,7 @@
 
 // IGNORE_LIGHT_ANALYSIS
 // IGNORE_BACKEND_K1: ANY
-// REASON: compile-time failure in K1/JVM (java.lang.AssertionError: Could not load module <Error module> @ org.jetbrains.kotlin.backend.common.serialization.KotlinIrLinker.resolveModuleDeserializer)
-// REASON: compile-time failure in K1/Native, K1/WASM, K1/JS (java.lang.NullPointerException @ org.jetbrains.kotlin.backend.common.serialization.mangle.descriptor.DescriptorExportCheckerVisitor.isExported)
+// Reason: red code
 
 fun box(): String {
     build {

compiler/testData/codegen/box/builderInference/issues/kt53109.kt

@@ -2,8 +2,7 @@
 
 // IGNORE_LIGHT_ANALYSIS
 // IGNORE_BACKEND_K1: ANY
-// REASON: compile-time failure in K1/JVM (java.lang.AssertionError: Could not load module <Error module> @ org.jetbrains.kotlin.backend.common.serialization.KotlinIrLinker.resolveModuleDeserializer)
-// REASON: compile-time failure in K1/Native, K1/WASM, K1/JS (java.lang.NullPointerException @ org.jetbrains.kotlin.backend.common.serialization.mangle.descriptor.DescriptorExportCheckerVisitor.isExported)
+// REASON: red code
 
 fun box(): String {
     build {

compiler/testData/codegen/box/builderInference/issues/kt53478.kt

@@ -0,0 +1,19 @@
+// ISSUE: KT-53478
+// IGNORE_BACKEND_K1: ANY
+// Reason: red code
+
+class UncompilingClass<T : Any>(
+    val block: (UncompilingClass<T>.() -> Unit)? = null,
+) {
+
+    var uncompilingFun: ((T) -> Unit)? = null
+}
+
+fun handleInt(arg: Int) = Unit
+
+fun box(): String {
+    val obj = UncompilingClass {
+        uncompilingFun = { handleInt(it) }
+    }
+    return "OK"
+}

compiler/testData/codegen/box/builderInference/issues/kt64066.kt

@@ -0,0 +1,13 @@
+// WITH_STDLIB
+// ISSUE: KT-64066
+// IGNORE_BACKEND_K1: ANY
+// Reason: red code
+
+fun box(): String {
+    val map = buildMap {
+        put(1, 1)
+        for (v in values) {}
+    }
+    if (map[1] != 1) return "FAIL"
+    return "OK"
+}

compiler/testData/diagnostics/tests/builderInference/issues/kt52757.fir.kt

@@ -0,0 +1,28 @@
+// ISSUE: KT-52757
+// CHECK_TYPE_WITH_EXACT
+
+fun test() {
+    val buildee = build {
+        typeVariableConsumer = ::consumeTargetType
+        typeVariableConsumer = {}
+    }
+    // exact type equality check — turns unexpected compile-time behavior into red code
+    // considered to be non-user-reproducible code for the purposes of these tests
+    checkExactType<Buildee<TargetType>>(buildee)
+}
+
+
+
+
+class TargetType
+
+fun consumeTargetType(value: TargetType) {}
+
+class Buildee<TV> {
+    var typeVariableConsumer: (TV) -> Unit = { storage = it }
+    private var storage: TV = null!!
+}
+
+fun <PTV> build(instructions: Buildee<PTV>.() -> Unit): Buildee<PTV> {
+    return Buildee<PTV>().apply(instructions)
+}

compiler/testData/diagnostics/tests/builderInference/issues/kt52757.kt

@@ -1,11 +1,10 @@
 // ISSUE: KT-52757
-// FIR_IDENTICAL
 // CHECK_TYPE_WITH_EXACT
 
 fun test() {
     val buildee = build {
         typeVariableConsumer = ::consumeTargetType
-        typeVariableConsumer = {}
+        typeVariableConsumer = <!BUILDER_INFERENCE_STUB_PARAMETER_TYPE!>{}<!>
     }
     // exact type equality check — turns unexpected compile-time behavior into red code
     // considered to be non-user-reproducible code for the purposes of these tests

compiler/testData/diagnostics/tests/builderInference/issues/kt53109.fir.kt

@@ -0,0 +1,27 @@
+// ISSUE: KT-53109
+// CHECK_TYPE_WITH_EXACT
+
+fun test() {
+    val buildee = build {
+        typeVariableConsumer = { consumeTargetType(it) }
+    }
+    // exact type equality check — turns unexpected compile-time behavior into red code
+    // considered to be non-user-reproducible code for the purposes of these tests
+    checkExactType<Buildee<TargetType>>(buildee)
+}
+
+
+
+
+class TargetType
+
+fun consumeTargetType(value: TargetType) {}
+
+class Buildee<TV> {
+    var typeVariableConsumer: (TV) -> Unit = { storage = it }
+    private var storage: TV = null!!
+}
+
+fun <PTV> build(instructions: Buildee<PTV>.() -> Unit): Buildee<PTV> {
+    return Buildee<PTV>().apply(instructions)
+}

compiler/testData/diagnostics/tests/builderInference/issues/kt53109.kt

@@ -1,10 +1,9 @@
 // ISSUE: KT-53109
-// FIR_IDENTICAL
 // CHECK_TYPE_WITH_EXACT
 
 fun test() {
     val buildee = build {
-        typeVariableConsumer = { consumeTargetType(it) }
+        typeVariableConsumer = <!BUILDER_INFERENCE_STUB_PARAMETER_TYPE!>{ consumeTargetType(it) }<!>
     }
     // exact type equality check — turns unexpected compile-time behavior into red code
     // considered to be non-user-reproducible code for the purposes of these tests

compiler/testData/diagnostics/tests/builderInference/issues/kt53478.fir.kt

@@ -0,0 +1,16 @@
+// ISSUE: KT-53478
+
+class UncompilingClass<T : Any>(
+    val block: (UncompilingClass<T>.() -> Unit)? = null,
+) {
+
+    var uncompilingFun: ((T) -> Unit)? = null
+}
+
+fun handleInt(arg: Int) = Unit
+
+fun box() {
+    val obj = UncompilingClass {
+        uncompilingFun = { handleInt(it) }
+    }
+}

compiler/testData/diagnostics/tests/builderInference/issues/kt53478.kt

@@ -0,0 +1,16 @@
+// ISSUE: KT-53478
+
+class UncompilingClass<T : Any>(
+    val block: (UncompilingClass<T>.() -> Unit)? = null,
+) {
+
+    var uncompilingFun: ((T) -> Unit)? = null
+}
+
+fun handleInt(arg: Int) = Unit
+
+fun box() {
+    val obj = UncompilingClass {
+        uncompilingFun = <!BUILDER_INFERENCE_STUB_PARAMETER_TYPE!>{ handleInt(it) }<!>
+    }
+}

compiler/testData/diagnostics/tests/builderInference/issues/kt64066.fir.kt

@@ -0,0 +1,9 @@
+// WITH_STDLIB
+// ISSUE: KT-64066
+
+fun box() {
+    val map = buildMap {
+        put(1, 1)
+        for (v in values) {}
+    }
+}

compiler/testData/diagnostics/tests/builderInference/issues/kt64066.kt

@@ -0,0 +1,9 @@
+// WITH_STDLIB
+// ISSUE: KT-64066
+
+fun box() {
+    val map = buildMap {
+        put(1, 1)
+        for (<!BUILDER_INFERENCE_STUB_PARAMETER_TYPE!>v<!> in values) {}
+    }
+}

compiler/testData/diagnostics/tests/builderInference/issues/kt64066disabled.kt

@@ -0,0 +1,26 @@
+// FIR_IDENTICAL
+// WITH_STDLIB
+// ISSUE: KT-64066, KT-53478
+// LANGUAGE: +NoBuilderInferenceWithoutAnnotationRestriction
+
+fun box1() {
+    val map = buildMap {
+        put(1, 1)
+        for (v in values) {}
+    }
+}
+
+class UncompilingClass<T : Any>(
+    val block: (UncompilingClass<T>.() -> Unit)? = null,
+) {
+
+    var uncompilingFun: ((T) -> Unit)? = null
+}
+
+fun handleInt(arg: Int) = Unit
+
+fun box2() {
+    val obj = UncompilingClass {
+        uncompilingFun = { handleInt(it) }
+    }
+}

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/DiagnosticTestGenerated.java

@@ -4049,6 +4049,12 @@ public class DiagnosticTestGenerated extends AbstractDiagnosticTest {
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -4235,6 +4241,18 @@ public class DiagnosticTestGenerated extends AbstractDiagnosticTest {
           runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt63841.kt");
         }
 
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066.kt");
+        }
+
+        @Test
+        @TestMetadata("kt64066disabled.kt")
+        public void testKt64066disabled() {
+          runTest("compiler/testData/diagnostics/tests/builderInference/issues/kt64066disabled.kt");
+        }
+
         @Test
         @TestMetadata("kt64077.kt")
         public void testKt64077() {

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/JvmAbiConsistencyTestBoxGenerated.java

@@ -3666,6 +3666,12 @@ public class JvmAbiConsistencyTestBoxGenerated extends AbstractJvmAbiConsistency
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3839,6 +3845,12 @@ public class JvmAbiConsistencyTestBoxGenerated extends AbstractJvmAbiConsistency
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/codegen/BlackBoxCodegenTestGenerated.java

@@ -3468,6 +3468,12 @@ public class BlackBoxCodegenTestGenerated extends AbstractBlackBoxCodegenTest {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3641,6 +3647,12 @@ public class BlackBoxCodegenTestGenerated extends AbstractBlackBoxCodegenTest {
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/codegen/IrBlackBoxCodegenTestGenerated.java

@@ -3666,6 +3666,12 @@ public class IrBlackBoxCodegenTestGenerated extends AbstractIrBlackBoxCodegenTes
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3839,6 +3845,12 @@ public class IrBlackBoxCodegenTestGenerated extends AbstractIrBlackBoxCodegenTes
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/codegen/IrBlackBoxCodegenWithIrInlinerTestGenerated.java

@@ -3666,6 +3666,12 @@ public class IrBlackBoxCodegenWithIrInlinerTestGenerated extends AbstractIrBlack
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3839,6 +3845,12 @@ public class IrBlackBoxCodegenWithIrInlinerTestGenerated extends AbstractIrBlack
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/tests-common-new/tests-gen/org/jetbrains/kotlin/test/runners/codegen/inlineScopes/FirBlackBoxCodegenTestWithInlineScopesGenerated.java

@@ -3666,6 +3666,12 @@ public class FirBlackBoxCodegenTestWithInlineScopesGenerated extends AbstractFir
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -3839,6 +3845,12 @@ public class FirBlackBoxCodegenTestWithInlineScopesGenerated extends AbstractFir
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

compiler/tests-gen/org/jetbrains/kotlin/codegen/LightAnalysisModeTestGenerated.java

@@ -3190,6 +3190,11 @@ public class LightAnalysisModeTestGenerated extends AbstractLightAnalysisModeTes
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53553.kt");
@@ -3334,6 +3339,11 @@ public class LightAnalysisModeTestGenerated extends AbstractLightAnalysisModeTes
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @TestMetadata("compiler/testData/codegen/box/builderInference/oneParameter")

js/js.tests/tests-gen/org/jetbrains/kotlin/js/test/fir/FirJsCodegenBoxTestGenerated.java

@@ -2754,6 +2754,12 @@ public class FirJsCodegenBoxTestGenerated extends AbstractFirJsCodegenBoxTest {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -2921,6 +2927,12 @@ public class FirJsCodegenBoxTestGenerated extends AbstractFirJsCodegenBoxTest {
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

js/js.tests/tests-gen/org/jetbrains/kotlin/js/test/fir/FirJsES6CodegenBoxTestGenerated.java

@@ -2754,6 +2754,12 @@ public class FirJsES6CodegenBoxTestGenerated extends AbstractFirJsES6CodegenBoxT
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -2921,6 +2927,12 @@ public class FirJsES6CodegenBoxTestGenerated extends AbstractFirJsES6CodegenBoxT
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

js/js.tests/tests-gen/org/jetbrains/kotlin/js/test/ir/IrJsCodegenBoxTestGenerated.java

@@ -2754,6 +2754,12 @@ public class IrJsCodegenBoxTestGenerated extends AbstractIrJsCodegenBoxTest {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -2921,6 +2927,12 @@ public class IrJsCodegenBoxTestGenerated extends AbstractIrJsCodegenBoxTest {
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

js/js.tests/tests-gen/org/jetbrains/kotlin/js/test/ir/IrJsES6CodegenBoxTestGenerated.java

@@ -2754,6 +2754,12 @@ public class IrJsES6CodegenBoxTestGenerated extends AbstractIrJsES6CodegenBoxTes
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -2921,6 +2927,12 @@ public class IrJsES6CodegenBoxTestGenerated extends AbstractIrJsES6CodegenBoxTes
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

native/native.tests/tests-gen/org/jetbrains/kotlin/konan/test/blackbox/FirNativeCodegenBoxTestGenerated.java

@@ -2831,6 +2831,12 @@ public class FirNativeCodegenBoxTestGenerated extends AbstractNativeCodegenBoxTe
           runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -2998,6 +3004,12 @@ public class FirNativeCodegenBoxTestGenerated extends AbstractNativeCodegenBoxTe
         public void testKt63840c() {
           runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
         }
+
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+        }
       }
 
       @Nested

native/native.tests/tests-gen/org/jetbrains/kotlin/konan/test/blackbox/FirNativeCodegenBoxTestNoPLGenerated.java

@@ -2889,6 +2889,12 @@ public class FirNativeCodegenBoxTestNoPLGenerated extends AbstractNativeCodegenB
           runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -3056,6 +3062,12 @@ public class FirNativeCodegenBoxTestNoPLGenerated extends AbstractNativeCodegenB
         public void testKt63840c() {
           runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
         }
+
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+        }
       }
 
       @Nested

native/native.tests/tests-gen/org/jetbrains/kotlin/konan/test/blackbox/NativeCodegenBoxTestGenerated.java

@@ -2773,6 +2773,12 @@ public class NativeCodegenBoxTestGenerated extends AbstractNativeCodegenBoxTest
           runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -2940,6 +2946,12 @@ public class NativeCodegenBoxTestGenerated extends AbstractNativeCodegenBoxTest
         public void testKt63840c() {
           runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
         }
+
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+        }
       }
 
       @Nested

native/native.tests/tests-gen/org/jetbrains/kotlin/konan/test/blackbox/NativeCodegenBoxTestNoPLGenerated.java

@@ -2832,6 +2832,12 @@ public class NativeCodegenBoxTestNoPLGenerated extends AbstractNativeCodegenBoxT
           runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
         }
 
+        @Test
+        @TestMetadata("kt53478.kt")
+        public void testKt53478() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+        }
+
         @Test
         @TestMetadata("kt53553.kt")
         public void testKt53553() {
@@ -2999,6 +3005,12 @@ public class NativeCodegenBoxTestNoPLGenerated extends AbstractNativeCodegenBoxT
         public void testKt63840c() {
           runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
         }
+
+        @Test
+        @TestMetadata("kt64066.kt")
+        public void testKt64066() {
+          runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+        }
       }
 
       @Nested

wasm/wasm.tests/tests-gen/org/jetbrains/kotlin/wasm/test/FirWasmJsCodegenBoxTestGenerated.java

@@ -2742,6 +2742,12 @@ public class FirWasmJsCodegenBoxTestGenerated extends AbstractFirWasmJsCodegenBo
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -2909,6 +2915,12 @@ public class FirWasmJsCodegenBoxTestGenerated extends AbstractFirWasmJsCodegenBo
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested

wasm/wasm.tests/tests-gen/org/jetbrains/kotlin/wasm/test/K1WasmCodegenBoxTestGenerated.java

@@ -2742,6 +2742,12 @@ public class K1WasmCodegenBoxTestGenerated extends AbstractK1WasmCodegenBoxTest
         runTest("compiler/testData/codegen/box/builderInference/issues/kt53422b.kt");
       }
 
+      @Test
+      @TestMetadata("kt53478.kt")
+      public void testKt53478() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt53478.kt");
+      }
+
       @Test
       @TestMetadata("kt53553.kt")
       public void testKt53553() {
@@ -2909,6 +2915,12 @@ public class K1WasmCodegenBoxTestGenerated extends AbstractK1WasmCodegenBoxTest
       public void testKt63840c() {
         runTest("compiler/testData/codegen/box/builderInference/issues/kt63840c.kt");
       }
+
+      @Test
+      @TestMetadata("kt64066.kt")
+      public void testKt64066() {
+        runTest("compiler/testData/codegen/box/builderInference/issues/kt64066.kt");
+      }
     }
 
     @Nested