1  /* Test case for preserved AVX512 registers in dynamic linker.
       2     Copyright (C) 2012-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_audit10_aux (void);
      22  
      23  static int
      24  avx512_enabled (void)
      25  {
      26  #ifdef bit_AVX512F
      27    unsigned int eax, ebx, ecx, edx;
      28  
      29    if (__get_cpuid (1, &eax, &ebx, &ecx, &edx) == 0
      30        || (ecx & (bit_AVX | bit_OSXSAVE)) != (bit_AVX | bit_OSXSAVE))
      31      return 0;
      32  
      33    __cpuid_count (7, 0, eax, ebx, ecx, edx);
      34    if (!(ebx & bit_AVX512F))
      35      return 0;
      36  
      37    asm ("xgetbv" : "=a" (eax), "=d" (edx) : "c" (0));
      38  
      39    /* Verify that ZMM, YMM and XMM states are enabled.  */
      40    return (eax & 0xe6) == 0xe6;
      41  #else
      42    return 0;
      43  #endif
      44  }
      45  
      46  static int
      47  do_test (void)
      48  {
      49    /* Run AVX512 test only if AVX512 is supported.  */
      50    if (avx512_enabled ())
      51      return tst_audit10_aux ();
      52    else
      53      return 77;
      54  }
      55  
      56  #define TEST_FUNCTION do_test ()
      57  #include "../../test-skeleton.c"