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

DecoderMaskedMultiHeadAttention CPU kernel. #22292

Merged
merged 14 commits into from
Oct 12, 2024
Merged

DecoderMaskedMultiHeadAttention CPU kernel. #22292

merged 14 commits into from
Oct 12, 2024

Conversation

mindest
Copy link
Contributor

@mindest mindest commented Oct 2, 2024

Description

DecoderMaskedMultiHeadAttention CPU kernel.

@tianleiwu
Copy link
Contributor

@mindest mindest marked this pull request as ready for review October 11, 2024 09:08
@mindest mindest changed the title [WIP] DecoderMaskedMultiHeadAttention CPU kernel. DecoderMaskedMultiHeadAttention CPU kernel. Oct 11, 2024
tianleiwu
tianleiwu previously approved these changes Oct 12, 2024
@tianleiwu tianleiwu merged commit 1fa219d into main Oct 12, 2024
91 checks passed
@tianleiwu tianleiwu deleted the linmin/cpu_dmmha branch October 12, 2024 20:43
guschmue pushed a commit that referenced this pull request Oct 18, 2024
### Description
DecoderMaskedMultiHeadAttention CPU kernel.
ishwar-raut1 pushed a commit to ishwar-raut1/onnxruntime that referenced this pull request Nov 19, 2024
### Description
DecoderMaskedMultiHeadAttention CPU kernel.
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.

3 participants