glibc/elf/tst-audit6.c
Ulrich Drepper 5644ef5461 Fix check for AVX enablement
The AVX bit is set if the CPU supports AVX.  But this doesn't mean the
kernel does.  Add checks according to Intel's documentation.
2011-07-20 21:21:03 -04:00

43 lines
926 B
C

/* Test case for x86-64 preserved registers in dynamic linker. */
#include <stdlib.h>
#include <string.h>
#include <cpuid.h>
#include <emmintrin.h>
extern __m128i audit_test (__m128i, __m128i, __m128i, __m128i,
__m128i, __m128i, __m128i, __m128i);
static int
avx_enabled (void)
{
unsigned int eax, ebx, ecx, edx;
if (__get_cpuid (1, &eax, &ebx, &ecx, &edx) == 0
|| (ecx & (bit_AVX | bit_OSXSAVE)) != (bit_AVX | bit_OSXSAVE))
return 0;
/* Check the OS has AVX and SSE saving enabled. */
asm ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (0));
return (eax & 6) == 6;
}
int
main (void)
{
/* Run AVX test only if AVX is supported. */
if (avx_enabled ())
{
__m128i xmm = _mm_setzero_si128 ();
__m128i ret = audit_test (xmm, xmm, xmm, xmm, xmm, xmm, xmm, xmm);
xmm = _mm_set1_epi32 (0x98abcdef);
if (memcmp (&xmm, &ret, sizeof (ret)))
abort ();
}
return 0;
}