From 53fe540884e7afc38f52f6a65f6c3ad238740a5f Mon Sep 17 00:00:00 2001
From: Anton Bannykh <anton.bannykh@jetbrains.com>
Date: Wed, 1 Dec 2021 16:48:39 +0300
Subject: [PATCH] [JS IR] use correct factory for new IR2JS

---
 .../src/org/jetbrains/kotlin/cli/js/K2JsIrCompiler.kt     | 8 +++++++-
 1 file changed, 7 insertions(+), 1 deletion(-)

diff --git a/compiler/cli/cli-js/src/org/jetbrains/kotlin/cli/js/K2JsIrCompiler.kt b/compiler/cli/cli-js/src/org/jetbrains/kotlin/cli/js/K2JsIrCompiler.kt
index f23e2a56804..5b14c713a4e 100644
--- a/compiler/cli/cli-js/src/org/jetbrains/kotlin/cli/js/K2JsIrCompiler.kt
+++ b/compiler/cli/cli-js/src/org/jetbrains/kotlin/cli/js/K2JsIrCompiler.kt
@@ -381,10 +381,16 @@ class K2JsIrCompiler : CLICompiler<K2JSCompilerArguments>() {
                 else -> JsGenerationGranularity.WHOLE_PROGRAM
             }
             try {
+                val irFactory = when {
+                    arguments.irDceDriven -> PersistentIrFactory()
+                    arguments.irNewIr2Js -> IrFactoryImplForJsIC(WholeWorldStageController())
+                    else -> IrFactoryImpl
+                }
+
                 val ir = compile(
                     module,
                     phaseConfig,
-                    if (arguments.irDceDriven) PersistentIrFactory() else IrFactoryImpl,
+                    irFactory,
                     dceRuntimeDiagnostic = RuntimeDiagnostic.resolve(
                         arguments.irDceRuntimeDiagnostic,
                         messageCollector