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

[JS/WebGPU] GroupQueryAttention rewrite #20946

Merged
merged 68 commits into from
Oct 23, 2024

Conversation

satyajandhyala
Copy link
Contributor

@satyajandhyala satyajandhyala commented Jun 6, 2024

Description

Implement JSEP GroupQueryAttention

Motivation and Context

Required to enable certain LLM models to run using WebGPU.

@satyajandhyala satyajandhyala changed the title [JS/WebGPU] group query attention update [JS/WebGPU] group query attention rewrite Oct 11, 2024
Copy link
Contributor

@github-actions github-actions bot left a comment

Choose a reason for hiding this comment

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

You can commit the suggested changes from lintrunner.

onnxruntime/contrib_ops/js/bert/group_query_attention.h Outdated Show resolved Hide resolved
@guschmue
Copy link
Contributor

looking

@satyajandhyala satyajandhyala changed the title [JS/WebGPU] group query attention rewrite [JS/WebGPU] GroupQueryAttention rewrite Oct 18, 2024
@satyajandhyala satyajandhyala dismissed github-actions[bot]’s stale review October 21, 2024 23:15

Already made the required changes.

@satyajandhyala satyajandhyala added the ep:WebGPU ort-web webgpu provider label Oct 23, 2024
@satyajandhyala satyajandhyala merged commit fd8ee48 into main Oct 23, 2024
91 checks passed
@satyajandhyala satyajandhyala deleted the sajandhy/webgpu_group_query_attention_update branch October 23, 2024 17:14
ishwar-raut1 pushed a commit to ishwar-raut1/onnxruntime that referenced this pull request Nov 19, 2024
### Description
Implement JSEP GroupQueryAttention



### Motivation and Context
Required to enable certain LLM models to run using WebGPU.
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
ep:WebGPU ort-web webgpu provider
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants