diff options
Diffstat (limited to 'thirdparty/embree-aarch64/common/simd/vllong8_avx512.h')
-rw-r--r-- | thirdparty/embree-aarch64/common/simd/vllong8_avx512.h | 381 |
1 files changed, 0 insertions, 381 deletions
diff --git a/thirdparty/embree-aarch64/common/simd/vllong8_avx512.h b/thirdparty/embree-aarch64/common/simd/vllong8_avx512.h deleted file mode 100644 index 76dddd8991..0000000000 --- a/thirdparty/embree-aarch64/common/simd/vllong8_avx512.h +++ /dev/null @@ -1,381 +0,0 @@ -// Copyright 2009-2020 Intel Corporation -// SPDX-License-Identifier: Apache-2.0 - -#pragma once - -namespace embree -{ - /* 8-wide AVX-512 64-bit long long type */ - template<> - struct vllong<8> - { - ALIGNED_STRUCT_(64); - - typedef vboold8 Bool; - - enum { size = 8 }; // number of SIMD elements - union { // data - __m512i v; - long long i[8]; - }; - - //////////////////////////////////////////////////////////////////////////////// - /// Constructors, Assignment & Cast Operators - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vllong() {} - __forceinline vllong(const vllong8& t) { v = t.v; } - __forceinline vllong8& operator =(const vllong8& f) { v = f.v; return *this; } - - __forceinline vllong(const __m512i& t) { v = t; } - __forceinline operator __m512i() const { return v; } - __forceinline operator __m256i() const { return _mm512_castsi512_si256(v); } - - __forceinline vllong(long long i) { - v = _mm512_set1_epi64(i); - } - - __forceinline vllong(long long a, long long b, long long c, long long d) { - v = _mm512_set4_epi64(d,c,b,a); - } - - __forceinline vllong(long long a0, long long a1, long long a2, long long a3, - long long a4, long long a5, long long a6, long long a7) - { - v = _mm512_set_epi64(a7,a6,a5,a4,a3,a2,a1,a0); - } - - __forceinline vllong(const vllong<4>& i) { - v = _mm512_broadcast_i64x4(i); - } - - //////////////////////////////////////////////////////////////////////////////// - /// Constants - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vllong(ZeroTy) : v(_mm512_setzero_epi32()) {} - __forceinline vllong(OneTy) : v(_mm512_set1_epi64(1)) {} - __forceinline vllong(StepTy) : v(_mm512_set_epi64(7,6,5,4,3,2,1,0)) {} - __forceinline vllong(ReverseStepTy) : v(_mm512_setr_epi64(7,6,5,4,3,2,1,0)) {} - - //////////////////////////////////////////////////////////////////////////////// - /// Loads and Stores - //////////////////////////////////////////////////////////////////////////////// - - static __forceinline void store_nt(void* __restrict__ ptr, const vllong8& a) { - _mm512_stream_si512((__m512i*)ptr,a); - } - - static __forceinline vllong8 loadu(const void* addr) { - return _mm512_loadu_si512(addr); - } - - static __forceinline vllong8 load(const vllong8* addr) { - return _mm512_load_si512(addr); - } - - static __forceinline vllong8 load(const long long* addr) { - return _mm512_load_si512(addr); - } - - static __forceinline vllong8 load(const uint8_t* ptr) { - return _mm512_cvtepu8_epi64(*(__m128i*)ptr); - } - - static __forceinline void store(void* ptr, const vllong8& v) { - _mm512_store_si512(ptr,v); - } - - static __forceinline void storeu(void* ptr, const vllong8& v) { - _mm512_storeu_si512(ptr,v); - } - - static __forceinline void storeu(const vboold8& mask, long long* ptr, const vllong8& f) { - _mm512_mask_storeu_epi64(ptr,mask,f); - } - - static __forceinline void store(const vboold8& mask, void* addr, const vllong8& v2) { - _mm512_mask_store_epi64(addr,mask,v2); - } - - /* pass by value to avoid compiler generating inefficient code */ - static __forceinline void storeu_compact(const vboold8 mask, void* addr, const vllong8& reg) { - _mm512_mask_compressstoreu_epi64(addr,mask,reg); - } - - static __forceinline vllong8 compact64bit(const vboold8& mask, vllong8& v) { - return _mm512_mask_compress_epi64(v,mask,v); - } - - static __forceinline vllong8 compact64bit(const vboold8& mask, vllong8& dest, const vllong8& source) { - return _mm512_mask_compress_epi64(dest,mask,source); - } - - static __forceinline vllong8 compact(const vboold8& mask, vllong8& v) { - return _mm512_mask_compress_epi64(v,mask,v); - } - - static __forceinline vllong8 compact(const vboold8& mask, const vllong8& a, vllong8& b) { - return _mm512_mask_compress_epi64(a,mask,b); - } - - static __forceinline vllong8 expand(const vboold8& mask, const vllong8& a, vllong8& b) { - return _mm512_mask_expand_epi64(b,mask,a); - } - - static __forceinline vllong8 broadcast64bit(size_t v) { - return _mm512_set1_epi64(v); - } - - static __forceinline size_t extract64bit(const vllong8& v) - { - return _mm_cvtsi128_si64(_mm512_castsi512_si128(v)); - } - - - //////////////////////////////////////////////////////////////////////////////// - /// Array Access - //////////////////////////////////////////////////////////////////////////////// - - __forceinline long long& operator [](size_t index) { assert(index < 8); return i[index]; } - __forceinline const long long& operator [](size_t index) const { assert(index < 8); return i[index]; } - - }; - - //////////////////////////////////////////////////////////////////////////////// - /// Unary Operators - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vboold8 asBool(const vllong8& a) { return _mm512_movepi64_mask(a); } - - __forceinline vllong8 operator +(const vllong8& a) { return a; } - __forceinline vllong8 operator -(const vllong8& a) { return _mm512_sub_epi64(_mm512_setzero_epi32(), a); } - - //////////////////////////////////////////////////////////////////////////////// - /// Binary Operators - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vllong8 operator +(const vllong8& a, const vllong8& b) { return _mm512_add_epi64(a, b); } - __forceinline vllong8 operator +(const vllong8& a, long long b) { return a + vllong8(b); } - __forceinline vllong8 operator +(long long a, const vllong8& b) { return vllong8(a) + b; } - - __forceinline vllong8 operator -(const vllong8& a, const vllong8& b) { return _mm512_sub_epi64(a, b); } - __forceinline vllong8 operator -(const vllong8& a, long long b) { return a - vllong8(b); } - __forceinline vllong8 operator -(long long a, const vllong8& b) { return vllong8(a) - b; } - - __forceinline vllong8 operator *(const vllong8& a, const vllong8& b) { return _mm512_mullo_epi64(a, b); } - __forceinline vllong8 operator *(const vllong8& a, long long b) { return a * vllong8(b); } - __forceinline vllong8 operator *(long long a, const vllong8& b) { return vllong8(a) * b; } - - __forceinline vllong8 operator &(const vllong8& a, const vllong8& b) { return _mm512_and_epi64(a, b); } - __forceinline vllong8 operator &(const vllong8& a, long long b) { return a & vllong8(b); } - __forceinline vllong8 operator &(long long a, const vllong8& b) { return vllong8(a) & b; } - - __forceinline vllong8 operator |(const vllong8& a, const vllong8& b) { return _mm512_or_epi64(a, b); } - __forceinline vllong8 operator |(const vllong8& a, long long b) { return a | vllong8(b); } - __forceinline vllong8 operator |(long long a, const vllong8& b) { return vllong8(a) | b; } - - __forceinline vllong8 operator ^(const vllong8& a, const vllong8& b) { return _mm512_xor_epi64(a, b); } - __forceinline vllong8 operator ^(const vllong8& a, long long b) { return a ^ vllong8(b); } - __forceinline vllong8 operator ^(long long a, const vllong8& b) { return vllong8(a) ^ b; } - - __forceinline vllong8 operator <<(const vllong8& a, long long n) { return _mm512_slli_epi64(a, n); } - __forceinline vllong8 operator >>(const vllong8& a, long long n) { return _mm512_srai_epi64(a, n); } - - __forceinline vllong8 operator <<(const vllong8& a, const vllong8& n) { return _mm512_sllv_epi64(a, n); } - __forceinline vllong8 operator >>(const vllong8& a, const vllong8& n) { return _mm512_srav_epi64(a, n); } - - __forceinline vllong8 sll (const vllong8& a, long long b) { return _mm512_slli_epi64(a, b); } - __forceinline vllong8 sra (const vllong8& a, long long b) { return _mm512_srai_epi64(a, b); } - __forceinline vllong8 srl (const vllong8& a, long long b) { return _mm512_srli_epi64(a, b); } - - __forceinline vllong8 min(const vllong8& a, const vllong8& b) { return _mm512_min_epi64(a, b); } - __forceinline vllong8 min(const vllong8& a, long long b) { return min(a,vllong8(b)); } - __forceinline vllong8 min(long long a, const vllong8& b) { return min(vllong8(a),b); } - - __forceinline vllong8 max(const vllong8& a, const vllong8& b) { return _mm512_max_epi64(a, b); } - __forceinline vllong8 max(const vllong8& a, long long b) { return max(a,vllong8(b)); } - __forceinline vllong8 max(long long a, const vllong8& b) { return max(vllong8(a),b); } - - __forceinline vllong8 mask_add(const vboold8& m, const vllong8& c, const vllong8& a, const vllong8& b) { return _mm512_mask_add_epi64(c,m,a,b); } - __forceinline vllong8 mask_sub(const vboold8& m, const vllong8& c, const vllong8& a, const vllong8& b) { return _mm512_mask_sub_epi64(c,m,a,b); } - - __forceinline vllong8 mask_and(const vboold8& m, const vllong8& c, const vllong8& a, const vllong8& b) { return _mm512_mask_and_epi64(c,m,a,b); } - __forceinline vllong8 mask_or (const vboold8& m, const vllong8& c, const vllong8& a, const vllong8& b) { return _mm512_mask_or_epi64(c,m,a,b); } - - //////////////////////////////////////////////////////////////////////////////// - /// Assignment Operators - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vllong8& operator +=(vllong8& a, const vllong8& b) { return a = a + b; } - __forceinline vllong8& operator +=(vllong8& a, long long b) { return a = a + b; } - - __forceinline vllong8& operator -=(vllong8& a, const vllong8& b) { return a = a - b; } - __forceinline vllong8& operator -=(vllong8& a, long long b) { return a = a - b; } - - __forceinline vllong8& operator *=(vllong8& a, const vllong8& b) { return a = a * b; } - __forceinline vllong8& operator *=(vllong8& a, long long b) { return a = a * b; } - - __forceinline vllong8& operator &=(vllong8& a, const vllong8& b) { return a = a & b; } - __forceinline vllong8& operator &=(vllong8& a, long long b) { return a = a & b; } - - __forceinline vllong8& operator |=(vllong8& a, const vllong8& b) { return a = a | b; } - __forceinline vllong8& operator |=(vllong8& a, long long b) { return a = a | b; } - - __forceinline vllong8& operator <<=(vllong8& a, long long b) { return a = a << b; } - __forceinline vllong8& operator >>=(vllong8& a, long long b) { return a = a >> b; } - - //////////////////////////////////////////////////////////////////////////////// - /// Comparison Operators + Select - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vboold8 operator ==(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_EQ); } - __forceinline vboold8 operator ==(const vllong8& a, long long b) { return a == vllong8(b); } - __forceinline vboold8 operator ==(long long a, const vllong8& b) { return vllong8(a) == b; } - - __forceinline vboold8 operator !=(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_NE); } - __forceinline vboold8 operator !=(const vllong8& a, long long b) { return a != vllong8(b); } - __forceinline vboold8 operator !=(long long a, const vllong8& b) { return vllong8(a) != b; } - - __forceinline vboold8 operator < (const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_LT); } - __forceinline vboold8 operator < (const vllong8& a, long long b) { return a < vllong8(b); } - __forceinline vboold8 operator < (long long a, const vllong8& b) { return vllong8(a) < b; } - - __forceinline vboold8 operator >=(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_GE); } - __forceinline vboold8 operator >=(const vllong8& a, long long b) { return a >= vllong8(b); } - __forceinline vboold8 operator >=(long long a, const vllong8& b) { return vllong8(a) >= b; } - - __forceinline vboold8 operator > (const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_GT); } - __forceinline vboold8 operator > (const vllong8& a, long long b) { return a > vllong8(b); } - __forceinline vboold8 operator > (long long a, const vllong8& b) { return vllong8(a) > b; } - - __forceinline vboold8 operator <=(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_LE); } - __forceinline vboold8 operator <=(const vllong8& a, long long b) { return a <= vllong8(b); } - __forceinline vboold8 operator <=(long long a, const vllong8& b) { return vllong8(a) <= b; } - - __forceinline vboold8 eq(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_EQ); } - __forceinline vboold8 ne(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_NE); } - __forceinline vboold8 lt(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_LT); } - __forceinline vboold8 ge(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_GE); } - __forceinline vboold8 gt(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_GT); } - __forceinline vboold8 le(const vllong8& a, const vllong8& b) { return _mm512_cmp_epi64_mask(a,b,_MM_CMPINT_LE); } - - __forceinline vboold8 eq(const vboold8 mask, const vllong8& a, const vllong8& b) { return _mm512_mask_cmp_epi64_mask(mask,a,b,_MM_CMPINT_EQ); } - __forceinline vboold8 ne(const vboold8 mask, const vllong8& a, const vllong8& b) { return _mm512_mask_cmp_epi64_mask(mask,a,b,_MM_CMPINT_NE); } - __forceinline vboold8 lt(const vboold8 mask, const vllong8& a, const vllong8& b) { return _mm512_mask_cmp_epi64_mask(mask,a,b,_MM_CMPINT_LT); } - __forceinline vboold8 ge(const vboold8 mask, const vllong8& a, const vllong8& b) { return _mm512_mask_cmp_epi64_mask(mask,a,b,_MM_CMPINT_GE); } - __forceinline vboold8 gt(const vboold8 mask, const vllong8& a, const vllong8& b) { return _mm512_mask_cmp_epi64_mask(mask,a,b,_MM_CMPINT_GT); } - __forceinline vboold8 le(const vboold8 mask, const vllong8& a, const vllong8& b) { return _mm512_mask_cmp_epi64_mask(mask,a,b,_MM_CMPINT_LE); } - - __forceinline vllong8 select(const vboold8& m, const vllong8& t, const vllong8& f) { - return _mm512_mask_or_epi64(f,m,t,t); - } - - __forceinline void xchg(const vboold8& m, vllong8& a, vllong8& b) { - const vllong8 c = a; a = select(m,b,a); b = select(m,c,b); - } - - __forceinline vboold8 test(const vboold8& m, const vllong8& a, const vllong8& b) { - return _mm512_mask_test_epi64_mask(m,a,b); - } - - __forceinline vboold8 test(const vllong8& a, const vllong8& b) { - return _mm512_test_epi64_mask(a,b); - } - - //////////////////////////////////////////////////////////////////////////////// - // Movement/Shifting/Shuffling Functions - //////////////////////////////////////////////////////////////////////////////// - - template<int i0, int i1> - __forceinline vllong8 shuffle(const vllong8& v) { - return _mm512_castpd_si512(_mm512_permute_pd(_mm512_castsi512_pd(v), (i1 << 7) | (i0 << 6) | (i1 << 5) | (i0 << 4) | (i1 << 3) | (i0 << 2) | (i1 << 1) | i0)); - } - - template<int i> - __forceinline vllong8 shuffle(const vllong8& v) { - return shuffle<i, i>(v); - } - - template<int i0, int i1, int i2, int i3> - __forceinline vllong8 shuffle(const vllong8& v) { - return _mm512_permutex_epi64(v, _MM_SHUFFLE(i3, i2, i1, i0)); - } - - template<int i0, int i1> - __forceinline vllong8 shuffle4(const vllong8& v) { - return _mm512_shuffle_i64x2(v, v, _MM_SHUFFLE(i1*2+1, i1*2, i0*2+1, i0*2)); - } - - template<int i> - __forceinline vllong8 shuffle4(const vllong8& v) { - return shuffle4<i, i>(v); - } - - template<int i> - __forceinline vllong8 align_shift_right(const vllong8& a, const vllong8& b) { - return _mm512_alignr_epi64(a, b, i); - }; - - __forceinline long long toScalar(const vllong8& v) { - return _mm_cvtsi128_si64(_mm512_castsi512_si128(v)); - } - - __forceinline vllong8 zeroExtend32Bit(const __m512i& a) { - return _mm512_cvtepu32_epi64(_mm512_castsi512_si256(a)); - } - - //////////////////////////////////////////////////////////////////////////////// - /// Reductions - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vllong8 vreduce_min2(vllong8 x) { return min(x, shuffle<1,0,3,2>(x)); } - __forceinline vllong8 vreduce_min4(vllong8 x) { x = vreduce_min2(x); return min(x, shuffle<2,3,0,1>(x)); } - __forceinline vllong8 vreduce_min (vllong8 x) { x = vreduce_min4(x); return min(x, shuffle4<1,0>(x)); } - - __forceinline vllong8 vreduce_max2(vllong8 x) { return max(x, shuffle<1,0,3,2>(x)); } - __forceinline vllong8 vreduce_max4(vllong8 x) { x = vreduce_max2(x); return max(x, shuffle<2,3,0,1>(x)); } - __forceinline vllong8 vreduce_max (vllong8 x) { x = vreduce_max4(x); return max(x, shuffle4<1,0>(x)); } - - __forceinline vllong8 vreduce_and2(vllong8 x) { return x & shuffle<1,0,3,2>(x); } - __forceinline vllong8 vreduce_and4(vllong8 x) { x = vreduce_and2(x); return x & shuffle<2,3,0,1>(x); } - __forceinline vllong8 vreduce_and (vllong8 x) { x = vreduce_and4(x); return x & shuffle4<1,0>(x); } - - __forceinline vllong8 vreduce_or2(vllong8 x) { return x | shuffle<1,0,3,2>(x); } - __forceinline vllong8 vreduce_or4(vllong8 x) { x = vreduce_or2(x); return x | shuffle<2,3,0,1>(x); } - __forceinline vllong8 vreduce_or (vllong8 x) { x = vreduce_or4(x); return x | shuffle4<1,0>(x); } - - __forceinline vllong8 vreduce_add2(vllong8 x) { return x + shuffle<1,0,3,2>(x); } - __forceinline vllong8 vreduce_add4(vllong8 x) { x = vreduce_add2(x); return x + shuffle<2,3,0,1>(x); } - __forceinline vllong8 vreduce_add (vllong8 x) { x = vreduce_add4(x); return x + shuffle4<1,0>(x); } - - __forceinline long long reduce_min(const vllong8& v) { return toScalar(vreduce_min(v)); } - __forceinline long long reduce_max(const vllong8& v) { return toScalar(vreduce_max(v)); } - __forceinline long long reduce_and(const vllong8& v) { return toScalar(vreduce_and(v)); } - __forceinline long long reduce_or (const vllong8& v) { return toScalar(vreduce_or (v)); } - __forceinline long long reduce_add(const vllong8& v) { return toScalar(vreduce_add(v)); } - - //////////////////////////////////////////////////////////////////////////////// - /// Memory load and store operations - //////////////////////////////////////////////////////////////////////////////// - - __forceinline vllong8 permute(const vllong8& v, const vllong8& index) { - return _mm512_permutexvar_epi64(index,v); - } - - __forceinline vllong8 reverse(const vllong8& a) { - return permute(a,vllong8(reverse_step)); - } - - //////////////////////////////////////////////////////////////////////////////// - /// Output Operators - //////////////////////////////////////////////////////////////////////////////// - - __forceinline embree_ostream operator <<(embree_ostream cout, const vllong8& v) - { - cout << "<" << v[0]; - for (size_t i=1; i<8; i++) cout << ", " << v[i]; - cout << ">"; - return cout; - } -} |