n/futokb
1
0
Fork 0
mirror of https://gitlab.futo.org/keyboard/latinime.git synced 2024-09-28 14:54:30 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Disable transformer when super stressed

Browse Source
This commit is contained in:
Aleksandras Kostarevas 2024-04-29 20:58:37 -04:00
parent ecaf9df340
commit 21800d37e2
2 changed files with 17 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
java/src/org/futo/inputmethod/latin/LatinIME.kt
View File

@ -381,6 +381,7 @@ class LatinIME : InputMethodService(), LifecycleOwner, ViewModelStoreOwner, Save
override fun onStartInput(attribute: EditorInfo?, restarting: Boolean) { override fun onStartInput(attribute: EditorInfo?, restarting: Boolean) {
super.onStartInput(attribute, restarting) super.onStartInput(attribute, restarting)
latinIMELegacy.onStartInput(attribute, restarting) latinIMELegacy.onStartInput(attribute, restarting)
languageModelFacilitator.onStartInput()
} }
override fun onStartInputView(info: EditorInfo?, restarting: Boolean) { override fun onStartInputView(info: EditorInfo?, restarting: Boolean) {

16
java/src/org/futo/inputmethod/latin/xlm/LanguageModelFacilitator.kt
View File

@ -133,6 +133,9 @@ public class LanguageModelFacilitator(
public fun hasPendingUpdate(): Boolean = public fun hasPendingUpdate(): Boolean =
computationSemaphore.availablePermits == 0 computationSemaphore.availablePermits == 0
private var numConsecutiveTimeouts = 0
private var transformerDisabled = false
public fun blockUntilComplete() { public fun blockUntilComplete() {
runBlocking { runBlocking {
try { try {
@ -145,8 +148,14 @@ public class LanguageModelFacilitator(
} }
} }
numConsecutiveTimeouts = 0
} catch(e: TimeoutCancellationException) { } catch(e: TimeoutCancellationException) {
Log.d("LanguageModelFacilitator", "Failed to complete prediction within 1000ms!") Log.d("LanguageModelFacilitator", "Failed to complete prediction within 1000ms!")
numConsecutiveTimeouts += 1
if(numConsecutiveTimeouts > 5) {
transformerDisabled = true
Log.w("LanguageModelFacilitator", "Temporarily disabling transformer due to continuous timeouts")
}
} }
} }
} }
@ -171,6 +180,8 @@ public class LanguageModelFacilitator(
} }
private suspend fun runLanguageModel(values: PredictionInputValues): ArrayList<SuggestedWordInfo>? { private suspend fun runLanguageModel(values: PredictionInputValues): ArrayList<SuggestedWordInfo>? {
if(transformerDisabled) return null
val locale = dictionaryFacilitator.locale ?: return null val locale = dictionaryFacilitator.locale ?: return null
if (languageModel == null || (languageModel?.locale?.language != locale.language)) { if (languageModel == null || (languageModel?.locale?.language != locale.language)) {
Log.d( Log.d(
@ -602,4 +613,9 @@ public class LanguageModelFacilitator(
assert(historyLog.isEmpty()) assert(historyLog.isEmpty())
loadHistoryLogBackup(context, historyLog) loadHistoryLogBackup(context, historyLog)
} }
public fun onStartInput() {
transformerDisabled = false
numConsecutiveTimeouts = 0
}
} }