1 | /* Test case for preserved AVX512 registers in dynamic linker. |
2 | Copyright (C) 2017-2022 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_avx512_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 (leaf: 1, eax: &eax, ebx: &ebx, ecx: &ecx, edx: &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_avx512_aux (); |
52 | else |
53 | return 77; |
54 | } |
55 | |
56 | #define TEST_FUNCTION do_test () |
57 | #include "../../test-skeleton.c" |
58 | |