Test from_fn autovectorizes

scottmcm · workingjubilee · commit 374f5a8091f5 · 2023-05-30T20:45:40.000-07:00
diff --git a/tests/codegen/autovectorize-f32x4.rs b/tests/codegen/autovectorize-f32x4.rs
@@ -30,3 +30,13 @@ pub fn auto_vectorize_loop(a: [f32; 4], b: [f32; 4]) -> [f32; 4] {
     }
     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])
+}

Original file line number	Diff line number	Diff line change
`@@ -30,3 +30,13 @@ pub fn auto_vectorize_loop(a: [f32; 4], b: [f32; 4]) -> [f32; 4] {`
`30`	`30`	`}`
`31`	`31`	`c`
`32`	`32`	`}`
	`33`	`+`
	`34`	`+// CHECK-LABEL: @auto_vectorize_array_from_fn`
	`35`	`+#[no_mangle]`
	`36`	`+pub fn auto_vectorize_array_from_fn(a: [f32; 4], b: [f32; 4]) -> [f32; 4] {`
	`37`	`+// CHECK: load <4 x float>`
	`38`	`+// CHECK: load <4 x float>`
	`39`	`+// CHECK: fadd <4 x float>`
	`40`	`+// CHECK: store <4 x float>`
	`41`	`+ std::array::from_fn(\|i\| a[i] + b[i])`
	`42`	`+}`