3 ms·
Routing in a MoE model might fit.
by naasking 6mo ago
Routing in a MoE model might fit.
- zozbot234 6mo agoYou want routing to be as quick as possible, because there are dependent loads of expert MoE weights (at least from CPU in most setups, potentially from storage) downstream of it. So that ultimately depends on what the bottleneck on that part of the model is: compute, memory throughput or both? If it's throughput, the NPU might be a bad fit.