<< All versions

Skill v1.0.1

currentAutomated scan100/100
google-ai-edge/litert-samples/compiled-model-app-scaffolding

+1 new

──Details
PublishedOctober 28, 2026 at 09:07 AM
Content Hashsha256:20c73050a04e1de6...
Git SHAcf837da83f55
Bump Typepatch
Compare with v1.0.0
──Files
Files (1 file, 9.2 KB)
SKILL.md9.2 KBactive
SKILL.md · 175 lines · 9.2 KB

name: compiled-model-app-scaffolding description: Build a new Android app (Kotlin, Compose) around a verified LiteRT model using the CompiledModel API - the app architecture, the inference-layer lifecycle rules, model delivery, and the UI traps that masquerade as model bugs. Use when turning a converted and device-verified model into a demo or product app, when an app's inference layer leaks memory or blocks the UI, or when a model that verified clean looks wrong inside an app. metadata: version: "1.0.1"


CompiledModel app scaffolding

An app around a verified model is done when three things hold, in this order:

  1. the app reproduces the model recipe's verification numbers on the

accelerator the recipe verified — before any UI exists,

  1. inference is confined and leak-free: one dispatcher owns the model,

every buffer is closed, benchmarks include the readback,

  1. the inference code is liftable — another app could take the helper

file unchanged.

Scope: this scaffolds a new app from a model recipe. Migrating an existing TFLite Interpreter app to CompiledModel is a different task with its own skill, litert-compiled-model-migration. LLMs are served through the LiteRT-LM Engine rather than raw CompiledModel — samples/litert/text_to_speech_lm is the reference for that case; everything below is the non-LLM CompiledModel app.

Step 0: prove parity before building UI

The first milestone has no UI: a bare harness that loads the .tflite, runs the recipe's verification input on the target accelerator, and reproduces the numbers recorded in the recipe README (correlation, residency). If they do not reproduce, stop — that is an on-device-verification problem, and no amount of app code fixes it. Only then build the ViewModel and the screen.

The shape

One sample = one standalone Gradle project, four layers:

app/src/main/java/<pkg>/
<Task>Helper.kt inference infra: owns the CompiledModel and its
buffers; pre/post-processing; NO Android UI types
MainViewModel.kt state machine: drives the helper on a confined
dispatcher, exposes UiState
UiState.kt one immutable data class
MainActivity.kt ComponentActivity + setContent, nothing else
view/ Screen.kt, Theme.kt, Color.kt — Compose only
app/src/main/res/values/ strings.xml etc. — no UI strings in Kotlin

The worked example of the full shape is samples/litert/image_segmentation/kotlin_cpu_gpu/android (litert 2.3.0: the GPU accelerator ships in litert-gpu, so declare litert:2.3.0 and litert-gpu:2.3.0; without the second the runtime logs GPU accelerator could not be loaded and registered and a GPU + CPU set runs on the CPU — Galaxy S26. The sample keeps Kotlin 2.2.21 with -Xskip-metadata-version-check; the 2.3.0 classes carry Kotlin 2.4 metadata). The layer boundary that matters most is the helper's: pre/post-processing is part of the model contract (it must match what the recipe exported against), so it lives with the model, not in the screen. Reusable inference code is worth more than UI polish — a reader will lift <Task>Helper.kt and delete the rest.

Inference-layer rules

  1. One serial dispatcher owns the model.

Dispatchers.IO.limitedParallelism(1, "ModelDispatcher") in the helper (the named form needs kotlinx-coroutines 1.9+; litert-api 2.3.0 pulls in 1.8.0); create, run, and close the model only inside withContext(modelDispatcher) — it serializes calls without pinning a thread. Not a bare executor in the Activity, and never the main thread.

  1. Buffers are created once, reused every frame, and closed.

TensorBuffer is AutoCloseable; forgetting the buffers leaks native memory even when the model itself is closed. Close in onCleared() on the dispatcher of rule 1 (viewModelScope is already cancelled there, so a launch never runs). The vendored CompiledModelRunner (below) gets the buffer lifecycle right.

  1. `run()` enqueues; the readback waits. run() may return before

the GPU finishes — the output read is the synchronization point. Benchmark run + readback together, never run() alone.

  1. Warm up once at init. The first GPU inference includes shader

compilation. Run one inference on dummy input right after create, so the first user action is not the compile, and no first-run number ever gets quoted as latency.

  1. Ask for the strict accelerator the recipe verified.

Accelerator.GPU alone makes create throw when the GPU cannot take the graph (Kotlin LiteRtException Failed to compile model, Galaxy S26, litert 2.2.0 and 2.3.0). Surface it as a visible error and route it back to the model recipe; do not paper over it with a CPU fallback that makes a slow app look like a working one. The exception: a node the delegate declines while building (TopK requires src tensor C dimension to be divisible by 4, 2.2.0 and 2.3.0 wheels on a Mac) runs on the CPU with only a log line — also read the runtime's operations will run on the GPU line.

  1. Stateful and multi-graph models: move references, not data. Feed

step N's output buffers as step N+1's inputs (buffer ping-pong) instead of copying state through the host. When one app creates many CompiledModels (pipelines, chunked models), create one Environment and pass it to every CompiledModel.create call, and close per-run buffers. An Environment created and closed per model reloads the GPU accelerator library each time (one reload aborted the process, once in 50 loads, Galaxy S26, litert 2.2.0). Treat the GPU program-cache option (GpuOptions key SERIALIZE_PROGRAM_CACHE) as untested per device, and note that the compiler-cache environment option targets NPU JIT, not GPU shader caching.

Vendor the helpers, don't rewrite them

utilities/common/kotlin/ holds the canonical copies of the code every app needs and every app gets subtly wrong when written from scratch:

FileWhat it provides
CompiledModelRunner.ktthe buffer lifecycle of rules 2–3 (its factories take no Environment; create the model yourself when rule 6 applies)
ImageTensor.ktBitmap → float tensor; mean/std, NCHW/NHWC, RGB/BGR, letterbox with coordinate mapping back
AudioCapture.kt16 kHz mono AudioRecord loop delivering float chunks
RealtimeCameraPipeline.ktCameraX capture → pooled Bitmap incl. rotation
MathOps.ktsoftmax, argmax, IoU, NMS

Vendor a copy with utilities/tools/sync_common.py --apply (it writes the provenance line), keep model-specific values in constructor arguments, and fix bugs in the canonical first — --check is the drift gate (exit 1 on a diverged copy). The preprocessing arguments are the model contract: a wrong mean/std or RGB/BGR looks exactly like a broken model, and it is the first thing to diff against the recipe's export script when app output is subtly wrong.

Model delivery

Weights are never committed.

  • Bundleable models: fetch at build with a download_model.gradle

(Hugging Face URL → assets/), and set androidResources { noCompress += "tflite" } (Kotlin DSL) so the asset stays mmappable. Worked example: samples/litert/image_segmentation/kotlin_cpu_gpu/android.

  • Models too big to bundle: stage into the app's private filesDir

with an install_to_device.sh (adb push to /data/local/tmp, then run-as <pkg> cp, debug builds only), and load with the from-file path. Worked example: samples/litert/text_to_speech/kotlin_cpu_gpu/android.

Link the model recipe (models/<family>/<model>/) from the app README; conversion and verification scripts belong to the recipe, not the app.

UI traps that masquerade as model bugs

  • The tiny-output trap. Image(contentScale = Fit) with

fillMaxWidth() inside a verticalScroll renders the bitmap at native pixel size: with no height constraint the row is exactly as tall as the bitmap and Fit never upscales, so a 256-px model output looks broken on a 1080-px screen. Drop the scroll and give each image weight(1f) in the Column.

  • Audio I/O belongs to the ViewModel. Mic capture and AudioTrack

playback run on their own serial dispatcher in the ViewModel, not the model's (a blocking AudioRecord.read loop would hold inference); the screen only requests permission (rememberLauncherForActivityResult) and renders state. For file input use OpenDocument with audio MIME types — the photo picker cannot see audio.

  • A "slow model" is often the UI thread. If the app stutters, check

that every helper call site goes through the ViewModel's dispatcher before profiling the model — a verified model that benchmarked fine in step 0 did not get slower by being in an app.

Output layout

samples/litert/<task>/kotlin_cpu_gpu/android/
app/src/main/java/... the four layers above
app/download_model.gradle or install_to_device.sh at the app root
README.md what it demos, which model recipe it consumes,
device + accelerator it was verified on

State the device the app was verified on, the same way the model recipe does. An app README that names no device inherits none of the recipe's verification.

← v1.0.0All versions