azalea/kotlin-fork
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[K/N][IR] Used liveness analysis in the coroutines lowering

Browse Source 
Liveness analysis results are used for spilling variables before
suspending a coroutine
This commit is contained in:
Igor Chevdar
2023-04-05 18:02:42 +03:00
parent 4e661a6337
commit 3e0196da89
1 changed files with 3 additions and 30 deletions
Download Patch File Download Diff File Expand all files Collapse all files
kotlin-native/backend.native/compiler/ir/backend.native/src/org/jetbrains/kotlin/backend/konan/lower/NativeSuspendFunctionLowering.kt
+3 -30
View File
@@ -2,6 +2,7 @@ package org.jetbrains.kotlin.backend.konan.lower
import org.jetbrains.kotlin.backend.common.descriptors.synthesizedName
import org.jetbrains.kotlin.backend.common.lower.*
import org.jetbrains.kotlin.backend.common.lower.optimizations.LivenessAnalysis
import org.jetbrains.kotlin.backend.konan.Context
import org.jetbrains.kotlin.backend.konan.NativeGenerationState
import org.jetbrains.kotlin.descriptors.Modality
@@ -14,10 +15,7 @@ import org.jetbrains.kotlin.ir.declarations.*
import org.jetbrains.kotlin.ir.declarations.impl.IrFunctionImpl
import org.jetbrains.kotlin.ir.declarations.impl.IrVariableImpl
import org.jetbrains.kotlin.ir.expressions.*
import org.jetbrains.kotlin.ir.expressions.impl.IrGetValueImpl
import org.jetbrains.kotlin.ir.expressions.impl.IrSetValueImpl
import org.jetbrains.kotlin.ir.expressions.impl.IrSuspendableExpressionImpl
import org.jetbrains.kotlin.ir.expressions.impl.IrSuspensionPointImpl
import org.jetbrains.kotlin.ir.expressions.impl.*
import org.jetbrains.kotlin.ir.symbols.IrClassSymbol
import org.jetbrains.kotlin.ir.symbols.IrVariableSymbol
import org.jetbrains.kotlin.ir.symbols.impl.IrSimpleFunctionSymbolImpl
@@ -85,7 +83,7 @@ internal class NativeSuspendFunctionsLowering(
// Extract all suspend calls to temporaries in order to make correct jumps to them.
originalBody.transformChildrenVoid(ExpressionSlicer(labelField.type, transformingFunction, saveState, restoreState, suspendResult, resultArgument))
val liveLocals = computeLivenessAtSuspensionPoints(originalBody)
val liveLocals = LivenessAnalysis.run(originalBody) { it is IrSuspensionPoint }
val immutableLiveLocals = liveLocals.values.flatten().filterNot { it.isVar }.toSet()
val localsMap = immutableLiveLocals.associate { it to irVar(it.name, it.type, true) }
@@ -238,31 +236,6 @@ internal class NativeSuspendFunctionsLowering(
})
}
private fun computeLivenessAtSuspensionPoints(body: IrBody): Map<IrSuspensionPoint, List<IrVariable>> {
// TODO: data flow analysis.
// Just save all visible for now.
val result = mutableMapOf<IrSuspensionPoint, List<IrVariable>>()
body.acceptChildrenVoid(object: VariablesScopeTracker() {
override fun visitExpression(expression: IrExpression) {
val suspensionPoint = expression as? IrSuspensionPoint
if (suspensionPoint == null) {
super.visitExpression(expression)
return
}
suspensionPoint.result.acceptChildrenVoid(this)
suspensionPoint.resumeResult.acceptChildrenVoid(this)
val visibleVariables = mutableListOf<IrVariable>()
scopeStack.forEach { visibleVariables += it }
result[suspensionPoint] = visibleVariables
}
})
return result
}
private inner class ExpressionSlicer(val suspensionPointIdType: IrType, val irFunction: IrFunction,
val saveState: IrFunction, val restoreState: IrFunction,
val suspendResult: IrVariable, val resultArgument: IrValueParameter): IrElementTransformerVoid() {