Test from_fn autovectorizes

2023-05-30 20:38:07 -07:00 · 2023-05-30 20:38:07 -07:00 · 374f5a8091
commit 374f5a8091
parent 472230d192
1 changed files with 10 additions and 0 deletions
--- a/tests/codegen/autovectorize-f32x4.rs
+++ b/tests/codegen/autovectorize-f32x4.rs
@ -30,3 +30,13 @@
    }
    c
 }
+
+// CHECK-LABEL: @auto_vectorize_array_from_fn
+#[no_mangle]
+pub fn auto_vectorize_array_from_fn(a: [f32; 4], b: [f32; 4]) -> [f32; 4] {
+// CHECK: load <4 x float>
+// CHECK: load <4 x float>
+// CHECK: fadd <4 x float>
+// CHECK: store <4 x float>
+    std::array::from_fn(|i| a[i] + b[i])
+}