Android + OpenCL implementation — runs on-device on non-flagship phones (Adreno 6xx)

#23
by a8nova - opened

Hi! I wanted to share an Android + OpenCL implementation of FunctionGemma-270M-IT, in case anyone wants to run it on a phone:

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 it decodes at ~8.9 tok/s in fp16, and greedy output is bit-identical to the PyTorch reference.

Hope it's useful — happy to answer questions!

Sign up or log in to comment