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

enable GQA on CPU #270

Merged
merged 7 commits into from
May 1, 2024
Merged

enable GQA on CPU #270

merged 7 commits into from
May 1, 2024

Conversation

yufenglee
Copy link
Member

Enable GQA on CPU. It depends on ORT PR: microsoft/onnxruntime#20299. DO NOT MERGE this PR before that one.

@yufenglee yufenglee changed the title [DO NOT MERGE] enable GQA on CPU enable GQA on CPU May 1, 2024
@yufenglee yufenglee merged commit a652001 into main May 1, 2024
12 checks passed
@yufenglee yufenglee deleted the yufeng/gqa_cpu branch May 1, 2024 23:35
baijumeswani pushed a commit that referenced this pull request May 1, 2024
Enable GQA on CPU. It depends on ORT PR:
microsoft/onnxruntime#20299. DO NOT MERGE this
PR before that one.
kunal-vaishnavi added a commit that referenced this pull request May 9, 2024
### Description

This PR adds `GroupQueryAttention` to ONNX models generated for CPU.

### Motivation and Context

This PR is a follow up to [this
PR](#270).
baijumeswani pushed a commit that referenced this pull request May 10, 2024
### Description

This PR adds `GroupQueryAttention` to ONNX models generated for CPU.

### Motivation and Context

This PR is a follow up to [this
PR](#270).
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