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 1af9c4d853c..8bfc79d116a 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 @@ -139,6 +139,15 @@ fun BodyLoweringPass.runOnFilePostfix(irFile: IrFile, withLocalDeclarations: Boo } } } + + override fun visitScript(declaration: IrScript, data: IrDeclaration?) { + ArrayList(declaration.declarations).forEach { it.accept(this, declaration) } + if (withLocalDeclarations) { + declaration.statements.forEach { it.accept(this, null) } + } + declaration.thisReceiver.accept(this, declaration) + } + }, null) } } @@ -239,6 +248,17 @@ fun DeclarationTransformer.runPostfix(withLocalDeclarations: Boolean = false): D declaration.declarations.transformFlat(this@runPostfix::transformFlatRestricted) } + + override fun visitScript(declaration: IrScript) { + ArrayList(declaration.declarations).forEach { it.accept(this, null) } + declaration.declarations.transformFlat(this@runPostfix::transformFlatRestricted) + + if (withLocalDeclarations) { + declaration.statements.forEach { it.accept(this, null) } + } + + declaration.thisReceiver.accept(this, null) + } }) return this@runPostfix.transformFlatRestricted(declaration)