Function make_gpu_tensor

cake-core/benches/bench_utils.rs:17–21

(shape: &[usize], seed: u64)

}

fn make_gpu_tensor(shape: &[usize], seed: u64) -> Tensor {
    super::bench_helpers::make_tensor(shape, seed)
        .to_device(&gpu_device())
        .unwrap()
}

#[divan::bench(args = [256, 1024, 4096])]
fn rms_norm_forward(bencher: divan::Bencher, hidden_size: usize) {