(bencher: divan::Bencher)
| 102 | |
| 103 | #[divan::bench] |
| 104 | fn cache_clear(bencher: divan::Bencher) { |
| 105 | let cfg = test_config(); |
| 106 | let head_dim = cfg.hidden_size / cfg.num_attention_heads; |
| 107 | let k = make_tensor(&[1, cfg.num_key_value_heads, 4, head_dim], 60); |
| 108 | let v = make_tensor(&[1, cfg.num_key_value_heads, 4, head_dim], 61); |
| 109 | bencher.bench_local(move || { |
| 110 | let mut cache = make_cache(&cfg); |
| 111 | cache.process_kv(0, k.clone(), v.clone()).unwrap(); |
| 112 | cache.clear(); |
| 113 | }); |
| 114 | } |
nothing calls this directly
no test coverage detected