1 #include <stdlib.h>
2 #include "cpuid.h"
3 #include "m512-check.h"
4 #include "avx512f-os-support.h"
5
6 #ifndef DO_TEST
7 #define DO_TEST do_test
8 #ifdef AVX512VL
9 static void test_256 (void);
10 static void test_128 (void);
11 #else
12 static void test_512 (void);
13 #endif
14
15 __attribute__ ((noinline))
16 static void
do_test(void)17 do_test (void)
18 {
19 #ifdef AVX512VL
20 test_256 ();
21 test_128 ();
22 #else
23 test_512 ();
24 #endif
25 }
26 #endif
27
28 static int
check_osxsave(void)29 check_osxsave (void)
30 {
31 unsigned int eax, ebx, ecx, edx;
32
33 __cpuid (1, eax, ebx, ecx, edx);
34 return (ecx & bit_OSXSAVE) != 0;
35 }
36
37 int
main()38 main ()
39 {
40 unsigned int eax, ebx, ecx, edx;
41
42 if (!__get_cpuid_count (7, 0, &eax, &ebx, &ecx, &edx))
43 return 0;
44
45 /* Run AVX512 test only if host has ISA support. */
46 if (check_osxsave ()
47 && (ebx & bit_AVX512F)
48 #ifdef AVX512VL
49 && (ebx & bit_AVX512VL)
50 #endif
51 #ifdef AVX512ER
52 && (ebx & bit_AVX512ER)
53 #endif
54 #ifdef AVX512CD
55 && (ebx & bit_AVX512CD)
56 #endif
57 #ifdef AVX512DQ
58 && (ebx & bit_AVX512DQ)
59 #endif
60 #ifdef AVX512BW
61 && (ebx & bit_AVX512BW)
62 #endif
63 #ifdef AVX512IFMA
64 && (ebx & bit_AVX512IFMA)
65 #endif
66 #ifdef AVX512VBMI
67 && (ecx & bit_AVX512VBMI)
68 #endif
69 #ifdef AVX5124FMAPS
70 && (edx & bit_AVX5124FMAPS)
71 #endif
72 #ifdef AVX5124VNNIW
73 && (edx & bit_AVX5124VNNIW)
74 #endif
75 #ifdef AVX512VPOPCNTDQ
76 && (ecx & bit_AVX512VPOPCNTDQ)
77 #endif
78 #ifdef AVX512BITALG
79 && (ecx & bit_AVX512BITALG)
80 #endif
81 #ifdef GFNI
82 && (ecx & bit_GFNI)
83 #endif
84 #ifdef AVX512VBMI2
85 && (ecx & bit_AVX512VBMI2)
86 #endif
87 #ifdef AVX512VNNI
88 && (ecx & bit_AVX512VNNI)
89 #endif
90 #ifdef VAES
91 && (ecx & bit_VAES)
92 #endif
93 #ifdef VPCLMULQDQ
94 && (ecx & bit_VPCLMULQDQ)
95 #endif
96 && avx512f_os_support ())
97 {
98 DO_TEST ();
99 #ifdef DEBUG
100 printf ("PASSED\n");
101 #endif
102 return 0;
103 }
104
105 #ifdef DEBUG
106 printf ("SKIPPED\n");
107 #endif
108 return 0;
109 }
110