speculative: draft-mtp for dots3note - #27618
Conversation
|
Marking this draft while I do more code review and testing -- happy to accept comments meanwhile though :) |
|
@ngxson since you wrote #27060 would you have time to review this? Or maybe @fairydreaming |
|
your PR says that it's related to dots3note, but file changes extended to 10 more models. I cannot review this PR as-is per contribution guidelines, unrelated changes require separate PR. |
sorry, I had done a TODO to refactor out some code instead of copy-n-paste for the 10th time. changes to other models reverted, now patch is just dots3note and core files like llama-graph & speculative. |
Feel free to submit that as a follow-up PR. Would be appreciated. :) |
Assisted-by: deepseek-v4-flash
Overview
draft-mtp support for dots3note model arch.
Not compatible with vision/mtmd -- hits existing TODO in common/speculative.cpp.
Additional information
Tested, gives sizable performance boost on Strix Halo (10t/s -> 14-16t/s). Acceptance rate is pretty good on some coding prompts I tried.
Tested w/ both ROCm and Vulkan.
Also tested CUDA with layer split against 2x RTX A5000, ofc heavy offload (-ncmoe 22), but still meaningful boost:
Requirements