Feat/wgpu/repeat

[louisfd]

Pull Request Template

Checklist

• Confirmed that run-checks all script has been executed.
• Made sure the book is up to date with changes in this PR.

Related Issues/PRs

In #1049 Repeat was a bottleneck

Changes

Made a wgpu kernel for repeat operation, so we don't need to call many slice_assign kernels

Testing

Tests for the kernel in several dims, a larger one that exceeds the workgroup size, and tests for int and bool versions.

[codecov]

Codecov Report

Attention: 1 lines in your changes are missing coverage. Please review.

Comparison is base (2ca1fd5) 85.54% compared to head (99399f5) 85.59%.

Files	Patch %	Lines
burn-wgpu/src/kernel/index/repeat.rs	98.97%	1 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #1068      +/-   ##
==========================================
+ Coverage   85.54%   85.59%   +0.04%     
==========================================
  Files         508      509       +1     
  Lines       54019    54161     +142     
==========================================
+ Hits        46211    46357     +146     
+ Misses       7808     7804       -4     

☔ View full report in Codecov by Sentry.
📢 Have feedback on the report? Share it here.