onnxruntime/js/web/lib/wasm/jsep/webgpu
Jiajia Qin fffefb1c22
[js/webgpu] Optimize matmul (#16969)
### Description
Changes in this PR:
1) use the optimized version `makeMatMulPacked[Vec4]Source` to support
matmul.
2) enable the conv2dByMatMul path.
3) support broadcast
4) use IndicesHelper.

MatMul with M = 512, K = 512, N = 512 becomes 2ms from 15ms when
enabling profilingMode on my ADL.
2023-08-29 12:40:57 -07:00
..
ops [js/webgpu] Optimize matmul (#16969) 2023-08-29 12:40:57 -07:00
attribute-with-cache-key.ts
gpu-data-manager.ts [js/webgpu] add kernel Not and Equal (#17306) 2023-08-27 19:50:17 -07:00
op-resolve-rules.ts [JS/WebGPU] Support GatherElements kernel (#17243) 2023-08-28 09:55:25 -07:00
program-manager.ts [js/webgpu] add label for some webgpu APIs (#17291) 2023-08-25 12:12:56 -07:00
types.ts [js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 2023-08-08 09:09:37 -07:00