(bencher: divan::Bencher, hidden: usize)
| 246 | |
| 247 | #[divan::bench(args = [1024, 4096, 16384])] |
| 248 | fn fused_add3_gpu(bencher: divan::Bencher, hidden: usize) { |
| 249 | let dev = gpu_device(); |
| 250 | let a = make_gpu_tensor(&[1, 64, hidden], 500); |
| 251 | let b = make_gpu_tensor(&[1, 64, hidden], 501); |
| 252 | let c = make_gpu_tensor(&[1, 64, hidden], 502); |
| 253 | let backend = cake_core::backends::create_backend(&dev); |
| 254 | bencher.bench_local(|| backend.add3(&a, &b, &c).unwrap()); |
| 255 | } |
| 256 | |
| 257 | #[divan::bench(args = [1024, 4096])] |
| 258 | fn fused_add_scaled_gpu(bencher: divan::Bencher, channels: usize) { |
nothing calls this directly
no test coverage detected