search

webonnx / wonnx

A WebGPU-accelerated ONNX inference run-time written 100% in Rust, ready for native and the web
Other
1.61k stars 59 forks source link

Allow larger output sizes for Concat #180

Closed mayjs closed 1 year ago

mayjs commented 1 year ago

UNet caused errors because the Concat operator could not handle the large output sizes of some Concat operations. The solution for this is to change the workgroup dimensions to also use the y dimension. Since there is a limit of threads per dimension, this allows us quadratically increase the maximum possible output size.

Fixes #178.

pixelspark commented 1 year ago

:+1: Let's await the CI run but otherwise good to merge