blob: 259a7c1ef7099bc89b30cdb0ebbd3a311fa88e18 [file] [log] [blame]
Borislav Petkovd61931d2010-03-05 17:34:46 +01001#ifndef _ASM_X86_HWEIGHT_H
2#define _ASM_X86_HWEIGHT_H
3
4#ifdef CONFIG_64BIT
H. Peter Anvinc59bd562010-05-17 15:13:23 -07005/* popcnt %edi, %eax -- redundant REX prefix for alignment */
6#define POPCNT32 ".byte 0xf3,0x40,0x0f,0xb8,0xc7"
Borislav Petkovd61931d2010-03-05 17:34:46 +01007/* popcnt %rdi, %rax */
H. Peter Anvinc59bd562010-05-17 15:13:23 -07008#define POPCNT64 ".byte 0xf3,0x48,0x0f,0xb8,0xc7"
Borislav Petkovd61931d2010-03-05 17:34:46 +01009#define REG_IN "D"
10#define REG_OUT "a"
11#else
12/* popcnt %eax, %eax */
H. Peter Anvinc59bd562010-05-17 15:13:23 -070013#define POPCNT32 ".byte 0xf3,0x0f,0xb8,0xc0"
Borislav Petkovd61931d2010-03-05 17:34:46 +010014#define REG_IN "a"
15#define REG_OUT "a"
16#endif
17
18/*
19 * __sw_hweightXX are called from within the alternatives below
20 * and callee-clobbered registers need to be taken care of. See
21 * ARCH_HWEIGHT_CFLAGS in <arch/x86/Kconfig> for the respective
22 * compiler switches.
23 */
Denys Vlasenkod14edb12015-08-04 16:15:15 +020024static __always_inline unsigned int __arch_hweight32(unsigned int w)
Borislav Petkovd61931d2010-03-05 17:34:46 +010025{
26 unsigned int res = 0;
27
H. Peter Anvinc59bd562010-05-17 15:13:23 -070028 asm (ALTERNATIVE("call __sw_hweight32", POPCNT32, X86_FEATURE_POPCNT)
Borislav Petkovd61931d2010-03-05 17:34:46 +010029 : "="REG_OUT (res)
30 : REG_IN (w));
31
32 return res;
33}
34
35static inline unsigned int __arch_hweight16(unsigned int w)
36{
37 return __arch_hweight32(w & 0xffff);
38}
39
40static inline unsigned int __arch_hweight8(unsigned int w)
41{
42 return __arch_hweight32(w & 0xff);
43}
44
Denys Vlasenkod14edb12015-08-04 16:15:15 +020045#ifdef CONFIG_X86_32
Borislav Petkovd61931d2010-03-05 17:34:46 +010046static inline unsigned long __arch_hweight64(__u64 w)
47{
Denys Vlasenkod14edb12015-08-04 16:15:15 +020048 return __arch_hweight32((u32)w) +
49 __arch_hweight32((u32)(w >> 32));
50}
51#else
52static __always_inline unsigned long __arch_hweight64(__u64 w)
53{
Borislav Petkovd61931d2010-03-05 17:34:46 +010054 unsigned long res = 0;
55
H. Peter Anvinc59bd562010-05-17 15:13:23 -070056 asm (ALTERNATIVE("call __sw_hweight64", POPCNT64, X86_FEATURE_POPCNT)
Borislav Petkovd61931d2010-03-05 17:34:46 +010057 : "="REG_OUT (res)
58 : REG_IN (w));
Borislav Petkovd61931d2010-03-05 17:34:46 +010059
60 return res;
61}
Denys Vlasenkod14edb12015-08-04 16:15:15 +020062#endif /* CONFIG_X86_32 */
Borislav Petkovd61931d2010-03-05 17:34:46 +010063
64#endif