Skip to content

[Native WebGPU] Support shared memory version of ReduceOps#24399

Merged
satyajandhyala merged 10 commits intomainfrom
sajandhy/webgpu-ep-shader-reduction-op
Apr 15, 2025
Merged

[Native WebGPU] Support shared memory version of ReduceOps#24399
satyajandhyala merged 10 commits intomainfrom
sajandhy/webgpu-ep-shader-reduction-op

Conversation

@satyajandhyala
Copy link
Contributor

Description

Support shared memory version of ReduceOps

Motivation and Context

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.

@satyajandhyala satyajandhyala marked this pull request as ready for review April 11, 2025 20:54
@satyajandhyala satyajandhyala added the ep:WebGPU ort-web webgpu provider label Apr 11, 2025
guschmue
guschmue previously approved these changes Apr 14, 2025
@guschmue
Copy link
Contributor

ReductionOpTest.ReductionVariationTest has some diff

fs-eire
fs-eire previously approved these changes Apr 14, 2025
Copy link
Contributor

@fs-eire fs-eire left a comment

Choose a reason for hiding this comment

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

I think table driven is better than the previous virtual function approach (GetOpSpecificCode).

There are slightly overhead but it should be OK considering they are only happening when generating shader.

@satyajandhyala
Copy link
Contributor Author

satyajandhyala commented Apr 14, 2025

I think table driven is better than the previous virtual function approach (GetOpSpecificCode).

There are slightly overhead but it should be OK considering they are only happening when generating shader.

In some cases the table driven is not possible, or at least not as simple as this, in the case of naive because there is some computation involved.

@guschmue
Copy link
Contributor

there is a ut failing on macos that we should fix

@satyajandhyala satyajandhyala merged commit ff607b4 into main Apr 15, 2025
84 of 89 checks passed
@satyajandhyala satyajandhyala deleted the sajandhy/webgpu-ep-shader-reduction-op branch April 15, 2025 16:48
fs-eire added a commit that referenced this pull request Apr 16, 2025
fs-eire added a commit that referenced this pull request Apr 16, 2025
fs-eire added a commit that referenced this pull request Apr 16, 2025
ashrit-ms pushed a commit that referenced this pull request Apr 24, 2025
### Description
 Support shared memory version of ReduceOps



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->
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.

3 participants