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

[kernels] CUDA version of block_sparse_attn #3

Merged
merged 6 commits into from
Feb 16, 2025

Conversation

Zhichenzzz
Copy link
Collaborator

This PR introduces block sparse attention kernel to our SeerAttention using CUDA/CUTLASS, resulting in enhanced computational efficiency compared with previous Triton-based one.

  • Kernel efficiency test benchmark
  • Add this kernel in modeling_llama_seerattn.py

@Zhichenzzz
Copy link
Collaborator Author

@Zhichenzzz please read the following Contributor License Agreement(CLA). If you agree with the CLA, please reply with the following information.

@microsoft-github-policy-service agree [company="{your company}"]

Options:

  • (default - no company specified) I have sole ownership of intellectual property rights to my Submissions and I am not making Submissions in the course of work for my employer.
@microsoft-github-policy-service agree
  • (when company given) I am making Submissions in the course of work for my employer (or my employer has intellectual property rights in my Submissions by contract or applicable law). I have permission from my employer to make Submissions and enter into this Agreement on behalf of my employer. By signing below, the defined term “You” includes me and my employer.
@microsoft-github-policy-service agree company="Microsoft"

Contributor License Agreement

@microsoft-github-policy-service agree

@Zhichenzzz Zhichenzzz requested a review from YizhaoGao February 16, 2025 05:52
@Zhichenzzz Zhichenzzz marked this pull request as draft February 16, 2025 05:59
@Zhichenzzz Zhichenzzz marked this pull request as ready for review February 16, 2025 06:11
@Zhichenzzz Zhichenzzz self-assigned this Feb 16, 2025
@YizhaoGao YizhaoGao merged commit 9a8a3ba into microsoft:main Feb 16, 2025
3 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