summaryrefslogtreecommitdiff
path: root/gcc/testsuite/gcc.target/i386/avx-os-support.h
diff options
context:
space:
mode:
Diffstat (limited to 'gcc/testsuite/gcc.target/i386/avx-os-support.h')
-rw-r--r--gcc/testsuite/gcc.target/i386/avx-os-support.h12
1 files changed, 10 insertions, 2 deletions
diff --git a/gcc/testsuite/gcc.target/i386/avx-os-support.h b/gcc/testsuite/gcc.target/i386/avx-os-support.h
index 21d332f7d3..fb1ce7562d 100644
--- a/gcc/testsuite/gcc.target/i386/avx-os-support.h
+++ b/gcc/testsuite/gcc.target/i386/avx-os-support.h
@@ -1,10 +1,18 @@
/* Check if the OS supports executing AVX instructions. */
+#define XCR_XFEATURE_ENABLED_MASK 0x0
+
+#define XSTATE_FP 0x1
+#define XSTATE_SSE 0x2
+#define XSTATE_YMM 0x4
+
static int
avx_os_support (void)
{
unsigned int eax, edx;
+ unsigned int ecx = XCR_XFEATURE_ENABLED_MASK;
+
+ __asm__ ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (ecx));
- __asm__ ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (0));
- return (eax & 6) == 6;
+ return (eax & (XSTATE_SSE | XSTATE_YMM)) == (XSTATE_SSE | XSTATE_YMM);
}