Commit caff39a
committed
docs(megatron-vs-mlx): direct seq=2048/4096 path_b measurement — fits-but-impractically-slow
Proven by direct gb10 runs (mamba3-chunk + grad-checkpoint-correctness fixes, no grad-checkpoint):
- bs1×seq1024: 191 tok/s, 33G — largest practical point.
- bs1×seq2048: peak 76G (fits 100G budget) but ~5min/step, did not finish 6 steps in 30min.
- bs1×seq4096: 93G steady/107G peak (>100G budget) + ~5min/step.
The chunked backward that saves memory is compute-heavy → MLX-eager trades OOM for impractical
slowness at long seq. Literal bs4×seq4096 MLX number is fundamentally impractical, now MEASURED
not assumed. C++/Megatron = 3399 tok/s live @ ~26G. MLX ceiling = 191 tok/s @ bs1×seq1024.1 parent 16d1c44 commit caff39a
1 file changed
Lines changed: 23 additions & 0 deletions
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
217 | 217 | | |
218 | 218 | | |
219 | 219 | | |
| 220 | + | |
| 221 | + | |
| 222 | + | |
| 223 | + | |
| 224 | + | |
| 225 | + | |
| 226 | + | |
| 227 | + | |
| 228 | + | |
| 229 | + | |
| 230 | + | |
| 231 | + | |
| 232 | + | |
| 233 | + | |
| 234 | + | |
| 235 | + | |
| 236 | + | |
| 237 | + | |
| 238 | + | |
| 239 | + | |
| 240 | + | |
| 241 | + | |
| 242 | + | |
220 | 243 | | |
221 | 244 | | |
222 | 245 | | |
| |||
0 commit comments