[JVM_IR] Contain 'invokeSuspend' rewriting to the suspend function.

Do not rewrite returns in methods in nested lambda implementations.
That leads to unintended non-local returns from nested non-suspend
lambdas.

compiler/ir/backend.jvm/src/org/jetbrains/kotlin/backend/jvm/lower/AddContinuationLowering.kt

@@ -187,6 +187,13 @@ private class AddContinuationLowering(private val context: JvmBackendContext) :
                         }
                     }
 
+                    // If the suspend lambda body contains declarations of other classes (for other lambdas),
+                    // do not rewrite those. In particular, that could lead to rewriting of returns in nested
+                    // lambdas to unintended non-local returns.
+                    override fun visitClass(declaration: IrClass): IrStatement {
+                        return declaration
+                    }
+
                     override fun visitReturn(expression: IrReturn): IrExpression {
                         val ret = super.visitReturn(expression) as IrReturn
                         return IrReturnImpl(ret.startOffset, ret.endOffset, context.irBuiltIns.anyType, function.symbol, ret.value)

compiler/testData/codegen/box/coroutines/await.kt

@@ -1,5 +1,4 @@
 // KJS_WITH_FULL_RUNTIME
-// IGNORE_BACKEND: JVM_IR
 // WITH_RUNTIME
 // WITH_COROUTINES
 // COMMON_COROUTINES_TEST

compiler/testData/codegen/box/coroutines/controllerAccessFromInnerLambda.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JVM_IR
 // WITH_RUNTIME
 // WITH_COROUTINES
 // COMMON_COROUTINES_TEST

compiler/testData/codegen/box/coroutines/featureIntersection/tailrec/realIteratorFoldl.kt

@@ -1,4 +1,3 @@
-// IGNORE_BACKEND: JVM_IR
 // WITH_RUNTIME
 // WITH_COROUTINES
 // COMMON_COROUTINES_TEST