Skip to content

[Optimization] Improve MTP module test coverage #5729

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
kesmeey wants to merge 27 commits into
base: develop
Choose a base branch
from
Open

[Optimization] Improve MTP module test coverage #5729

Changes from all commits
Commits
Show all changes
27 commits
Select commit Hold shift + click to select a range
f839869
Add unit tests for MTPProposer class in spec_decode/mtp.py
kesmeey Dec 12, 2025
f771cb1
fix: remove non-existent QuantizationConfig import in test_mtp_proposer
kesmeey Dec 12, 2025
8520193
fix: add logprobs_mode attribute to FakeModelConfig
kesmeey Dec 12, 2025
dd3513d
fix: fix test failures in test_mtp_proposer - fix Mock setup, remove …
kesmeey Dec 12, 2025
6fe7374
fix: add seq_lens_this_time initialization and kv_cache init before i…
kesmeey Dec 13, 2025
7a6a1ed
fix: check pos_emb_type attribute existence before assertion
kesmeey Dec 13, 2025
636e7c6
test: add minimal coverage for mtp cache type, mm init, preempted
kesmeey Dec 15, 2025
7f6273f
test: fix cache_type_branches unsupported platform on 12
kesmeey Dec 15, 2025
59ca57d
test: refine MTPProposer tests for cache type, requests and chunked p…
kesmeey Dec 16, 2025
b6ef68f
chore: remove stray spec_decode copy
kesmeey Dec 16, 2025
703ae7c
improve coverage
kesmeey Dec 23, 2025
07e5bc8
Merge branch 'develop' into 12
kesmeey Dec 23, 2025
f73b6e9
fix bugs
kesmeey Dec 23, 2025
3f850f3
Revert "fix bugs"
kesmeey Dec 23, 2025
8d75bf7
fix bugs
kesmeey Dec 23, 2025
c733ad2
fix bugs
kesmeey Dec 24, 2025
e0d6d84
Merge branch 'develop' into 12
EmmonsCurse Dec 24, 2025
dca5df7
Add test_propose_cuda_path to improve mtp.py coverage
kesmeey Dec 24, 2025
6ade2f0
Fix sampler mock in test_propose_cuda_path
kesmeey Dec 24, 2025
db8853c
Add comprehensive test coverage for mtp.py
kesmeey Dec 24, 2025
7bf1a98
Fix failing tests in test_mtp_proposer.py
kesmeey Dec 24, 2025
a72561f
Fix test_chunked_prefill_enabled: ensure seq_lens_this_time is initia…
kesmeey Dec 24, 2025
9ac655f
Add comprehensive test coverage for remaining mtp.py branches
kesmeey Dec 25, 2025
c86feb3
Fix test_xpu_propose_path: correct XPU function mocks
kesmeey Dec 25, 2025
b475218
Fix test_xpu_propose_path: mock XPU functions in mtp module
kesmeey Dec 25, 2025
b2c9159
Remove complex XPU test that was difficult to implement correctly
kesmeey Dec 25, 2025
6fcbd86
Merge branch 'develop' into 12
kesmeey Dec 26, 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
Loading
Loading