Chris@10
|
1 /*
|
Chris@10
|
2 * Copyright (c) 2003, 2007-11 Matteo Frigo
|
Chris@10
|
3 * Copyright (c) 2003, 2007-11 Massachusetts Institute of Technology
|
Chris@10
|
4 *
|
Chris@10
|
5 * This program is free software; you can redistribute it and/or modify
|
Chris@10
|
6 * it under the terms of the GNU General Public License as published by
|
Chris@10
|
7 * the Free Software Foundation; either version 2 of the License, or
|
Chris@10
|
8 * (at your option) any later version.
|
Chris@10
|
9 *
|
Chris@10
|
10 * This program is distributed in the hope that it will be useful,
|
Chris@10
|
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
Chris@10
|
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
Chris@10
|
13 * GNU General Public License for more details.
|
Chris@10
|
14 *
|
Chris@10
|
15 * You should have received a copy of the GNU General Public License
|
Chris@10
|
16 * along with this program; if not, write to the Free Software
|
Chris@10
|
17 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
|
Chris@10
|
18 *
|
Chris@10
|
19 */
|
Chris@10
|
20
|
Chris@10
|
21
|
Chris@10
|
22 #ifdef _MSC_VER
|
Chris@10
|
23 #ifndef inline
|
Chris@10
|
24 #define inline __inline
|
Chris@10
|
25 #endif
|
Chris@10
|
26 #endif
|
Chris@10
|
27
|
Chris@10
|
28 #ifdef _MSC_VER
|
Chris@10
|
29 #include <intrin.h>
|
Chris@10
|
30 #if (_MSC_VER >= 1600) && !defined(__INTEL_COMPILER)
|
Chris@10
|
31 #include <immintrin.h>
|
Chris@10
|
32 #endif
|
Chris@10
|
33 #endif
|
Chris@10
|
34
|
Chris@10
|
35 static inline int cpuid_ecx(int op)
|
Chris@10
|
36 {
|
Chris@10
|
37 # ifdef _MSC_VER
|
Chris@10
|
38 # ifdef __INTEL_COMPILER
|
Chris@10
|
39 int result;
|
Chris@10
|
40 _asm {
|
Chris@10
|
41 push rbx
|
Chris@10
|
42 mov eax,op
|
Chris@10
|
43 cpuid
|
Chris@10
|
44 mov result,ecx
|
Chris@10
|
45 pop rbx
|
Chris@10
|
46 }
|
Chris@10
|
47 return result;
|
Chris@10
|
48 # else
|
Chris@10
|
49 int cpu_info[4];
|
Chris@10
|
50 __cpuid(cpu_info,op);
|
Chris@10
|
51 return cpu_info[2];
|
Chris@10
|
52 # endif
|
Chris@10
|
53 # else
|
Chris@10
|
54 int eax, ecx, edx;
|
Chris@10
|
55
|
Chris@10
|
56 __asm__("pushq %%rbx\n\tcpuid\n\tpopq %%rbx"
|
Chris@10
|
57 : "=a" (eax), "=c" (ecx), "=d" (edx)
|
Chris@10
|
58 : "a" (op));
|
Chris@10
|
59 return ecx;
|
Chris@10
|
60 # endif
|
Chris@10
|
61 }
|
Chris@10
|
62
|
Chris@10
|
63 static inline int xgetbv_eax(int op)
|
Chris@10
|
64 {
|
Chris@10
|
65 # ifdef _MSC_VER
|
Chris@10
|
66 # ifdef __INTEL_COMPILER
|
Chris@10
|
67 int veax, vedx;
|
Chris@10
|
68 _asm {
|
Chris@10
|
69 mov ecx,op
|
Chris@10
|
70 xgetbv
|
Chris@10
|
71 mov veax,eax
|
Chris@10
|
72 mov vedx,edx
|
Chris@10
|
73 }
|
Chris@10
|
74 return veax;
|
Chris@10
|
75 # else
|
Chris@10
|
76 # if defined(_MSC_VER) && (_MSC_VER >= 1600)
|
Chris@10
|
77 unsigned __int64 result;
|
Chris@10
|
78 result = _xgetbv(op);
|
Chris@10
|
79 return (int)result;
|
Chris@10
|
80 # else
|
Chris@10
|
81 # error "Need at least Visual Studio 10 SP1 for AVX support"
|
Chris@10
|
82 # endif
|
Chris@10
|
83 # endif
|
Chris@10
|
84 # else
|
Chris@10
|
85 int eax, edx;
|
Chris@10
|
86 __asm__ (".byte 0x0f, 0x01, 0xd0" : "=a"(eax), "=d"(edx) : "c" (op));
|
Chris@10
|
87 return eax;
|
Chris@10
|
88 #endif
|
Chris@10
|
89 }
|