Commit b0737f7
committed
tenstorrent: disable paged_fused_update_cache prefill path (K/V overlap)
paged_fused_update_cache rejects overlapping K/V sharded input grids
("input_tensor1 and input_tensor2 must not overlap"). Both K and V shards
land on the same C cores via MakeHeightShardedUpdateInput. Make
TryDevicePagedFusedUpdateBatch return false so TryDevicePagedPushPair falls
back to two separate TryDevicePagedUpdateBatch calls (K then V), each on
its own core set.
Pre-existing issue in the VT_TT_HOST_FREE_DECODE prefill path. The fused
path is an optimization; the separate-call fallback is correct (just 2x
the launches). A proper fix would shard K and V on disjoint core sets.
Following-Agents-Protocol: true
AI-Assisted: true
Assisted-by: Maki:mistral/zai-glm-5-2 [Maki]1 parent 95f20ad commit b0737f7
4 files changed
Lines changed: 26 additions & 65 deletions
File tree
- scripts
- src
- capi
- vllm/model_executor/models
- vt/tenstorrent
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
299 | 299 | | |
300 | 300 | | |
301 | 301 | | |
302 | | - | |
303 | | - | |
304 | | - | |
305 | | - | |
306 | | - | |
307 | | - | |
308 | | - | |
309 | | - | |
| 302 | + | |
| 303 | + | |
| 304 | + | |
| 305 | + | |
| 306 | + | |
| 307 | + | |
| 308 | + | |
| 309 | + | |
| 310 | + | |
| 311 | + | |
| 312 | + | |
| 313 | + | |
| 314 | + | |
| 315 | + | |
| 316 | + | |
| 317 | + | |
310 | 318 | | |
311 | 319 | | |
312 | 320 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
502 | 502 | | |
503 | 503 | | |
504 | 504 | | |
505 | | - | |
506 | | - | |
507 | | - | |
508 | | - | |
509 | | - | |
510 | | - | |
511 | | - | |
512 | | - | |
513 | | - | |
514 | | - | |
515 | | - | |
516 | | - | |
517 | | - | |
518 | | - | |
519 | | - | |
520 | | - | |
521 | 505 | | |
522 | 506 | | |
523 | 507 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
950 | 950 | | |
951 | 951 | | |
952 | 952 | | |
953 | | - | |
| 953 | + | |
954 | 954 | | |
955 | 955 | | |
956 | 956 | | |
| |||
| Original file line number | Diff line number | Diff line change | |
|---|---|---|---|
| |||
617 | 617 | | |
618 | 618 | | |
619 | 619 | | |
620 | | - | |
| 620 | + | |
621 | 621 | | |
622 | 622 | | |
623 | | - | |
624 | | - | |
625 | | - | |
626 | | - | |
627 | | - | |
628 | | - | |
629 | | - | |
630 | | - | |
631 | | - | |
632 | | - | |
633 | | - | |
634 | | - | |
635 | | - | |
636 | | - | |
637 | | - | |
638 | | - | |
639 | | - | |
640 | | - | |
641 | | - | |
642 | | - | |
643 | | - | |
644 | | - | |
645 | | - | |
646 | | - | |
647 | | - | |
648 | | - | |
649 | | - | |
650 | | - | |
651 | | - | |
652 | | - | |
653 | | - | |
654 | | - | |
655 | | - | |
656 | | - | |
657 | | - | |
658 | | - | |
659 | | - | |
660 | | - | |
661 | | - | |
| 623 | + | |
| 624 | + | |
| 625 | + | |
| 626 | + | |
| 627 | + | |
| 628 | + | |
| 629 | + | |
| 630 | + | |
662 | 631 | | |
663 | 632 | | |
664 | 633 | | |
| |||
0 commit comments