about summary refs log tree commit diff
path: root/tests/codegen-llvm/target-feature-inline-closure.rs
diff options
context:
space:
mode:
Diffstat (limited to 'tests/codegen-llvm/target-feature-inline-closure.rs')
-rw-r--r--tests/codegen-llvm/target-feature-inline-closure.rs33
1 files changed, 33 insertions, 0 deletions
diff --git a/tests/codegen-llvm/target-feature-inline-closure.rs b/tests/codegen-llvm/target-feature-inline-closure.rs
new file mode 100644
index 00000000000..5d54444f994
--- /dev/null
+++ b/tests/codegen-llvm/target-feature-inline-closure.rs
@@ -0,0 +1,33 @@
+//@ only-x86_64
+// Set the base cpu explicitly, in case the default has been changed.
+//@ compile-flags: -Copt-level=3 -Ctarget-cpu=x86-64
+
+#![crate_type = "lib"]
+
+#[cfg(target_arch = "x86_64")]
+use std::arch::x86_64::*;
+
+// CHECK-LABEL: @with_avx
+#[no_mangle]
+#[cfg(target_arch = "x86_64")]
+#[target_feature(enable = "avx")]
+fn with_avx(x: __m256) -> __m256 {
+    // CHECK: fadd <8 x float>
+    let add = {
+        #[inline(always)]
+        |x, y| unsafe { _mm256_add_ps(x, y) }
+    };
+    add(x, x)
+}
+
+// CHECK-LABEL: @without_avx
+#[no_mangle]
+#[cfg(target_arch = "x86_64")]
+unsafe fn without_avx(x: __m256) -> __m256 {
+    // CHECK-NOT: fadd <8 x float>
+    let add = {
+        #[inline(always)]
+        |x, y| unsafe { _mm256_add_ps(x, y) }
+    };
+    add(x, x)
+}