From 3644082a82bc48145649c332dbb97bfa9de3a66b Mon Sep 17 00:00:00 2001 From: Han Yin Date: Fri, 10 Oct 2025 22:41:01 -0700 Subject: [PATCH] lib: perform engine state check inclusively instead of exclusively --- .../java/com/arm/aichat/InferenceEngine.kt | 20 ++++++++++--------- 1 file changed, 11 insertions(+), 9 deletions(-) diff --git a/examples/llama.android/lib/src/main/java/com/arm/aichat/InferenceEngine.kt b/examples/llama.android/lib/src/main/java/com/arm/aichat/InferenceEngine.kt index a40b584362..44852fa828 100644 --- a/examples/llama.android/lib/src/main/java/com/arm/aichat/InferenceEngine.kt +++ b/examples/llama.android/lib/src/main/java/com/arm/aichat/InferenceEngine.kt @@ -72,16 +72,18 @@ interface InferenceEngine { } val State.isUninterruptible - get() = this !is State.Initialized && - this !is State.ModelReady && - this !is State.Generating && - this !is State.Error + get() = this is State.Initializing || + this is State.LoadingModel || + this is State.UnloadingModel || + this is State.Benchmarking || + this is State.ProcessingSystemPrompt || + this is State.ProcessingUserPrompt val State.isModelLoaded: Boolean - get() = this !is State.Uninitialized && - this !is State.Initializing && - this !is State.Initialized && - this !is State.LoadingModel && - this !is State.UnloadingModel + get() = this is State.ModelReady || + this is State.Benchmarking || + this is State.ProcessingSystemPrompt || + this is State.ProcessingUserPrompt || + this is State.Generating class UnsupportedArchitectureException : Exception()