diff --git a/src/util/arch/common/popcount.h b/src/util/arch/common/popcount.h new file mode 100644 index 00000000..0bd1e837 --- /dev/null +++ b/src/util/arch/common/popcount.h @@ -0,0 +1,60 @@ +/* + * Copyright (c) 2015-2017, Intel Corporation + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * * Neither the name of Intel Corporation nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ + +/** \file + * \brief Platform specific popcount functions + */ + +#ifndef POPCOUNT_ARCH_COMMON_H +#define POPCOUNT_ARCH_COMMON_H + +static really_inline +u32 popcount32_impl_c(u32 x) { + // Fast branch-free version from bit-twiddling hacks as older Intel + // processors do not have a POPCNT instruction. + x -= (x >> 1) & 0x55555555; + x = (x & 0x33333333) + ((x >> 2) & 0x33333333); + return (((x + (x >> 4)) & 0xf0f0f0f) * 0x1010101) >> 24; +} + +static really_inline +u32 popcount64_impl_c(u64a x) { +#if defined(ARCH_64_BIT) + // Fast branch-free version from bit-twiddling hacks as older Intel + // processors do not have a POPCNT instruction. + x -= (x >> 1) & 0x5555555555555555; + x = (x & 0x3333333333333333) + ((x >> 2) & 0x3333333333333333); + x = (x + (x >> 4)) & 0x0f0f0f0f0f0f0f0f; + return (x * 0x0101010101010101) >> 56; +#else + // Synthesise from two 32-bit cases. + return popcount32_impl(x >> 32) + popcount32_impl(x); +#endif +} + +#endif // POPCOUNT_ARCH_COMMON_H \ No newline at end of file diff --git a/src/util/arch/x86/popcount.h b/src/util/arch/x86/popcount.h new file mode 100644 index 00000000..86929ede --- /dev/null +++ b/src/util/arch/x86/popcount.h @@ -0,0 +1,67 @@ +/* + * Copyright (c) 2015-2017, Intel Corporation + * + * Redistribution and use in source and binary forms, with or without + * modification, are permitted provided that the following conditions are met: + * + * * Redistributions of source code must retain the above copyright notice, + * this list of conditions and the following disclaimer. + * * Redistributions in binary form must reproduce the above copyright + * notice, this list of conditions and the following disclaimer in the + * documentation and/or other materials provided with the distribution. + * * Neither the name of Intel Corporation nor the names of its contributors + * may be used to endorse or promote products derived from this software + * without specific prior written permission. + * + * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" + * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE + * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE + * ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT OWNER OR CONTRIBUTORS BE + * LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR + * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF + * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS + * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN + * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) + * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE + * POSSIBILITY OF SUCH DAMAGE. + */ + +/** \file + * \brief Platform specific popcount functions + */ + +#ifndef POPCOUNT_ARCH_X86_H +#define POPCOUNT_ARCH_X86_H + +#include "ue2common.h" +#include "util/arch.h" +#include "util/intrinsics.h" + +#include "util/arch/common/popcount.h" + +static really_inline +u32 popcount32_impl(u32 x) { +#if defined(HAVE_POPCOUNT_INSTR) + // Single-instruction builtin. + return _mm_popcnt_u32(x); +#else + return popcount32_impl_c(x); +#endif +} + +static really_inline +u32 popcount64_impl(u64a x) { +#if defined(ARCH_X86_64) +# if defined(HAVE_POPCOUNT_INSTR) + // Single-instruction builtin. + return (u32)_mm_popcnt_u64(x); +# else + return popcount64_impl_c(x); +# endif +#else + // Synthesise from two 32-bit cases. + return popcount32_impl(x >> 32) + popcount32_impl(x); +#endif +} + +#endif // POPCOUNT_ARCH_X86_h \ No newline at end of file diff --git a/src/util/popcount.h b/src/util/popcount.h index eb08f6b1..932fc2cf 100644 --- a/src/util/popcount.h +++ b/src/util/popcount.h @@ -33,41 +33,22 @@ #ifndef UTIL_POPCOUNT_H_ #define UTIL_POPCOUNT_H_ +#include "config.h" #include "ue2common.h" #include "util/arch.h" +#if defined(ARCH_IA32) || defined(ARCH_X86_64) +#include "util/arch/x86/popcount.h" +#endif + static really_inline u32 popcount32(u32 x) { -#if defined(HAVE_POPCOUNT_INSTR) - // Single-instruction builtin. - return _mm_popcnt_u32(x); -#else - // Fast branch-free version from bit-twiddling hacks as older Intel - // processors do not have a POPCNT instruction. - x -= (x >> 1) & 0x55555555; - x = (x & 0x33333333) + ((x >> 2) & 0x33333333); - return (((x + (x >> 4)) & 0xf0f0f0f) * 0x1010101) >> 24; -#endif + return popcount32_impl(x); } static really_inline -u32 popcount64(u64a x) { -#if defined(ARCH_X86_64) -# if defined(HAVE_POPCOUNT_INSTR) - // Single-instruction builtin. - return (u32)_mm_popcnt_u64(x); -# else - // Fast branch-free version from bit-twiddling hacks as older Intel - // processors do not have a POPCNT instruction. - x -= (x >> 1) & 0x5555555555555555; - x = (x & 0x3333333333333333) + ((x >> 2) & 0x3333333333333333); - x = (x + (x >> 4)) & 0x0f0f0f0f0f0f0f0f; - return (x * 0x0101010101010101) >> 56; -# endif -#else - // Synthesise from two 32-bit cases. - return popcount32(x >> 32) + popcount32(x); -#endif +u32 popcount64(u32 x) { + return popcount64_impl(x); } #endif /* UTIL_POPCOUNT_H_ */