Skip to content

feat: rocm python mtp#668

Merged
LLLLKKKK merged 6 commits intomainfrom
yilin/rocm_python_mtp
Mar 18, 2026
Merged

feat: rocm python mtp#668
LLLLKKKK merged 6 commits intomainfrom
yilin/rocm_python_mtp

Conversation

@amd-yilizhao
Copy link
Copy Markdown
Collaborator

@amd-yilizhao amd-yilizhao commented Feb 3, 2026

Add ROCM Python MTP support
Triton Paged Attention for ROCM platform to support speculative decode

@wht21
Copy link
Copy Markdown
Collaborator

wht21 commented Feb 3, 2026

internal source has been updated, please review the changes!

@wht21
Copy link
Copy Markdown
Collaborator

wht21 commented Feb 24, 2026

internal source has been updated, please review the changes!

@amd-yilizhao amd-yilizhao force-pushed the yilin/rocm_python_mtp branch 4 times, most recently from c2ccfdd to a649521 Compare March 2, 2026 08:07
@liaocz liaocz force-pushed the yilin/rocm_python_mtp branch from a649521 to 9fb17e9 Compare March 5, 2026 07:19
@amd-yilizhao amd-yilizhao force-pushed the yilin/rocm_python_mtp branch from 9fb17e9 to 3bbc43c Compare March 9, 2026 10:47
@liaocz liaocz force-pushed the yilin/rocm_python_mtp branch 4 times, most recently from 521f0e1 to 3e6387b Compare March 15, 2026 10:03
@liaocz liaocz force-pushed the yilin/rocm_python_mtp branch from 3e6387b to 1a5714e Compare March 17, 2026 06:43
@LLLLKKKK LLLLKKKK merged commit cf76cab into main Mar 18, 2026
7 of 8 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

5 participants