msluszniak commited on
Commit
8e2a215
·
verified ·
1 Parent(s): f9beeac

perf(coreml): run decode on the CPU, not the Neural Engine

Browse files

decode is a single-token program whose cost on device is dispatch rather than arithmetic, so it was paying Neural Engine overhead on every call under ExecuTorch's ALL default. Pinning only decode to CPU_ONLY (encode stays on ALL) measures 6.7 ms/step against 20.0 on an iPhone 16.

On 14.5s of real speech the whole transcribe is 1.5-3.1x faster depending on size, with a byte-identical transcript: this removes overhead, it does not change numerics.

Exported from export-scripts MR !18, built on main pins (ExecuTorch 1.4.1). Replaces an ExecuTorch 1.2.0 build of the same change; transcripts verified byte-identical on device at all six sizes.

coreml/whisper_tiny_coreml_fp16.pte CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:8a743e18747f925b49acea2a2be62c10a037cc2d6d63e3cb3740f08cdf9b70f5
3
- size 76478434
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e3859a444c730926104c63b0fd5756190f4d3004ed9141cb3ba20549a829dc75
3
+ size 76477794