bjoernager/rust
bjoernager/rust
1
Fork 0
Code Activity

Optimize Take::{fold, for_each} when wrapping TrustedRandomAccess iterators

Browse source
This commit is contained in:
The 8472 2023-08-25 23:29:18 +02:00
parent f320f42c59
commit 72b01d5cca
2 changed files with 97 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

14
tests/codegen/lib-optimizations/iter-sum.rs Normal file
View file

@ -0,0 +1,14 @@
// ignore-debug: the debug assertions get in the way
// compile-flags: -O
#![crate_type = "lib"]
// Ensure that slice + take + sum gets vectorized.
// Currently this relies on the slice::Iter::try_fold implementation
// CHECK-LABEL: @slice_take_sum
#[no_mangle]
pub fn slice_take_sum(s: &[u64], l: usize) -> u64 {
// CHECK: vector.body:
// CHECK: ret
s.iter().take(l).sum()
}