Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
Show all changes
41 commits
Select commit Hold shift + click to select a range
74bc13f
use BTreeSet, and allow for push_front (preemption)
PeaBrane May 27, 2025
f2343d5
preemption is push_front
PeaBrane May 27, 2025
6fe3154
use Hongkuan's quadratic formulas for decode and prefill
PeaBrane May 27, 2025
cccebad
cleaner scheduling + generation separation, and waterline bug fix
PeaBrane May 28, 2025
793d1d1
Merge branch 'main' into rupei/mocker-v0
PeaBrane May 28, 2025
394c2bf
restore printing out fwd pass metrics in test
PeaBrane May 30, 2025
f5ab2e1
Merge remote-tracking branch 'origin/main' into rupei/mocker-v0
PeaBrane Jun 11, 2025
dad183f
multi-dp mocker engine
PeaBrane Jun 11, 2025
009ec78
fixed prefill cost, and more conservative watermarking
PeaBrane Jun 12, 2025
ee11427
fwd pass metrics
PeaBrane Jun 12, 2025
8e8d0b4
can emit kv event, not tested
PeaBrane Jun 13, 2025
e96f810
move block resp test in kv manager
PeaBrane Jun 13, 2025
c09f007
basic test passes for both load metrics and kv events
PeaBrane Jun 14, 2025
4502e5e
better tracing
PeaBrane Jun 14, 2025
fe20aa3
async engine core
PeaBrane Jun 16, 2025
2fbf998
hook up with dynamo run
PeaBrane Jun 17, 2025
b548050
docs
PeaBrane Jun 17, 2025
c7c4be5
fmt
PeaBrane Jun 17, 2025
1845a8d
Merge branch 'main' into rupei/mocker-v0
PeaBrane Jun 17, 2025
3ad7780
refactor
PeaBrane Jun 17, 2025
c78bef2
works with kv router
PeaBrane Jun 17, 2025
a206569
actually load extra mocker args in guide
PeaBrane Jun 17, 2025
d3730ff
free blocks if failed to send (receiver dropped)
PeaBrane Jun 23, 2025
68d822a
do not regenereate tokens after pre-emption
PeaBrane Jun 23, 2025
d69edcf
evictor cleanup
PeaBrane Jun 30, 2025
c08f9ea
only need runtime in dynamic arms
PeaBrane Jun 30, 2025
dee1413
no separate extra-mocker-args
PeaBrane Jun 30, 2025
082bcec
Merge branch 'main' into rupei/mocker-v0
PeaBrane Jun 30, 2025
99fd3f2
update to match batched tokens
PeaBrane Jun 30, 2025
85c7ccf
max-num-seqs
PeaBrane Jun 30, 2025
ec1f360
enable_prefix_caching arg
PeaBrane Jun 30, 2025
94abc0d
only publish kv events if enable_prefix_caching set true
PeaBrane Jun 30, 2025
35da284
small note on chunked prefill being false for now
PeaBrane Jun 30, 2025
c7c072d
revert flags
PeaBrane Jul 1, 2025
de54247
revert dynamo-run changes
PeaBrane Jul 1, 2025
81c12aa
tiny reversion
PeaBrane Jul 1, 2025
b959df4
another reversion
PeaBrane Jul 1, 2025
f07e28d
Merge remote-tracking branch 'origin/main' into rupei/mocker-v0
PeaBrane Jul 1, 2025
b15070a
usize reversion
PeaBrane Jul 1, 2025
3a20b9d
clippy
PeaBrane Jul 1, 2025
c747606
more clippy
PeaBrane Jul 1, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
1 change: 1 addition & 0 deletions lib/llm/src/mocker.rs
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
// See the License for the specific language governing permissions and
// limitations under the License.

pub mod engine;
pub mod evictor;
pub mod kv_manager;
pub mod protocols;
Expand Down
Loading
Loading