From 72b01d5cca0604e1ea5818d90b9feefae92a0093 Mon Sep 17 00:00:00 2001 From: The 8472 Date: Fri, 25 Aug 2023 23:29:18 +0200 Subject: Optimize Take::{fold, for_each} when wrapping TrustedRandomAccess iterators --- tests/codegen/lib-optimizations/iter-sum.rs | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 tests/codegen/lib-optimizations/iter-sum.rs (limited to 'tests/codegen/lib-optimizations') diff --git a/tests/codegen/lib-optimizations/iter-sum.rs b/tests/codegen/lib-optimizations/iter-sum.rs new file mode 100644 index 00000000000..d6ea4cd74d5 --- /dev/null +++ b/tests/codegen/lib-optimizations/iter-sum.rs @@ -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() +} -- cgit 1.4.1-3-g733a5 From f93e1258287e189aaf080d7e5336bac75633eb58 Mon Sep 17 00:00:00 2001 From: The 8472 Date: Sat, 2 Sep 2023 13:42:58 +0200 Subject: restrict test to x86-64 --- tests/codegen/lib-optimizations/iter-sum.rs | 1 + 1 file changed, 1 insertion(+) (limited to 'tests/codegen/lib-optimizations') diff --git a/tests/codegen/lib-optimizations/iter-sum.rs b/tests/codegen/lib-optimizations/iter-sum.rs index d6ea4cd74d5..ff7ca6ef6c1 100644 --- a/tests/codegen/lib-optimizations/iter-sum.rs +++ b/tests/codegen/lib-optimizations/iter-sum.rs @@ -1,5 +1,6 @@ // ignore-debug: the debug assertions get in the way // compile-flags: -O +// only-x86_64 (vectorization varies between architectures) #![crate_type = "lib"] -- cgit 1.4.1-3-g733a5