Android + OpenCL implementation — text-to-music on-device on non-flagship phones (Adreno 6xx)
#36
by a8nova - opened
Hi! I wanted to share an Android + OpenCL implementation of MusicGen-small, in case anyone wants to run it on a phone:
- Try it: Edgi on Google Play — runs fully on-device, no cloud.
- Open source: the app is built on top of the open-source adreno-llms inference engine — https://github.com/a8nova/adreno-llms — pure C++/OpenCL with hand-written kernels tuned for Adreno, running the full T5 + token LM + EnCodec pipeline on the GPU.
It's tuned and tested on Adreno 6xx GPUs — the GPU class in mid-range and older Android phones (verified on a 2020 Motorola Razr / Adreno 620) — and should run on most arm64 Android phones with OpenCL, though the optimizations are Adreno-specific. On the Adreno 620 (an entry-tier GPU) it decodes at ~11 tok/s — a 5 s clip takes ~34 s wall; faster Adreno tiers close the gap toward real time.
Hope it's useful — happy to answer questions!