From 72b01d5cca0604e1ea5818d90b9feefae92a0093 Mon Sep 17 00:00:00 2001
From: The 8472 <git@infinite-source.de>
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