onnxruntime
[Native WebGPU] Add Conv, ConTranspose and FusedConv
#24186
Merged

[Native WebGPU] Add Conv, ConTranspose and FusedConv #24186

satyajandhyala merged 73 commits into main from sajandhy/webgpu-ep-add-conv
satyajandhyala
satyajandhyala Added Conv
de80fadf
satyajandhyala lint
4490299b
satyajandhyala Added Conv1D
598a7604
satyajandhyala Added grouped conv
27e8bf87
satyajandhyala Fix shader code.
856f4d37
satyajandhyala Removed fused_utils.cc and fused_utils.h
c6e2d1a0
satyajandhyala Tmp
27a03df0
satyajandhyala tmp2
3c713de7
satyajandhyala Handle conv1d output
fd994bc4
satyajandhyala Modified CreateMatMulProgram signature.
0c9273bb
satyajandhyala Fix shader code.
c34324db
satyajandhyala Fix compilation errors.
0ffc9925
satyajandhyala Added FusedConv and Activation.
24b40362
satyajandhyala Removed DoTranspose, Added shader i32 casts, and input_output_shapes …
171b1013
satyajandhyala Fix input_output_shapes usage.
f9b5ae9a
satyajandhyala Add FusedConv
a946098b
satyajandhyala Enabled WebGpuEP FusedConv tests.
df8f1387
satyajandhyala Layout transformation should skip FusedConv for WebGpuEp
234f719b
satyajandhyala Revert "Layout transformation should skip FusedConv for WebGpuEp"
02c16ee1
satyajandhyala Added sequentially_access_by_threads and transpose_a flags to matmul …
94c6f444
satyajandhyala typo
7b9897c1
satyajandhyala Added reduced shapes when creating Conv2DMMProgram.
298467fd
satyajandhyala Added FusedConv related changes.
8ab59310
satyajandhyala fixed merge error
4fa8223c
satyajandhyala Added ConvTranspose
2297bc16
satyajandhyala More changes to fix issues.
d7a5fed5
satyajandhyala Fixed data type in the shader code.
61ab3d38
satyajandhyala satyajandhyala added ep:WebGPU
satyajandhyala fix compilation error
3db88f5a
github-actions
github-actions commented on 2025-03-26
github-advanced-security
github-advanced-security commented on 2025-03-26
satyajandhyala Fix activation
55bf4c3b
satyajandhyala Make ConvTranspose test run.
b559ec95
github-actions
github-actions commented on 2025-03-26
github-advanced-security
github-advanced-security commented on 2025-03-26
satyajandhyala Removed code dividing offset by the components in the shader code.
a82ee44d
satyajandhyala Fix unused variable error.
087b407c
satyajandhyala lint
b839a7bb
satyajandhyala Revert "Fix unused variable error."
ccd43c38
satyajandhyala Added activation_snippet
b919e2f7
satyajandhyala Added channels last support in Conv and MatMul
bd5322f0
satyajandhyala Added ReduceShapeByComponents
0962d299
satyajandhyala Added permutation parameter to TransposeKernel function.
da78cca3
satyajandhyala Fixed the ordered of AddInput calls.
78f9eb29
satyajandhyala typo
5ad45559
satyajandhyala Calculate outputIndices from global_idx without multiplying with comp…
9ddbc7d4
satyajandhyala convert activation to string
62fade7b
satyajandhyala satyajandhyala force pushed from b16eb648 to 62fade7b 299 days ago
satyajandhyala Removed unused variables/members, fixed conv-transpose1d
6fa622d1
satyajandhyala satyajandhyala force pushed from 490259b9 to 6fa622d1 298 days ago
satyajandhyala Fix ConvTranspose asymmetric pads
3520e256
satyajandhyala Fix conv-transpoes padding
ca810e87
satyajandhyala Removed ComputeOutputShape defined in Conv, and using functionality a…
65b2e62d
satyajandhyala Disabled Conv/ConvTranspose 3D test case and empty input test cases.
27a18e10
satyajandhyala Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
eadf892a
satyajandhyala Enabled ConvTranspoes_onnx_group testcase.
c1b1af67
satyajandhyala Added activation hint to Conv/MatMul.
5627190d
satyajandhyala satyajandhyala marked this pull request as ready for review 297 days ago
satyajandhyala Fixed compilation issues.
bf2800e6
github-actions
github-actions commented on 2025-03-30
satyajandhyala lint
de4f84c7
satyajandhyala Fixed activation code to handle vector data types
143f2ed3
github-actions
github-actions commented on 2025-03-30
github-advanced-security
github-advanced-security commented on 2025-03-30
satyajandhyala lint
bc5aed22
satyajandhyala Use x_element_t instead of x_value_t
55874c2d
satyajandhyala Reverted unintended changes.
4da5aa2c
satyajandhyala typo
83296206
satyajandhyala typo
c43d1832
satyajandhyala Removed default arguments causing f32 vs. f16 related shader code err…
8ee6129f
satyajandhyala Removed unused local variable.
08856bf6
satyajandhyala Use explicit scope when accessing the template base class member.
36aeffaf
satyajandhyala Fixed sigmoid activation, fixed matmul to apply activation after addi…
657ec018
satyajandhyala Refactored and fixed issue with calling malmul from conv in one scena…
0d78a8df
github-actions
github-actions commented on 2025-04-02
satyajandhyala lint
d5d45442
satyajandhyala clean-up
900d0007
satyajandhyala Add missing bias input
705e8bc6
satyajandhyala clean-up2
5870ed46
satyajandhyala More indexing/components related changes/fixes
4970c7ef
satyajandhyala Added stringify
6be25e44
satyajandhyala Fixed typos and improved ReduceShapeByComponents functionality to div…
78af3177
satyajandhyala Merge branch 'main' of https://github.com/microsoft/onnxruntime into …
1434be4c
satyajandhyala Fixed ConvTranspose typed test.
5eb067f7
satyajandhyala Enabled some activation fusion test cases.
f51a2046
satyajandhyala satyajandhyala changed the title [WIP][Native WebGPU] Add Conv, ConTranspose and FusedConv [Native WebGPU] Add Conv, ConTranspose and FusedConv 292 days ago
fs-eire
fs-eire commented on 2025-04-04
fs-eire
fs-eire approved these changes on 2025-04-04
satyajandhyala satyajandhyala merged 9102aaee into main 291 days ago
satyajandhyala satyajandhyala deleted the sajandhy/webgpu-ep-add-conv branch 291 days ago

Login to write a write a comment.

Login via GitHub

Assignees
No one assigned
Labels
Milestone