mirror of
https://github.com/JayDDee/cpuminer-opt.git
synced 2025-09-17 23:44:27 +00:00
814 lines
25 KiB
C
814 lines
25 KiB
C
#if !defined(SIMD_128_H__)
|
|
#define SIMD_128_H__ 1
|
|
|
|
#if defined(__SSE2__)
|
|
|
|
///////////////////////////////////////////////////////////////////////////
|
|
//
|
|
// 128 bit SSE vectors
|
|
//
|
|
// SSE2 is required for 128 bit integer support. Some functions are also
|
|
// optimized with SSSE3, SSE4.1 or AVX. Some of these more optimized
|
|
// functions don't have SSE2 equivalents and their use would break SSE2
|
|
// compatibility.
|
|
//
|
|
// Constants are an issue with simd. Simply put, immediate constants don't
|
|
// exist. All simd constants either reside in memory or a register and
|
|
// must be loaded from memory or generated at run time.
|
|
//
|
|
// Due to the cost of generating constants it is more efficient to
|
|
// define a local const for repeated references to the same constant.
|
|
//
|
|
// One common use for simd constants is as a control index for vector
|
|
// instructions like blend and shuffle. Alhough the ultimate instruction
|
|
// may execute in a single clock cycle, generating the control index adds
|
|
// several more cycles to the entire operation.
|
|
//
|
|
// All of the utilities here assume all data is in registers except
|
|
// in rare cases where arguments are pointers.
|
|
//
|
|
// Some constants are generated using a memory overlay on the stack.
|
|
//
|
|
// Intrinsics automatically promote from REX to VEX when AVX is available
|
|
// but ASM needs to be done manually.
|
|
//
|
|
///////////////////////////////////////////////////////////////////////////
|
|
|
|
|
|
// Used instead if casting.
|
|
typedef union
|
|
{
|
|
__m128i m128;
|
|
uint32_t u32[4];
|
|
} __attribute__ ((aligned (16))) m128_ovly;
|
|
|
|
// Efficient and convenient moving between GP & low bits of XMM.
|
|
// Use VEX when available to give access to xmm8-15 and zero extend for
|
|
// larger vectors.
|
|
|
|
static inline __m128i mm128_mov64_128( const uint64_t n )
|
|
{
|
|
__m128i a;
|
|
#if defined(__AVX__)
|
|
asm( "vmovq %1, %0\n\t" : "=x"(a) : "r"(n) );
|
|
#else
|
|
asm( "movq %1, %0\n\t" : "=x"(a) : "r"(n) );
|
|
#endif
|
|
return a;
|
|
}
|
|
|
|
static inline __m128i mm128_mov32_128( const uint32_t n )
|
|
{
|
|
__m128i a;
|
|
#if defined(__AVX__)
|
|
asm( "vmovd %1, %0\n\t" : "=x"(a) : "r"(n) );
|
|
#else
|
|
asm( "movd %1, %0\n\t" : "=x"(a) : "r"(n) );
|
|
#endif
|
|
return a;
|
|
}
|
|
|
|
// Inconstant naming, prefix should reflect return value:
|
|
// u64_mov128_64
|
|
|
|
static inline uint64_t u64_mov128_64( const __m128i a )
|
|
{
|
|
uint64_t n;
|
|
#if defined(__AVX__)
|
|
asm( "vmovq %1, %0\n\t" : "=r"(n) : "x"(a) );
|
|
#else
|
|
asm( "movq %1, %0\n\t" : "=r"(n) : "x"(a) );
|
|
#endif
|
|
return n;
|
|
}
|
|
|
|
static inline uint32_t u32_mov128_32( const __m128i a )
|
|
{
|
|
uint32_t n;
|
|
#if defined(__AVX__)
|
|
asm( "vmovd %1, %0\n\t" : "=r"(n) : "x"(a) );
|
|
#else
|
|
asm( "movd %1, %0\n\t" : "=r"(n) : "x"(a) );
|
|
#endif
|
|
return n;
|
|
}
|
|
|
|
// Equivalent of set1, broadcast integer to all elements.
|
|
#define m128_const_i128( i ) mm128_mov64_128( i )
|
|
#define m128_const1_64( i ) _mm_shuffle_epi32( mm128_mov64_128( i ), 0x44 )
|
|
#define m128_const1_32( i ) _mm_shuffle_epi32( mm128_mov32_128( i ), 0x00 )
|
|
|
|
#if defined(__SSE4_1__)
|
|
|
|
// Assign 64 bit integers to respective elements: {hi, lo}
|
|
#define m128_const_64( hi, lo ) \
|
|
_mm_insert_epi64( mm128_mov64_128( lo ), hi, 1 )
|
|
|
|
#else // No insert in SSE2
|
|
|
|
#define m128_const_64 _mm_set_epi64x
|
|
|
|
#endif
|
|
|
|
// Pseudo constants
|
|
|
|
#define m128_zero _mm_setzero_si128()
|
|
#define m128_one_128 mm128_mov64_128( 1 )
|
|
#define m128_one_64 _mm_shuffle_epi32( mm128_mov64_128( 1 ), 0x44 )
|
|
#define m128_one_32 _mm_shuffle_epi32( mm128_mov32_128( 1 ), 0x00 )
|
|
#define m128_one_16 _mm_shuffle_epi32( \
|
|
mm128_mov32_128( 0x00010001 ), 0x00 )
|
|
#define m128_one_8 _mm_shuffle_epi32( \
|
|
mm128_mov32_128( 0x01010101 ), 0x00 )
|
|
|
|
// ASM avoids the need to initialize return variable to avoid compiler warning.
|
|
// Macro abstracts function parentheses to look like an identifier.
|
|
|
|
static inline __m128i mm128_neg1_fn()
|
|
{
|
|
__m128i a;
|
|
#if defined(__AVX__)
|
|
asm( "vpcmpeqq %0, %0, %0\n\t" : "=x"(a) );
|
|
#else
|
|
asm( "pcmpeqq %0, %0\n\t" : "=x"(a) );
|
|
#endif
|
|
return a;
|
|
}
|
|
#define m128_neg1 mm128_neg1_fn()
|
|
|
|
#if defined(__SSE4_1__)
|
|
|
|
/////////////////////////////
|
|
//
|
|
// _mm_insert_ps( _mm128i v1, __m128i v2, imm8 c )
|
|
//
|
|
// Fast and powerful but very limited in its application.
|
|
// It requires SSE4.1 but only works with 128 bit vectors with 32 bit
|
|
// elements. There is no equivalent instruction for 256 bit or 512 bit vectors.
|
|
// There's no integer version. There's no 64 bit, 16 bit or byte element
|
|
// sizing. It's unique.
|
|
//
|
|
// It can:
|
|
// - zero 32 bit elements of a 128 bit vector.
|
|
// - extract any 32 bit element from one 128 bit vector and insert the
|
|
// data to any 32 bit element of another 128 bit vector, or the same vector.
|
|
// - do both simultaneoulsly.
|
|
//
|
|
// It can be used as a more efficient replacement for _mm_insert_epi32
|
|
// or _mm_extract_epi32.
|
|
//
|
|
// Control byte definition:
|
|
// c[3:0] zero mask
|
|
// c[5:4] destination element selector
|
|
// c[7:6] source element selector
|
|
|
|
// Convert type and abbreviate name: e"x"tract "i"nsert "m"ask
|
|
#define mm128_xim_32( v1, v2, c ) \
|
|
_mm_castps_si128( _mm_insert_ps( _mm_castsi128_ps( v1 ), \
|
|
_mm_castsi128_ps( v2 ), c ) )
|
|
|
|
// Some examples of simple operations:
|
|
|
|
// Insert 32 bit integer into v at element c and return modified v.
|
|
static inline __m128i mm128_insert_32( const __m128i v, const uint32_t i,
|
|
const int c )
|
|
{ return mm128_xim_32( v, mm128_mov32_128( i ), c<<4 ); }
|
|
|
|
// Extract 32 bit element c from v and return as integer.
|
|
static inline uint32_t mm128_extract_32( const __m128i v, const int c )
|
|
{ return u32_mov128_32( mm128_xim_32( v, v, c<<6 ) ); }
|
|
|
|
// Clear (zero) 32 bit elements based on bits set in 4 bit mask.
|
|
static inline __m128i mm128_mask_32( const __m128i v, const int m )
|
|
{ return mm128_xim_32( v, v, m ); }
|
|
|
|
// Move element i2 of v2 to element i1 of v1. For reference and convenience,
|
|
// it's faster to precalculate the index.
|
|
#define mm128_shuflmov_32( v1, i1, v2, i2 ) \
|
|
mm128_xim_32( v1, v2, ( (i1)<<4 ) | ( (i2)<<6 ) )
|
|
|
|
#endif // SSE4_1
|
|
|
|
//
|
|
// Basic operations without equivalent SIMD intrinsic
|
|
|
|
// Bitwise not (~v)
|
|
#define mm128_not( v ) _mm_xor_si128( v, m128_neg1 )
|
|
|
|
// Unary negation of elements (-v)
|
|
#define mm128_negate_64( v ) _mm_sub_epi64( m128_zero, v )
|
|
#define mm128_negate_32( v ) _mm_sub_epi32( m128_zero, v )
|
|
#define mm128_negate_16( v ) _mm_sub_epi16( m128_zero, v )
|
|
|
|
|
|
// Add 4 values, fewer dependencies than sequential addition.
|
|
#define mm128_add4_64( a, b, c, d ) \
|
|
_mm_add_epi64( _mm_add_epi64( a, b ), _mm_add_epi64( c, d ) )
|
|
|
|
#define mm128_add4_32( a, b, c, d ) \
|
|
_mm_add_epi32( _mm_add_epi32( a, b ), _mm_add_epi32( c, d ) )
|
|
|
|
#define mm128_add4_16( a, b, c, d ) \
|
|
_mm_add_epi16( _mm_add_epi16( a, b ), _mm_add_epi16( c, d ) )
|
|
|
|
#define mm128_add4_8( a, b, c, d ) \
|
|
_mm_add_epi8( _mm_add_epi8( a, b ), _mm_add_epi8( c, d ) )
|
|
|
|
#define mm128_xor4( a, b, c, d ) \
|
|
_mm_xor_si128( _mm_xor_si128( a, b ), _mm_xor_si128( c, d ) )
|
|
|
|
|
|
//
|
|
// Vector pointer cast
|
|
|
|
// p = any aligned pointer
|
|
// returns p as pointer to vector type
|
|
#define castp_m128i(p) ((__m128i*)(p))
|
|
|
|
// p = any aligned pointer
|
|
// returns *p, watch your pointer arithmetic
|
|
#define cast_m128i(p) (*((__m128i*)(p)))
|
|
|
|
// p = any aligned pointer, i = scaled array index
|
|
// returns value p[i]
|
|
#define casti_m128i(p,i) (((__m128i*)(p))[(i)])
|
|
|
|
// p = any aligned pointer, o = scaled offset
|
|
// returns pointer p+o
|
|
#define casto_m128i(p,o) (((__m128i*)(p))+(o))
|
|
|
|
|
|
// Memory functions
|
|
// Mostly for convenience, avoids calculating bytes.
|
|
// Assumes data is alinged and integral.
|
|
// n = number of __m128i, bytes/16
|
|
|
|
static inline void memset_zero_128( __m128i *dst, const int n )
|
|
{ for ( int i = 0; i < n; i++ ) dst[i] = m128_zero; }
|
|
|
|
static inline void memset_128( __m128i *dst, const __m128i a, const int n )
|
|
{ for ( int i = 0; i < n; i++ ) dst[i] = a; }
|
|
|
|
static inline void memcpy_128( __m128i *dst, const __m128i *src, const int n )
|
|
{ for ( int i = 0; i < n; i ++ ) dst[i] = src[i]; }
|
|
|
|
#if defined(__AVX512VL__)
|
|
|
|
// a ^ b ^ c
|
|
#define mm128_xor3( a, b, c ) \
|
|
_mm_ternarylogic_epi64( a, b, c, 0x96 )
|
|
|
|
// a ^ ( b & c )
|
|
#define mm128_xorand( a, b, c ) \
|
|
_mm_ternarylogic_epi64( a, b, c, 0x78 )
|
|
|
|
#else
|
|
|
|
#define mm128_xor3( a, b, c ) \
|
|
_mm_xor_si128( a, _mm_xor_si128( b, c ) )
|
|
|
|
#define mm128_xorand( a, b, c ) \
|
|
_mm_xor_si128( a, _mm_and_si128( b, c ) )
|
|
|
|
#endif
|
|
|
|
// Mask making
|
|
// Equivalent of AVX512 _mm_movepi64_mask & _mm_movepi32_mask.
|
|
// Returns 2 or 4 bit integer mask from MSB of 64 or 32 bit elements.
|
|
// Effectively a sign test.
|
|
|
|
#define mm_movmask_64( v ) \
|
|
_mm_castpd_si128( _mm_movmask_pd( _mm_castsi128_pd( v ) ) )
|
|
|
|
#define mm_movmask_32( v ) \
|
|
_mm_castps_si128( _mm_movmask_ps( _mm_castsi128_ps( v ) ) )
|
|
|
|
|
|
// Diagonal blend
|
|
|
|
// Blend 4 32 bit elements from 4 vectors
|
|
|
|
#if defined (__AVX2__)
|
|
|
|
#define mm128_diagonal_32( v3, v2, v1, v0 ) \
|
|
mm_blend_epi32( _mm_blend_epi32( s3, s2, 0x4 ), \
|
|
_mm_blend_epi32( s1, s0, 0x1 ), 0x3 )
|
|
|
|
#elif defined(__SSE4_1__)
|
|
|
|
#define mm128_diagonal_32( v3, v2, v1, v0 ) \
|
|
mm_blend_epi16( _mm_blend_epi16( s3, s2, 0x30 ), \
|
|
_mm_blend_epi16( s1, s0, 0x03 ), 0x0f )
|
|
|
|
#endif
|
|
|
|
/*
|
|
//
|
|
// Extended bit shift for concatenated packed elements from 2 vectors.
|
|
// Shift right returns low half, shift left return high half.
|
|
|
|
#if defined(__AVX512VBMI2__) && defined(__AVX512VL__)
|
|
|
|
#define mm128_shl2_64( v1, v2, c ) _mm_shldi_epi64( v1, v2, c )
|
|
#define mm128_shr2_64( v1, v2, c ) _mm_shrdi_epi64( v1, v2, c )
|
|
|
|
#define mm128_shl2_32( v1, v2, c ) _mm_shldi_epi32( v1, v2, c )
|
|
#define mm128_shr2_32( v1, v2, c ) _mm_shrdi_epi32( v1, v2, c )
|
|
|
|
#define mm128_shl2_16( v1, v2, c ) _mm_shldi_epi16( v1, v2, c )
|
|
#define mm128_shr2_16( v1, v2, c ) _mm_shrdi_epi16( v1, v2, c )
|
|
|
|
#else
|
|
|
|
#define mm128_shl2_64( v1, v2, c ) \
|
|
_mm_or_si128( _mm_slli_epi64( v1, c ), _mm_srli_epi64( v2, 64 - (c) ) )
|
|
|
|
#define mm128_shr2_64( v1, v2, c ) \
|
|
_mm_or_si128( _mm_srli_epi64( v2, c ), _mm_slli_epi64( v1, 64 - (c) ) )
|
|
|
|
#define mm128_shl2_32( v1, v2, c ) \
|
|
_mm_or_si128( _mm_slli_epi32( v1, c ), _mm_srli_epi32( v2, 32 - (c) ) )
|
|
|
|
#define mm128_shr2_32( v1, v2, c ) \
|
|
_mm_or_si128( _mm_srli_epi32( v2, c ), _mm_slli_epi32( v1, 32 - (c) ) )
|
|
|
|
#define mm128_shl2_16( v1, v2, c ) \
|
|
_mm_or_si128( _mm_slli_epi16( v1, c ), _mm_srli_epi16( v2, 16 - (c) ) )
|
|
|
|
#define mm128_shr2_16( v1, v2, c ) \
|
|
_mm_or_si128( _mm_srli_epi16( v2, c ), _mm_slli_epi16( v1, 16 - (c) ) )
|
|
|
|
#endif
|
|
*/
|
|
|
|
//
|
|
// Bit rotations
|
|
|
|
// x2 rotates elements in 2 individual vectors in a double buffered
|
|
// optimization for SSE2, does nothing for AVX512 but is there for
|
|
// transparency.
|
|
|
|
#if defined(__AVX512VL__)
|
|
|
|
#define mm128_ror_64 _mm_ror_epi64
|
|
#define mm128_rol_64 _mm_rol_epi64
|
|
#define mm128_ror_32 _mm_ror_epi32
|
|
#define mm128_rol_32 _mm_rol_epi32
|
|
|
|
#define mm128_rorx2_64( v1, v0, c ) \
|
|
_mm_ror_epi64( v0, c ); \
|
|
_mm_ror_epi64( v1, c )
|
|
|
|
#define mm128_rolx2_64( v1, v0, c ) \
|
|
_mm_rol_epi64( v0, c ); \
|
|
_mm_rol_epi64( v1, c )
|
|
|
|
#define mm128_rorx2_32( v1, v0, c ) \
|
|
_mm_ror_epi32( v0, c ); \
|
|
_mm_ror_epi32( v1, c )
|
|
|
|
#define mm128_rolx2_32( v1, v0, c ) \
|
|
_mm_rol_epi32( v0, c ); \
|
|
_mm_rol_epi32( v1, c )
|
|
|
|
#else // SSE2
|
|
|
|
#define mm128_ror_64( v, c ) \
|
|
_mm_or_si128( _mm_srli_epi64( v, c ), _mm_slli_epi64( v, 64-(c) ) )
|
|
|
|
#define mm128_rol_64( v, c ) \
|
|
_mm_or_si128( _mm_slli_epi64( v, c ), _mm_srli_epi64( v, 64-(c) ) )
|
|
|
|
#define mm128_ror_32( v, c ) \
|
|
_mm_or_si128( _mm_srli_epi32( v, c ), _mm_slli_epi32( v, 32-(c) ) )
|
|
|
|
#define mm128_rol_32( v, c ) \
|
|
_mm_or_si128( _mm_slli_epi32( v, c ), _mm_srli_epi32( v, 32-(c) ) )
|
|
|
|
#define mm128_rorx2_64( v1, v0, c ) \
|
|
{ \
|
|
__m128i t0 = _mm_srli_epi64( v0, c ); \
|
|
__m128i t1 = _mm_srli_epi64( v1, c ); \
|
|
v0 = _mm_slli_epi64( v0, 64-(c) ); \
|
|
v1 = _mm_slli_epi64( v1, 64-(c) ); \
|
|
v0 = _mm_or_si256( v0, t0 ); \
|
|
v1 = _mm_or_si256( v1, t1 ); \
|
|
}
|
|
|
|
#define mm128_rolx2_64( v1, v0, c ) \
|
|
{ \
|
|
__m128i t0 = _mm_slli_epi64( v0, c ); \
|
|
__m128i t1 = _mm_slli_epi64( v1, c ); \
|
|
v0 = _mm_srli_epi64( v0, 64-(c) ); \
|
|
v1 = _mm_srli_epi64( v1, 64-(c) ); \
|
|
v0 = _mm_or_si256( v0, t0 ); \
|
|
v1 = _mm_or_si256( v1, t1 ); \
|
|
}
|
|
|
|
#define mm128_rorx2_32( v1, v0, c ) \
|
|
{ \
|
|
__m128i t0 = _mm_srli_epi32( v0, c ); \
|
|
__m128i t1 = _mm_srli_epi32( v1, c ); \
|
|
v0 = _mm_slli_epi32( v0, 32-(c) ); \
|
|
v1 = _mm_slli_epi32( v1, 32-(c) ); \
|
|
v0 = _mm_or_si256( v0, t0 ); \
|
|
v1 = _mm_or_si256( v1, t1 ); \
|
|
}
|
|
|
|
#define mm128_rolx2_32( v1, v0, c ) \
|
|
{ \
|
|
__m128i t0 = _mm_slli_epi32( v0, c ); \
|
|
__m128i t1 = _mm_slli_epi32( v1, c ); \
|
|
v0 = _mm_srli_epi32( v0, 32-(c) ); \
|
|
v1 = _mm_srli_epi32( v1, 32-(c) ); \
|
|
v0 = _mm_or_si256( v0, t0 ); \
|
|
v1 = _mm_or_si256( v1, t1 ); \
|
|
}
|
|
|
|
#endif // AVX512 else SSE2
|
|
|
|
#define mm128_ror_16( v, c ) \
|
|
_mm_or_si128( _mm_srli_epi16( v, c ), _mm_slli_epi16( v, 16-(c) ) )
|
|
|
|
#define mm128_rol_16( v, c ) \
|
|
_mm_or_si128( _mm_slli_epi16( v, c ), _mm_srli_epi16( v, 16-(c) ) )
|
|
|
|
// Deprecated.
|
|
#define mm128_rol_var_32( v, c ) \
|
|
_mm_or_si128( _mm_slli_epi32( v, c ), _mm_srli_epi32( v, 32-(c) ) )
|
|
|
|
//
|
|
// Limited 2 input shuffle, combines shuffle with blend. The destination low
|
|
// half is always taken from src a, and the high half from src b.
|
|
#define mm128_shuffle2_64( v1, v2, c ) \
|
|
_mm_castpd_si128( _mm_shuffle_pd( _mm_castsi128_pd( v1 ), \
|
|
_mm_castsi128_pd( v2 ), c ) );
|
|
|
|
#define mm128_shuffle2_32( v1, v2, c ) \
|
|
_mm_castps_si128( _mm_shuffle_ps( _mm_castsi128_ps( v1 ), \
|
|
_mm_castsi128_ps( v2 ), c ) );
|
|
|
|
//
|
|
// Rotate vector elements accross all lanes
|
|
|
|
#define mm128_swap_64( v ) _mm_shuffle_epi32( v, 0x4e )
|
|
#define mm128_shuflr_64 mm128_swap_64
|
|
#define mm128_shufll_64 mm128_swap_64
|
|
|
|
#define mm128_shuflr_32( v ) _mm_shuffle_epi32( v, 0x39 )
|
|
#define mm128_shufll_32( v ) _mm_shuffle_epi32( v, 0x93 )
|
|
|
|
#if defined(__SSSE3__)
|
|
|
|
// Rotate right by c bytes, no SSE2 equivalent.
|
|
static inline __m128i mm128_shuflr_x8( const __m128i v, const int c )
|
|
{ return _mm_alignr_epi8( v, v, c ); }
|
|
|
|
#endif
|
|
|
|
// Rotate byte elements within 64 or 32 bit lanes, AKA optimized bit rotations
|
|
// for multiples of 8 bits. Uses ror/rol macros when AVX512 is available
|
|
// (unlikely but faster), or when SSSE3 is not available (slower).
|
|
|
|
#define mm128_swap64_32( v ) _mm_shuffle_epi32( v, 0xb1 )
|
|
#define mm128_shuflr64_32 mm128_swap64_32
|
|
#define mm128_shufll64_32 mm128_swap64_32
|
|
|
|
#if defined(__SSSE3__) && !defined(__AVX512VL__)
|
|
#define mm128_shuflr64_24( v ) \
|
|
_mm_shuffle_epi8( v, _mm_set_epi64x( \
|
|
0x0a09080f0e0d0c0b, 0x0201000706050403 ) )
|
|
#else
|
|
#define mm128_shuflr64_24( v ) mm128_ror_64( v, 24 )
|
|
#endif
|
|
|
|
#if defined(__SSSE3__) && !defined(__AVX512VL__)
|
|
#define mm128_shuflr64_16( v ) \
|
|
_mm_shuffle_epi8( v, _mm_set_epi64x( \
|
|
0x09080f0e0d0c0b0a, 0x0100070605040302 ) )
|
|
#else
|
|
#define mm128_shuflr64_16( v ) mm128_ror_64( v, 16 )
|
|
#endif
|
|
|
|
#if defined(__SSSE3__) && !defined(__AVX512VL__)
|
|
#define mm128_swap32_16( v ) \
|
|
_mm_shuffle_epi8( v, _mm_set_epi64x( \
|
|
0x0d0c0f0e09080b0a, 0x0504070601000302 ) )
|
|
#else
|
|
#define mm128_swap32_16( v ) mm128_ror_32( v, 16 )
|
|
#endif
|
|
#define mm128_shuflr32_16 mm128_swap32_16
|
|
#define mm128_shufll32_16 mm128_swap32_16
|
|
|
|
#if defined(__SSSE3__) && !defined(__AVX512VL__)
|
|
#define mm128_shuflr32_8( v ) \
|
|
_mm_shuffle_epi8( v, _mm_set_epi64x( \
|
|
0x0c0f0e0d080b0a09, 0x0407060500030201 ) )
|
|
#else
|
|
#define mm128_shuflr32_8( v ) mm128_ror_32( v, 8 )
|
|
#endif
|
|
|
|
//
|
|
// Endian byte swap.
|
|
|
|
#if defined(__SSSE3__)
|
|
|
|
#define mm128_bswap_64( v ) \
|
|
_mm_shuffle_epi8( v, m128_const_64( 0x08090a0b0c0d0e0f, \
|
|
0x0001020304050607 ) )
|
|
|
|
#define mm128_bswap_32( v ) \
|
|
_mm_shuffle_epi8( v, m128_const_64( 0x0c0d0e0f08090a0b, \
|
|
0x0405060700010203 ) )
|
|
|
|
#define mm128_bswap_16( v ) \
|
|
_mm_shuffle_epi8( v, m128_const_64( 0x0e0f0c0d0a0b0809, \
|
|
0x0607040502030001 )
|
|
|
|
// 8 byte qword * 8 qwords * 2 lanes = 128 bytes
|
|
#define mm128_block_bswap_64( d, s ) do \
|
|
{ \
|
|
__m128i ctl = m128_const_64( 0x08090a0b0c0d0e0f, 0x0001020304050607 ); \
|
|
casti_m128i( d, 0 ) = _mm_shuffle_epi8( casti_m128i( s, 0 ), ctl ); \
|
|
casti_m128i( d, 1 ) = _mm_shuffle_epi8( casti_m128i( s, 1 ), ctl ); \
|
|
casti_m128i( d, 2 ) = _mm_shuffle_epi8( casti_m128i( s, 2 ), ctl ); \
|
|
casti_m128i( d, 3 ) = _mm_shuffle_epi8( casti_m128i( s, 3 ), ctl ); \
|
|
casti_m128i( d, 4 ) = _mm_shuffle_epi8( casti_m128i( s, 4 ), ctl ); \
|
|
casti_m128i( d, 5 ) = _mm_shuffle_epi8( casti_m128i( s, 5 ), ctl ); \
|
|
casti_m128i( d, 6 ) = _mm_shuffle_epi8( casti_m128i( s, 6 ), ctl ); \
|
|
casti_m128i( d, 7 ) = _mm_shuffle_epi8( casti_m128i( s, 7 ), ctl ); \
|
|
} while(0)
|
|
|
|
// 4 byte dword * 8 dwords * 4 lanes = 128 bytes
|
|
#define mm128_block_bswap_32( d, s ) do \
|
|
{ \
|
|
__m128i ctl = m128_const_64( 0x0c0d0e0f08090a0b, 0x0405060700010203 ); \
|
|
casti_m128i( d, 0 ) = _mm_shuffle_epi8( casti_m128i( s, 0 ), ctl ); \
|
|
casti_m128i( d, 1 ) = _mm_shuffle_epi8( casti_m128i( s, 1 ), ctl ); \
|
|
casti_m128i( d, 2 ) = _mm_shuffle_epi8( casti_m128i( s, 2 ), ctl ); \
|
|
casti_m128i( d, 3 ) = _mm_shuffle_epi8( casti_m128i( s, 3 ), ctl ); \
|
|
casti_m128i( d, 4 ) = _mm_shuffle_epi8( casti_m128i( s, 4 ), ctl ); \
|
|
casti_m128i( d, 5 ) = _mm_shuffle_epi8( casti_m128i( s, 5 ), ctl ); \
|
|
casti_m128i( d, 6 ) = _mm_shuffle_epi8( casti_m128i( s, 6 ), ctl ); \
|
|
casti_m128i( d, 7 ) = _mm_shuffle_epi8( casti_m128i( s, 7 ), ctl ); \
|
|
} while(0)
|
|
|
|
#else // SSE2
|
|
|
|
static inline __m128i mm128_bswap_64( __m128i v )
|
|
{
|
|
v = _mm_or_si128( _mm_slli_epi16( v, 8 ), _mm_srli_epi16( v, 8 ) );
|
|
v = _mm_shufflelo_epi16( v, _MM_SHUFFLE( 0, 1, 2, 3 ) );
|
|
return _mm_shufflehi_epi16( v, _MM_SHUFFLE( 0, 1, 2, 3 ) );
|
|
}
|
|
|
|
static inline __m128i mm128_bswap_32( __m128i v )
|
|
{
|
|
v = _mm_or_si128( _mm_slli_epi16( v, 8 ), _mm_srli_epi16( v, 8 ) );
|
|
v = _mm_shufflelo_epi16( v, _MM_SHUFFLE( 2, 3, 0, 1 ) );
|
|
return _mm_shufflehi_epi16( v, _MM_SHUFFLE( 2, 3, 0, 1 ) );
|
|
}
|
|
|
|
static inline __m128i mm128_bswap_16( __m128i v )
|
|
{
|
|
return _mm_or_si128( _mm_slli_epi16( v, 8 ), _mm_srli_epi16( v, 8 ) );
|
|
}
|
|
|
|
static inline void mm128_block_bswap_64( __m128i *d, const __m128i *s )
|
|
{
|
|
d[0] = mm128_bswap_64( s[0] );
|
|
d[1] = mm128_bswap_64( s[1] );
|
|
d[2] = mm128_bswap_64( s[2] );
|
|
d[3] = mm128_bswap_64( s[3] );
|
|
d[4] = mm128_bswap_64( s[4] );
|
|
d[5] = mm128_bswap_64( s[5] );
|
|
d[6] = mm128_bswap_64( s[6] );
|
|
d[7] = mm128_bswap_64( s[7] );
|
|
}
|
|
|
|
static inline void mm128_block_bswap_32( __m128i *d, const __m128i *s )
|
|
{
|
|
d[0] = mm128_bswap_32( s[0] );
|
|
d[1] = mm128_bswap_32( s[1] );
|
|
d[2] = mm128_bswap_32( s[2] );
|
|
d[3] = mm128_bswap_32( s[3] );
|
|
d[4] = mm128_bswap_32( s[4] );
|
|
d[5] = mm128_bswap_32( s[5] );
|
|
d[6] = mm128_bswap_32( s[6] );
|
|
d[7] = mm128_bswap_32( s[7] );
|
|
}
|
|
|
|
#endif // SSSE3 else SSE2
|
|
|
|
//
|
|
// Rotate in place concatenated 128 bit vectors as one 256 bit vector.
|
|
|
|
// Swap 128 bit vectors.
|
|
// This should be avoided, it's more efficient to switch references.
|
|
#define mm128_swap256_128( v1, v2 ) \
|
|
v1 = _mm_xor_si128( v1, v2 ); \
|
|
v2 = _mm_xor_si128( v1, v2 ); \
|
|
v1 = _mm_xor_si128( v1, v2 );
|
|
|
|
|
|
// Two input shuffle-rotate.
|
|
// Concatenate v1 & v2 and byte rotate as a 256 bit vector.
|
|
// Function macros with two inputs and one output, inputs are preserved.
|
|
// Returns the high 128 bits, ie updated v1.
|
|
|
|
#if defined(__SSSE3__)
|
|
|
|
#define mm128_shufl2r_64( v1, v2 ) _mm_alignr_epi8( v2, v1, 8 )
|
|
#define mm128_shufl2l_64( v1, v2 ) _mm_alignr_epi8( v1, v2, 8 )
|
|
|
|
/*
|
|
#define mm128_shufl2r_32( v1, v2 ) _mm_alignr_epi8( v2, v1, 4 )
|
|
#define mm128_shufl2l_32( v1, v2 ) _mm_alignr_epi8( v1, v2, 4 )
|
|
|
|
#define mm128_shufl2r_16( v1, v2 ) _mm_alignr_epi8( v2, v1, 2 )
|
|
#define mm128_shufl2l_16( v1, v2 ) _mm_alignr_epi8( v1, v2, 2 )
|
|
|
|
#define mm128_shufl2r_8( v1, v2 ) _mm_alignr_epi8( v2, v1, 1 )
|
|
#define mm128_shufl2l_8( v1, v2 ) _mm_alignr_epi8( v1, v2, 1 )
|
|
*/
|
|
|
|
#else
|
|
|
|
#define mm128_shufl2r_64( v1, v2 ) \
|
|
_mm_or_si128( _mm_srli_si128( v1, 8 ), \
|
|
_mm_slli_si128( v2, 8 ) )
|
|
|
|
#define mm128_shufl2l_64( v1, v2 ) \
|
|
_mm_or_si128( _mm_slli_si128( v1, 8 ), \
|
|
_mm_srli_si128( v2, 8 ) )
|
|
/*
|
|
#define mm128_shufl2r_32( v1, v2 ) \
|
|
_mm_or_si128( _mm_srli_si128( v1, 4 ), \
|
|
_mm_slli_si128( v2, 12 ) )
|
|
|
|
#define mm128_shufl2l_32( v1, v2 ) \
|
|
_mm_or_si128( _mm_slli_si128( v1, 4 ), \
|
|
_mm_srli_si128( v2, 12 ) )
|
|
|
|
#define mm128_shufl2r_16( v1, v2 ) \
|
|
_mm_or_si128( _mm_srli_si128( v1, 2 ), \
|
|
_mm_slli_si128( v2, 14 ) )
|
|
|
|
#define mm128_shufl2l_16( v1, v2 ) \
|
|
_mm_or_si128( _mm_slli_si128( v1, 2 ), \
|
|
_mm_srli_si128( v2, 14 ) )
|
|
|
|
#define mm128_shufl2r_8( v1, v2 ) \
|
|
_mm_or_si128( _mm_srli_si128( v1, 1 ), \
|
|
_mm_slli_si128( v2, 15 ) )
|
|
|
|
#define mm128_shufl2l_8( v1, v2 ) \
|
|
_mm_or_si128( _mm_slli_si128( v1, 1 ), \
|
|
_mm_srli_si128( v2, 15 ) )
|
|
*/
|
|
#endif
|
|
|
|
// Procedure macros with 2 inputs and 2 outputs, input args are overwritten.
|
|
// vrol & vror are deprecated and do not exist for larger vectors.
|
|
// Their only use is by lyra2 blake2b when AVX2 is not available and is
|
|
// grandfathered.
|
|
|
|
#if defined(__SSSE3__)
|
|
|
|
#define mm128_vror256_64( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 8 ); \
|
|
v1 = _mm_alignr_epi8( v2, v1, 8 ); \
|
|
v2 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_64( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 8 ); \
|
|
v2 = _mm_alignr_epi8( v2, v1, 8 ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
/*
|
|
#define mm128_vror256_32( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 4 ); \
|
|
v1 = _mm_alignr_epi8( v2, v1, 4 ); \
|
|
v2 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_32( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 12 ); \
|
|
v2 = _mm_alignr_epi8( v2, v1, 12 ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vror256_16( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 2 ); \
|
|
v1 = _mm_alignr_epi8( v2, v1, 2 ); \
|
|
v2 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_16( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 14 ); \
|
|
v2 = _mm_alignr_epi8( v2, v1, 14 ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vror256_8( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 1 ); \
|
|
v1 = _mm_alignr_epi8( v2, v1, 1 ); \
|
|
v2 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_8( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_alignr_epi8( v1, v2, 15 ); \
|
|
v2 = _mm_alignr_epi8( v2, v1, 15 ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
*/
|
|
|
|
#else // SSE2
|
|
|
|
#define mm128_vror256_64( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_srli_si128( v1, 8 ), \
|
|
_mm_slli_si128( v2, 8 ) ); \
|
|
v2 = _mm_or_si128( _mm_srli_si128( v2, 8 ), \
|
|
_mm_slli_si128( v1, 8 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_64( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_slli_si128( v1, 8 ), \
|
|
_mm_srli_si128( v2, 8 ) ); \
|
|
v2 = _mm_or_si128( _mm_slli_si128( v2, 8 ), \
|
|
_mm_srli_si128( v1, 8 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
/*
|
|
#define mm128_vror256_32( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_srli_si128( v1, 4 ), \
|
|
_mm_slli_si128( v2, 12 ) ); \
|
|
v2 = _mm_or_si128( _mm_srli_si128( v2, 4 ), \
|
|
_mm_slli_si128( v1, 12 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_32( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_slli_si128( v1, 4 ), \
|
|
_mm_srli_si128( v2, 12 ) ); \
|
|
v2 = _mm_or_si128( _mm_slli_si128( v2, 4 ), \
|
|
_mm_srli_si128( v1, 12 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vror256_16( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_srli_si128( v1, 2 ), \
|
|
_mm_slli_si128( v2, 14 ) ); \
|
|
v2 = _mm_or_si128( _mm_srli_si128( v2, 2 ), \
|
|
_mm_slli_si128( v1, 14 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_16( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_slli_si128( v1, 2 ), \
|
|
_mm_srli_si128( v2, 14 ) ); \
|
|
v2 = _mm_or_si128( _mm_slli_si128( v2, 2 ), \
|
|
_mm_srli_si128( v1, 14 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vror256_8( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_srli_si128( v1, 1 ), \
|
|
_mm_slli_si128( v2, 15 ) ); \
|
|
v2 = _mm_or_si128( _mm_srli_si128( v2, 1 ), \
|
|
_mm_slli_si128( v1, 15 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
|
|
#define mm128_vrol256_8( v1, v2 ) \
|
|
do { \
|
|
__m128i t = _mm_or_si128( _mm_slli_si128( v1, 1 ), \
|
|
_mm_srli_si128( v2, 15 ) ); \
|
|
v2 = _mm_or_si128( _mm_slli_si128( v2, 1 ), \
|
|
_mm_srli_si128( v1, 15 ) ); \
|
|
v1 = t; \
|
|
} while(0)
|
|
*/
|
|
#endif // SSE4.1 else SSE2
|
|
|
|
#endif // __SSE2__
|
|
#endif // SIMD_128_H__
|