onnxruntime/js/web/lib/wasm/jsep/webgpu
Jiajia Qin abdf8b7c3f
[js/webgpu] Optimize broadcast binary. (#18185)
### Description
Currently, the binary algorithms are divided into the vectorize one
(efficient) and non-vectorize one (less efficient). Below situations
will go to the vectorize one:
1) A or B's shape length is 1.
2) The shared dimensions length of A and B are divisible by 4.
3) A and B have same shape.

This PR adds another situation as below to go to the vectorize
algorithm.
4. A or B's last dimension is divisible by 4.

With this change, the aggerate time of Add in sam-b-encoder becomes
309.65 ms from 409.12 ms on Intel ADL.
2023-11-20 16:52:17 -08:00
..
ops [js/webgpu] Optimize broadcast binary. (#18185) 2023-11-20 16:52:17 -08:00
attribute-with-cache-key.ts [js] optimize eslint config (#18460) 2023-11-20 12:00:56 -08:00
gpu-data-manager.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
op-resolve-rules.ts [js/web] JSEP Attention & MultiHeadAttention (#17742) 2023-11-17 12:23:52 -08:00
program-manager.ts [js/web] fix typescript type check (#18343) 2023-11-10 16:03:38 -08:00
types.ts [js/webgpu] revise uniform support (#17871) 2023-10-11 16:41:46 -07:00