3 use test
::{black_box, Bencher}
;
6 fn bench_new(b
: &mut Bencher
) {
7 b
.iter(|| Vec
::<u32>::new())
10 fn do_bench_with_capacity(b
: &mut Bencher
, src_len
: usize) {
11 b
.bytes
= src_len
as u64;
13 b
.iter(|| Vec
::<u32>::with_capacity(src_len
))
17 fn bench_with_capacity_0000(b
: &mut Bencher
) {
18 do_bench_with_capacity(b
, 0)
22 fn bench_with_capacity_0010(b
: &mut Bencher
) {
23 do_bench_with_capacity(b
, 10)
27 fn bench_with_capacity_0100(b
: &mut Bencher
) {
28 do_bench_with_capacity(b
, 100)
32 fn bench_with_capacity_1000(b
: &mut Bencher
) {
33 do_bench_with_capacity(b
, 1000)
36 fn do_bench_from_fn(b
: &mut Bencher
, src_len
: usize) {
37 b
.bytes
= src_len
as u64;
39 b
.iter(|| (0..src_len
).collect
::<Vec
<_
>>())
43 fn bench_from_fn_0000(b
: &mut Bencher
) {
44 do_bench_from_fn(b
, 0)
48 fn bench_from_fn_0010(b
: &mut Bencher
) {
49 do_bench_from_fn(b
, 10)
53 fn bench_from_fn_0100(b
: &mut Bencher
) {
54 do_bench_from_fn(b
, 100)
58 fn bench_from_fn_1000(b
: &mut Bencher
) {
59 do_bench_from_fn(b
, 1000)
62 fn do_bench_from_elem(b
: &mut Bencher
, src_len
: usize) {
63 b
.bytes
= src_len
as u64;
65 b
.iter(|| repeat(5).take(src_len
).collect
::<Vec
<usize>>())
69 fn bench_from_elem_0000(b
: &mut Bencher
) {
70 do_bench_from_elem(b
, 0)
74 fn bench_from_elem_0010(b
: &mut Bencher
) {
75 do_bench_from_elem(b
, 10)
79 fn bench_from_elem_0100(b
: &mut Bencher
) {
80 do_bench_from_elem(b
, 100)
84 fn bench_from_elem_1000(b
: &mut Bencher
) {
85 do_bench_from_elem(b
, 1000)
88 fn do_bench_from_slice(b
: &mut Bencher
, src_len
: usize) {
89 let src
: Vec
<_
> = FromIterator
::from_iter(0..src_len
);
91 b
.bytes
= src_len
as u64;
93 b
.iter(|| src
.as_slice().to_vec());
97 fn bench_from_slice_0000(b
: &mut Bencher
) {
98 do_bench_from_slice(b
, 0)
102 fn bench_from_slice_0010(b
: &mut Bencher
) {
103 do_bench_from_slice(b
, 10)
107 fn bench_from_slice_0100(b
: &mut Bencher
) {
108 do_bench_from_slice(b
, 100)
112 fn bench_from_slice_1000(b
: &mut Bencher
) {
113 do_bench_from_slice(b
, 1000)
116 fn do_bench_from_iter(b
: &mut Bencher
, src_len
: usize) {
117 let src
: Vec
<_
> = FromIterator
::from_iter(0..src_len
);
119 b
.bytes
= src_len
as u64;
122 let dst
: Vec
<_
> = FromIterator
::from_iter(src
.iter().cloned());
128 fn bench_from_iter_0000(b
: &mut Bencher
) {
129 do_bench_from_iter(b
, 0)
133 fn bench_from_iter_0010(b
: &mut Bencher
) {
134 do_bench_from_iter(b
, 10)
138 fn bench_from_iter_0100(b
: &mut Bencher
) {
139 do_bench_from_iter(b
, 100)
143 fn bench_from_iter_1000(b
: &mut Bencher
) {
144 do_bench_from_iter(b
, 1000)
147 fn do_bench_extend(b
: &mut Bencher
, dst_len
: usize, src_len
: usize) {
148 let dst
: Vec
<_
> = FromIterator
::from_iter(0..dst_len
);
149 let src
: Vec
<_
> = FromIterator
::from_iter(dst_len
..dst_len
+ src_len
);
151 b
.bytes
= src_len
as u64;
154 let mut dst
= dst
.clone();
155 dst
.extend(src
.clone());
161 fn bench_extend_0000_0000(b
: &mut Bencher
) {
162 do_bench_extend(b
, 0, 0)
166 fn bench_extend_0000_0010(b
: &mut Bencher
) {
167 do_bench_extend(b
, 0, 10)
171 fn bench_extend_0000_0100(b
: &mut Bencher
) {
172 do_bench_extend(b
, 0, 100)
176 fn bench_extend_0000_1000(b
: &mut Bencher
) {
177 do_bench_extend(b
, 0, 1000)
181 fn bench_extend_0010_0010(b
: &mut Bencher
) {
182 do_bench_extend(b
, 10, 10)
186 fn bench_extend_0100_0100(b
: &mut Bencher
) {
187 do_bench_extend(b
, 100, 100)
191 fn bench_extend_1000_1000(b
: &mut Bencher
) {
192 do_bench_extend(b
, 1000, 1000)
195 fn do_bench_extend_from_slice(b
: &mut Bencher
, dst_len
: usize, src_len
: usize) {
196 let dst
: Vec
<_
> = FromIterator
::from_iter(0..dst_len
);
197 let src
: Vec
<_
> = FromIterator
::from_iter(dst_len
..dst_len
+ src_len
);
199 b
.bytes
= src_len
as u64;
202 let mut dst
= dst
.clone();
203 dst
.extend_from_slice(&src
);
209 fn bench_extend_recycle(b
: &mut Bencher
) {
210 let mut data
= vec
![0; 1000];
213 let tmp
= std
::mem
::take(&mut data
);
214 let mut to_extend
= black_box(Vec
::new());
215 to_extend
.extend(tmp
.into_iter());
216 data
= black_box(to_extend
);
223 fn bench_extend_from_slice_0000_0000(b
: &mut Bencher
) {
224 do_bench_extend_from_slice(b
, 0, 0)
228 fn bench_extend_from_slice_0000_0010(b
: &mut Bencher
) {
229 do_bench_extend_from_slice(b
, 0, 10)
233 fn bench_extend_from_slice_0000_0100(b
: &mut Bencher
) {
234 do_bench_extend_from_slice(b
, 0, 100)
238 fn bench_extend_from_slice_0000_1000(b
: &mut Bencher
) {
239 do_bench_extend_from_slice(b
, 0, 1000)
243 fn bench_extend_from_slice_0010_0010(b
: &mut Bencher
) {
244 do_bench_extend_from_slice(b
, 10, 10)
248 fn bench_extend_from_slice_0100_0100(b
: &mut Bencher
) {
249 do_bench_extend_from_slice(b
, 100, 100)
253 fn bench_extend_from_slice_1000_1000(b
: &mut Bencher
) {
254 do_bench_extend_from_slice(b
, 1000, 1000)
257 fn do_bench_clone(b
: &mut Bencher
, src_len
: usize) {
258 let src
: Vec
<usize> = FromIterator
::from_iter(0..src_len
);
260 b
.bytes
= src_len
as u64;
262 b
.iter(|| src
.clone());
266 fn bench_clone_0000(b
: &mut Bencher
) {
271 fn bench_clone_0010(b
: &mut Bencher
) {
272 do_bench_clone(b
, 10)
276 fn bench_clone_0100(b
: &mut Bencher
) {
277 do_bench_clone(b
, 100)
281 fn bench_clone_1000(b
: &mut Bencher
) {
282 do_bench_clone(b
, 1000)
285 fn do_bench_clone_from(b
: &mut Bencher
, times
: usize, dst_len
: usize, src_len
: usize) {
286 let dst
: Vec
<_
> = FromIterator
::from_iter(0..src_len
);
287 let src
: Vec
<_
> = FromIterator
::from_iter(dst_len
..dst_len
+ src_len
);
289 b
.bytes
= (times
* src_len
) as u64;
292 let mut dst
= dst
.clone();
295 dst
.clone_from(&src
);
296 dst
= black_box(dst
);
303 fn bench_clone_from_01_0000_0000(b
: &mut Bencher
) {
304 do_bench_clone_from(b
, 1, 0, 0)
308 fn bench_clone_from_01_0000_0010(b
: &mut Bencher
) {
309 do_bench_clone_from(b
, 1, 0, 10)
313 fn bench_clone_from_01_0000_0100(b
: &mut Bencher
) {
314 do_bench_clone_from(b
, 1, 0, 100)
318 fn bench_clone_from_01_0000_1000(b
: &mut Bencher
) {
319 do_bench_clone_from(b
, 1, 0, 1000)
323 fn bench_clone_from_01_0010_0010(b
: &mut Bencher
) {
324 do_bench_clone_from(b
, 1, 10, 10)
328 fn bench_clone_from_01_0100_0100(b
: &mut Bencher
) {
329 do_bench_clone_from(b
, 1, 100, 100)
333 fn bench_clone_from_01_1000_1000(b
: &mut Bencher
) {
334 do_bench_clone_from(b
, 1, 1000, 1000)
338 fn bench_clone_from_01_0010_0100(b
: &mut Bencher
) {
339 do_bench_clone_from(b
, 1, 10, 100)
343 fn bench_clone_from_01_0100_1000(b
: &mut Bencher
) {
344 do_bench_clone_from(b
, 1, 100, 1000)
348 fn bench_clone_from_01_0010_0000(b
: &mut Bencher
) {
349 do_bench_clone_from(b
, 1, 10, 0)
353 fn bench_clone_from_01_0100_0010(b
: &mut Bencher
) {
354 do_bench_clone_from(b
, 1, 100, 10)
358 fn bench_clone_from_01_1000_0100(b
: &mut Bencher
) {
359 do_bench_clone_from(b
, 1, 1000, 100)
363 fn bench_clone_from_10_0000_0000(b
: &mut Bencher
) {
364 do_bench_clone_from(b
, 10, 0, 0)
368 fn bench_clone_from_10_0000_0010(b
: &mut Bencher
) {
369 do_bench_clone_from(b
, 10, 0, 10)
373 fn bench_clone_from_10_0000_0100(b
: &mut Bencher
) {
374 do_bench_clone_from(b
, 10, 0, 100)
378 fn bench_clone_from_10_0000_1000(b
: &mut Bencher
) {
379 do_bench_clone_from(b
, 10, 0, 1000)
383 fn bench_clone_from_10_0010_0010(b
: &mut Bencher
) {
384 do_bench_clone_from(b
, 10, 10, 10)
388 fn bench_clone_from_10_0100_0100(b
: &mut Bencher
) {
389 do_bench_clone_from(b
, 10, 100, 100)
393 fn bench_clone_from_10_1000_1000(b
: &mut Bencher
) {
394 do_bench_clone_from(b
, 10, 1000, 1000)
398 fn bench_clone_from_10_0010_0100(b
: &mut Bencher
) {
399 do_bench_clone_from(b
, 10, 10, 100)
403 fn bench_clone_from_10_0100_1000(b
: &mut Bencher
) {
404 do_bench_clone_from(b
, 10, 100, 1000)
408 fn bench_clone_from_10_0010_0000(b
: &mut Bencher
) {
409 do_bench_clone_from(b
, 10, 10, 0)
413 fn bench_clone_from_10_0100_0010(b
: &mut Bencher
) {
414 do_bench_clone_from(b
, 10, 100, 10)
418 fn bench_clone_from_10_1000_0100(b
: &mut Bencher
) {
419 do_bench_clone_from(b
, 10, 1000, 100)
422 macro_rules
! bench_in_place
{
423 ($
($fname
:ident
, $
type:ty
, $count
:expr
, $init
:expr
);*) => {
426 fn $
fname(b
: &mut Bencher
) {
428 let src
: Vec
<$
type> = black_box(vec
![$init
; $count
]);
431 .map(|(idx
, e
)| idx
as $
type ^ e
)
432 .collect
::<Vec
<$
type>>()
440 bench_in_place_xxu8_0010_i0
, u8, 10, 0;
441 bench_in_place_xxu8_0100_i0
, u8, 100, 0;
442 bench_in_place_xxu8_1000_i0
, u8, 1000, 0;
443 bench_in_place_xxu8_0010_i1
, u8, 10, 1;
444 bench_in_place_xxu8_0100_i1
, u8, 100, 1;
445 bench_in_place_xxu8_1000_i1
, u8, 1000, 1;
446 bench_in_place_xu32_0010_i0
, u32, 10, 0;
447 bench_in_place_xu32_0100_i0
, u32, 100, 0;
448 bench_in_place_xu32_1000_i0
, u32, 1000, 0;
449 bench_in_place_xu32_0010_i1
, u32, 10, 1;
450 bench_in_place_xu32_0100_i1
, u32, 100, 1;
451 bench_in_place_xu32_1000_i1
, u32, 1000, 1;
452 bench_in_place_u128_0010_i0
, u128
, 10, 0;
453 bench_in_place_u128_0100_i0
, u128
, 100, 0;
454 bench_in_place_u128_1000_i0
, u128
, 1000, 0;
455 bench_in_place_u128_0010_i1
, u128
, 10, 1;
456 bench_in_place_u128_0100_i1
, u128
, 100, 1;
457 bench_in_place_u128_1000_i1
, u128
, 1000, 1
461 fn bench_in_place_recycle(b
: &mut Bencher
) {
462 let mut data
= vec
![0; 1000];
465 let tmp
= std
::mem
::take(&mut data
);
469 .map(|(idx
, e
)| idx
.wrapping_add(e
))
471 .collect
::<Vec
<usize>>(),
477 fn bench_in_place_zip_recycle(b
: &mut Bencher
) {
478 let mut data
= vec
![0u8; 1000];
479 let mut rng
= crate::bench_rng();
480 let mut subst
= vec
![0u8; 1000];
481 rng
.fill_bytes(&mut subst
[..]);
484 let tmp
= std
::mem
::take(&mut data
);
487 .zip(subst
.iter().copied())
489 .map(|(i
, (d
, s
))| d
.wrapping_add(i
as u8) ^ s
)
490 .collect
::<Vec
<_
>>();
491 data
= black_box(mangled
);
496 fn bench_in_place_zip_iter_mut(b
: &mut Bencher
) {
497 let mut data
= vec
![0u8; 256];
498 let mut rng
= crate::bench_rng();
499 let mut subst
= vec
![0u8; 1000];
500 rng
.fill_bytes(&mut subst
[..]);
503 data
.iter_mut().enumerate().for_each(|(i
, d
)| {
504 *d
= d
.wrapping_add(i
as u8) ^ subst
[i
];
511 pub fn vec_cast
<T
, U
>(input
: Vec
<T
>) -> Vec
<U
> {
512 input
.into_iter().map(|e
| unsafe { std::mem::transmute_copy(&e) }
).collect()
516 fn bench_transmute(b
: &mut Bencher
) {
517 let mut vec
= vec
![10u32; 100];
518 b
.bytes
= 800; // 2 casts x 4 bytes x 100
520 let v
= std
::mem
::take(&mut vec
);
521 let v
= black_box(vec_cast
::<u32, i32>(v
));
522 let v
= black_box(vec_cast
::<i32, u32>(v
));
528 struct Droppable(usize);
530 impl Drop
for Droppable
{
537 fn bench_in_place_collect_droppable(b
: &mut Bencher
) {
538 let v
: Vec
<Droppable
> = std
::iter
::repeat_with(|| Droppable(0)).take(1000).collect();
544 .map(|(i
, e
)| Droppable(i ^ e
.0))
549 const LEN
: usize = 16384;
552 fn bench_chain_collect(b
: &mut Bencher
) {
553 let data
= black_box([0; LEN
]);
554 b
.iter(|| data
.iter().cloned().chain([1]).collect
::<Vec
<_
>>());
558 fn bench_chain_chain_collect(b
: &mut Bencher
) {
559 let data
= black_box([0; LEN
]);
560 b
.iter(|| data
.iter().cloned().chain([1]).chain([2]).collect
::<Vec
<_
>>());
564 fn bench_nest_chain_chain_collect(b
: &mut Bencher
) {
565 let data
= black_box([0; LEN
]);
567 data
.iter().cloned().chain([1].iter().chain([2].iter()).cloned()).collect
::<Vec
<_
>>()
572 fn bench_range_map_collect(b
: &mut Bencher
) {
573 b
.iter(|| (0..LEN
).map(|_
| u32::default()).collect
::<Vec
<_
>>());
577 fn bench_chain_extend_ref(b
: &mut Bencher
) {
578 let data
= black_box([0; LEN
]);
580 let mut v
= Vec
::<u32>::with_capacity(data
.len() + 1);
581 v
.extend(data
.iter().chain([1].iter()));
587 fn bench_chain_extend_value(b
: &mut Bencher
) {
588 let data
= black_box([0; LEN
]);
590 let mut v
= Vec
::<u32>::with_capacity(data
.len() + 1);
591 v
.extend(data
.iter().cloned().chain(Some(1)));
597 fn bench_rev_1(b
: &mut Bencher
) {
598 let data
= black_box([0; LEN
]);
600 let mut v
= Vec
::<u32>::new();
601 v
.extend(data
.iter().rev());
607 fn bench_rev_2(b
: &mut Bencher
) {
608 let data
= black_box([0; LEN
]);
610 let mut v
= Vec
::<u32>::with_capacity(data
.len());
611 v
.extend(data
.iter().rev());
617 fn bench_map_regular(b
: &mut Bencher
) {
618 let data
= black_box([(0, 0); LEN
]);
620 let mut v
= Vec
::<u32>::new();
621 v
.extend(data
.iter().map(|t
| t
.1));
627 fn bench_map_fast(b
: &mut Bencher
) {
628 let data
= black_box([(0, 0); LEN
]);
630 let mut result
: Vec
<u32> = Vec
::with_capacity(data
.len());
631 for i
in 0..data
.len() {
633 *result
.as_mut_ptr().add(i
) = data
[i
].0;
641 fn random_sorted_fill(mut seed
: u32, buf
: &mut [u32]) {
642 let mask
= if buf
.len() < 8192 {
644 } else if buf
.len() < 200_000 {
650 for item
in buf
.iter_mut() {
661 fn bench_vec_dedup_old(b
: &mut Bencher
, sz
: usize) {
662 let mut template
= vec
![0u32; sz
];
663 b
.bytes
= std
::mem
::size_of_val(template
.as_slice()) as u64;
664 random_sorted_fill(0x43, &mut template
);
666 let mut vec
= template
.clone();
669 let (dedup
, _
) = vec
.partition_dedup();
674 black_box(vec
.first());
676 vec
.extend_from_slice(&template
);
680 fn bench_vec_dedup_new(b
: &mut Bencher
, sz
: usize) {
681 let mut template
= vec
![0u32; sz
];
682 b
.bytes
= std
::mem
::size_of_val(template
.as_slice()) as u64;
683 random_sorted_fill(0x43, &mut template
);
685 let mut vec
= template
.clone();
688 black_box(vec
.first());
690 vec
.extend_from_slice(&template
);
695 fn bench_dedup_old_100(b
: &mut Bencher
) {
696 bench_vec_dedup_old(b
, 100);
699 fn bench_dedup_new_100(b
: &mut Bencher
) {
700 bench_vec_dedup_new(b
, 100);
704 fn bench_dedup_old_1000(b
: &mut Bencher
) {
705 bench_vec_dedup_old(b
, 1000);
708 fn bench_dedup_new_1000(b
: &mut Bencher
) {
709 bench_vec_dedup_new(b
, 1000);
713 fn bench_dedup_old_10000(b
: &mut Bencher
) {
714 bench_vec_dedup_old(b
, 10000);
717 fn bench_dedup_new_10000(b
: &mut Bencher
) {
718 bench_vec_dedup_new(b
, 10000);
722 fn bench_dedup_old_100000(b
: &mut Bencher
) {
723 bench_vec_dedup_old(b
, 100000);
726 fn bench_dedup_new_100000(b
: &mut Bencher
) {
727 bench_vec_dedup_new(b
, 100000);
731 fn bench_flat_map_collect(b
: &mut Bencher
) {
732 let v
= vec
![777u32; 500000];
733 b
.iter(|| v
.iter().flat_map(|color
| color
.rotate_left(8).to_be_bytes()).collect
::<Vec
<_
>>());
736 /// Reference benchmark that `retain` has to compete with.
738 fn bench_retain_iter_100000(b
: &mut Bencher
) {
739 let mut v
= Vec
::with_capacity(100000);
742 let mut tmp
= std
::mem
::take(&mut v
);
744 tmp
.extend(black_box(1..=100000));
745 v
= tmp
.into_iter().filter(|x
| x
& 1 == 0).collect();
750 fn bench_retain_100000(b
: &mut Bencher
) {
751 let mut v
= Vec
::with_capacity(100000);
755 v
.extend(black_box(1..=100000));
756 v
.retain(|x
| x
& 1 == 0)
761 fn bench_retain_whole_100000(b
: &mut Bencher
) {
762 let mut v
= black_box(vec
![826u32; 100000]);
763 b
.iter(|| v
.retain(|x
| *x
== 826u32));
767 fn bench_next_chunk(b
: &mut Bencher
) {
768 let v
= vec
![13u8; 2048];
771 const CHUNK
: usize = 8;
773 let mut sum
= [0u32; CHUNK
];
774 let mut iter
= black_box(v
.clone()).into_iter();
776 while let Ok(chunk
) = iter
.next_chunk
::<CHUNK
>() {
778 sum
[i
] += chunk
[i
] as u32;