Skip to content
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

Fix 2pass sdpa kernel #73

Merged
merged 3 commits into from
Mar 1, 2025
Merged

Fix 2pass sdpa kernel #73

merged 3 commits into from
Mar 1, 2025

Conversation

brrr
Copy link

@brrr brrr commented Feb 16, 2025

2pass sdpa kernel miss some pieces of code, it will make sdpa_vector_2pass_1 no output.
At the mean time, add a unit test for GQA-2pass sdpa.

Thanks for your greet work!

@EricLBuehler
Copy link
Owner

Hi @brrr! Thanks for the PR.

I'm testing it now.

Copy link
Owner

@EricLBuehler EricLBuehler left a comment

Choose a reason for hiding this comment

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

Thank you.

@EricLBuehler EricLBuehler merged commit 76819e8 into EricLBuehler:main Mar 1, 2025
8 of 10 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.

2 participants