saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-02 03:55:34 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/lib/wasm/jsep/webgpu/ops
History
Yulong Wang d9b9c5a537
[js/webgpu] support using uniform buffer (#17803) 
### Description
support using uniform buffer.

This PR allows to use uniform buffer in shader program, so that some
runtime information (eg. input/output shape) is no longer need to be
hardcoded into shader code.

There are 2 commits in this PR:
-
[667f31c](667f31c83d):
framework changes to support uniform buffer, as well as updates in
program manager, gpu data manager and indices helper.
-
[09e1d2a](09e1d2ad1d):
an example change for operator `Transpose` to use input's rank-only
instead of dims as shader key. With this change, model mobilenetv2-12
shader compile times dropped from 71 to 52.
2023-10-10 00:31:12 -07:00
..
3rd-party [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
argminmax.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
bias-add.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
bias-split-gelu.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
binary-op.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
common.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
concat.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
conv-grouped.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
conv-transpose.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
conv.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
einsum.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
expand.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
fuse-utils.ts
gather-elements.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
gather.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
gemm.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
instance-norm.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
layer-norm.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
matmul.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
pad.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
pool.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
range.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
reduce.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
resize.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
skip-layer-norm.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
slice.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
softmax.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
split.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
tile.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
transpose.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
unary-op.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00
where.ts [js/webgpu] support using uniform buffer (#17803) 2023-10-10 00:31:12 -07:00