Skip to content

feat: Add performance data collectors for Mamba2 #297

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

Jump to bottom
Draft
Arsene12358 wants to merge 2 commits into
base: main
Choose a base branch
from
Draft

feat: Add performance data collectors for Mamba2 #297

Arsene12358 wants to merge 2 commits into from

Conversation

@Arsene12358
Copy link
Contributor

@Arsene12358 Arsene12358 commented Jan 27, 2026

Overview:

  1. Add data collector for mamba prefill
  2. Add data collector for mamba decode

yimingl (generated by with_the_same_user script) added 2 commits January 27, 2026 13:42
Add collectors for mamba2
7312a21 
Signed-off-by: yimingl (generated by with_the_same_user script) <yimingl@gb-nvl-147-compute02.nvidia.com>
fix shape and runtime error
5410a02 
Signed-off-by: yimingl (generated by with_the_same_user script) <yimingl@gb-nvl-147-compute02.nvidia.com>
@copy-pr-bot
Copy link

copy-pr-bot bot commented Jan 27, 2026

This pull request requires additional validation before any workflows can run on NVIDIA's runners.

Pull request vetters can view their responsibilities here.

Contributors can view more details about this message here.

@github-actions github-actions bot added the feat label Jan 27, 2026
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@YijiaZhao YijiaZhao Awaiting requested review from YijiaZhao YijiaZhao will be requested when the pull request is marked ready for review YijiaZhao is a code owner

@tianhaox tianhaox Awaiting requested review from tianhaox tianhaox will be requested when the pull request is marked ready for review tianhaox is a code owner

@jasonqinzhou jasonqinzhou Awaiting requested review from jasonqinzhou jasonqinzhou will be requested when the pull request is marked ready for review jasonqinzhou is a code owner

At least 1 approving review is required to merge this pull request.

Assignees

No one assigned

Labels

feat

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

1 participant

@Arsene12358