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

[Feature] Cascade attention kernels #1715

Open

merrymercy opened this issue Oct 19, 2024 · 2 comments

Labels

good first issue

Contributor

merrymercy commented Oct 19, 2024 •

edited

Loading

We would like to integrate the cascade attention kernel from flashinfer.

Code pointers:

Attention backend in sglang: https://github.com/sgl-project/sglang/blob/main/python/sglang/srt/layers/attention/flashinfer_backend.py
Usage of cascade: https://docs.flashinfer.ai/api/python/cascade.html

merrymercy added the good first issue label

ZelinMa557 commented Oct 24, 2024

I'm glad to work on it!

Contributor Author

merrymercy commented Oct 24, 2024 •

edited

Loading

@ZelinMa557 Are you in the slack channel? We can discuss it there.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment