onnxruntime/js/web/lib/wasm/jsep/webgpu
Jiajia Qin 7be006c466
[js/webgpu] Optimize convtranspose (#23302)
### Description
<!-- Describe your changes. -->
BUG #23273

With this change, I see the convTranspose time in that bug becomes ~7s
from ~90s on my Meteor Lake.

This PR does below things:
1. Use stride to update the increasement in the loop.
In the bug, the stride is 1024, which can greatly reduce the loop times.
2. Support components for A to reduce the memory access times.
3. When output channels is 1, the b components can be same with A to
further reduce the memory access times.
2025-01-09 11:24:42 -08:00
..
ops [js/webgpu] Optimize convtranspose (#23302) 2025-01-09 11:24:42 -08:00
attribute-with-cache-key.ts [js] change default formatter for JavaScript/TypeScript from clang-format to Prettier (#21728) 2024-08-14 16:51:22 -07:00
gpu-data-manager.ts [js/webgpu] Destroy staging buffers aggressively during weights uploading (#22726) 2024-11-06 08:55:15 -08:00
op-resolve-rules.ts [js/webgpu] Add GatherND (#22847) 2024-12-04 09:57:32 -08:00
program-manager.ts [JS/WebGPU] Creating devices with subgroup features enabled if possible (#21833) 2024-11-07 02:13:40 -08:00
types.ts [JS/WebGPU] Creating devices with subgroup features enabled if possible (#21833) 2024-11-07 02:13:40 -08:00