X-Git-Url: https://git.sesse.net/?p=stockfish;a=blobdiff_plain;f=src%2Fbitcount.h;h=3c5ba2a1fa4004e40905e0b2514d76d4ab1518ee;hp=7a05970d2abcb45a96333a34a070504941ac3b20;hb=7614501362896ddee902c1257f1b56ad148bc4d3;hpb=1ea70dd9ddb346aaa23e592edcff9774c305174d diff --git a/src/bitcount.h b/src/bitcount.h index 7a05970d..3c5ba2a1 100644 --- a/src/bitcount.h +++ b/src/bitcount.h @@ -1,7 +1,7 @@ /* Stockfish, a UCI chess playing engine derived from Glaurung 2.1 Copyright (C) 2004-2008 Tord Romstad (Glaurung author) - Copyright (C) 2008-2009 Marco Costalba + Copyright (C) 2008-2010 Marco Costalba, Joona Kiiski, Tord Romstad Stockfish is free software: you can redistribute it and/or modify it under the terms of the GNU General Public License as published by @@ -24,73 +24,23 @@ #include "types.h" -// Select type of intrinsic bit count instruction to use, see -// README.txt on how to pgo compile with POPCNT support. +enum BitCountType { + CNT64, + CNT64_MAX15, + CNT32, + CNT32_MAX15, + CNT_POPCNT +}; -#if defined(__INTEL_COMPILER) && defined(USE_POPCNT) // Intel compiler - -#include - -inline bool cpu_has_popcnt() { - - int CPUInfo[4] = {-1}; - __cpuid(CPUInfo, 0x00000001); - return (CPUInfo[2] >> 23) & 1; -} - -#define POPCNT_INTRINSIC(x) _mm_popcnt_u64(x) - -#elif defined(_MSC_VER) && defined(USE_POPCNT) // Microsoft compiler - -#include - -inline bool cpu_has_popcnt() { - - int CPUInfo[4] = {-1}; - __cpuid(CPUInfo, 0x00000001); - return (CPUInfo[2] >> 23) & 1; -} - -#define POPCNT_INTRINSIC(x) (int)__popcnt64(x) - -#elif defined(__GNUC__) && defined(USE_POPCNT) // Gcc compiler - -inline void __cpuid(unsigned int op, - unsigned int *eax, unsigned int *ebx, - unsigned int *ecx, unsigned int *edx) -{ - *eax = op; - *ecx = 0; - __asm__("cpuid" : "=a" (*eax), "=b" (*ebx), "=c" (*ecx), "=d" (*edx) - : "0" (*eax), "2" (*ecx)); -} - -inline bool cpu_has_popcnt() { - - unsigned int eax, ebx, ecx, edx; - __cpuid(1, &eax, &ebx, &ecx, &edx); - return (ecx >> 23) & 1; -} - -#define POPCNT_INTRINSIC(x) ({ \ - unsigned long __ret; \ - __asm__("popcnt %1, %0" : "=r" (__ret) : "r" (x)); \ - __ret; }) - -#else // Safe fallback for unsupported compilers or when USE_POPCNT is disabled - -inline bool cpu_has_popcnt() { return false; } - -#define POPCNT_INTRINSIC(x) 0 - -#endif // cpu_has_popcnt() and POPCNT_INTRINSIC() definitions - - -/// Software implementation of bit count functions - -#if defined(IS_64BIT) - -inline int count_1s(Bitboard b) { +/// count_1s() counts the number of nonzero bits in a bitboard. +/// We have different optimized versions according if platform +/// is 32 or 64 bits, and to the maximum number of nonzero bits. +/// We also support hardware popcnt instruction. See Readme.txt +/// on how to pgo compile with popcnt support. +template inline int count_1s(Bitboard); + +template<> +inline int count_1s(Bitboard b) { b -= ((b>>1) & 0x5555555555555555ULL); b = ((b>>2) & 0x3333333333333333ULL) + (b & 0x3333333333333333ULL); b = ((b>>4) + b) & 0x0F0F0F0F0F0F0F0FULL; @@ -98,16 +48,16 @@ inline int count_1s(Bitboard b) { return int(b >> 56); } -inline int count_1s_max_15(Bitboard b) { +template<> +inline int count_1s(Bitboard b) { b -= (b>>1) & 0x5555555555555555ULL; b = ((b>>2) & 0x3333333333333333ULL) + (b & 0x3333333333333333ULL); b *= 0x1111111111111111ULL; return int(b >> 60); } -#else // if !defined(IS_64BIT) - -inline int count_1s(Bitboard b) { +template<> +inline int count_1s(Bitboard b) { unsigned w = unsigned(b >> 32), v = unsigned(b); v -= (v >> 1) & 0x55555555; // 0-2 in 2 bits w -= (w >> 1) & 0x55555555; @@ -119,7 +69,8 @@ inline int count_1s(Bitboard b) { return int(v >> 24); } -inline int count_1s_max_15(Bitboard b) { +template<> +inline int count_1s(Bitboard b) { unsigned w = unsigned(b >> 32), v = unsigned(b); v -= (v >> 1) & 0x55555555; // 0-2 in 2 bits w -= (w >> 1) & 0x55555555; @@ -130,38 +81,47 @@ inline int count_1s_max_15(Bitboard b) { return int(v >> 28); } -#endif // BITCOUNT - - -/// count_1s() counts the number of nonzero bits in a bitboard. -/// If template parameter is true an intrinsic is called, otherwise -/// we fallback on a software implementation. - -template -inline int count_1s(Bitboard b) { - - return UseIntrinsic ? POPCNT_INTRINSIC(b) : count_1s(b); +template<> +inline int count_1s(Bitboard b) { +#if !defined(USE_POPCNT) + return int(b != 0); // Avoid 'b not used' warning +#elif defined(_MSC_VER) && defined(__INTEL_COMPILER) + return _mm_popcnt_u64(b); +#elif defined(_MSC_VER) + return __popcnt64(b); +#elif defined(__GNUC__) + unsigned long ret; + __asm__("popcnt %1, %0" : "=r" (ret) : "r" (b)); + return ret; +#endif } -template -inline int count_1s_max_15(Bitboard b) { - return UseIntrinsic ? POPCNT_INTRINSIC(b) : count_1s_max_15(b); +/// cpu_has_popcnt() detects support for popcnt instruction at runtime +inline bool cpu_has_popcnt() { + + int CPUInfo[4] = {-1}; + __cpuid(CPUInfo, 0x00000001); + return (CPUInfo[2] >> 23) & 1; } -// Global constant initialized at startup that is set to true if -// CPU on which application runs supports POPCNT intrinsic. Unless -// USE_POPCNT is not defined. +/// CpuHasPOPCNT is a global constant initialized at startup that +/// is set to true if CPU on which application runs supports popcnt +/// hardware instruction. Unless USE_POPCNT is not defined. +#if defined(USE_POPCNT) const bool CpuHasPOPCNT = cpu_has_popcnt(); +#else +const bool CpuHasPOPCNT = false; +#endif -// Global constant used to print info about the use of 64 optimized -// functions to verify that a 64 bit compile has been correctly built. +/// CpuIs64Bit is a global constant initialized at compile time that +/// is set to true if CPU on which application runs is a 64 bits. #if defined(IS_64BIT) -const bool CpuHas64BitPath = true; +const bool CpuIs64Bit = true; #else -const bool CpuHas64BitPath = false; +const bool CpuIs64Bit = false; #endif #endif // !defined(BITCOUNT_H_INCLUDED)