(bencher: divan::Bencher, size: usize)
| 175 | |
| 176 | #[divan::bench(args = [1024, 65536, 1048576])] |
| 177 | fn zstd_compress_chunk(bencher: divan::Bencher, size: usize) { |
| 178 | // Simulate safetensors data (pseudo-random F16 weights) |
| 179 | let data: Vec<u8> = (0..size).map(|i| (i % 251) as u8).collect(); |
| 180 | bencher |
| 181 | .counter(BytesCount::new(size)) |
| 182 | .bench_local(|| zstd::encode_all(data.as_slice(), 1).unwrap()); |
| 183 | } |
| 184 | |
| 185 | #[divan::bench(args = [1024, 65536, 1048576])] |
| 186 | fn zstd_decompress_chunk(bencher: divan::Bencher, size: usize) { |