rust/tests/codegen/lib-optimizations/iter-sum.rs
2025-02-11 13:41:35 -08:00

13 lines
409 B
Rust

//@ compile-flags: -Copt-level=3
//@ 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()
}