Fix rand interval

[laggui]

Checklist

• Confirmed that cargo run-checks command has been executed.

Related Issues/PRs

• Linked to Fix/random interval cubecl#744
• Issue discovered via tensor.roll() #3281 (see this CI run)

  failures:
  
  ---- tests::cube::kernel::scatter::tests::scatter_should_work_with_multiple_workgroups_3d_dim2 stdout ----
  
  thread 'tests::cube::kernel::scatter::tests::scatter_should_work_with_multiple_workgroups_3d_dim2' panicked at /home/runner/work/burn/burn/crates/burn-ndarray/src/ops/base.rs:593:23:
  ndarray: index [5, 256] is out of bounds for array of shape [36, 256]
  
  
  failures:
      tests::cube::kernel::scatter::tests::scatter_should_work_with_multiple_workgroups_3d_dim2
  
  test result: FAILED. 3059 passed; 1 failed; 22 ignored; 0 measured; 0 filtered out; finished in 663.28s

Changes

• Updated cubecl with the rand kernel fix
• Changed assert_within_range to support different element types for comparison

Testing

Given the randomness of the source, the issue is not always reproducible. Best way is to generate a bunch of random numbers many times, but this might not be suitable for actual unit tests:

#[test]
    fn values_all_within_interval_uniform() {
        TestBackend::seed(0);
        let device = Default::default();

        let low = 5.;
        let high = 12.;
        for i in 0..1000 {
            println!("Test #{i}");
            let tensor =
                TestTensor::<1>::random([100_000], Distribution::Uniform(low, high), &device);

            tensor
                .into_data()
                .assert_within_range::<FT>(low.elem()..high.elem());
        }
    }

[codecov]

Codecov Report

Attention: Patch coverage is 87.50000% with 4 lines in your changes missing coverage. Please review.

Project coverage is 46.30%. Comparing base (14811c6) to head (cad03da).
Report is 1 commits behind head on main.

Files with missing lines	Patch %	Lines
crates/burn-tensor/src/tensor/data.rs	33.33%	4 Missing ⚠️

Additional details and impacted files

@@            Coverage Diff             @@
##             main    #3321      +/-   ##
==========================================
- Coverage   46.31%   46.30%   -0.01%     
==========================================
  Files         414      414              
  Lines       60979    60976       -3     
==========================================
- Hits        28240    28234       -6     
- Misses      32739    32742       +3     

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

🚀 New features to boost your workflow:

• ❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

[laggui]

Linked: tracel-ai/cubecl#753

TODO: update rev

[laggui]

Marking it as draft before we accidentally merge this 😅 need to wrap some things up on the cubecl side to fix the no-std build which was previously failing due to new profiling stuff.