elf/tst-audit4.c

   1 /* Test case for x86-64 preserved registers in dynamic linker.  */
   2
   3 #ifdef __AVX__
   4 #include <stdlib.h>
   5 #include <string.h>
   6 #include <cpuid.h>
   7 #include <immintrin.h>
   8
   9
  10 static int
  11 avx_enabled (void)
  12 {
  13   unsigned int eax, ebx, ecx, edx;
  14
  15   if (__get_cpuid (1, &eax, &ebx, &ecx, &edx) == 0
  16       || (ecx & (bit_AVX | bit_OSXSAVE)) != (bit_AVX | bit_OSXSAVE))
  17     return 0;
  18
  19   /* Check the OS has AVX and SSE saving enabled.  */
  20   asm ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (0));
  21
  22   return (eax & 6) == 6;
  23 }
  24
  25
  26 extern __m256i audit_test (__m256i, __m256i, __m256i, __m256i,
  27                            __m256i, __m256i, __m256i, __m256i);
  28 int
  29 main (void)
  30 {
  31   /* Run AVX test only if AVX is supported.  */
  32   if (avx_enabled ())
  33     {
  34       __m256i ymm = _mm256_setzero_si256 ();
  35       __m256i ret = audit_test (ymm, ymm, ymm, ymm, ymm, ymm, ymm, ymm);
  36
  37       ymm =  _mm256_set1_epi32 (0x12349876);
  38       if (memcmp (&ymm, &ret, sizeof (ret)))
  39         abort ();
  40     }
  41   return 0;
  42 }
  43 #else
  44 int
  45 main (void)
  46 {
  47   return 0;
  48 }
  49 #endif