From 002f4210f72060aa033fe3c72e1fd541abd07e0e Mon Sep 17 00:00:00 2001
From: Anton Bannykh <anton.bannykh@jetbrains.com>
Date: Tue, 6 Jul 2021 17:30:18 +0300
Subject: [PATCH] IR: pass information about the declaration being lowered to
 the infrastructure

---
 .../src/org/jetbrains/kotlin/backend/common/Lower.kt  | 11 +++++++----
 1 file changed, 7 insertions(+), 4 deletions(-)

diff --git a/compiler/ir/backend.common/src/org/jetbrains/kotlin/backend/common/Lower.kt b/compiler/ir/backend.common/src/org/jetbrains/kotlin/backend/common/Lower.kt
index c10fc4cfa06..2c518fab5af 100644
--- a/compiler/ir/backend.common/src/org/jetbrains/kotlin/backend/common/Lower.kt
+++ b/compiler/ir/backend.common/src/org/jetbrains/kotlin/backend/common/Lower.kt
@@ -150,11 +150,14 @@ private class BodyLoweringVisitor(
 
     override fun visitBody(body: IrBody, data: IrDeclaration?) {
         if (withLocalDeclarations) body.acceptChildren(this, null)
-        if (allowDeclarationModification) {
-            loweringPass.lower(body, data!!)
-        } else {
-            data!!.factory.stageController.bodyLowering {
+        val stageController = data!!.factory.stageController
+        stageController.restrictTo(data) {
+            if (allowDeclarationModification) {
                 loweringPass.lower(body, data)
+            } else {
+                stageController.bodyLowering {
+                    loweringPass.lower(body, data)
+                }
             }
         }
     }