burn icon indicating copy to clipboard operation
burn copied to clipboard

Autotune WGPU: binary_elemwise kernel

Open louisfd opened this issue 1 year ago • 0 comments

Use the autotune mechanism in the WGPU backend to find the fastest kernel version of binary_elemewise and its inplace counterpart, by varying the WORKGROUP argument. Take inspiration from the burn-wgpu/src/kernel/matmul/tune folder.

louisfd avatar Oct 30 '23 11:10 louisfd