rust/tests/codegen/lib-optimizations/iter-sum.rs
2024-02-23 18:04:01 -05:00

15 lines
399 B
Rust

//@ compile-flags: -O
//@ only-x86_64 (vectorization varies between architectures)
#![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()
}