sysdeps/x86_64/tst-audit10.c

   1 /* Copyright (C) 2012-2015 Free Software Foundation, Inc.
   2    This file is part of the GNU C Library.
   3
   4    The GNU C Library is free software; you can redistribute it and/or
   5    modify it under the terms of the GNU Lesser General Public
   6    License as published by the Free Software Foundation; either
   7    version 2.1 of the License, or (at your option) any later version.
   8
   9    The GNU C Library is distributed in the hope that it will be useful,
  10    but WITHOUT ANY WARRANTY; without even the implied warranty of
  11    MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
  12    Lesser General Public License for more details.
  13
  14    You should have received a copy of the GNU Lesser General Public
  15    License along with the GNU C Library; if not, see
  16    <http://www.gnu.org/licenses/>.  */
  17
  18 /* Test case for x86-64 preserved registers in dynamic linker.  */
  19
  20 #ifdef __AVX512F__
  21 #include <stdlib.h>
  22 #include <string.h>
  23 #include <cpuid.h>
  24 #include <immintrin.h>
  25
  26 static int
  27 avx512_enabled (void)
  28 {
  29   unsigned int eax, ebx, ecx, edx;
  30
  31   if (__get_cpuid (1, &eax, &ebx, &ecx, &edx) == 0
  32       || (ecx & (bit_AVX | bit_OSXSAVE)) != (bit_AVX | bit_OSXSAVE))
  33     return 0;
  34
  35   __cpuid_count (7, 0, eax, ebx, ecx, edx);
  36   if (!(ebx & bit_AVX512F))
  37     return 0;
  38
  39   asm ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (0));
  40
  41   /* Verify that ZMM, YMM and XMM states are enabled.  */
  42   return (eax & 0xe6) == 0xe6;
  43 }
  44
  45
  46 extern __m512i audit_test (__m512i, __m512i, __m512i, __m512i,
  47                            __m512i, __m512i, __m512i, __m512i);
  48 static int
  49 do_test (void)
  50 {
  51   /* Run AVX512 test only if AVX512 is supported.  */
  52   if (avx512_enabled ())
  53     {
  54       __m512i zmm = _mm512_setzero_si512 ();
  55       __m512i ret = audit_test (zmm, zmm, zmm, zmm, zmm, zmm, zmm, zmm);
  56
  57       zmm = _mm512_set1_epi64 (0x12349876);
  58
  59       if (memcmp (&zmm, &ret, sizeof (ret)))
  60         abort ();
  61     }
  62   return 0;
  63 }
  64 #else
  65 static int
  66 do_test (void)
  67 {
  68   return 0;
  69 }
  70 #endif
  71
  72 #define TEST_FUNCTION do_test ()
  73 #include "../../test-skeleton.c"