saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-20 02:07:56 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions
onnxruntime/js/web/lib/wasm/jsep/webgpu
History
Arthur Islamov c3f04251c7
[js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 
### Description
Added two kernels for Layer and Instance norm

Also added maximum limits for `maxBufferSize` when requesting GPU device
as by default it's limited to 256mb and it fails allocating 600mb buffer
while running fp32 StableDiffusion weights.


### Motivation and Context
These two are used in StableDiffusion and many other networks
2023-08-08 09:09:37 -07:00
..
ops [js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 2023-08-08 09:09:37 -07:00
attribute-with-cache-key.ts
gpu-data-manager.ts [js/webgpu] Make sure only storage buffers are reused (#16893) 2023-08-04 13:40:52 -07:00
op-resolve-rules.ts [js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 2023-08-08 09:09:37 -07:00
program-manager.ts [js/webgpu] Make sure only storage buffers are reused (#16893) 2023-08-04 13:40:52 -07:00
types.ts [js/web] JSEP LayerNormalization and InstanceNormalizations kernels (#16830) 2023-08-08 09:09:37 -07:00