Skip to content
New issue

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

#

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

Already on GitHub? # to your account

Jump to bottom

add mamba causal-conv1d-update kernel #48

Open
wants to merge 2 commits into
base: main
Choose a base branch
from
Open

add mamba causal-conv1d-update kernel #48

wants to merge 2 commits into from

Conversation

thoangtrvn
Copy link
Contributor

@thoangtrvn thoangtrvn commented Mar 21, 2025 •
edited
Loading

This is related to the prior PR: #47

This adds the second Triton kernel (decode stage) to be used in mamba-based model for the inference purpose.

@facebook-github-bot facebook-github-bot added the CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. label Mar 21, 2025
# for free to join this conversation on GitHub. Already have an account? # to comment
Reviewers
No reviews
Assignees
No one assigned
Labels
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
Projects
None yet
Milestone
No milestone
Development

Successfully merging this pull request may close these issues.
3 participants
@thoangtrvn @facebook-github-bot @tmhoangt