[V1] [Hybrid] Some additional clean-up in Mamba2 prefix caching #26222
seq_lens_pending -> query_lens
7579fdd1
Simplification
1548bf76
Remove line break
141258a0
Remove line break
a517536c
Remove redundant branch
4a4ae236
Cleanup code for writing mamba states
561b5f78
Remove last_computed_offset
f8857640
context_lens -> num_computed_tokens
e3699c90
Fix indexing bug
bddbedaf
Fix issue for eager mode
b16a7883
Rename tensors
f671288d
Fix small bug
9a37a6a4
add shapes
1d20e4e8
Review comments
9bdd0c89
Simplify logic a bit
9f2e99c3
Merge branch 'main' into mamba-prefix-cache-postmerge
d0f23588
Merge branch 'main' into mamba-prefix-cache-postmerge
752b58a4
format my files
b5df5fbf
merge main
64024373
heheda12345
deleted the mamba-prefix-cache-postmerge branch 130 days ago
Assignees
No one assigned
Login to write a write a comment.
Login via GitHub