Skip to content

feat: add support for MTP+cuda_graph_padding #3096

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.

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

Merged

Conversation

lfr-0531
Copy link
Collaborator

Add support for MTP+cuda_graph_padding=True

@lfr-0531
Copy link
Collaborator Author

/bot run

@niukuo
Copy link
Collaborator

niukuo commented Mar 26, 2025

PR_Github #545 [ run ] triggered by Bot

@niukuo
Copy link
Collaborator

niukuo commented Mar 26, 2025

PR_Github #545 [ run ] completed with state SUCCESS
/LLM/main/L0_MergeRequest_PR pipeline #463 completed with status: 'SUCCESS'

@lfr-0531 lfr-0531 requested a review from dongxuy04 March 27, 2025 05:21
Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>
@lfr-0531 lfr-0531 force-pushed the user/fanrongl/add_mtp_cuda_graph_padding branch from 32fec39 to 7b8bbda Compare March 27, 2025 05:45
@lfr-0531
Copy link
Collaborator Author

/bot run

@tensorrt-cicd
Copy link
Collaborator

PR_Github #645 [ run ] triggered by Bot

@tensorrt-cicd
Copy link
Collaborator

PR_Github #645 [ run ] completed with state SUCCESS
/LLM/main/L0_MergeRequest_PR pipeline #544 completed with status: 'SUCCESS'

@dongxuy04 dongxuy04 merged commit 0976360 into NVIDIA:main Mar 27, 2025
2 checks passed
@lfr-0531 lfr-0531 deleted the user/fanrongl/add_mtp_cuda_graph_padding branch April 29, 2025 03:25
wu1du2 pushed a commit to wu1du2/TensorRT-LLM that referenced this pull request May 11, 2025
Signed-off-by: Fanrong Li <23290157+lfr-0531@users.noreply.github.com>
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.

4 participants