fit: warn when --cpu-moe is used on integrated/UMA GPUs; docs: add Jetson MTP benchmark#25883
fit: warn when --cpu-moe is used on integrated/UMA GPUs; docs: add Jetson MTP benchmark#25883hernandez42 wants to merge 1 commit into
Conversation
docs: add Jetson AGX Xavier MTP benchmark data On iGPU/UMA systems (NVIDIA Jetson, Intel ARC iGPUs, AMD APUs), moving MoE weights to system memory with --cpu-moe forces their computation onto the CPU, which is ~20x slower than GPU compute. Add a runtime warning when --cpu-moe would move MoE layers to the CPU on an integrated GPU device. Benchmark data from Jetson AGX Xavier (Volta 512-core, ~135 GB/s unified memory): Q3_K_M 35B MoE model with MTP heads achieves 23.14 t/s (+48% over baseline) at 97.7% draft acceptance rate. Key findings documented for the iGPU/embedded community.
|
Hi @hernandez42, thanks for your contribution! Per our contribution guidelines, the automated PR checker found the following issue(s) that need your attention:
Please note that maintainers reserve the right to make final decisions on PRs. If you believe there is a mistake, please comment below. |
|
Thanks for the automated check!
Happy to make any further changes needed to get this merged. |
Summary
Two changes:
1.
common/fit.cpp— iGPU warning for --cpu-moeWhen
--cpu-moewould move MoE experts out of GPU memory, add a runtime warning if the device is an integrated GPU (GGML_BACKEND_DEVICE_TYPE_IGPU). On iGPU/UMA systems such as the NVIDIA Jetson AGX Xavier, moving MoE weights to system memory forces their computation onto the CPU, which is ~20x slower than GPU compute (~6 t/s vs ~23 t/s). The warning suggests omitting--cpu-moeon integrated GPUs.2.
docs/speculative.md— Jetson AGX Xavier MTP benchmark dataAdds a new
Multi Token Prediction (draft-mtp)section with real-world benchmark data from an NVIDIA Jetson AGX Xavier (Volta 512-core, ~135 GB/s unified memory, 32 GB shared RAM, 30W TDP).Key findings documented:
--speculative-max 2achieves 23.14 t/s (+48%) with 97.7% draft acceptance rate--speculative-max 3produces ~75% acceptance on embedded GPUs — lowering to 2 is optimal--cpu-moeon iGPU/UMA systems causes ~4x slowdown vs GPU-only (6 vs 23 t/s)Motivation
NVIDIA Jetson devices (AGX Xavier, AGX Orin, NX, Nano) are increasingly popular for local LLM inference. The embedded GPU community lacks documented MTP performance data and can easily hit the
--cpu-moeperformance trap. These changes help users on integrated GPUs avoid common pitfalls.