onnxruntime/js/web/lib/wasm
Arthur Islamov d0519a7603
[js/web] BiasSplitGelu and BiasAdd kernels (#17161)
### Description
Two contrib kernels that supposed to speed-up StableDiffusion according
to this doc
https://github.com/microsoft/onnxruntime/blob/main/onnxruntime/python/tools/transformers/models/stable_diffusion/README.md

However, there is no noticable effect in speed or memory consumption. So
i guess the only way to make it faster is to implement
MultiHeadAttention but i'm not capable of doing that right now. So i'll
focus on existing PRs and finding the JSEP kernel that produces
incorrect results. It should be one of the old ones (i suspect Conv or
ConvTranspose), as SD was not generating images correctly on webgpu
since i started working on it. I hoped someone else would fix that by
the time i finish with kernels/optimizations 😅

---------

Co-authored-by: Guenther Schmuelling <guschmue@microsoft.com>
Co-authored-by: Yulong Wang <7679871+fs-eire@users.noreply.github.com>
2023-10-03 12:20:20 -07:00
..
binding [js/webgpu] support IO binding (#17480) 2023-09-29 11:24:42 -07:00
jsep [js/web] BiasSplitGelu and BiasAdd kernels (#17161) 2023-10-03 12:20:20 -07:00
proxy-worker [js/webgpu] support proxy for webgpu (#15851) 2023-05-15 16:23:13 -07:00
proxy-messages.ts [js/webgpu] support IO binding (#17480) 2023-09-29 11:24:42 -07:00
proxy-wrapper.ts [js/webgpu] support IO binding (#17480) 2023-09-29 11:24:42 -07:00
run-options.ts [js/web] enable ONNX Runtime Web error messages in JS (#16335) 2023-06-15 09:45:41 -07:00
session-handler.ts Add training interfaces to js/common (#17333) 2023-09-29 19:05:10 -07:00
session-options.ts [js/webgpu] allow specify preferredLayout (#17756) 2023-10-02 21:25:12 -07:00
wasm-common.ts [js/webgpu] support IO binding (#17480) 2023-09-29 11:24:42 -07:00
wasm-core-impl.ts [js/webgpu] support IO binding (#17480) 2023-09-29 11:24:42 -07:00
wasm-factory.ts [js/web] add target ort.webgpu.min.js (#15780) 2023-05-04 10:05:39 -07:00
wasm-utils.ts [js/web] enable ONNX Runtime Web error messages in JS (#16335) 2023-06-15 09:45:41 -07:00