ClickHouse/libs/libvectorclass/vectorf256.h

3167 lines
113 KiB
C
Raw Normal View History

/**************************** vectorf256.h *******************************
* Author: Agner Fog
* Date created: 2012-05-30
* Last modified: 2014-10-22
* Version: 1.16
* Project: vector classes
* Description:
* Header file defining 256-bit floating point vector classes as interface
* to intrinsic functions in x86 microprocessors with AVX instruction set.
*
* Instructions:
* Use Gnu, Intel or Microsoft C++ compiler. Compile for the desired
* instruction set, which must be at least AVX.
*
* The following vector classes are defined here:
* Vec8f Vector of 8 single precision floating point numbers
* Vec8fb Vector of 8 Booleans for use with Vec8f
* Vec4d Vector of 4 double precision floating point numbers
* Vec4db Vector of 4 Booleans for use with Vec4d
*
* Each vector object is represented internally in the CPU as a 256-bit register.
* This header file defines operators and functions for these vectors.
*
* For example:
* Vec4d a(1., 2., 3., 4.), b(5., 6., 7., 8.), c;
* c = a + b; // now c contains (6., 8., 10., 12.)
*
* For detailed instructions, see VectorClass.pdf
*
* (c) Copyright 2012 - 2014 GNU General Public License http://www.gnu.org/licenses
*****************************************************************************/
// check combination of header files
#if defined (VECTORF256_H)
#if VECTORF256_H != 2
#error Two different versions of vectorf256.h included
#endif
#else
#define VECTORF256_H 2
#if INSTRSET < 7 // AVX required
#error Please compile for the AVX instruction set or higher
#endif
#include "vectorf128.h" // Define 128-bit vectors
/*****************************************************************************
*
* select functions
*
*****************************************************************************/
// Select between two __m256 sources, element by element. Used in various functions
// and operators. Corresponds to this pseudocode:
// for (int i = 0; i < 8; i++) result[i] = s[i] ? a[i] : b[i];
// Each element in s must be either 0 (false) or 0xFFFFFFFF (true).
static inline __m256 selectf (__m256 const & s, __m256 const & a, __m256 const & b) {
return _mm256_blendv_ps (b, a, s);
}
// Same, with two __m256d sources.
// and operators. Corresponds to this pseudocode:
// for (int i = 0; i < 4; i++) result[i] = s[i] ? a[i] : b[i];
// Each element in s must be either 0 (false) or 0xFFFFFFFFFFFFFFFF (true). No other
// values are allowed.
static inline __m256d selectd (__m256d const & s, __m256d const & a, __m256d const & b) {
return _mm256_blendv_pd (b, a, s);
}
/*****************************************************************************
*
* Generate compile-time constant vector
*
*****************************************************************************/
// Generate a constant vector of 8 integers stored in memory,
// load as __m256
template <int i0, int i1, int i2, int i3, int i4, int i5, int i6, int i7>
static inline __m256 constant8f() {
static const union {
int i[8];
__m256 ymm;
} u = {{i0,i1,i2,i3,i4,i5,i6,i7}};
return u.ymm;
}
/*****************************************************************************
*
* Join two 128-bit vectors
*
*****************************************************************************/
#define set_m128r(lo,hi) _mm256_insertf128_ps(_mm256_castps128_ps256(lo),(hi),1)
// _mm256_set_m128(hi,lo); // not defined in all versions of immintrin.h
/*****************************************************************************
*
* Vec8fb: Vector of 8 Booleans for use with Vec8f
*
*****************************************************************************/
class Vec8fb {
protected:
__m256 ymm; // Float vector
public:
// Default constructor:
Vec8fb() {
}
// Constructor to build from all elements:
Vec8fb(bool b0, bool b1, bool b2, bool b3, bool b4, bool b5, bool b6, bool b7) {
#if INSTRSET >= 8 // AVX2
ymm = _mm256_castsi256_ps(_mm256_setr_epi32(-(int)b0, -(int)b1, -(int)b2, -(int)b3, -(int)b4, -(int)b5, -(int)b6, -(int)b7));
#else
__m128 blo = _mm_castsi128_ps(_mm_setr_epi32(-(int)b0, -(int)b1, -(int)b2, -(int)b3));
__m128 bhi = _mm_castsi128_ps(_mm_setr_epi32(-(int)b4, -(int)b5, -(int)b6, -(int)b7));
ymm = set_m128r(blo,bhi);
#endif
}
// Constructor to build from two Vec4fb:
Vec8fb(Vec4fb const & a0, Vec4fb const & a1) {
ymm = set_m128r(a0, a1);
}
// Constructor to convert from type __m256 used in intrinsics:
Vec8fb(__m256 const & x) {
ymm = x;
}
// Assignment operator to convert from type __m256 used in intrinsics:
Vec8fb & operator = (__m256 const & x) {
ymm = x;
return *this;
}
// Constructor to broadcast the same value into all elements:
Vec8fb(bool b) {
#if INSTRSET >= 8 // AVX2
ymm = _mm256_castsi256_ps(_mm256_set1_epi32(-(int)b));
#else
__m128 b1 = _mm_castsi128_ps(_mm_set1_epi32(-(int)b));
//ymm = _mm256_set_m128(b1,b1);
ymm = set_m128r(b1,b1);
#endif
}
// Assignment operator to broadcast scalar value:
Vec8fb & operator = (bool b) {
*this = Vec8fb(b);
return *this;
}
private: // Prevent constructing from int, etc.
Vec8fb(int b);
Vec8fb & operator = (int x);
public:
// Type cast operator to convert to __m256 used in intrinsics
operator __m256() const {
return ymm;
}
#if defined (VECTORI256_H)
#if VECTORI256_H >= 2 // AVX2 version
// Constructor to convert from type Vec8ib used as Boolean for integer vectors
Vec8fb(Vec8ib const & x) {
ymm = _mm256_castsi256_ps(x);
}
// Assignment operator to convert from type Vec8ib used as Boolean for integer vectors
Vec8fb & operator = (Vec8ib const & x) {
ymm = _mm256_castsi256_ps(x);
return *this;
}
#ifndef FIX_CLANG_VECTOR_ALIAS_AMBIGUITY
// Type cast operator to convert to type Vec8ib used as Boolean for integer vectors
operator Vec8ib() const {
return _mm256_castps_si256(ymm);
}
#endif
#else
// Constructor to convert from type Vec8ib used as Boolean for integer vectors
Vec8fb(Vec8ib const & x) {
ymm = set_m128r(_mm_castsi128_ps(x.get_low()), _mm_castsi128_ps(x.get_high()));
}
// Assignment operator to convert from type Vec8ib used as Boolean for integer vectors
Vec8fb & operator = (Vec8ib const & x) {
ymm = set_m128r(_mm_castsi128_ps(x.get_low()), _mm_castsi128_ps(x.get_high()));
return *this;
}
// Type cast operator to convert to type Vec8ib used as Boolean for integer vectors
operator Vec8ib() const {
return Vec8i(_mm_castps_si128(get_low()), _mm_castps_si128(get_high()));
}
#endif
#endif // VECTORI256_H
// Member function to change a single element in vector
// Note: This function is inefficient. Use load function if changing more than one element
Vec8fb const & insert(uint32_t index, bool value) {
static const int32_t maskl[16] = {0,0,0,0,0,0,0,0,-1,0,0,0,0,0,0,0};
__m256 mask = _mm256_loadu_ps((float const*)(maskl+8-(index & 7))); // mask with FFFFFFFF at index position
if (value) {
ymm = _mm256_or_ps(ymm,mask);
}
else {
ymm = _mm256_andnot_ps(mask,ymm);
}
return *this;
}
// Member function extract a single element from vector
bool extract(uint32_t index) const {
union {
float f[8];
int32_t i[8];
} u;
_mm256_storeu_ps(u.f, ymm);
return u.i[index & 7] != 0;
}
// Extract a single element. Operator [] can only read an element, not write.
bool operator [] (uint32_t index) const {
return extract(index);
}
// Member functions to split into two Vec4fb:
Vec4fb get_low() const {
return _mm256_castps256_ps128(ymm);
}
Vec4fb get_high() const {
return _mm256_extractf128_ps(ymm,1);
}
static int size () {
return 8;
}
};
/*****************************************************************************
*
* Operators for Vec8fb
*
*****************************************************************************/
// vector operator & : bitwise and
static inline Vec8fb operator & (Vec8fb const & a, Vec8fb const & b) {
return _mm256_and_ps(a, b);
}
static inline Vec8fb operator && (Vec8fb const & a, Vec8fb const & b) {
return a & b;
}
// vector operator &= : bitwise and
static inline Vec8fb & operator &= (Vec8fb & a, Vec8fb const & b) {
a = a & b;
return a;
}
// vector operator | : bitwise or
static inline Vec8fb operator | (Vec8fb const & a, Vec8fb const & b) {
return _mm256_or_ps(a, b);
}
static inline Vec8fb operator || (Vec8fb const & a, Vec8fb const & b) {
return a | b;
}
// vector operator |= : bitwise or
static inline Vec8fb & operator |= (Vec8fb & a, Vec8fb const & b) {
a = a | b;
return a;
}
// vector operator ^ : bitwise xor
static inline Vec8fb operator ^ (Vec8fb const & a, Vec8fb const & b) {
return _mm256_xor_ps(a, b);
}
// vector operator ^= : bitwise xor
static inline Vec8fb & operator ^= (Vec8fb & a, Vec8fb const & b) {
a = a ^ b;
return a;
}
// vector operator ~ : bitwise not
static inline Vec8fb operator ~ (Vec8fb const & a) {
return _mm256_xor_ps(a, constant8f<-1,-1,-1,-1,-1,-1,-1,-1>());
}
// vector operator ! : logical not
// (operator ! is less efficient than operator ~. Use only where not
// all bits in an element are the same)
static inline Vec8fb operator ! (Vec8fb const & a) {
return Vec8fb( !Vec8ib(a));
}
// Functions for Vec8fb
// andnot: a & ~ b
static inline Vec8fb andnot(Vec8fb const & a, Vec8fb const & b) {
return _mm256_andnot_ps(b, a);
}
/*****************************************************************************
*
* Horizontal Boolean functions
*
*****************************************************************************/
// horizontal_and. Returns true if all bits are 1
static inline bool horizontal_and (Vec8fb const & a) {
return _mm256_testc_ps(a,constant8f<-1,-1,-1,-1,-1,-1,-1,-1>()) != 0;
}
// horizontal_or. Returns true if at least one bit is 1
static inline bool horizontal_or (Vec8fb const & a) {
return ! _mm256_testz_ps(a,a);
}
/*****************************************************************************
*
* Vec4db: Vector of 4 Booleans for use with Vec4d
*
*****************************************************************************/
class Vec4db {
protected:
__m256d ymm; // double vector
public:
// Default constructor:
Vec4db() {
}
// Constructor to build from all elements:
Vec4db(bool b0, bool b1, bool b2, bool b3) {
#if INSTRSET >= 8 // AVX2
ymm = _mm256_castsi256_pd(_mm256_setr_epi64x(-(int64_t)b0, -(int64_t)b1, -(int64_t)b2, -(int64_t)b3));
#else
__m128 blo = _mm_castsi128_ps(_mm_setr_epi32(-(int)b0, -(int)b0, -(int)b1, -(int)b1));
__m128 bhi = _mm_castsi128_ps(_mm_setr_epi32(-(int)b2, -(int)b2, -(int)b3, -(int)b3));
ymm = _mm256_castps_pd(set_m128r(bhi,blo));
#endif
}
// Constructor to build from two Vec2db:
Vec4db(Vec2db const & a0, Vec2db const & a1) {
ymm = _mm256_castps_pd(set_m128r(_mm_castpd_ps(a0),_mm_castpd_ps(a1)));
//ymm = _mm256_set_m128d(a1, a0);
}
// Constructor to convert from type __m256d used in intrinsics:
Vec4db(__m256d const & x) {
ymm = x;
}
// Assignment operator to convert from type __m256d used in intrinsics:
Vec4db & operator = (__m256d const & x) {
ymm = x;
return *this;
}
// Constructor to broadcast the same value into all elements:
Vec4db(bool b) {
#if INSTRSET >= 8 // AVX2
ymm = _mm256_castsi256_pd(_mm256_set1_epi64x(-(int64_t)b));
#else
__m128 b1 = _mm_castsi128_ps(_mm_set1_epi32(-(int)b));
ymm = _mm256_castps_pd(set_m128r(b1,b1));
#endif
}
// Assignment operator to broadcast scalar value:
Vec4db & operator = (bool b) {
ymm = _mm256_castsi256_pd(_mm256_set1_epi32(-int32_t(b)));
return *this;
}
private: // Prevent constructing from int, etc.
Vec4db(int b);
Vec4db & operator = (int x);
public:
// Type cast operator to convert to __m256d used in intrinsics
operator __m256d() const {
return ymm;
}
#ifdef VECTORI256_H
#if VECTORI256_H == 2 // 256 bit integer vectors are available, AVX2
// Constructor to convert from type Vec4qb used as Boolean for integer vectors
Vec4db(Vec4qb const & x) {
ymm = _mm256_castsi256_pd(x);
}
// Assignment operator to convert from type Vec4qb used as Boolean for integer vectors
Vec4db & operator = (Vec4qb const & x) {
ymm = _mm256_castsi256_pd(x);
return *this;
}
#ifndef FIX_CLANG_VECTOR_ALIAS_AMBIGUITY
// Type cast operator to convert to type Vec4qb used as Boolean for integer vectors
operator Vec4qb() const {
return _mm256_castpd_si256(ymm);
}
#endif
#else // 256 bit integer vectors emulated without AVX2
// Constructor to convert from type Vec4qb used as Boolean for integer vectors
Vec4db(Vec4qb const & x) {
*this = Vec4db(_mm_castsi128_pd(x.get_low()), _mm_castsi128_pd(x.get_high()));
}
// Assignment operator to convert from type Vec4qb used as Boolean for integer vectors
Vec4db & operator = (Vec4qb const & x) {
*this = Vec4db(_mm_castsi128_pd(x.get_low()), _mm_castsi128_pd(x.get_high()));
return *this;
}
// Type cast operator to convert to type Vec4qb used as Boolean for integer vectors
operator Vec4qb() const {
return Vec4q(_mm_castpd_si128(get_low()), _mm_castpd_si128(get_high()));
}
#endif
#endif // VECTORI256_H
// Member function to change a single element in vector
// Note: This function is inefficient. Use load function if changing more than one element
Vec4db const & insert(uint32_t index, bool value) {
static const int32_t maskl[16] = {0,0,0,0,0,0,0,0,-1,-1,0,0,0,0,0,0};
__m256d mask = _mm256_loadu_pd((double const*)(maskl+8-(index&3)*2)); // mask with FFFFFFFFFFFFFFFF at index position
if (value) {
ymm = _mm256_or_pd(ymm,mask);
}
else {
ymm = _mm256_andnot_pd(mask,ymm);
}
return *this;
}
// Member function extract a single element from vector
bool extract(uint32_t index) const {
union {
double f[8];
int32_t i[16];
} u;
_mm256_storeu_pd(u.f, ymm);
return u.i[(index & 3) * 2 + 1] != 0;
}
// Extract a single element. Operator [] can only read an element, not write.
bool operator [] (uint32_t index) const {
return extract(index);
}
// Member functions to split into two Vec4fb:
Vec2db get_low() const {
return _mm256_castpd256_pd128(ymm);
}
Vec2db get_high() const {
return _mm256_extractf128_pd(ymm,1);
}
static int size () {
return 4;
}
};
/*****************************************************************************
*
* Operators for Vec4db
*
*****************************************************************************/
// vector operator & : bitwise and
static inline Vec4db operator & (Vec4db const & a, Vec4db const & b) {
return _mm256_and_pd(a, b);
}
static inline Vec4db operator && (Vec4db const & a, Vec4db const & b) {
return a & b;
}
// vector operator &= : bitwise and
static inline Vec4db & operator &= (Vec4db & a, Vec4db const & b) {
a = a & b;
return a;
}
// vector operator | : bitwise or
static inline Vec4db operator | (Vec4db const & a, Vec4db const & b) {
return _mm256_or_pd(a, b);
}
static inline Vec4db operator || (Vec4db const & a, Vec4db const & b) {
return a | b;
}
// vector operator |= : bitwise or
static inline Vec4db & operator |= (Vec4db & a, Vec4db const & b) {
a = a | b;
return a;
}
// vector operator ^ : bitwise xor
static inline Vec4db operator ^ (Vec4db const & a, Vec4db const & b) {
return _mm256_xor_pd(a, b);
}
// vector operator ^= : bitwise xor
static inline Vec4db & operator ^= (Vec4db & a, Vec4db const & b) {
a = a ^ b;
return a;
}
// vector operator ~ : bitwise not
static inline Vec4db operator ~ (Vec4db const & a) {
return _mm256_xor_pd(a, _mm256_castps_pd (constant8f<-1,-1,-1,-1,-1,-1,-1,-1>()));
}
// vector operator ! : logical not
// (operator ! is less efficient than operator ~. Use only where not
// all bits in an element are the same)
static inline Vec4db operator ! (Vec4db const & a) {
return Vec4db( ! Vec4qb(a));
}
// Functions for Vec8fb
// andnot: a & ~ b
static inline Vec4db andnot(Vec4db const & a, Vec4db const & b) {
return _mm256_andnot_pd(b, a);
}
/*****************************************************************************
*
* Horizontal Boolean functions
*
*****************************************************************************/
// horizontal_and. Returns true if all bits are 1
static inline bool horizontal_and (Vec4db const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
return horizontal_and(Vec256b(_mm256_castpd_si256(a)));
#else // split into 128 bit vectors
return horizontal_and(a.get_low() & a.get_high());
#endif
}
// horizontal_or. Returns true if at least one bit is 1
static inline bool horizontal_or (Vec4db const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
return horizontal_or(Vec256b(_mm256_castpd_si256(a)));
#else // split into 128 bit vectors
return horizontal_or(a.get_low() | a.get_high());
#endif
}
/*****************************************************************************
*
* Vec8f: Vector of 8 single precision floating point values
*
*****************************************************************************/
class Vec8f {
protected:
__m256 ymm; // Float vector
public:
// Default constructor:
Vec8f() {
}
// Constructor to broadcast the same value into all elements:
Vec8f(float f) {
ymm = _mm256_set1_ps(f);
}
// Constructor to build from all elements:
Vec8f(float f0, float f1, float f2, float f3, float f4, float f5, float f6, float f7) {
ymm = _mm256_setr_ps(f0, f1, f2, f3, f4, f5, f6, f7);
}
// Constructor to build from two Vec4f:
Vec8f(Vec4f const & a0, Vec4f const & a1) {
ymm = set_m128r(a0, a1);
//ymm = _mm256_set_m128(a1, a0);
}
// Constructor to convert from type __m256 used in intrinsics:
Vec8f(__m256 const & x) {
ymm = x;
}
// Assignment operator to convert from type __m256 used in intrinsics:
Vec8f & operator = (__m256 const & x) {
ymm = x;
return *this;
}
// Type cast operator to convert to __m256 used in intrinsics
operator __m256() const {
return ymm;
}
// Member function to load from array (unaligned)
Vec8f & load(float const * p) {
ymm = _mm256_loadu_ps(p);
return *this;
}
// Member function to load from array, aligned by 32
// You may use load_a instead of load if you are certain that p points to an address
// divisible by 32.
Vec8f & load_a(float const * p) {
ymm = _mm256_load_ps(p);
return *this;
}
// Member function to store into array (unaligned)
void store(float * p) const {
_mm256_storeu_ps(p, ymm);
}
// Member function to store into array, aligned by 32
// You may use store_a instead of store if you are certain that p points to an address
// divisible by 32.
void store_a(float * p) const {
_mm256_store_ps(p, ymm);
}
// Partial load. Load n elements and set the rest to 0
Vec8f & load_partial(int n, float const * p) {
if (n > 0 && n <= 4) {
*this = Vec8f(Vec4f().load_partial(n, p), _mm_setzero_ps());
// ymm = _mm256_castps128_ps256(Vec4f().load_partial<n>(p)); (this doesn't work on MS compiler due to sloppy definition of the cast)
}
else if (n > 4 && n <= 8) {
*this = Vec8f(Vec4f().load(p), Vec4f().load_partial(n - 4, p + 4));
}
else {
ymm = _mm256_setzero_ps();
}
return *this;
}
// Partial store. Store n elements
void store_partial(int n, float * p) const {
if (n <= 4) {
get_low().store_partial(n, p);
}
else if (n <= 8) {
get_low().store(p);
get_high().store_partial(n - 4, p + 4);
}
}
// cut off vector to n elements. The last 8-n elements are set to zero
Vec8f & cutoff(int n) {
if (uint32_t(n) >= 8) return *this;
static const union {
int32_t i[16];
float f[16];
} mask = {{-1,-1,-1,-1,-1,-1,-1,-1,0,0,0,0,0,0,0,0}};
*this = Vec8fb(*this) & Vec8fb(Vec8f().load(mask.f + 8 - n));
return *this;
}
// Member function to change a single element in vector
// Note: This function is inefficient. Use load function if changing more than one element
Vec8f const & insert(uint32_t index, float value) {
__m256 v0 = _mm256_broadcast_ss(&value);
switch (index) {
case 0:
ymm = _mm256_blend_ps (ymm, v0, 1); break;
case 1:
ymm = _mm256_blend_ps (ymm, v0, 2); break;
case 2:
ymm = _mm256_blend_ps (ymm, v0, 4); break;
case 3:
ymm = _mm256_blend_ps (ymm, v0, 8); break;
case 4:
ymm = _mm256_blend_ps (ymm, v0, 0x10); break;
case 5:
ymm = _mm256_blend_ps (ymm, v0, 0x20); break;
case 6:
ymm = _mm256_blend_ps (ymm, v0, 0x40); break;
default:
ymm = _mm256_blend_ps (ymm, v0, 0x80); break;
}
return *this;
}
// Member function extract a single element from vector
float extract(uint32_t index) const {
float x[8];
store(x);
return x[index & 7];
}
// Extract a single element. Use store function if extracting more than one element.
// Operator [] can only read an element, not write.
float operator [] (uint32_t index) const {
return extract(index);
}
// Member functions to split into two Vec4f:
Vec4f get_low() const {
return _mm256_castps256_ps128(ymm);
}
Vec4f get_high() const {
return _mm256_extractf128_ps(ymm,1);
}
static int size () {
return 8;
}
};
/*****************************************************************************
*
* Operators for Vec8f
*
*****************************************************************************/
// vector operator + : add element by element
static inline Vec8f operator + (Vec8f const & a, Vec8f const & b) {
return _mm256_add_ps(a, b);
}
// vector operator + : add vector and scalar
static inline Vec8f operator + (Vec8f const & a, float b) {
return a + Vec8f(b);
}
static inline Vec8f operator + (float a, Vec8f const & b) {
return Vec8f(a) + b;
}
// vector operator += : add
static inline Vec8f & operator += (Vec8f & a, Vec8f const & b) {
a = a + b;
return a;
}
// postfix operator ++
static inline Vec8f operator ++ (Vec8f & a, int) {
Vec8f a0 = a;
a = a + 1.0f;
return a0;
}
// prefix operator ++
static inline Vec8f & operator ++ (Vec8f & a) {
a = a + 1.0f;
return a;
}
// vector operator - : subtract element by element
static inline Vec8f operator - (Vec8f const & a, Vec8f const & b) {
return _mm256_sub_ps(a, b);
}
// vector operator - : subtract vector and scalar
static inline Vec8f operator - (Vec8f const & a, float b) {
return a - Vec8f(b);
}
static inline Vec8f operator - (float a, Vec8f const & b) {
return Vec8f(a) - b;
}
// vector operator - : unary minus
// Change sign bit, even for 0, INF and NAN
static inline Vec8f operator - (Vec8f const & a) {
return _mm256_xor_ps(a, constant8f<(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000> ());
}
// vector operator -= : subtract
static inline Vec8f & operator -= (Vec8f & a, Vec8f const & b) {
a = a - b;
return a;
}
// postfix operator --
static inline Vec8f operator -- (Vec8f & a, int) {
Vec8f a0 = a;
a = a - 1.0f;
return a0;
}
// prefix operator --
static inline Vec8f & operator -- (Vec8f & a) {
a = a - 1.0f;
return a;
}
// vector operator * : multiply element by element
static inline Vec8f operator * (Vec8f const & a, Vec8f const & b) {
return _mm256_mul_ps(a, b);
}
// vector operator * : multiply vector and scalar
static inline Vec8f operator * (Vec8f const & a, float b) {
return a * Vec8f(b);
}
static inline Vec8f operator * (float a, Vec8f const & b) {
return Vec8f(a) * b;
}
// vector operator *= : multiply
static inline Vec8f & operator *= (Vec8f & a, Vec8f const & b) {
a = a * b;
return a;
}
// vector operator / : divide all elements by same integer
static inline Vec8f operator / (Vec8f const & a, Vec8f const & b) {
return _mm256_div_ps(a, b);
}
// vector operator / : divide vector and scalar
static inline Vec8f operator / (Vec8f const & a, float b) {
return a / Vec8f(b);
}
static inline Vec8f operator / (float a, Vec8f const & b) {
return Vec8f(a) / b;
}
// vector operator /= : divide
static inline Vec8f & operator /= (Vec8f & a, Vec8f const & b) {
a = a / b;
return a;
}
// vector operator == : returns true for elements for which a == b
static inline Vec8fb operator == (Vec8f const & a, Vec8f const & b) {
return _mm256_cmp_ps(a, b, 0);
}
// vector operator != : returns true for elements for which a != b
static inline Vec8fb operator != (Vec8f const & a, Vec8f const & b) {
return _mm256_cmp_ps(a, b, 4);
}
// vector operator < : returns true for elements for which a < b
static inline Vec8fb operator < (Vec8f const & a, Vec8f const & b) {
return _mm256_cmp_ps(a, b, 1);
}
// vector operator <= : returns true for elements for which a <= b
static inline Vec8fb operator <= (Vec8f const & a, Vec8f const & b) {
return _mm256_cmp_ps(a, b, 2);
}
// vector operator > : returns true for elements for which a > b
static inline Vec8fb operator > (Vec8f const & a, Vec8f const & b) {
return b < a;
}
// vector operator >= : returns true for elements for which a >= b
static inline Vec8fb operator >= (Vec8f const & a, Vec8f const & b) {
return b <= a;
}
// Bitwise logical operators
// vector operator & : bitwise and
static inline Vec8f operator & (Vec8f const & a, Vec8f const & b) {
return _mm256_and_ps(a, b);
}
// vector operator &= : bitwise and
static inline Vec8f & operator &= (Vec8f & a, Vec8f const & b) {
a = a & b;
return a;
}
// vector operator & : bitwise and of Vec8f and Vec8fb
static inline Vec8f operator & (Vec8f const & a, Vec8fb const & b) {
return _mm256_and_ps(a, b);
}
static inline Vec8f operator & (Vec8fb const & a, Vec8f const & b) {
return _mm256_and_ps(a, b);
}
// vector operator | : bitwise or
static inline Vec8f operator | (Vec8f const & a, Vec8f const & b) {
return _mm256_or_ps(a, b);
}
// vector operator |= : bitwise or
static inline Vec8f & operator |= (Vec8f & a, Vec8f const & b) {
a = a | b;
return a;
}
// vector operator ^ : bitwise xor
static inline Vec8f operator ^ (Vec8f const & a, Vec8f const & b) {
return _mm256_xor_ps(a, b);
}
// vector operator ^= : bitwise xor
static inline Vec8f & operator ^= (Vec8f & a, Vec8f const & b) {
a = a ^ b;
return a;
}
// vector operator ! : logical not. Returns Boolean vector
static inline Vec8fb operator ! (Vec8f const & a) {
return a == Vec8f(0.0f);
}
/*****************************************************************************
*
* Functions for Vec8f
*
*****************************************************************************/
// Select between two operands. Corresponds to this pseudocode:
// for (int i = 0; i < 8; i++) result[i] = s[i] ? a[i] : b[i];
// Each byte in s must be either 0 (false) or 0xFFFFFFFF (true). No other values are allowed.
static inline Vec8f select (Vec8fb const & s, Vec8f const & a, Vec8f const & b) {
return _mm256_blendv_ps (b, a, s);
}
// Conditional add: For all vector elements i: result[i] = f[i] ? (a[i] + b[i]) : a[i]
static inline Vec8f if_add (Vec8fb const & f, Vec8f const & a, Vec8f const & b) {
return a + (Vec8f(f) & b);
}
// Conditional multiply: For all vector elements i: result[i] = f[i] ? (a[i] * b[i]) : a[i]
static inline Vec8f if_mul (Vec8fb const & f, Vec8f const & a, Vec8f const & b) {
return a * select(f, b, 1.f);
}
// General arithmetic functions, etc.
// Horizontal add: Calculates the sum of all vector elements.
static inline float horizontal_add (Vec8f const & a) {
__m256 t1 = _mm256_hadd_ps(a,a);
__m256 t2 = _mm256_hadd_ps(t1,t1);
__m128 t3 = _mm256_extractf128_ps(t2,1);
__m128 t4 = _mm_add_ss(_mm256_castps256_ps128(t2),t3);
return _mm_cvtss_f32(t4);
}
// function max: a > b ? a : b
static inline Vec8f max(Vec8f const & a, Vec8f const & b) {
return _mm256_max_ps(a,b);
}
// function min: a < b ? a : b
static inline Vec8f min(Vec8f const & a, Vec8f const & b) {
return _mm256_min_ps(a,b);
}
// function abs: absolute value
// Removes sign bit, even for -0.0f, -INF and -NAN
static inline Vec8f abs(Vec8f const & a) {
__m256 mask = constant8f<0x7FFFFFFF,0x7FFFFFFF,0x7FFFFFFF,0x7FFFFFFF,0x7FFFFFFF,0x7FFFFFFF,0x7FFFFFFF,0x7FFFFFFF> ();
return _mm256_and_ps(a,mask);
}
// function sqrt: square root
static inline Vec8f sqrt(Vec8f const & a) {
return _mm256_sqrt_ps(a);
}
// function square: a * a
static inline Vec8f square(Vec8f const & a) {
return a * a;
}
// pow(Vec8f, int):
template <typename TT> static Vec8f pow(Vec8f const & a, TT n);
// Raise floating point numbers to integer power n
template <>
inline Vec8f pow<int>(Vec8f const & x0, int n) {
return pow_template_i<Vec8f>(x0, n);
}
// allow conversion from unsigned int
template <>
inline Vec8f pow<uint32_t>(Vec8f const & x0, uint32_t n) {
return pow_template_i<Vec8f>(x0, (int)n);
}
// Raise floating point numbers to integer power n, where n is a compile-time constant
template <int n>
static inline Vec8f pow_n(Vec8f const & a) {
if (n < 0) return Vec8f(1.0f) / pow_n<-n>(a);
if (n == 0) return Vec8f(1.0f);
if (n >= 256) return pow(a, n);
Vec8f x = a; // a^(2^i)
Vec8f y; // accumulator
const int lowest = n - (n & (n-1));// lowest set bit in n
if (n & 1) y = x;
if (n < 2) return y;
x = x*x; // x^2
if (n & 2) {
if (lowest == 2) y = x; else y *= x;
}
if (n < 4) return y;
x = x*x; // x^4
if (n & 4) {
if (lowest == 4) y = x; else y *= x;
}
if (n < 8) return y;
x = x*x; // x^8
if (n & 8) {
if (lowest == 8) y = x; else y *= x;
}
if (n < 16) return y;
x = x*x; // x^16
if (n & 16) {
if (lowest == 16) y = x; else y *= x;
}
if (n < 32) return y;
x = x*x; // x^32
if (n & 32) {
if (lowest == 32) y = x; else y *= x;
}
if (n < 64) return y;
x = x*x; // x^64
if (n & 64) {
if (lowest == 64) y = x; else y *= x;
}
if (n < 128) return y;
x = x*x; // x^128
if (n & 128) {
if (lowest == 128) y = x; else y *= x;
}
return y;
}
template <int n>
static inline Vec8f pow(Vec8f const & a, Const_int_t<n>) {
return pow_n<n>(a);
}
// function round: round to nearest integer (even). (result as float vector)
static inline Vec8f round(Vec8f const & a) {
return _mm256_round_ps(a, 0);
}
// function truncate: round towards zero. (result as float vector)
static inline Vec8f truncate(Vec8f const & a) {
return _mm256_round_ps(a, 3);
}
// function floor: round towards minus infinity. (result as float vector)
static inline Vec8f floor(Vec8f const & a) {
return _mm256_round_ps(a, 1);
}
// function ceil: round towards plus infinity. (result as float vector)
static inline Vec8f ceil(Vec8f const & a) {
return _mm256_round_ps(a, 2);
}
#ifdef VECTORI256_H // 256 bit integer vectors are available
#if VECTORI256_H > 1 // AVX2
// function round_to_int: round to nearest integer (even). (result as integer vector)
static inline Vec8i round_to_int(Vec8f const & a) {
// Note: assume MXCSR control register is set to rounding
return _mm256_cvtps_epi32(a);
}
// function truncate_to_int: round towards zero. (result as integer vector)
static inline Vec8i truncate_to_int(Vec8f const & a) {
return _mm256_cvttps_epi32(a);
}
// function to_float: convert integer vector to float vector
static inline Vec8f to_float(Vec8i const & a) {
return _mm256_cvtepi32_ps(a);
}
#else // no AVX2
// function round_to_int: round to nearest integer (even). (result as integer vector)
static inline Vec8i round_to_int(Vec8f const & a) {
// Note: assume MXCSR control register is set to rounding
return Vec8i(_mm_cvtps_epi32(a.get_low()), _mm_cvtps_epi32(a.get_high()));
}
// function truncate_to_int: round towards zero. (result as integer vector)
static inline Vec8i truncate_to_int(Vec8f const & a) {
return Vec8i(_mm_cvttps_epi32(a.get_low()), _mm_cvttps_epi32(a.get_high()));
}
// function to_float: convert integer vector to float vector
static inline Vec8f to_float(Vec8i const & a) {
return Vec8f(_mm_cvtepi32_ps(a.get_low()), _mm_cvtepi32_ps(a.get_high()));
}
#endif
#endif // VECTORI256_H
// Fused multiply and add functions
// Multiply and add
static inline Vec8f mul_add(Vec8f const & a, Vec8f const & b, Vec8f const & c) {
#ifdef __FMA__
return _mm256_fmadd_ps(a, b, c);
#elif defined (__FMA4__)
return _mm256_macc_ps(a, b, c);
#else
return a * b + c;
#endif
}
// Multiply and subtract
static inline Vec8f mul_sub(Vec8f const & a, Vec8f const & b, Vec8f const & c) {
#ifdef __FMA__
return _mm256_fmsub_ps(a, b, c);
#elif defined (__FMA4__)
return _mm256_msub_ps(a, b, c);
#else
return a * b - c;
#endif
}
// Multiply and inverse subtract
static inline Vec8f nmul_add(Vec8f const & a, Vec8f const & b, Vec8f const & c) {
#ifdef __FMA__
return _mm256_fnmadd_ps(a, b, c);
#elif defined (__FMA4__)
return _mm256_nmacc_ps(a, b, c);
#else
return c - a * b;
#endif
}
// Multiply and subtract with extra precision on the intermediate calculations,
// even if FMA instructions not supported, using Veltkamp-Dekker split
static inline Vec8f mul_sub_x(Vec8f const & a, Vec8f const & b, Vec8f const & c) {
#ifdef __FMA__
return _mm256_fmsub_ps(a, b, c);
#elif defined (__FMA4__)
return _mm256_msub_ps(a, b, c);
#else
// calculate a * b - c with extra precision
const int b12 = -(1 << 12); // mask to remove lower 12 bits
Vec8f upper_mask = constant8f<b12,b12,b12,b12,b12,b12,b12,b12>();
Vec8f a_high = a & upper_mask; // split into high and low parts
Vec8f b_high = b & upper_mask;
Vec8f a_low = a - a_high;
Vec8f b_low = b - b_high;
Vec8f r1 = a_high * b_high; // this product is exact
Vec8f r2 = r1 - c; // subtract c from high product
Vec8f r3 = r2 + (a_high * b_low + b_high * a_low) + a_low * b_low; // add rest of product
return r3; // + ((r2 - r1) + c);
#endif
}
// Approximate math functions
// approximate reciprocal (Faster than 1.f / a. relative accuracy better than 2^-11)
static inline Vec8f approx_recipr(Vec8f const & a) {
return _mm256_rcp_ps(a);
}
// approximate reciprocal squareroot (Faster than 1.f / sqrt(a). Relative accuracy better than 2^-11)
static inline Vec8f approx_rsqrt(Vec8f const & a) {
return _mm256_rsqrt_ps(a);
}
// Math functions using fast bit manipulation
#ifdef VECTORI256_H // 256 bit integer vectors are available, AVX2
// Extract the exponent as an integer
// exponent(a) = floor(log2(abs(a)));
// exponent(1.0f) = 0, exponent(0.0f) = -127, exponent(INF) = +128, exponent(NAN) = +128
static inline Vec8i exponent(Vec8f const & a) {
#if VECTORI256_H > 1 // AVX2
Vec8ui t1 = _mm256_castps_si256(a);// reinterpret as 32-bit integer
Vec8ui t2 = t1 << 1; // shift out sign bit
Vec8ui t3 = t2 >> 24; // shift down logical to position 0
Vec8i t4 = Vec8i(t3) - 0x7F; // subtract bias from exponent
return t4;
#else // no AVX2
return Vec8i(exponent(a.get_low()), exponent(a.get_high()));
#endif
}
#endif
// Extract the fraction part of a floating point number
// a = 2^exponent(a) * fraction(a), except for a = 0
// fraction(1.0f) = 1.0f, fraction(5.0f) = 1.25f
static inline Vec8f fraction(Vec8f const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 2 // 256 bit integer vectors are available, AVX2
Vec8ui t1 = _mm256_castps_si256(a); // reinterpret as 32-bit integer
Vec8ui t2 = (t1 & 0x007FFFFF) | 0x3F800000; // set exponent to 0 + bias
return _mm256_castsi256_ps(t2);
#else
return Vec8f(fraction(a.get_low()), fraction(a.get_high()));
#endif
}
#ifdef VECTORI256_H // 256 bit integer vectors are available, AVX2
// Fast calculation of pow(2,n) with n integer
// n = 0 gives 1.0f
// n >= 128 gives +INF
// n <= -127 gives 0.0f
// This function will never produce denormals, and never raise exceptions
static inline Vec8f exp2(Vec8i const & n) {
#if VECTORI256_H > 1 // AVX2
Vec8i t1 = max(n, -0x7F); // limit to allowed range
Vec8i t2 = min(t1, 0x80);
Vec8i t3 = t2 + 0x7F; // add bias
Vec8i t4 = t3 << 23; // put exponent into position 23
return _mm256_castsi256_ps(t4); // reinterpret as float
#else
return Vec8f(exp2(n.get_low()), exp2(n.get_high()));
#endif
}
//static inline Vec8f exp2(Vec8f const & x); // defined in vectormath_exp.h
#endif // VECTORI256_H
// Categorization functions
// Function sign_bit: gives true for elements that have the sign bit set
// even for -0.0f, -INF and -NAN
// Note that sign_bit(Vec8f(-0.0f)) gives true, while Vec8f(-0.0f) < Vec8f(0.0f) gives false
// (the underscore in the name avoids a conflict with a macro in Intel's mathimf.h)
static inline Vec8fb sign_bit(Vec8f const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec8i t1 = _mm256_castps_si256(a); // reinterpret as 32-bit integer
Vec8i t2 = t1 >> 31; // extend sign bit
return _mm256_castsi256_ps(t2); // reinterpret as 32-bit Boolean
#else
return Vec8fb(sign_bit(a.get_low()), sign_bit(a.get_high()));
#endif
}
// Function sign_combine: changes the sign of a when b has the sign bit set
// same as select(sign_bit(b), -a, a)
static inline Vec8f sign_combine(Vec8f const & a, Vec8f const & b) {
Vec8f signmask = constant8f<(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000,(int)0x80000000>(); // -0.0
return a ^ (b & signmask);
}
// Function is_finite: gives true for elements that are normal, denormal or zero,
// false for INF and NAN
// (the underscore in the name avoids a conflict with a macro in Intel's mathimf.h)
static inline Vec8fb is_finite(Vec8f const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec8i t1 = _mm256_castps_si256(a); // reinterpret as 32-bit integer
Vec8i t2 = t1 << 1; // shift out sign bit
Vec8ib t3 = Vec8i(t2 & 0xFF000000) != 0xFF000000; // exponent field is not all 1s
return t3;
#else
return Vec8fb(is_finite(a.get_low()), is_finite(a.get_high()));
#endif
}
// Function is_inf: gives true for elements that are +INF or -INF
// false for finite numbers and NAN
// (the underscore in the name avoids a conflict with a macro in Intel's mathimf.h)
static inline Vec8fb is_inf(Vec8f const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec8i t1 = _mm256_castps_si256(a); // reinterpret as 32-bit integer
Vec8i t2 = t1 << 1; // shift out sign bit
return t2 == 0xFF000000; // exponent is all 1s, fraction is 0
#else
return Vec8fb(is_inf(a.get_low()), is_inf(a.get_high()));
#endif
}
// Function is_nan: gives true for elements that are +NAN or -NAN
// false for finite numbers and +/-INF
// (the underscore in the name avoids a conflict with a macro in Intel's mathimf.h)
static inline Vec8fb is_nan(Vec8f const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec8i t1 = _mm256_castps_si256(a); // reinterpret as 32-bit integer
Vec8i t2 = t1 << 1; // shift out sign bit
Vec8i t3 = 0xFF000000; // exponent mask
Vec8i t4 = t2 & t3; // exponent
Vec8i t5 = _mm256_andnot_si256(t3,t2);// fraction
return Vec8ib(t4 == t3 && t5 != 0);// exponent = all 1s and fraction != 0
#else
return Vec8fb(is_nan(a.get_low()), is_nan(a.get_high()));
#endif
}
// Function is_subnormal: gives true for elements that are denormal (subnormal)
// false for finite numbers, zero, NAN and INF
static inline Vec8fb is_subnormal(Vec8f const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec8i t1 = _mm256_castps_si256(a); // reinterpret as 32-bit integer
Vec8i t2 = t1 << 1; // shift out sign bit
Vec8i t3 = 0xFF000000; // exponent mask
Vec8i t4 = t2 & t3; // exponent
Vec8i t5 = _mm256_andnot_si256(t3,t2);// fraction
return Vec8ib(t4 == 0 && t5 != 0); // exponent = 0 and fraction != 0
#else
return Vec8fb(is_subnormal(a.get_low()), is_subnormal(a.get_high()));
#endif
}
// Function is_zero_or_subnormal: gives true for elements that are zero or subnormal (denormal)
// false for finite numbers, NAN and INF
static inline Vec8fb is_zero_or_subnormal(Vec8f const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec8i t = _mm256_castps_si256(a); // reinterpret as 32-bit integer
t &= 0x7F800000; // isolate exponent
return t == 0; // exponent = 0
#else
return Vec8fb(is_zero_or_subnormal(a.get_low()), is_zero_or_subnormal(a.get_high()));
#endif
}
// Function infinite4f: returns a vector where all elements are +INF
static inline Vec8f infinite8f() {
return constant8f<0x7F800000,0x7F800000,0x7F800000,0x7F800000,0x7F800000,0x7F800000,0x7F800000,0x7F800000>();
}
// Function nan4f: returns a vector where all elements are +NAN (quiet)
static inline Vec8f nan8f(int n = 0x10) {
return _mm256_castsi256_ps(_mm256_set1_epi32(0x7FC00000 + n));
}
// change signs on vectors Vec8f
// Each index i0 - i7 is 1 for changing sign on the corresponding element, 0 for no change
template <int i0, int i1, int i2, int i3, int i4, int i5, int i6, int i7>
static inline Vec8f change_sign(Vec8f const & a) {
if ((i0 | i1 | i2 | i3 | i4 | i5 | i6 | i7) == 0) return a;
__m256 mask = constant8f<i0 ? 0x80000000 : 0, i1 ? 0x80000000 : 0, i2 ? 0x80000000 : 0, i3 ? 0x80000000 : 0,
i4 ? 0x80000000 : 0, i5 ? 0x80000000 : 0, i6 ? 0x80000000 : 0, i7 ? 0x80000000 : 0> ();
return _mm256_xor_ps(a, mask);
}
/*****************************************************************************
*
* Vec4d: Vector of 4 double precision floating point values
*
*****************************************************************************/
class Vec4d {
protected:
__m256d ymm; // double vector
public:
// Default constructor:
Vec4d() {
}
// Constructor to broadcast the same value into all elements:
Vec4d(double d) {
ymm = _mm256_set1_pd(d);
}
// Constructor to build from all elements:
Vec4d(double d0, double d1, double d2, double d3) {
ymm = _mm256_setr_pd(d0, d1, d2, d3);
}
// Constructor to build from two Vec2d:
Vec4d(Vec2d const & a0, Vec2d const & a1) {
ymm = _mm256_castps_pd(set_m128r(_mm_castpd_ps(a0), _mm_castpd_ps(a1)));
//ymm = _mm256_set_m128d(a1, a0);
}
// Constructor to convert from type __m256d used in intrinsics:
Vec4d(__m256d const & x) {
ymm = x;
}
// Assignment operator to convert from type __m256d used in intrinsics:
Vec4d & operator = (__m256d const & x) {
ymm = x;
return *this;
}
// Type cast operator to convert to __m256d used in intrinsics
operator __m256d() const {
return ymm;
}
// Member function to load from array (unaligned)
Vec4d & load(double const * p) {
ymm = _mm256_loadu_pd(p);
return *this;
}
// Member function to load from array, aligned by 32
// You may use load_a instead of load if you are certain that p points to an address
// divisible by 32
Vec4d & load_a(double const * p) {
ymm = _mm256_load_pd(p);
return *this;
}
// Member function to store into array (unaligned)
void store(double * p) const {
_mm256_storeu_pd(p, ymm);
}
// Member function to store into array, aligned by 32
// You may use store_a instead of store if you are certain that p points to an address
// divisible by 32
void store_a(double * p) const {
_mm256_store_pd(p, ymm);
}
// Partial load. Load n elements and set the rest to 0
Vec4d & load_partial(int n, double const * p) {
if (n > 0 && n <= 2) {
*this = Vec4d(Vec2d().load_partial(n, p), _mm_setzero_pd());
}
else if (n > 2 && n <= 4) {
*this = Vec4d(Vec2d().load(p), Vec2d().load_partial(n - 2, p + 2));
}
else {
ymm = _mm256_setzero_pd();
}
return *this;
}
// Partial store. Store n elements
void store_partial(int n, double * p) const {
if (n <= 2) {
get_low().store_partial(n, p);
}
else if (n <= 4) {
get_low().store(p);
get_high().store_partial(n - 2, p + 2);
}
}
// cut off vector to n elements. The last 4-n elements are set to zero
Vec4d & cutoff(int n) {
ymm = _mm256_castps_pd(Vec8f(_mm256_castpd_ps(ymm)).cutoff(n*2));
return *this;
}
// Member function to change a single element in vector
// Note: This function is inefficient. Use load function if changing more than one element
Vec4d const & insert(uint32_t index, double value) {
__m256d v0 = _mm256_broadcast_sd(&value);
switch (index) {
case 0:
ymm = _mm256_blend_pd (ymm, v0, 1); break;
case 1:
ymm = _mm256_blend_pd (ymm, v0, 2); break;
case 2:
ymm = _mm256_blend_pd (ymm, v0, 4); break;
default:
ymm = _mm256_blend_pd (ymm, v0, 8); break;
}
return *this;
}
// Member function extract a single element from vector
double extract(uint32_t index) const {
double x[4];
store(x);
return x[index & 3];
}
// Extract a single element. Use store function if extracting more than one element.
// Operator [] can only read an element, not write.
double operator [] (uint32_t index) const {
return extract(index);
}
// Member functions to split into two Vec2d:
Vec2d get_low() const {
return _mm256_castpd256_pd128(ymm);
}
Vec2d get_high() const {
return _mm256_extractf128_pd(ymm,1);
}
static int size () {
return 4;
}
};
/*****************************************************************************
*
* Operators for Vec4d
*
*****************************************************************************/
// vector operator + : add element by element
static inline Vec4d operator + (Vec4d const & a, Vec4d const & b) {
return _mm256_add_pd(a, b);
}
// vector operator + : add vector and scalar
static inline Vec4d operator + (Vec4d const & a, double b) {
return a + Vec4d(b);
}
static inline Vec4d operator + (double a, Vec4d const & b) {
return Vec4d(a) + b;
}
// vector operator += : add
static inline Vec4d & operator += (Vec4d & a, Vec4d const & b) {
a = a + b;
return a;
}
// postfix operator ++
static inline Vec4d operator ++ (Vec4d & a, int) {
Vec4d a0 = a;
a = a + 1.0;
return a0;
}
// prefix operator ++
static inline Vec4d & operator ++ (Vec4d & a) {
a = a + 1.0;
return a;
}
// vector operator - : subtract element by element
static inline Vec4d operator - (Vec4d const & a, Vec4d const & b) {
return _mm256_sub_pd(a, b);
}
// vector operator - : subtract vector and scalar
static inline Vec4d operator - (Vec4d const & a, double b) {
return a - Vec4d(b);
}
static inline Vec4d operator - (double a, Vec4d const & b) {
return Vec4d(a) - b;
}
// vector operator - : unary minus
// Change sign bit, even for 0, INF and NAN
static inline Vec4d operator - (Vec4d const & a) {
return _mm256_xor_pd(a, _mm256_castps_pd(constant8f<0,(int)0x80000000,0,(int)0x80000000,0,(int)0x80000000,0,(int)0x80000000> ()));
}
// vector operator -= : subtract
static inline Vec4d & operator -= (Vec4d & a, Vec4d const & b) {
a = a - b;
return a;
}
// postfix operator --
static inline Vec4d operator -- (Vec4d & a, int) {
Vec4d a0 = a;
a = a - 1.0;
return a0;
}
// prefix operator --
static inline Vec4d & operator -- (Vec4d & a) {
a = a - 1.0;
return a;
}
// vector operator * : multiply element by element
static inline Vec4d operator * (Vec4d const & a, Vec4d const & b) {
return _mm256_mul_pd(a, b);
}
// vector operator * : multiply vector and scalar
static inline Vec4d operator * (Vec4d const & a, double b) {
return a * Vec4d(b);
}
static inline Vec4d operator * (double a, Vec4d const & b) {
return Vec4d(a) * b;
}
// vector operator *= : multiply
static inline Vec4d & operator *= (Vec4d & a, Vec4d const & b) {
a = a * b;
return a;
}
// vector operator / : divide all elements by same integer
static inline Vec4d operator / (Vec4d const & a, Vec4d const & b) {
return _mm256_div_pd(a, b);
}
// vector operator / : divide vector and scalar
static inline Vec4d operator / (Vec4d const & a, double b) {
return a / Vec4d(b);
}
static inline Vec4d operator / (double a, Vec4d const & b) {
return Vec4d(a) / b;
}
// vector operator /= : divide
static inline Vec4d & operator /= (Vec4d & a, Vec4d const & b) {
a = a / b;
return a;
}
// vector operator == : returns true for elements for which a == b
static inline Vec4db operator == (Vec4d const & a, Vec4d const & b) {
return _mm256_cmp_pd(a, b, 0);
}
// vector operator != : returns true for elements for which a != b
static inline Vec4db operator != (Vec4d const & a, Vec4d const & b) {
return _mm256_cmp_pd(a, b, 4);
}
// vector operator < : returns true for elements for which a < b
static inline Vec4db operator < (Vec4d const & a, Vec4d const & b) {
return _mm256_cmp_pd(a, b, 1);
}
// vector operator <= : returns true for elements for which a <= b
static inline Vec4db operator <= (Vec4d const & a, Vec4d const & b) {
return _mm256_cmp_pd(a, b, 2);
}
// vector operator > : returns true for elements for which a > b
static inline Vec4db operator > (Vec4d const & a, Vec4d const & b) {
return b < a;
}
// vector operator >= : returns true for elements for which a >= b
static inline Vec4db operator >= (Vec4d const & a, Vec4d const & b) {
return b <= a;
}
// Bitwise logical operators
// vector operator & : bitwise and
static inline Vec4d operator & (Vec4d const & a, Vec4d const & b) {
return _mm256_and_pd(a, b);
}
// vector operator &= : bitwise and
static inline Vec4d & operator &= (Vec4d & a, Vec4d const & b) {
a = a & b;
return a;
}
// vector operator & : bitwise and of Vec4d and Vec4db
static inline Vec4d operator & (Vec4d const & a, Vec4db const & b) {
return _mm256_and_pd(a, b);
}
static inline Vec4d operator & (Vec4db const & a, Vec4d const & b) {
return _mm256_and_pd(a, b);
}
// vector operator | : bitwise or
static inline Vec4d operator | (Vec4d const & a, Vec4d const & b) {
return _mm256_or_pd(a, b);
}
// vector operator |= : bitwise or
static inline Vec4d & operator |= (Vec4d & a, Vec4d const & b) {
a = a | b;
return a;
}
// vector operator ^ : bitwise xor
static inline Vec4d operator ^ (Vec4d const & a, Vec4d const & b) {
return _mm256_xor_pd(a, b);
}
// vector operator ^= : bitwise xor
static inline Vec4d & operator ^= (Vec4d & a, Vec4d const & b) {
a = a ^ b;
return a;
}
// vector operator ! : logical not. Returns Boolean vector
static inline Vec4db operator ! (Vec4d const & a) {
return a == Vec4d(0.0);
}
/*****************************************************************************
*
* Functions for Vec4d
*
*****************************************************************************/
// Select between two operands. Corresponds to this pseudocode:
// for (int i = 0; i < 2; i++) result[i] = s[i] ? a[i] : b[i];
// Each byte in s must be either 0 (false) or 0xFFFFFFFFFFFFFFFF (true).
// No other values are allowed.
static inline Vec4d select (Vec4db const & s, Vec4d const & a, Vec4d const & b) {
return _mm256_blendv_pd(b, a, s);
}
// Conditional add: For all vector elements i: result[i] = f[i] ? (a[i] + b[i]) : a[i]
static inline Vec4d if_add (Vec4db const & f, Vec4d const & a, Vec4d const & b) {
return a + (Vec4d(f) & b);
}
// Conditional multiply: For all vector elements i: result[i] = f[i] ? (a[i] * b[i]) : a[i]
static inline Vec4d if_mul (Vec4db const & f, Vec4d const & a, Vec4d const & b) {
return a * select(f, b, 1.);
}
// General arithmetic functions, etc.
// Horizontal add: Calculates the sum of all vector elements.
static inline double horizontal_add (Vec4d const & a) {
__m256d t1 = _mm256_hadd_pd(a,a);
__m128d t2 = _mm256_extractf128_pd(t1,1);
__m128d t3 = _mm_add_sd(_mm256_castpd256_pd128(t1),t2);
return _mm_cvtsd_f64(t3);
}
// function max: a > b ? a : b
static inline Vec4d max(Vec4d const & a, Vec4d const & b) {
return _mm256_max_pd(a,b);
}
// function min: a < b ? a : b
static inline Vec4d min(Vec4d const & a, Vec4d const & b) {
return _mm256_min_pd(a,b);
}
// function abs: absolute value
// Removes sign bit, even for -0.0f, -INF and -NAN
static inline Vec4d abs(Vec4d const & a) {
__m256d mask = _mm256_castps_pd(constant8f<-1,0x7FFFFFFF,-1,0x7FFFFFFF,-1,0x7FFFFFFF,-1,0x7FFFFFFF> ());
return _mm256_and_pd(a,mask);
}
// function sqrt: square root
static inline Vec4d sqrt(Vec4d const & a) {
return _mm256_sqrt_pd(a);
}
// function square: a * a
static inline Vec4d square(Vec4d const & a) {
return a * a;
}
// pow(Vec4d, int):
template <typename TT> static Vec4d pow(Vec4d const & a, TT n);
// Raise floating point numbers to integer power n
template <>
inline Vec4d pow<int>(Vec4d const & x0, int n) {
return pow_template_i<Vec4d>(x0, n);
}
// allow conversion from unsigned int
template <>
inline Vec4d pow<uint32_t>(Vec4d const & x0, uint32_t n) {
return pow_template_i<Vec4d>(x0, (int)n);
}
// Raise floating point numbers to integer power n, where n is a compile-time constant
template <int n>
static inline Vec4d pow_n(Vec4d const & a) {
if (n < 0) return Vec4d(1.0) / pow_n<-n>(a);
if (n == 0) return Vec4d(1.0);
if (n >= 256) return pow(a, n);
Vec4d x = a; // a^(2^i)
Vec4d y; // accumulator
const int lowest = n - (n & (n-1));// lowest set bit in n
if (n & 1) y = x;
if (n < 2) return y;
x = x*x; // x^2
if (n & 2) {
if (lowest == 2) y = x; else y *= x;
}
if (n < 4) return y;
x = x*x; // x^4
if (n & 4) {
if (lowest == 4) y = x; else y *= x;
}
if (n < 8) return y;
x = x*x; // x^8
if (n & 8) {
if (lowest == 8) y = x; else y *= x;
}
if (n < 16) return y;
x = x*x; // x^16
if (n & 16) {
if (lowest == 16) y = x; else y *= x;
}
if (n < 32) return y;
x = x*x; // x^32
if (n & 32) {
if (lowest == 32) y = x; else y *= x;
}
if (n < 64) return y;
x = x*x; // x^64
if (n & 64) {
if (lowest == 64) y = x; else y *= x;
}
if (n < 128) return y;
x = x*x; // x^128
if (n & 128) {
if (lowest == 128) y = x; else y *= x;
}
return y;
}
template <int n>
static inline Vec4d pow(Vec4d const & a, Const_int_t<n>) {
return pow_n<n>(a);
}
// function round: round to nearest integer (even). (result as double vector)
static inline Vec4d round(Vec4d const & a) {
return _mm256_round_pd(a, 0);
}
// function truncate: round towards zero. (result as double vector)
static inline Vec4d truncate(Vec4d const & a) {
return _mm256_round_pd(a, 3);
}
// function floor: round towards minus infinity. (result as double vector)
static inline Vec4d floor(Vec4d const & a) {
return _mm256_round_pd(a, 1);
}
// function ceil: round towards plus infinity. (result as double vector)
static inline Vec4d ceil(Vec4d const & a) {
return _mm256_round_pd(a, 2);
}
// function round_to_int: round to nearest integer (even). (result as integer vector)
static inline Vec4i round_to_int(Vec4d const & a) {
// Note: assume MXCSR control register is set to rounding
return _mm256_cvtpd_epi32(a);
}
// function truncate_to_int: round towards zero. (result as integer vector)
static inline Vec4i truncate_to_int(Vec4d const & a) {
return _mm256_cvttpd_epi32(a);
}
#ifdef VECTORI256_H // 256 bit integer vectors are available
// function truncate_to_int64: round towards zero. (inefficient)
static inline Vec4q truncate_to_int64(Vec4d const & a) {
double aa[4];
a.store(aa);
return Vec4q(int64_t(aa[0]), int64_t(aa[1]), int64_t(aa[2]), int64_t(aa[3]));
}
// function truncate_to_int64_limited: round towards zero.
// result as 64-bit integer vector, but with limited range
static inline Vec4q truncate_to_int64_limited(Vec4d const & a) {
#if VECTORI256_H > 1
// Note: assume MXCSR control register is set to rounding
Vec2q b = _mm256_cvttpd_epi32(a); // round to 32-bit integers
__m256i c = permute4q<0,-256,1,-256>(Vec4q(b,b)); // get bits 64-127 to position 128-191
__m256i s = _mm256_srai_epi32(c, 31); // sign extension bits
return _mm256_unpacklo_epi32(c, s); // interleave with sign extensions
#else
return Vec4q(truncate_to_int64_limited(a.get_low()), truncate_to_int64_limited(a.get_high()));
#endif
}
// function round_to_int64: round to nearest or even. (inefficient)
static inline Vec4q round_to_int64(Vec4d const & a) {
return truncate_to_int64(round(a));
}
// function round_to_int64_limited: round to nearest integer (even)
// result as 64-bit integer vector, but with limited range
static inline Vec4q round_to_int64_limited(Vec4d const & a) {
#if VECTORI256_H > 1
// Note: assume MXCSR control register is set to rounding
Vec2q b = _mm256_cvtpd_epi32(a); // round to 32-bit integers
__m256i c = permute4q<0,-256,1,-256>(Vec4q(b,b)); // get bits 64-127 to position 128-191
__m256i s = _mm256_srai_epi32(c, 31); // sign extension bits
return _mm256_unpacklo_epi32(c, s); // interleave with sign extensions
#else
return Vec4q(round_to_int64_limited(a.get_low()), round_to_int64_limited(a.get_high()));
#endif
}
// function to_double: convert integer vector elements to double vector (inefficient)
static inline Vec4d to_double(Vec4q const & a) {
int64_t aa[4];
a.store(aa);
return Vec4d(double(aa[0]), double(aa[1]), double(aa[2]), double(aa[3]));
}
// function to_double_limited: convert integer vector elements to double vector
// limited to abs(x) < 2^31
static inline Vec4d to_double_limited(Vec4q const & x) {
Vec8i compressed = permute8i<0,2,4,6,-256,-256,-256,-256>(Vec8i(x));
return _mm256_cvtepi32_pd(compressed.get_low()); // AVX
}
#endif // VECTORI256_H
// function to_double: convert integer vector to double vector
static inline Vec4d to_double(Vec4i const & a) {
return _mm256_cvtepi32_pd(a);
}
// function compress: convert two Vec4d to one Vec8f
static inline Vec8f compress (Vec4d const & low, Vec4d const & high) {
__m128 t1 = _mm256_cvtpd_ps(low);
__m128 t2 = _mm256_cvtpd_ps(high);
return Vec8f(t1, t2);
}
// Function extend_low : convert Vec8f vector elements 0 - 3 to Vec4d
static inline Vec4d extend_low(Vec8f const & a) {
return _mm256_cvtps_pd(_mm256_castps256_ps128(a));
}
// Function extend_high : convert Vec8f vector elements 4 - 7 to Vec4d
static inline Vec4d extend_high (Vec8f const & a) {
return _mm256_cvtps_pd(_mm256_extractf128_ps(a,1));
}
// Fused multiply and add functions
// Multiply and add
static inline Vec4d mul_add(Vec4d const & a, Vec4d const & b, Vec4d const & c) {
#ifdef __FMA__
return _mm256_fmadd_pd(a, b, c);
#elif defined (__FMA4__)
return _mm256_macc_pd(a, b, c);
#else
return a * b + c;
#endif
}
// Multiply and subtract
static inline Vec4d mul_sub(Vec4d const & a, Vec4d const & b, Vec4d const & c) {
#ifdef __FMA__
return _mm256_fmsub_pd(a, b, c);
#elif defined (__FMA4__)
return _mm256_msub_pd(a, b, c);
#else
return a * b - c;
#endif
}
// Multiply and inverse subtract
static inline Vec4d nmul_add(Vec4d const & a, Vec4d const & b, Vec4d const & c) {
#ifdef __FMA__
return _mm256_fnmadd_pd(a, b, c);
#elif defined (__FMA4__)
return _mm256_nmacc_pd(a, b, c);
#else
return c - a * b;
#endif
}
// Multiply and subtract with extra precision on the intermediate calculations,
// even if FMA instructions not supported, using Veltkamp-Dekker split
static inline Vec4d mul_sub_x(Vec4d const & a, Vec4d const & b, Vec4d const & c) {
#ifdef __FMA__
return _mm256_fmsub_pd(a, b, c);
#elif defined (__FMA4__)
return _mm256_msub_pd(a, b, c);
#else
// calculate a * b - c with extra precision
// mask to remove lower 27 bits
Vec4d upper_mask = _mm256_castps_pd(constant8f<(int)0xF8000000,-1,(int)0xF8000000,-1,(int)0xF8000000,-1,(int)0xF8000000,-1>());
Vec4d a_high = a & upper_mask; // split into high and low parts
Vec4d b_high = b & upper_mask;
Vec4d a_low = a - a_high;
Vec4d b_low = b - b_high;
Vec4d r1 = a_high * b_high; // this product is exact
Vec4d r2 = r1 - c; // subtract c from high product
Vec4d r3 = r2 + (a_high * b_low + b_high * a_low) + a_low * b_low; // add rest of product
return r3; // + ((r2 - r1) + c);
#endif
}
// Math functions using fast bit manipulation
#ifdef VECTORI256_H // 256 bit integer vectors are available
// Extract the exponent as an integer
// exponent(a) = floor(log2(abs(a)));
// exponent(1.0) = 0, exponent(0.0) = -1023, exponent(INF) = +1024, exponent(NAN) = +1024
static inline Vec4q exponent(Vec4d const & a) {
#if VECTORI256_H > 1 // AVX2
Vec4uq t1 = _mm256_castpd_si256(a);// reinterpret as 64-bit integer
Vec4uq t2 = t1 << 1; // shift out sign bit
Vec4uq t3 = t2 >> 53; // shift down logical to position 0
Vec4q t4 = Vec4q(t3) - 0x3FF; // subtract bias from exponent
return t4;
#else
return Vec4q(exponent(a.get_low()), exponent(a.get_high()));
#endif
}
// Extract the fraction part of a floating point number
// a = 2^exponent(a) * fraction(a), except for a = 0
// fraction(1.0) = 1.0, fraction(5.0) = 1.25
static inline Vec4d fraction(Vec4d const & a) {
#if VECTORI256_H > 1 // AVX2
Vec4uq t1 = _mm256_castpd_si256(a); // reinterpret as 64-bit integer
Vec4uq t2 = Vec4uq((t1 & 0x000FFFFFFFFFFFFF) | 0x3FF0000000000000); // set exponent to 0 + bias
return _mm256_castsi256_pd(t2);
#else
return Vec4d(fraction(a.get_low()), fraction(a.get_high()));
#endif
}
// Fast calculation of pow(2,n) with n integer
// n = 0 gives 1.0
// n >= 1024 gives +INF
// n <= -1023 gives 0.0
// This function will never produce denormals, and never raise exceptions
static inline Vec4d exp2(Vec4q const & n) {
#if VECTORI256_H > 1 // AVX2
Vec4q t1 = max(n, -0x3FF); // limit to allowed range
Vec4q t2 = min(t1, 0x400);
Vec4q t3 = t2 + 0x3FF; // add bias
Vec4q t4 = t3 << 52; // put exponent into position 52
return _mm256_castsi256_pd(t4); // reinterpret as double
#else
return Vec4d(exp2(n.get_low()), exp2(n.get_high()));
#endif
}
//static inline Vec4d exp2(Vec4d const & x); // defined in vectormath_exp.h
#endif
// Categorization functions
// Function sign_bit: gives true for elements that have the sign bit set
// even for -0.0, -INF and -NAN
// Note that sign_bit(Vec4d(-0.0)) gives true, while Vec4d(-0.0) < Vec4d(0.0) gives false
static inline Vec4db sign_bit(Vec4d const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec4q t1 = _mm256_castpd_si256(a); // reinterpret as 64-bit integer
Vec4q t2 = t1 >> 63; // extend sign bit
return _mm256_castsi256_pd(t2); // reinterpret as 64-bit Boolean
#else
return Vec4db(sign_bit(a.get_low()),sign_bit(a.get_high()));
#endif
}
// Function sign_combine: changes the sign of a when b has the sign bit set
// same as select(sign_bit(b), -a, a)
static inline Vec4d sign_combine(Vec4d const & a, Vec4d const & b) {
Vec4d signmask = _mm256_castps_pd(constant8f<0,(int)0x80000000,0,(int)0x80000000,0,(int)0x80000000,0,(int)0x80000000>()); // -0.0
return a ^ (b & signmask);
}
// Function is_finite: gives true for elements that are normal, denormal or zero,
// false for INF and NAN
static inline Vec4db is_finite(Vec4d const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec4q t1 = _mm256_castpd_si256(a); // reinterpret as 64-bit integer
Vec4q t2 = t1 << 1; // shift out sign bit
Vec4q t3 = 0xFFE0000000000000; // exponent mask
Vec4qb t4 = Vec4q(t2 & t3) != t3; // exponent field is not all 1s
return t4;
#else
return Vec4db(is_finite(a.get_low()),is_finite(a.get_high()));
#endif
}
// Function is_inf: gives true for elements that are +INF or -INF
// false for finite numbers and NAN
static inline Vec4db is_inf(Vec4d const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec4q t1 = _mm256_castpd_si256(a); // reinterpret as 64-bit integer
Vec4q t2 = t1 << 1; // shift out sign bit
return t2 == 0xFFE0000000000000; // exponent is all 1s, fraction is 0
#else
return Vec4db(is_inf(a.get_low()),is_inf(a.get_high()));
#endif
}
// Function is_nan: gives true for elements that are +NAN or -NAN
// false for finite numbers and +/-INF
static inline Vec4db is_nan(Vec4d const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec4q t1 = _mm256_castpd_si256(a); // reinterpret as 64-bit integer
Vec4q t2 = t1 << 1; // shift out sign bit
Vec4q t3 = 0xFFE0000000000000; // exponent mask
Vec4q t4 = t2 & t3; // exponent
Vec4q t5 = _mm256_andnot_si256(t3,t2);// fraction
return Vec4qb(t4 == t3 && t5 != 0);// exponent = all 1s and fraction != 0
#else
return Vec4db(is_nan(a.get_low()),is_nan(a.get_high()));
#endif
}
// Function is_subnormal: gives true for elements that are denormal (subnormal)
// false for finite numbers, zero, NAN and INF
static inline Vec4db is_subnormal(Vec4d const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec4q t1 = _mm256_castpd_si256(a); // reinterpret as 64-bit integer
Vec4q t2 = t1 << 1; // shift out sign bit
Vec4q t3 = 0xFFE0000000000000; // exponent mask
Vec4q t4 = t2 & t3; // exponent
Vec4q t5 = _mm256_andnot_si256(t3,t2);// fraction
return Vec4qb(t4 == 0 && t5 != 0); // exponent = 0 and fraction != 0
#else
return Vec4db(is_subnormal(a.get_low()),is_subnormal(a.get_high()));
#endif
}
// Function is_zero_or_subnormal: gives true for elements that are zero or subnormal (denormal)
// false for finite numbers, NAN and INF
static inline Vec4db is_zero_or_subnormal(Vec4d const & a) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
Vec4q t = _mm256_castpd_si256(a); // reinterpret as 32-bit integer
t &= 0x7FF0000000000000ll; // isolate exponent
return t == 0; // exponent = 0
#else
return Vec4db(is_zero_or_subnormal(a.get_low()),is_zero_or_subnormal(a.get_high()));
#endif
}
// Function infinite2d: returns a vector where all elements are +INF
static inline Vec4d infinite4d() {
return _mm256_castps_pd(constant8f<0,0x7FF00000,0,0x7FF00000,0,0x7FF00000,0,0x7FF00000>());
}
// Function nan4d: returns a vector where all elements are +NAN (quiet)
static inline Vec4d nan4d(int n = 0x10) {
#if defined (VECTORI256_H) && VECTORI256_H > 1 // 256 bit integer vectors are available, AVX2
return _mm256_castsi256_pd(Vec4q(0x7FF8000000000000 + n));
#else
return Vec4d(nan2d(n),nan2d(n));
#endif
}
// change signs on vectors Vec4d
// Each index i0 - i3 is 1 for changing sign on the corresponding element, 0 for no change
template <int i0, int i1, int i2, int i3>
static inline Vec4d change_sign(Vec4d const & a) {
if ((i0 | i1 | i2 | i3) == 0) return a;
__m256 mask = constant8f<0, i0 ? 0x80000000 : 0, 0, i1 ? 0x80000000 : 0, 0, i2 ? 0x80000000 : 0, 0, i3 ? 0x80000000 : 0> ();
return _mm256_xor_pd(a, _mm256_castps_pd(mask));
}
/*****************************************************************************
*
* Functions for reinterpretation between vector types
*
*****************************************************************************/
#if defined (VECTORI256_H) && VECTORI256_H >= 2
// AVX2 vectors defined
// ABI version 4 or later needed on Gcc for correct mangling of 256-bit intrinsic vectors.
// It is recommended to compile with -fabi-version=0 to get the latest abi version
#if !defined (GCC_VERSION) || (defined (__GXX_ABI_VERSION) && __GXX_ABI_VERSION >= 1004)
static inline __m256i reinterpret_i (__m256i const & x) {
return x;
}
static inline __m256i reinterpret_i (__m256 const & x) {
return _mm256_castps_si256(x);
}
static inline __m256i reinterpret_i (__m256d const & x) {
return _mm256_castpd_si256(x);
}
static inline __m256 reinterpret_f (__m256i const & x) {
return _mm256_castsi256_ps(x);
}
static inline __m256 reinterpret_f (__m256 const & x) {
return x;
}
static inline __m256 reinterpret_f (__m256d const & x) {
return _mm256_castpd_ps(x);
}
static inline __m256d reinterpret_d (__m256i const & x) {
return _mm256_castsi256_pd(x);
}
static inline __m256d reinterpret_d (__m256 const & x) {
return _mm256_castps_pd(x);
}
static inline __m256d reinterpret_d (__m256d const & x) {
return x;
}
#else // __GXX_ABI_VERSION < 1004
static inline __m256i reinterpret_i (Vec32c const & x) {
return x;
}
static inline __m256i reinterpret_i (Vec16s const & x) {
return x;
}
static inline __m256i reinterpret_i (Vec8i const & x) {
return x;
}
static inline __m256i reinterpret_i (Vec4q const & x) {
return x;
}
static inline __m256i reinterpret_i (Vec8f const & x) {
return _mm256_castps_si256(x);
}
static inline __m256i reinterpret_i (Vec4d const & x) {
return _mm256_castpd_si256(x);
}
static inline __m256 reinterpret_f (Vec32c const & x) {
return _mm256_castsi256_ps(x);
}
static inline __m256 reinterpret_f (Vec16s const & x) {
return _mm256_castsi256_ps(x);
}
static inline __m256 reinterpret_f (Vec8i const & x) {
return _mm256_castsi256_ps(x);
}
static inline __m256 reinterpret_f (Vec4q const & x) {
return _mm256_castsi256_ps(x);
}
static inline __m256 reinterpret_f (Vec8f const & x) {
return x;
}
static inline __m256 reinterpret_f (Vec4d const & x) {
return _mm256_castpd_ps(x);
}
static inline __m256d reinterpret_d (Vec32c const & x) {
return _mm256_castsi256_pd(x);
}
static inline __m256d reinterpret_d (Vec16s const & x) {
return _mm256_castsi256_pd(x);
}
static inline __m256d reinterpret_d (Vec8i const & x) {
return _mm256_castsi256_pd(x);
}
static inline __m256d reinterpret_d (Vec4q const & x) {
return _mm256_castsi256_pd(x);
}
static inline __m256d reinterpret_d (Vec8f const & x) {
return _mm256_castps_pd(x);
}
static inline __m256d reinterpret_d (Vec4d const & x) {
return x;
}
#endif // __GXX_ABI_VERSION
#else
// AVX2 emulated in vectori256e.h, AVX supported
// ABI version 4 or later needed on Gcc for correct mangling of 256-bit intrinsic vectors.
// It is recommended to compile with -fabi-version=0 to get the latest abi version
#if !defined (GCC_VERSION) || (defined (__GXX_ABI_VERSION) && __GXX_ABI_VERSION >= 1004)
static inline Vec256ie reinterpret_i (__m256 const & x) {
Vec8f xx(x);
return Vec256ie(reinterpret_i(xx.get_low()), reinterpret_i(xx.get_high()));
}
static inline Vec256ie reinterpret_i (__m256d const & x) {
Vec4d xx(x);
return Vec256ie(reinterpret_i(xx.get_low()), reinterpret_i(xx.get_high()));
}
static inline __m256 reinterpret_f (__m256 const & x) {
return x;
}
static inline __m256 reinterpret_f (__m256d const & x) {
return _mm256_castpd_ps(x);
}
static inline __m256d reinterpret_d (__m256 const & x) {
return _mm256_castps_pd(x);
}
static inline __m256d reinterpret_d (__m256d const & x) {
return x;
}
#else // __GXX_ABI_VERSION < 1004
static inline Vec256ie reinterpret_i (Vec8f const & x) {
Vec8f xx(x);
return Vec256ie(reinterpret_i(xx.get_low()), reinterpret_i(xx.get_high()));
}
static inline Vec256ie reinterpret_i (Vec4d const & x) {
Vec4d xx(x);
return Vec256ie(reinterpret_i(xx.get_low()), reinterpret_i(xx.get_high()));
}
static inline __m256 reinterpret_f (Vec8f const & x) {
return x;
}
static inline __m256 reinterpret_f (Vec4d const & x) {
return _mm256_castpd_ps(x);
}
static inline __m256d reinterpret_d (Vec8f const & x) {
return _mm256_castps_pd(x);
}
static inline __m256d reinterpret_d (Vec4d const & x) {
return x;
}
#endif // __GXX_ABI_VERSION
static inline Vec256ie reinterpret_i (Vec256ie const & x) {
return x;
}
static inline __m256 reinterpret_f (Vec256ie const & x) {
return Vec8f(Vec4f(reinterpret_f(x.get_low())), Vec4f(reinterpret_f(x.get_high())));
}
static inline __m256d reinterpret_d (Vec256ie const & x) {
return Vec4d(Vec2d(reinterpret_d(x.get_low())), Vec2d(reinterpret_d(x.get_high())));
}
#endif // VECTORI256_H
/*****************************************************************************
*
* Vector permute and blend functions
*
******************************************************************************
*
* The permute function can reorder the elements of a vector and optionally
* set some elements to zero.
*
* The indexes are inserted as template parameters in <>. These indexes must be
* constants. Each template parameter is an index to the element you want to
* select. An index of -1 will generate zero. An index of -256 means don't care.
*
* Example:
* Vec4d a(10., 11., 12., 13.); // a is (10, 11, 12, 13)
* Vec4d b;
* b = permute4d<1,0,-1,3>(a); // b is (11, 10, 0, 13)
*
*
* The blend function can mix elements from two different vectors and
* optionally set some elements to zero.
*
* The indexes are inserted as template parameters in <>. These indexes must be
* constants. Each template parameter is an index to the element you want to
* select, where indexes 0 - 3 indicate an element from the first source
* vector and indexes 4 - 7 indicate an element from the second source vector.
* A negative index will generate zero.
*
*
* Example:
* Vec4d a(10., 11., 12., 13.); // a is (10, 11, 12, 13)
* Vec4d b(20., 21., 22., 23.); // a is (20, 21, 22, 23)
* Vec4d c;
* c = blend4d<4,3,7,-1> (a,b); // c is (20, 13, 23, 0)
*
* A lot of the code here is metaprogramming aiming to find the instructions
* that best fit the template parameters and instruction set. The metacode
* will be reduced out to leave only a few vector instructions in release
* mode with optimization on.
*****************************************************************************/
// permute vector Vec4d
template <int i0, int i1, int i2, int i3>
static inline Vec4d permute4d(Vec4d const & a) {
const int ior = i0 | i1 | i2 | i3; // OR indexes
// is zeroing needed
const bool do_zero = ior < 0 && (ior & 0x80); // at least one index is negative, and not -0x100
// is shuffling needed
const bool do_shuffle = (i0>0) || (i1!=1 && i1>=0) || (i2!=2 && i2>=0) || (i3!=3 && i3>=0);
if (!do_shuffle) { // no shuffling needed
if (do_zero) { // zeroing
if ((i0 & i1 & i2 & i3) < 0) {
return _mm256_setzero_pd(); // zero everything
}
// zero some elements
__m256d const mask = _mm256_castps_pd (
constant8f< -int(i0>=0), -int(i0>=0), -int(i1>=0), -int(i1>=0), -int(i2>=0), -int(i2>=0), -int(i3>=0), -int(i3>=0) > ());
return _mm256_and_pd(a, mask); // zero with AND mask
}
else {
return a; // do nothing
}
}
#if INSTRSET >= 8 // AVX2: use VPERMPD
__m256d x = _mm256_permute4x64_pd(a, (i0&3) | (i1&3)<<2 | (i2&3)<<4 | (i3&3)<<6);
if (do_zero) { // zeroing
// zero some elements
__m256d const mask2 = _mm256_castps_pd (
constant8f< -int(i0>=0), -int(i0>=0), -int(i1>=0), -int(i1>=0), -int(i2>=0), -int(i2>=0), -int(i3>=0), -int(i3>=0) > ());
x = _mm256_and_pd(x, mask2); // zero with AND mask
}
return x;
#else // AVX
// Needed contents of low/high part of each source register in VSHUFPD
// 0: a.low, 1: a.high, 3: zero
const int s1 = (i0 < 0 ? 3 : (i0 & 2) >> 1) | (i2 < 0 ? 0x30 : (i2 & 2) << 3);
const int s2 = (i1 < 0 ? 3 : (i1 & 2) >> 1) | (i3 < 0 ? 0x30 : (i3 & 2) << 3);
// permute mask
const int sm = (i0 < 0 ? 0 : (i0 & 1)) | (i1 < 0 ? 1 : (i1 & 1)) << 1 | (i2 < 0 ? 0 : (i2 & 1)) << 2 | (i3 < 0 ? 1 : (i3 & 1)) << 3;
if (s1 == 0x01 || s1 == 0x11 || s2 == 0x01 || s2 == 0x11) {
// too expensive to use 256 bit permute, split into two 128 bit permutes
Vec2d alo = a.get_low();
Vec2d ahi = a.get_high();
Vec2d rlo = blend2d<i0, i1> (alo, ahi);
Vec2d rhi = blend2d<i2, i3> (alo, ahi);
return Vec4d(rlo, rhi);
}
// make operands for VSHUFPD
__m256d r1, r2;
switch (s1) {
case 0x00: // LL
r1 = _mm256_insertf128_pd(a,_mm256_castpd256_pd128(a),1); break;
case 0x03: // LZ
r1 = _mm256_insertf128_pd(do_zero ? _mm256_setzero_pd() : __m256d(a), _mm256_castpd256_pd128(a), 1);
break;
case 0x10: // LH
r1 = a; break;
case 0x13: // ZH
r1 = do_zero ? _mm256_and_pd(a, _mm256_castps_pd(constant8f<0,0,0,0,-1,-1,-1,-1>())) : __m256d(a); break;
case 0x30: // LZ
if (do_zero) {
__m128d t = _mm256_castpd256_pd128(a);
t = _mm_and_pd(t,t);
r1 = _mm256_castpd128_pd256(t);
}
else r1 = a;
break;
case 0x31: // HZ
r1 = _mm256_castpd128_pd256(_mm256_extractf128_pd(a,1)); break;
case 0x33: // ZZ
r1 = do_zero ? _mm256_setzero_pd() : __m256d(a); break;
}
if (s2 == s1) {
if (sm == 0x0A) return r1;
r2 = r1;
}
else {
switch (s2) {
case 0x00: // LL
r2 = _mm256_insertf128_pd(a,_mm256_castpd256_pd128(a),1); break;
case 0x03: // ZL
r2 = _mm256_insertf128_pd(do_zero ? _mm256_setzero_pd() : __m256d(a), _mm256_castpd256_pd128(a), 1);
break;
case 0x10: // LH
r2 = a; break;
case 0x13: // ZH
r2 = do_zero ? _mm256_and_pd(a,_mm256_castps_pd(constant8f<0,0,0,0,-1,-1,-1,-1>())) : __m256d(a); break;
case 0x30: // LZ
if (do_zero) {
__m128d t = _mm256_castpd256_pd128(a);
t = _mm_and_pd(t,t);
r2 = _mm256_castpd128_pd256(t);
}
else r2 = a;
break;
case 0x31: // HZ
r2 = _mm256_castpd128_pd256(_mm256_extractf128_pd(a,1)); break;
case 0x33: // ZZ
r2 = do_zero ? _mm256_setzero_pd() : __m256d(a); break;
}
}
return _mm256_shuffle_pd(r1, r2, sm);
#endif // INSTRSET >= 8
}
// blend vectors Vec4d
template <int i0, int i1, int i2, int i3>
static inline Vec4d blend4d(Vec4d const & a, Vec4d const & b) {
// Combine all the indexes into a single bitfield, with 8 bits for each
const int m1 = (i0 & 7) | (i1 & 7) << 8 | (i2 & 7) << 16 | (i3 & 7) << 24;
// Mask to zero out negative indexes
const uint32_t mz = (i0 < 0 ? 0 : 0xFF) | (i1 < 0 ? 0 : 0xFF) << 8 | (i2 < 0 ? 0 : 0xFF) << 16 | (i3 < 0 ? 0 : 0xFF) << 24;
if (mz == 0) return _mm256_setzero_pd(); // all zero
__m256d t1;
if ((((m1 & 0xFEFEFEFE) ^ 0x06020400) & mz) == 0) {
// fits VSHUFPD(a,b)
t1 = _mm256_shuffle_pd(a, b, (i0 & 1) | (i1 & 1) << 1 | (i2 & 1) << 2 | (i3 & 1) << 3);
if (mz == 0xFFFFFFFF) return t1;
return permute4d<i0 < 0 ? -1 : 0, i1 < 0 ? -1 : 1, i2 < 0 ? -1 : 2, i3 < 0 ? -1 : 3> (t1);
}
if ((((m1 & 0xFEFEFEFE) ^0x02060004) & mz) == 0) {
// fits VSHUFPD(b,a)
t1 = _mm256_shuffle_pd(b, a, (i0 & 1) | (i1 & 1) << 1 | (i2 & 1) << 2 | (i3 & 1) << 3);
if (mz == 0xFFFFFFFF) return t1;
return permute4d<i0 < 0 ? -1 : 0, i1 < 0 ? -1 : 1, i2 < 0 ? -1 : 2, i3 < 0 ? -1 : 3> (t1);
}
if ((((m1 & 0x03030303) ^ 0x03020100) & mz) == 0) {
// blend and zero, no permute
if ((m1 & 0x04040404 & mz) == 0) {
t1 = a;
}
else if (((m1 ^ 0x04040404) & 0x04040404 & mz) == 0) {
t1 = b;
}
else {
t1 = _mm256_blend_pd(a, b, (i0&4)>>2 | (i1&4)>>1 | (i2&4) | (i3&4) << 1);
}
if (mz == 0xFFFFFFFF) return t1;
return permute4d<i0 < 0 ? -1 : 0, i1 < 0 ? -1 : 1, i2 < 0 ? -1 : 2, i3 < 0 ? -1 : 3> (t1);
}
if ((m1 & 0x04040404 & mz) == 0) {
// all from a
return permute4d<i0, i1, i2, i3> (a);
}
if (((m1 ^ 0x04040404) & 0x04040404 & mz) == 0) {
// all from b
return permute4d<i0 ^ 4, i1 ^ 4, i2 ^ 4, i3 ^ 4> (b);
}
// check if we can do 128-bit blend/permute
if (((m1 ^ 0x01000100) & 0x01010101 & mz) == 0) {
const uint32_t j0 = uint32_t((i0 >= 0 ? i0 : i1 >= 0 ? i1 : -1) >> 1);
const uint32_t j1 = uint32_t((i2 >= 0 ? i2 : i3 >= 0 ? i3 : -1) >> 1);
if (((m1 ^ ((j0 & 3) * 0x00000202 | (j1 & 3) * 0x02020000)) & 0x06060606 & mz) == 0) {
t1 = _mm256_permute2f128_pd(a, b, (j0 & 0x0F) | (j1 & 0x0F) << 4);
const bool partialzero = (((i0 | i1) ^ j0) & 0x80) != 0 || (((i2 | i3) ^ j1) & 0x80) != 0;
if (partialzero) {
// zero some elements
__m256d mask = _mm256_castps_pd (constant8f <
i0 < 0 ? 0 : -1, i0 < 0 ? 0 : -1, i1 < 0 ? 0 : -1, i1 < 0 ? 0 : -1,
i2 < 0 ? 0 : -1, i2 < 0 ? 0 : -1, i3 < 0 ? 0 : -1, i3 < 0 ? 0 : -1 > ());
return _mm256_and_pd(t1, mask);
}
else return t1;
}
}
// general case. combine two permutes
Vec4d a1 = permute4d <
(uint32_t)i0 < 4 ? i0 : -0x100,
(uint32_t)i1 < 4 ? i1 : -0x100,
(uint32_t)i2 < 4 ? i2 : -0x100,
(uint32_t)i3 < 4 ? i3 : -0x100 > (a);
Vec4d b1 = permute4d <
(uint32_t)(i0^4) < 4 ? (i0^4) : -0x100,
(uint32_t)(i1^4) < 4 ? (i1^4) : -0x100,
(uint32_t)(i2^4) < 4 ? (i2^4) : -0x100,
(uint32_t)(i3^4) < 4 ? (i3^4) : -0x100 > (b);
t1 = _mm256_blend_pd(a1, b1, (i0&4)>>2 | (i1&4)>>1 | (i2&4) | (i3&4) << 1);
if (mz == 0xFFFFFFFF) return t1;
return permute4d<i0 < 0 ? -1 : 0, i1 < 0 ? -1 : 1, i2 < 0 ? -1 : 2, i3 < 0 ? -1 : 3> (t1);
}
/*****************************************************************************
*
* Vector Vec8f permute and blend functions
*
*****************************************************************************/
// permute vector Vec8f
template <int i0, int i1, int i2, int i3, int i4, int i5, int i6, int i7>
static inline Vec8f permute8f(Vec8f const & a) {
__m256 t1, mask;
const int ior = i0 | i1 | i2 | i3 | i4 | i5 | i6 | i7; // OR indexes
// is zeroing needed
const bool do_zero = ior < 0 && (ior & 0x80); // at least one index is negative, and not -0x100
// is shuffling needed
const bool do_shuffle = (i0>0) || (i1!=1 && i1>=0) || (i2!=2 && i2>=0) || (i3!=3 && i3>=0) ||
(i4!=4 && i4>=0) || (i5!=5 && i5>=0) || (i6!=6 && i6>=0) || (i7!=7 && i7>=0);
if (!do_shuffle) { // no shuffling needed
if (do_zero) { // zeroing
if ((i0 & i1 & i2 & i3 & i4 & i5 & i6 & i7) < 0) {
return _mm256_setzero_ps(); // zero everything
}
// zero some elements
mask = constant8f< -int(i0>=0), -int(i1>=0), -int(i2>=0), -int(i3>=0), -int(i4>=0), -int(i5>=0), -int(i6>=0), -int(i7>=0) > ();
return _mm256_and_ps(a, mask); // zero with AND mask
}
else {
return a; // do nothing
}
}
#if INSTRSET >= 8 // AVX2: use VPERMPS
if (do_shuffle) { // shuffling
mask = constant8f< i0 & 7, i1 & 7, i2 & 7, i3 & 7, i4 & 7, i5 & 7, i6 & 7, i7 & 7 > ();
#if defined (_MSC_VER) && _MSC_VER < 1700 && ! defined(__INTEL_COMPILER)
// bug in MS VS 11 beta: operands in wrong order. fixed in 11.0
t1 = _mm256_permutevar8x32_ps(mask, _mm256_castps_si256(a)); // problem in immintrin.h
#elif defined (GCC_VERSION) && GCC_VERSION <= 40700 && !defined(__INTEL_COMPILER) && !defined(__clang__)
// Gcc 4.7.0 has wrong parameter type and operands in wrong order. fixed in version 4.7.1
t1 = _mm256_permutevar8x32_ps(mask, a);
#else // no bug version
t1 = _mm256_permutevar8x32_ps(a, _mm256_castps_si256(mask));
#endif
}
else {
t1 = a; // no shuffling
}
if (do_zero) { // zeroing
if ((i0 & i1 & i2 & i3 & i4 & i5 & i6 & i7) < 0) {
return _mm256_setzero_ps(); // zero everything
}
// zero some elements
mask = constant8f< -int(i0>=0), -int(i1>=0), -int(i2>=0), -int(i3>=0), -int(i4>=0), -int(i5>=0), -int(i6>=0), -int(i7>=0) > ();
t1 = _mm256_and_ps(t1, mask); // zero with AND mask
}
return t1;
#else // AVX
// Combine all the indexes into a single bitfield, with 4 bits for each
const int m1 = (i0&7) | (i1&7)<<4 | (i2&7)<<8 | (i3&7)<<12 | (i4&7)<<16 | (i5&7)<<20 | (i6&7)<<24 | (i7&7)<<28;
// Mask to zero out negative indexes
const int m2 = (i0<0?0:0xF) | (i1<0?0:0xF)<<4 | (i2<0?0:0xF)<<8 | (i3<0?0:0xF)<<12 | (i4<0?0:0xF)<<16 | (i5<0?0:0xF)<<20 | (i6<0?0:0xF)<<24 | (i7<0?0:0xF)<<28;
// Check if it is possible to use VSHUFPS. Index n must match index n+4 on bit 0-1, and even index n must match odd index n+1 on bit 2
const bool sps = ((m1 ^ (m1 >> 16)) & 0x3333 & m2 & (m2 >> 16)) == 0 && ((m1 ^ (m1 >> 4)) & 0x04040404 & m2 & m2 >> 4) == 0;
if (sps) { // can use VSHUFPS
// Index of each pair (i[n],i[n+1])
const int j0 = i0 >= 0 ? i0 : i1;
const int j1 = i2 >= 0 ? i2 : i3;
const int j2 = i4 >= 0 ? i4 : i5;
const int j3 = i6 >= 0 ? i6 : i7;
// Index of each pair (i[n],i[n+4])
const int k0 = i0 >= 0 ? i0 : i4;
const int k1 = i1 >= 0 ? i1 : i5;
const int k2 = i2 >= 0 ? i2 : i6;
const int k3 = i3 >= 0 ? i3 : i7;
// Needed contents of low/high part of each source register in VSHUFPS
// 0: a.low, 1: a.high, 3: zero or don't care
const int s1 = (j0 < 0 ? 3 : (j0 & 4) >> 2) | (j2 < 0 ? 0x30 : (j2 & 4) << 2);
const int s2 = (j1 < 0 ? 3 : (j1 & 4) >> 2) | (j3 < 0 ? 0x30 : (j3 & 4) << 2);
// calculate cost of using VSHUFPS
const int cost1 = (s1 == 0x01 || s1 == 0x11) ? 2 : (s1 == 0x00 || s1 == 0x03 || s1 == 0x31) ? 1 : 0;
const int cost2 = (s2 == s1) ? 0 : (s2 == 0x01 || s2 == 0x11) ? 2 : (s2 == 0x00 || (s2 == 0x03 && (s1 & 0xF0) != 0x00) || (s2 == 0x31 && (s1 & 0x0F) != 0x01)) ? 1 : 0;
if (cost1 + cost2 <= 3) {
// permute mask
const int sm = (k0 < 0 ? 0 : (k0 & 3)) | (k1 < 0 ? 1 : (k1 & 3)) << 2 | (k2 < 0 ? 2 : (k2 & 3)) << 4 | (k3 < 0 ? 3 : (k3 & 3)) << 6;
// make operands for VSHUFPS
__m256 r1, r2;
switch (s1) {
case 0x00: // LL
case 0x03: // ZL
r1 = _mm256_insertf128_ps(a,_mm256_castps256_ps128(a),1); break;
case 0x01: // HL
r1 = _mm256_castps128_ps256(_mm256_extractf128_ps(a,1));
r1 = _mm256_insertf128_ps(r1,_mm256_castps256_ps128(a),1); break;
case 0x10: // LH
case 0x13: // ZH
case 0x30: // LZ
case 0x33: // ZZ
r1 = a; break;
case 0x11: // HH
r1 = _mm256_castps128_ps256(_mm256_extractf128_ps(a,1));
r1 = _mm256_insertf128_ps(r1,_mm256_castps256_ps128(r1),1); break;
case 0x31: // HZ
r1 = _mm256_castps128_ps256(_mm256_extractf128_ps(a,1)); break;
}
if (s2 == s1) {
if (sm == 0xE4) return r1;
r2 = r1;
}
else {
switch (s2) {
case 0x00: // LL
r2 = _mm256_insertf128_ps(a,_mm256_castps256_ps128(a),1); break;
case 0x03: // ZL
if ((s1 & 0xF0) == 0x00) r2 = r1;
else {
r2 = _mm256_insertf128_ps(a,_mm256_castps256_ps128(a),1);
}
break;
case 0x01: // HL
r2 = _mm256_castps128_ps256(_mm256_extractf128_ps(a,1));
r2 = _mm256_insertf128_ps(r1,_mm256_castps256_ps128(a),1); break;
case 0x10: // LH
case 0x13: // ZH
case 0x30: // LZ
case 0x33: // ZZ
r2 = a; break;
case 0x11: // HH
r2 = _mm256_castps128_ps256(_mm256_extractf128_ps(a,1));
r2 = _mm256_insertf128_ps(r2,_mm256_castps256_ps128(r2),1); break;
case 0x31: // HZ
if ((s1 & 0x0F) == 0x01) r2 = r1;
else {
r2 = _mm256_castps128_ps256(_mm256_extractf128_ps(a,1));
}
break;
}
}
// now the permute instruction
t1 = _mm256_shuffle_ps(r1, r2, sm);
if (do_zero) {
// zero some elements
mask = constant8f< -int(i0>=0), -int(i1>=0), -int(i2>=0), -int(i3>=0), -int(i4>=0), -int(i5>=0), -int(i6>=0), -int(i7>=0) > ();
t1 = _mm256_and_ps(t1, mask); // zero with AND mask
}
return t1;
}
}
// not using VSHUFPS. Split into low and high part
Vec4f alo = a.get_low();
Vec4f ahi = a.get_high();
Vec4f rlo = blend4f<i0, i1, i2, i3> (alo, ahi);
Vec4f rhi = blend4f<i4, i5, i6, i7> (alo, ahi);
return Vec8f(rlo, rhi);
#endif
}
// blend vectors Vec8f
template <int i0, int i1, int i2, int i3, int i4, int i5, int i6, int i7>
static inline Vec8f blend8f(Vec8f const & a, Vec8f const & b) {
const int ior = i0 | i1 | i2 | i3 | i4 | i5 | i6 | i7; // OR indexes
// is zeroing needed
const bool do_zero = ior < 0 && (ior & 0x80); // at least one index is negative, and not -0x100
// Combine all the indexes into a single bitfield, with 4 bits for each
const int m1 = (i0&0xF) | (i1&0xF)<<4 | (i2&0xF)<<8 | (i3&0xF)<<12 | (i4&0xF)<<16 | (i5&0xF)<<20 | (i6&0xF)<<24 | (i7&0xF)<<28;
// Mask to zero out negative indexes
const int mz = (i0<0?0:0xF) | (i1<0?0:0xF)<<4 | (i2<0?0:0xF)<<8 | (i3<0?0:0xF)<<12 | (i4<0?0:0xF)<<16 | (i5<0?0:0xF)<<20 | (i6<0?0:0xF)<<24 | (i7<0?0:0xF)<<28;
__m256 t1, mask;
if (mz == 0) return _mm256_setzero_ps(); // all zero
if ((m1 & 0x88888888 & mz) == 0) {
// all from a
return permute8f<i0, i1, i2, i3, i4, i5, i6, i7> (a);
}
if (((m1 ^ 0x88888888) & 0x88888888 & mz) == 0) {
// all from b
return permute8f<i0&~8, i1&~8, i2&~8, i3&~8, i4&~8, i5&~8, i6&~8, i7&~8> (b);
}
if ((((m1 & 0x77777777) ^ 0x76543210) & mz) == 0) {
// blend and zero, no permute
mask = constant8f<(i0&8)?0:-1, (i1&8)?0:-1, (i2&8)?0:-1, (i3&8)?0:-1, (i4&8)?0:-1, (i5&8)?0:-1, (i6&8)?0:-1, (i7&8)?0:-1> ();
t1 = select(mask, a, b);
if (!do_zero) return t1;
// zero some elements
mask = constant8f< (i0<0&&(i0&8)) ? 0 : -1, (i1<0&&(i1&8)) ? 0 : -1, (i2<0&&(i2&8)) ? 0 : -1, (i3<0&&(i3&8)) ? 0 : -1,
(i4<0&&(i4&8)) ? 0 : -1, (i5<0&&(i5&8)) ? 0 : -1, (i6<0&&(i6&8)) ? 0 : -1, (i7<0&&(i7&8)) ? 0 : -1 > ();
return _mm256_and_ps(t1, mask);
}
// check if we can do 128-bit blend/permute
if (((m1 ^ 0x32103210) & 0x33333333 & mz) == 0) {
const uint32_t j0 = (i0 >= 0 ? i0 : i1 >= 0 ? i1 : i2 >= 0 ? i2 : i3 >= 0 ? i3 : -1) >> 2;
const uint32_t j1 = (i4 >= 0 ? i4 : i5 >= 0 ? i5 : i6 >= 0 ? i6 : i7 >= 0 ? i7 : -1) >> 2;
if (((m1 ^ ((j0 & 3) * 0x00004444 | (j1 & 3) * 0x44440000)) & 0xCCCCCCCC & mz) == 0) {
t1 = _mm256_permute2f128_ps(a, b, (j0 & 0x0F) | (j1 & 0x0F) << 4);
const bool partialzero = (((i0 | i1 | i2 | i3) ^ j0) & 0x80) != 0 || (((i4 | i5 | i6 | i7) ^ j1) & 0x80) != 0;
if (partialzero) {
// zero some elements
mask = constant8f< i0 < 0 ? 0 : -1, i1 < 0 ? 0 : -1, i2 < 0 ? 0 : -1, i3 < 0 ? 0 : -1,
i4 < 0 ? 0 : -1, i5 < 0 ? 0 : -1, i6 < 0 ? 0 : -1, i7 < 0 ? 0 : -1 > ();
return _mm256_and_ps(t1, mask);
}
else return t1;
}
}
// Not checking special cases for vunpckhps, vunpcklps: they are too rare
// Check if it is possible to use VSHUFPS.
// Index n must match index n+4 on bit 0-1, and even index n must match odd index n+1 on bit 2-3
const bool sps = ((m1 ^ (m1 >> 16)) & 0x3333 & mz & (mz >> 16)) == 0 && ((m1 ^ (m1 >> 4)) & 0x0C0C0C0C & mz & mz >> 4) == 0;
if (sps) { // can use VSHUFPS
// Index of each pair (i[n],i[n+1])
const int j0 = i0 >= 0 ? i0 : i1;
const int j1 = i2 >= 0 ? i2 : i3;
const int j2 = i4 >= 0 ? i4 : i5;
const int j3 = i6 >= 0 ? i6 : i7;
// Index of each pair (i[n],i[n+4])
const int k0 = i0 >= 0 ? i0 : i4;
const int k1 = i1 >= 0 ? i1 : i5;
const int k2 = i2 >= 0 ? i2 : i6;
const int k3 = i3 >= 0 ? i3 : i7;
// Needed contents of low/high part of each source register in VSHUFPS
// 0: a.low, 1: a.high, 2: b.low, 3: b.high, 4: zero or don't care
const int s1 = (j0 < 0 ? 4 : (j0 & 0xC) >> 2) | (j2 < 0 ? 0x30 : (j2 & 0xC) << 2);
const int s2 = (j1 < 0 ? 3 : (j1 & 0xC) >> 2) | (j3 < 0 ? 0x30 : (j3 & 0xC) << 2);
// permute mask
const int sm = (k0 < 0 ? 0 : (k0 & 3)) | (k1 < 0 ? 1 : (k1 & 3)) << 2 | (k2 < 0 ? 2 : (k2 & 3)) << 4 | (k3 < 0 ? 3 : (k3 & 3)) << 6;
__m256 r1, r2;
__m128 ahi = _mm256_extractf128_ps(a,1); // 1
__m128 bhi = _mm256_extractf128_ps(b,1); // 3
switch (s1) {
case 0x00: case 0x04:
r1 = _mm256_insertf128_ps(a,_mm256_castps256_ps128(a),1); break;
case 0x01: case 0x41:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),_mm256_castps256_ps128(a),1); break;
case 0x02:
r1 = _mm256_insertf128_ps(b,_mm256_castps256_ps128(a),1); break;
case 0x03:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),_mm256_castps256_ps128(a),1); break;
case 0x10: case 0x14: case 0x40: case 0x44:
r1 = a; break;
case 0x11:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),ahi,1); break;
case 0x12:
r1 = _mm256_insertf128_ps(b,ahi,1); break;
case 0x13:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),ahi,1); break;
case 0x20:
r1 = _mm256_insertf128_ps(a,_mm256_castps256_ps128(b),1); break;
case 0x21:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),_mm256_castps256_ps128(b),1); break;
case 0x22: case 0x24: case 0x42:
r1 = _mm256_insertf128_ps(b,_mm256_castps256_ps128(b),1); break;
case 0x23: case 0x43:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),_mm256_castps256_ps128(b),1); break;
case 0x30:
r1 = _mm256_insertf128_ps(a,bhi,1); break;
case 0x31:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),bhi,1); break;
case 0x32: case 0x34:
r1 = b; break;
case 0x33:
r1 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),bhi,1); break;
}
if (s2 == s1 || ((s2 & 0x04) && ((s1 ^ s2) & 0xF0) == 0) || ((s2 & 0x40) && ((s1 ^ s2) & 0x0F) == 0)) {
// can use r2 = r1
if (sm == 0xE4) return r1; // no shuffling needed
r2 = r1;
}
else {
switch (s2) {
case 0x00: case 0x04:
r2 = _mm256_insertf128_ps(a,_mm256_castps256_ps128(a),1); break;
case 0x01: case 0x41:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),_mm256_castps256_ps128(a),1); break;
case 0x02:
r2 = _mm256_insertf128_ps(b,_mm256_castps256_ps128(a),1); break;
case 0x03:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),_mm256_castps256_ps128(a),1); break;
case 0x10: case 0x14: case 0x40: case 0x44:
r2 = a; break;
case 0x11:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),ahi,1); break;
case 0x12:
r2 = _mm256_insertf128_ps(b,ahi,1); break;
case 0x13:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),ahi,1); break;
case 0x20:
r2 = _mm256_insertf128_ps(a,_mm256_castps256_ps128(b),1); break;
case 0x21:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),_mm256_castps256_ps128(b),1); break;
case 0x22: case 0x24: case 0x42:
r2 = _mm256_insertf128_ps(b,_mm256_castps256_ps128(b),1); break;
case 0x23: case 0x43:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),_mm256_castps256_ps128(b),1); break;
case 0x30:
r2 = _mm256_insertf128_ps(a,bhi,1); break;
case 0x31:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(ahi),bhi,1); break;
case 0x32: case 0x34:
r2 = b; break;
case 0x33:
r2 = _mm256_insertf128_ps(_mm256_castps128_ps256(bhi),bhi,1); break;
}
}
// now the shuffle instruction
t1 = _mm256_shuffle_ps(r1, r2, sm);
if (do_zero) {
// zero some elements
mask = constant8f< -int(i0>=0), -int(i1>=0), -int(i2>=0), -int(i3>=0), -int(i4>=0), -int(i5>=0), -int(i6>=0), -int(i7>=0) > ();
t1 = _mm256_and_ps(t1, mask); // zero with AND mask
}
return t1;
}
// Check if we can use 64-bit blend. Even numbered indexes must be even and odd numbered
// indexes must be equal to the preceding index + 1, except for negative indexes.
if (((m1 ^ 0x10101010) & 0x11111111 & mz) == 0 && ((m1 ^ m1 >> 4) & 0x0E0E0E0E & mz & mz >> 4) == 0) {
const bool partialzero = int((i0 ^ i1) | (i2 ^ i3) | (i4 ^ i5) | (i6 ^ i7)) < 0; // part of a 64-bit block is zeroed
const int blank1 = partialzero ? -0x100 : -1; // ignore or zero
const int n0 = i0 > 0 ? i0/2 : i1 > 0 ? i1/2 : blank1; // indexes for 64 bit blend
const int n1 = i2 > 0 ? i2/2 : i3 > 0 ? i3/2 : blank1;
const int n2 = i4 > 0 ? i4/2 : i5 > 0 ? i5/2 : blank1;
const int n3 = i6 > 0 ? i6/2 : i7 > 0 ? i7/2 : blank1;
t1 = _mm256_castpd_ps (blend4d<n0,n1,n2,n3> (_mm256_castps_pd(a), _mm256_castps_pd(b)));
if (blank1 == -1 || !do_zero) {
return t1;
}
// need more zeroing
mask = constant8f< -int(i0>=0), -int(i1>=0), -int(i2>=0), -int(i3>=0), -int(i4>=0), -int(i5>=0), -int(i6>=0), -int(i7>=0) > ();
return _mm256_and_ps(t1, mask); // zero with AND mask
}
// general case: permute and blend and possible zero
const int blank2 = do_zero ? -1 : -0x100; // ignore or zero
Vec8f ta = permute8f <
(uint32_t)i0 < 8 ? i0 : blank2,
(uint32_t)i1 < 8 ? i1 : blank2,
(uint32_t)i2 < 8 ? i2 : blank2,
(uint32_t)i3 < 8 ? i3 : blank2,
(uint32_t)i4 < 8 ? i4 : blank2,
(uint32_t)i5 < 8 ? i5 : blank2,
(uint32_t)i6 < 8 ? i6 : blank2,
(uint32_t)i7 < 8 ? i7 : blank2 > (a);
Vec8f tb = permute8f <
(uint32_t)(i0^8) < 8 ? (i0^8) : blank2,
(uint32_t)(i1^8) < 8 ? (i1^8) : blank2,
(uint32_t)(i2^8) < 8 ? (i2^8) : blank2,
(uint32_t)(i3^8) < 8 ? (i3^8) : blank2,
(uint32_t)(i4^8) < 8 ? (i4^8) : blank2,
(uint32_t)(i5^8) < 8 ? (i5^8) : blank2,
(uint32_t)(i6^8) < 8 ? (i6^8) : blank2,
(uint32_t)(i7^8) < 8 ? (i7^8) : blank2 > (b);
if (blank2 == -1) {
return _mm256_or_ps(ta, tb);
}
// no zeroing, need to blend
const int maskb = ((i0 >> 3) & 1) | ((i1 >> 2) & 2) | ((i2 >> 1) & 4) | (i3 & 8) |
((i4 << 1) & 0x10) | ((i5 << 2) & 0x20) | ((i6 << 3) & 0x40) | ((i7 << 4) & 0x80);
return _mm256_blend_ps(ta, tb, maskb); // blend
}
/*****************************************************************************
*
* Vector lookup functions
*
******************************************************************************
*
* These functions use vector elements as indexes into a table.
* The table is given as one or more vectors or as an array.
*
* This can be used for several purposes:
* - table lookup
* - permute or blend with variable indexes
* - blend from more than two sources
* - gather non-contiguous data
*
* An index out of range may produce any value - the actual value produced is
* implementation dependent and may be different for different instruction
* sets. An index out of range does not produce an error message or exception.
*
* Example:
* Vec4i a(2,0,0,3); // index a is ( 2, 0, 0, 3)
* Vec4f b(1.0f,1.1f,1.2f,1.3f); // table b is (1.0, 1.1, 1.2, 1.3)
* Vec4f c;
* c = lookup4 (a,b); // result c is (1.2, 1.0, 1.0, 1.3)
*
*****************************************************************************/
#ifdef VECTORI256_H // Vec8i and Vec4q must be defined
static inline Vec8f lookup8(Vec8i const & index, Vec8f const & table) {
#if INSTRSET >= 8 && VECTORI256_H > 1 // AVX2
#if defined (_MSC_VER) && _MSC_VER < 1700 && ! defined(__INTEL_COMPILER)
// bug in MS VS 11 beta: operands in wrong order. fixed in 11.0
return _mm256_permutevar8x32_ps(_mm256_castsi256_ps(index), _mm256_castps_si256(table));
#elif defined (GCC_VERSION) && GCC_VERSION <= 40700 && !defined(__INTEL_COMPILER) && !defined(__clang__)
// Gcc 4.7.0 has wrong parameter type and operands in wrong order. fixed in version 4.7.1
return _mm256_permutevar8x32_ps(_mm256_castsi256_ps(index), table);
#else
// no bug version
return _mm256_permutevar8x32_ps(table, index);
#endif
#else // AVX
// swap low and high part of table
__m256 t1 = _mm256_castps128_ps256(_mm256_extractf128_ps(table, 1));
__m256 t2 = _mm256_insertf128_ps(t1, _mm256_castps256_ps128(table), 1);
// join index parts
__m256i index2 = _mm256_insertf128_si256(_mm256_castsi128_si256(index.get_low()), index.get_high(), 1);
// permute within each 128-bit part
__m256 r0 = _mm256_permutevar_ps(table, index2);
__m256 r1 = _mm256_permutevar_ps(t2, index2);
// high index bit for blend
__m128i k1 = _mm_slli_epi32(index.get_high() ^ 4, 29);
__m128i k0 = _mm_slli_epi32(index.get_low(), 29);
__m256 kk = _mm256_insertf128_ps(_mm256_castps128_ps256(_mm_castsi128_ps(k0)), _mm_castsi128_ps(k1), 1);
// blend the two permutes
return _mm256_blendv_ps(r0, r1, kk);
#endif
}
template <int n>
static inline Vec8f lookup(Vec8i const & index, float const * table) {
if (n <= 0) return 0;
if (n <= 4) {
Vec4f table1 = Vec4f().load(table);
return Vec8f(
lookup4 (index.get_low(), table1),
lookup4 (index.get_high(), table1));
}
#if INSTRSET < 8 // not AVX2
if (n <= 8) {
return lookup8(index, Vec8f().load(table));
}
#endif
// Limit index
Vec8ui index1;
if ((n & (n-1)) == 0) {
// n is a power of 2, make index modulo n
index1 = Vec8ui(index) & (n-1);
}
else {
// n is not a power of 2, limit to n-1
index1 = min(Vec8ui(index), n-1);
}
#if INSTRSET >= 8 && VECTORI256_H > 1 // AVX2
return _mm256_i32gather_ps(table, index1, 4);
#else // AVX
return Vec8f(table[index1[0]],table[index1[1]],table[index1[2]],table[index1[3]],
table[index1[4]],table[index1[5]],table[index1[6]],table[index1[7]]);
#endif
}
static inline Vec4d lookup4(Vec4q const & index, Vec4d const & table) {
#if INSTRSET >= 8 && VECTORI256_H > 1 // AVX2
// We can't use VPERMPD because it has constant indexes.
// Convert the index to fit VPERMPS
Vec8i index1 = permute8i<0,0,2,2,4,4,6,6> (Vec8i(index+index));
Vec8i index2 = index1 + Vec8i(constant8i<0,1,0,1,0,1,0,1>());
#if defined (_MSC_VER) && _MSC_VER < 1700 && ! defined(__INTEL_COMPILER)
// bug in MS VS 11 beta: operands in wrong order. fixed in 11.0
return _mm256_castps_pd(_mm256_permutevar8x32_ps(_mm256_castsi256_ps(index2), _mm256_castpd_si256(table)));
#elif defined (GCC_VERSION) && GCC_VERSION <= 40700 && !defined(__INTEL_COMPILER) && !defined(__clang__)
// Gcc 4.7.0 has wrong parameter type and operands in wrong order
return _mm256_castps_pd(_mm256_permutevar8x32_ps(_mm256_castsi256_ps(index2), _mm256_castpd_ps(table)));
#else
// no bug version
return _mm256_castps_pd(_mm256_permutevar8x32_ps(_mm256_castpd_ps(table), index2));
#endif
#else // AVX
// swap low and high part of table
__m256d t1 = _mm256_castpd128_pd256(_mm256_extractf128_pd(table, 1));
__m256d t2 = _mm256_insertf128_pd(t1, _mm256_castpd256_pd128(table), 1);
// index << 1
__m128i index2lo = index.get_low() + index.get_low();
__m128i index2hi = index.get_high() + index.get_high();
// join index parts
__m256i index3 = _mm256_insertf128_si256(_mm256_castsi128_si256(index2lo), index2hi, 1);
// permute within each 128-bit part
__m256d r0 = _mm256_permutevar_pd(table, index3);
__m256d r1 = _mm256_permutevar_pd(t2, index3);
// high index bit for blend
__m128i k1 = _mm_slli_epi64(index.get_high() ^ 2, 62);
__m128i k0 = _mm_slli_epi64(index.get_low(), 62);
__m256d kk = _mm256_insertf128_pd(_mm256_castpd128_pd256(_mm_castsi128_pd(k0)), _mm_castsi128_pd(k1), 1);
// blend the two permutes
return _mm256_blendv_pd(r0, r1, kk);
#endif
}
template <int n>
static inline Vec4d lookup(Vec4q const & index, double const * table) {
if (n <= 0) return 0;
if (n <= 2) {
Vec2d table1 = Vec2d().load(table);
return Vec4d(
lookup2 (index.get_low(), table1),
lookup2 (index.get_high(), table1));
}
#if INSTRSET < 8 // not AVX2
if (n <= 4) {
return lookup4(index, Vec4d().load(table));
}
#endif
// Limit index
Vec8ui index1;
if ((n & (n-1)) == 0) {
// n is a power of 2, make index modulo n
index1 = Vec8ui(index) & constant8i<n-1, 0, n-1, 0, n-1, 0, n-1, 0>();
}
else {
// n is not a power of 2, limit to n-1
index1 = min(Vec8ui(index), constant8i<n-1, 0, n-1, 0, n-1, 0, n-1, 0>() );
}
#if INSTRSET >= 8 && VECTORI256_H > 1 // AVX2
return _mm256_i64gather_pd(table, index1, 8);
#else // AVX
Vec4q index2 = Vec4q(index1);
return Vec4d(table[index2[0]],table[index2[1]],table[index2[2]],table[index2[3]]);
#endif
}
#endif // VECTORI256_H
/*****************************************************************************
*
* Gather functions with fixed indexes
*
*****************************************************************************/
// Load elements from array a with indices i0, i1, i2, i3, ..
template <int i0, int i1, int i2, int i3, int i4, int i5, int i6, int i7>
static inline Vec8f gather8f(void const * a) {
return reinterpret_f(gather8i<i0, i1, i2, i3, i4, i5, i6, i7>(a));
}
// Load elements from array a with indices i0, i1, i2, i3
template <int i0, int i1, int i2, int i3>
static inline Vec4d gather4d(void const * a) {
return reinterpret_d(gather4q<i0, i1, i2, i3>(a));
}
/*****************************************************************************
*
* Horizontal scan functions
*
*****************************************************************************/
// Get index to the first element that is true. Return -1 if all are false
static inline int horizontal_find_first(Vec8fb const & x) {
return horizontal_find_first(Vec8ib(x));
}
static inline int horizontal_find_first(Vec4db const & x) {
return horizontal_find_first(Vec4qb(x));
}
// Count the number of elements that are true
static inline uint32_t horizontal_count(Vec8fb const & x) {
return horizontal_count(Vec8ib(x));
}
static inline uint32_t horizontal_count(Vec4db const & x) {
return horizontal_count(Vec4qb(x));
}
/*****************************************************************************
*
* Boolean <-> bitfield conversion functions
*
*****************************************************************************/
// to_bits: convert boolean vector to integer bitfield
static inline uint8_t to_bits(Vec8fb const & x) {
return to_bits(Vec8ib(x));
}
// to_Vec8fb: convert integer bitfield to boolean vector
static inline Vec8fb to_Vec8fb(uint8_t x) {
return Vec8fb(to_Vec8ib(x));
}
// to_bits: convert boolean vector to integer bitfield
static inline uint8_t to_bits(Vec4db const & x) {
return to_bits(Vec4qb(x));
}
// to_Vec4db: convert integer bitfield to boolean vector
static inline Vec4db to_Vec4db(uint8_t x) {
return Vec4db(to_Vec4qb(x));
}
#endif // VECTORF256_H