[WGPU Backend] Weird burn-jit/fusion/kernel panic bug in latest commit about burn HOT 5 CLOSED

Thanks for the MWE! This will help.

I'll investigate to see what's the cause.

Thought a recent PR might have been the root of the issue but if it works on ndarray (and I checked with a commit just before said PR), so it's something else.

/edit: seems to be the tensor + 1 elementwise op that fails on wgpu.

from burn.

@laggui Just tested it out with the linked PR branch and it works fine, thanks for that. I'll just keep burn patched with that branch until it's merged.

(will leave this issue open so it can get closed by the PR)

from burn.

Update: Just tested with ndarray backend and it's fine (no panic).

from burn.

@jclmnop should be fixed with the linked PR. If you wanna give it a shot before it merges you can check out the branch.

from burn.

@jclmnop should be fixed with the linked PR. If you wanna give it a shot before it merges you can check out the branch.

Nice one thanks, I'll give it a go either tonight or tomorrow.

from burn.