1  /* Test case for preserved AVX registers in dynamic linker.
       2     Copyright (C) 2017-2023 Free Software Foundation, Inc.
       3     This file is part of the GNU C Library.
       4  
       5     The GNU C Library is free software; you can redistribute it and/or
       6     modify it under the terms of the GNU Lesser General Public
       7     License as published by the Free Software Foundation; either
       8     version 2.1 of the License, or (at your option) any later version.
       9  
      10     The GNU C Library is distributed in the hope that it will be useful,
      11     but WITHOUT ANY WARRANTY; without even the implied warranty of
      12     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
      13     Lesser General Public License for more details.
      14  
      15     You should have received a copy of the GNU Lesser General Public
      16     License along with the GNU C Library; if not, see
      17     <https://www.gnu.org/licenses/>.  */
      18  
      19  #include <cpuid.h>
      20  
      21  int tst_avx_aux (void);
      22  
      23  static int
      24  avx_enabled (void)
      25  {
      26    unsigned int eax, ebx, ecx, edx;
      27  
      28    if (__get_cpuid (1, &eax, &ebx, &ecx, &edx) == 0
      29        || (ecx & (bit_AVX | bit_OSXSAVE)) != (bit_AVX | bit_OSXSAVE))
      30      return 0;
      31  
      32    /* Check the OS has AVX and SSE saving enabled.  */
      33    asm ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (0));
      34  
      35    return (eax & 6) == 6;
      36  }
      37  
      38  static int
      39  do_test (void)
      40  {
      41    /* Run AVX test only if AVX is supported.  */
      42    if (avx_enabled ())
      43      return tst_avx_aux ();
      44    else
      45      return 77;
      46  }
      47  
      48  #define TEST_FUNCTION do_test ()
      49  #include "../../test-skeleton.c"