(bencher: divan::Bencher, hidden: usize)
| 101 | |
| 102 | #[divan::bench(args = [1024, 4096, 16384])] |
| 103 | fn fused_stable_softplus_gpu(bencher: divan::Bencher, hidden: usize) { |
| 104 | let dev = gpu_device(); |
| 105 | let x = make_gpu_tensor(&[1, 64, hidden], 260); |
| 106 | let backend = cake_core::backends::create_backend(&dev); |
| 107 | bencher.bench_local(|| backend.stable_softplus(&x).unwrap()); |
| 108 | } |
| 109 | |
| 110 | #[divan::bench(args = [1024, 4096, 16384])] |
| 111 | fn fused_rms_norm_gated_gpu(bencher: divan::Bencher, hidden: usize) { |
nothing calls this directly
no test coverage detected