Skip to content

feat: add Hy3 support, MTP, and FP8 optimizations - #4815

Open
yidingcheng0206 wants to merge 21 commits into
InternLM:mainfrom
yidingcheng0206:feat/hy3-mtp-submit
Open

feat: add Hy3 support, MTP, and FP8 optimizations#4815
yidingcheng0206 wants to merge 21 commits into
InternLM:mainfrom
yidingcheng0206:feat/hy3-mtp-submit

Conversation

@yidingcheng0206

Copy link
Copy Markdown
Collaborator

Draft PR for Hy3 model support, embedded MTP speculative decoding, static FP8 support, and opt-in kernel optimizations. Detailed description and known issues will be added next.

@yidingcheng0206
yidingcheng0206 marked this pull request as ready for review August 3, 2026 12:13
@lvhan028 lvhan028 added the enhancement New feature or request label Aug 3, 2026
@lvhan028
lvhan028 requested a review from grimoire August 4, 2026 06:08
Comment thread lmdeploy/pytorch/backends/cuda/attention/fa3.py Outdated
Comment thread lmdeploy/pytorch/weight_loader/model_weight_loader.py Outdated
Comment thread lmdeploy/pytorch/backends/cuda/static_fp8_modules.py Outdated
Comment thread lmdeploy/pytorch/backends/cuda/static_fp8_modules.py Outdated
main_stream = torch.cuda.current_stream(
hidden_states.device,
)
shared_stream = _get_hy3_shared_expert_stream(

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

I can accept this for this PR. But this implementation is dirty and device related.
We should find a way to make this an engine feature so it can be used for different model/backend.

Comment thread tests/pytorch/test_model_weight_loader.py Outdated
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

enhancement New feature or request

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants