fixed lshr (i64 ...) function implementation

This commit is contained in:
Anton Mitrokhin
2014-11-27 18:33:44 +04:00
parent 0a987ad06f
commit 6f518dfcc9

View File

@@ -509,7 +509,7 @@ static FORCEINLINE __vec16_i32 __add(__vec16_i32 a, __vec16_i32 b) {
return _mm512_add_epi32(a, b);
}
static FORCEINLINE __vec16_i32 __sub(__vec16_i32 a, __vec16_i32 b) {
static FORCEINLINE __vec16_i32 __sub(__vec16_i32 a, __vec16_i32 b) {
return _mm512_sub_epi32(a, b);
}
@@ -964,7 +964,11 @@ static FORCEINLINE __vec16_i64 __lshr(__vec16_i64 a, __vec16_i64 b) {
}
static FORCEINLINE __vec16_i64 __lshr(__vec16_i64 a, unsigned long long b) {
__vec16_i32 xfer = _mm512_and_epi32(_mm512_slli_epi32(__ispc_ffffffff, 32-b), _mm512_slli_epi32(a.v_hi, 32-b));
/* this is a safety gate in case b-shift >= 32 */
__vec16_i32 xfer;
if (32 < b) xfer = __lshr(a.v_hi, b-32);
else xfer = _mm512_and_epi32(_mm512_slli_epi32(__ispc_ffffffff, 32-b), _mm512_slli_epi32(a.v_hi, 32-b));
__vec16_i32 hi = _mm512_srli_epi32(a.v_hi, b);
__vec16_i32 lo = _mm512_or_epi32(xfer, _mm512_srli_epi32(a.v_lo, b));
return __vec16_i64(lo, hi);