PR #24358 [WebGPU EP] Add EINSUM implementation

[WebGPU EP] Add EINSUM implementation #24358

fs-eire merged 18 commits into microsoft:main from feich-ms:user/feich-ms/migrate_einsum_op_to_native

github-advanced-security commented on 2025-04-09

satyajandhyala added ep:WebGPU

initial changes by cline LLM model

f90c2859

update einsum.h and einsum.cc

c64cdfe6

write the dynamical j_max uniform variable to shader directly

a5cb8162

fix a test break

929f9cc4

revert changes in CMakeLists.txt

5726c055

fix a test break

62da03be

feich-ms force pushed from ce7ce7fe to 62da03be 1 year ago

fix some test cases

68216be4

optimize

454d12af

fix ci breaks

7b784625

fix ci breaks caused by implicit conversion loses

93407d1f

use input_shape as uniform variable

e7445766

optimize logic to handle all test cases

1dfa521d

revert bad changes in einsum_test.cc

4a45829b

feich-ms marked this pull request as ready for review 1 year ago

fix lint issue

88a395cb

fs-eire commented on 2025-04-18

use input{i}_indices_t offerred by indices helper

0170b1ad

fix warning issue

d94d63e3

add comments for potential optimization

14d6e801

fix lint issue

7266acf6

fs-eire requested a review from

copilot-pull-request-reviewer 1 year ago

copilot-pull-request-reviewer commented on 2025-04-28

fs-eire approved these changes on 2025-04-28

fs-eire merged e14bcd34 into main 1 year ago

feich-ms deleted the user/feich-ms/migrate_einsum_op_to_native branch 1 year ago

Reviewers

fs-eire

copilot-pull-request-reviewer

github-advanced-security

Assignees

No one assigned

Labels

ep:WebGPU

Milestone

No milestone

onnxruntime [WebGPU EP] Add EINSUM implementation #24358 Merged

[WebGPU EP] Add EINSUM implementation #24358

onnxruntime
[WebGPU EP] Add EINSUM implementation
#24358

Merged