mirror of
https://github.com/paboyle/Grid.git
synced 2024-11-10 07:55:35 +00:00
355 lines
10 KiB
C++
355 lines
10 KiB
C++
//----------------------------------------------------------------------
|
|
/*! @file Grid_knc.h
|
|
@brief Optimization libraries for AVX512 instructions set for KNC
|
|
|
|
Using intrinsics
|
|
*/
|
|
// Time-stamp: <2015-06-09 14:27:28 neo>
|
|
//----------------------------------------------------------------------
|
|
|
|
#include <immintrin.h>
|
|
#ifndef KNC_ONLY_STORES
|
|
#define _mm512_storenrngo_ps _mm512_store_ps // not present in AVX512
|
|
#define _mm512_storenrngo_pd _mm512_store_pd // not present in AVX512
|
|
#endif
|
|
|
|
|
|
namespace Optimization {
|
|
|
|
struct Vsplat{
|
|
//Complex float
|
|
inline __m512 operator()(float a, float b){
|
|
return _mm512_set_ps(b,a,b,a,b,a,b,a,b,a,b,a,b,a,b,a);
|
|
}
|
|
// Real float
|
|
inline __m512 operator()(float a){
|
|
return _mm512_set1_ps(a);
|
|
}
|
|
//Complex double
|
|
inline __m512d operator()(double a, double b){
|
|
return _mm512_set_pd(b,a,b,a,b,a,b,a);
|
|
}
|
|
//Real double
|
|
inline __m512d operator()(double a){
|
|
return _mm512_set1_pd(a);
|
|
}
|
|
//Integer
|
|
inline __m512i operator()(Integer a){
|
|
return _mm512_set1_epi32(a);
|
|
}
|
|
};
|
|
|
|
struct Vstore{
|
|
//Float
|
|
inline void operator()(__m512 a, float* F){
|
|
_mm512_store_ps(F,a);
|
|
}
|
|
//Double
|
|
inline void operator()(__m512d a, double* D){
|
|
_mm512_store_pd(D,a);
|
|
}
|
|
//Integer
|
|
inline void operator()(__m512i a, Integer* I){
|
|
_mm512_store_si512((__m512i *)I,a);
|
|
}
|
|
|
|
};
|
|
|
|
|
|
struct Vstream{
|
|
//Float
|
|
inline void operator()(float * a, __m512 b){
|
|
_mm512_storenrngo_ps(a,b);
|
|
}
|
|
//Double
|
|
inline void operator()(double * a, __m512d b){
|
|
_mm512_storenrngo_pd(a,b);
|
|
}
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
struct Vset{
|
|
// Complex float
|
|
inline __m512 operator()(Grid::ComplexF *a){
|
|
return _mm512_set_ps(a[7].imag(),a[7].real(),a[6].imag(),a[6].real(),
|
|
a[5].imag(),a[5].real(),a[4].imag(),a[4].real(),
|
|
a[3].imag(),a[3].real(),a[2].imag(),a[2].real(),
|
|
a[1].imag(),a[1].real(),a[0].imag(),a[0].real());
|
|
}
|
|
// Complex double
|
|
inline __m512d operator()(Grid::ComplexD *a){
|
|
return _mm512_set_pd(a[3].imag(),a[3].real(),a[2].imag(),a[2].real(),
|
|
a[1].imag(),a[1].real(),a[0].imag(),a[0].real());
|
|
}
|
|
// Real float
|
|
inline __m512 operator()(float *a){
|
|
return _mm512_set_ps( a[15],a[14],a[13],a[12],a[11],a[10],a[9],a[8],
|
|
a[7],a[6],a[5],a[4],a[3],a[2],a[1],a[0]);
|
|
}
|
|
// Real double
|
|
inline __m512d operator()(double *a){
|
|
return _mm512_set_pd(a[7],a[6],a[5],a[4],a[3],a[2],a[1],a[0]);
|
|
}
|
|
// Integer
|
|
inline __m512i operator()(Integer *a){
|
|
return _mm512_set_epi32( a[15],a[14],a[13],a[12],a[11],a[10],a[9],a[8],
|
|
a[7],a[6],a[5],a[4],a[3],a[2],a[1],a[0]);
|
|
}
|
|
|
|
|
|
};
|
|
|
|
template <typename Out_type, typename In_type>
|
|
struct Reduce{
|
|
//Need templated class to overload output type
|
|
//General form must generate error if compiled
|
|
inline Out_type operator()(In_type in){
|
|
printf("Error, using wrong Reduce function\n");
|
|
exit(1);
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
|
|
|
|
|
|
/////////////////////////////////////////////////////
|
|
// Arithmetic operations
|
|
/////////////////////////////////////////////////////
|
|
struct Sum{
|
|
//Complex/Real float
|
|
inline __m512 operator()(__m512 a, __m512 b){
|
|
return _mm512_add_ps(a,b);
|
|
}
|
|
//Complex/Real double
|
|
inline __m512d operator()(__m512d a, __m512d b){
|
|
return _mm512_add_pd(a,b);
|
|
}
|
|
//Integer
|
|
inline __m512i operator()(__m512i a, __m512i b){
|
|
return _mm512_add_epi32(a,b);
|
|
}
|
|
};
|
|
|
|
struct Sub{
|
|
//Complex/Real float
|
|
inline __m512 operator()(__m512 a, __m512 b){
|
|
return _mm512_sub_ps(a,b);
|
|
}
|
|
//Complex/Real double
|
|
inline __m512d operator()(__m512d a, __m512d b){
|
|
return _mm512_sub_pd(a,b);
|
|
}
|
|
//Integer
|
|
inline __m512i operator()(__m512i a, __m512i b){
|
|
return _mm512_sub_epi32(a,b);
|
|
}
|
|
};
|
|
|
|
|
|
struct MultComplex{
|
|
// Complex float
|
|
inline __m512 operator()(__m512 a, __m512 b){
|
|
__m512 vzero,ymm0,ymm1,real, imag;
|
|
vzero = _mm512_setzero_ps();
|
|
ymm0 = _mm512_swizzle_ps(a, _MM_SWIZ_REG_CDAB); //
|
|
real = (__m512)_mm512_mask_or_epi32((__m512i)a, 0xAAAA,(__m512i)vzero,(__m512i)ymm0);
|
|
imag = _mm512_mask_sub_ps(a, 0x5555,vzero, ymm0);
|
|
ymm1 = _mm512_mul_ps(real, b);
|
|
ymm0 = _mm512_swizzle_ps(b, _MM_SWIZ_REG_CDAB); // OK
|
|
return _mm512_fmadd_ps(ymm0,imag,ymm1);
|
|
}
|
|
// Complex double
|
|
inline __m512d operator()(__m512d a, __m512d b){
|
|
/* This is from
|
|
* Automatic SIMD Vectorization of Fast Fourier Transforms for the Larrabee and AVX Instruction Sets
|
|
* @inproceedings{McFarlin:2011:ASV:1995896.1995938,
|
|
* author = {McFarlin, Daniel S. and Arbatov, Volodymyr and Franchetti, Franz and P\"{u}schel, Markus},
|
|
* title = {Automatic SIMD Vectorization of Fast Fourier Transforms for the Larrabee and AVX Instruction Sets},
|
|
* booktitle = {Proceedings of the International Conference on Supercomputing},
|
|
* series = {ICS '11},
|
|
* year = {2011},
|
|
* isbn = {978-1-4503-0102-2},
|
|
* location = {Tucson, Arizona, USA},
|
|
* pages = {265--274},
|
|
* numpages = {10},
|
|
* url = {http://doi.acm.org/10.1145/1995896.1995938},
|
|
* doi = {10.1145/1995896.1995938},
|
|
* acmid = {1995938},
|
|
* publisher = {ACM},
|
|
* address = {New York, NY, USA},
|
|
* keywords = {autovectorization, fourier transform, program generation, simd, super-optimization},
|
|
* }
|
|
*/
|
|
__m512d vzero,ymm0,ymm1,real,imag;
|
|
vzero =_mm512_setzero_pd();
|
|
ymm0 = _mm512_swizzle_pd(a, _MM_SWIZ_REG_CDAB); //
|
|
real =(__m512d)_mm512_mask_or_epi64((__m512i)a, 0xAA,(__m512i)vzero,(__m512i) ymm0);
|
|
imag = _mm512_mask_sub_pd(a, 0x55,vzero, ymm0);
|
|
ymm1 = _mm512_mul_pd(real, b);
|
|
ymm0 = _mm512_swizzle_pd(b, _MM_SWIZ_REG_CDAB); // OK
|
|
return _mm512_fmadd_pd(ymm0,imag,ymm1);
|
|
}
|
|
};
|
|
|
|
struct Mult{
|
|
// Real float
|
|
inline __m512 operator()(__m512 a, __m512 b){
|
|
return _mm512_mul_ps(a,b);
|
|
}
|
|
// Real double
|
|
inline __m512d operator()(__m512d a, __m512d b){
|
|
return _mm512_mul_pd(a,b);
|
|
}
|
|
// Integer
|
|
inline __m512i operator()(__m512i a, __m512i b){
|
|
return _mm512_mullo_epi32(a,b);
|
|
}
|
|
};
|
|
|
|
|
|
struct Conj{
|
|
// Complex single
|
|
inline __m512 operator()(__m512 in){
|
|
return _mm512_mask_sub_ps(in,0xaaaa,_mm512_setzero_ps(),in); // Zero out 0+real 0-imag
|
|
}
|
|
// Complex double
|
|
inline __m512d operator()(__m512d in){
|
|
return _mm512_mask_sub_pd(in, 0xaa,_mm512_setzero_pd(), in);
|
|
}
|
|
// do not define for integer input
|
|
};
|
|
|
|
struct TimesMinusI{
|
|
//Complex single
|
|
inline __m512 operator()(__m512 in, __m512 ret){
|
|
__m512 tmp = _mm512_mask_sub_ps(in,0xaaaa,_mm512_setzero_ps(),in); // real -imag
|
|
return _mm512_swizzle_ps(tmp, _MM_SWIZ_REG_CDAB);// OK
|
|
}
|
|
//Complex double
|
|
inline __m512d operator()(__m512d in, __m512d ret){
|
|
__m512d tmp = _mm512_mask_sub_pd(in,0xaa,_mm512_setzero_pd(),in); // real -imag
|
|
return _mm512_swizzle_pd(tmp, _MM_SWIZ_REG_CDAB);// OK
|
|
}
|
|
|
|
|
|
};
|
|
|
|
struct TimesI{
|
|
//Complex single
|
|
inline __m512 operator()(__m512 in, __m512 ret){
|
|
__m512 tmp = _mm512_swizzle_ps(in, _MM_SWIZ_REG_CDAB);// OK
|
|
return _mm512_mask_sub_ps(tmp,0xaaaa,_mm512_setzero_ps(),tmp); // real -imag
|
|
}
|
|
//Complex double
|
|
inline __m512d operator()(__m512d in, __m512d ret){
|
|
__m512d tmp = _mm512_swizzle_pd(in, _MM_SWIZ_REG_CDAB);// OK
|
|
return _mm512_mask_sub_pd(tmp,0xaa,_mm512_setzero_pd(),tmp); // real -imag
|
|
}
|
|
|
|
|
|
};
|
|
|
|
|
|
|
|
|
|
|
|
//////////////////////////////////////////////
|
|
// Some Template specialization
|
|
|
|
//Complex float Reduce
|
|
template<>
|
|
inline Grid::ComplexF Reduce<Grid::ComplexF, __m512>::operator()(__m512 in){
|
|
return Grid::ComplexF(_mm512_mask_reduce_add_ps(0x5555, in),_mm512_mask_reduce_add_ps(0xAAAA, in));
|
|
}
|
|
//Real float Reduce
|
|
template<>
|
|
inline Grid::RealF Reduce<Grid::RealF, __m512>::operator()(__m512 in){
|
|
return _mm512_reduce_add_ps(in);
|
|
}
|
|
|
|
|
|
//Complex double Reduce
|
|
template<>
|
|
inline Grid::ComplexD Reduce<Grid::ComplexD, __m512d>::operator()(__m512d in){
|
|
return Grid::ComplexD(_mm512_mask_reduce_add_pd(0x55, in),_mm512_mask_reduce_add_pd(0xAA, in));
|
|
}
|
|
|
|
//Real double Reduce
|
|
template<>
|
|
inline Grid::RealD Reduce<Grid::RealD, __m512d>::operator()(__m512d in){
|
|
return _mm512_reduce_add_pd(in);
|
|
}
|
|
|
|
//Integer Reduce
|
|
template<>
|
|
inline Integer Reduce<Integer, __m512i>::operator()(__m512i in){
|
|
// FIXME unimplemented
|
|
printf("Reduce : Missing integer implementation -> FIX\n");
|
|
assert(0);
|
|
}
|
|
|
|
|
|
}
|
|
|
|
//////////////////////////////////////////////////////////////////////////////////////
|
|
// Here assign types
|
|
namespace Grid {
|
|
typedef __m512 SIMD_Ftype; // Single precision type
|
|
typedef __m512d SIMD_Dtype; // Double precision type
|
|
typedef __m512i SIMD_Itype; // Integer type
|
|
|
|
// prefecth
|
|
inline void v_prefetch0(int size, const char *ptr){
|
|
for(int i=0;i<size;i+=64){ // Define L1 linesize above
|
|
_mm_prefetch(ptr+i+4096,_MM_HINT_T1);
|
|
_mm_prefetch(ptr+i+512,_MM_HINT_T0);
|
|
}
|
|
}
|
|
inline void prefetch_HINT_T0(const char *ptr){
|
|
_mm_prefetch(ptr,_MM_HINT_T0);
|
|
}
|
|
|
|
|
|
|
|
|
|
// Gpermute utilities consider coalescing into 1 Gpermute
|
|
template < typename VectorSIMD >
|
|
inline void Gpermute(VectorSIMD &y,const VectorSIMD &b, int perm ) {
|
|
union {
|
|
__m512 f;
|
|
decltype(VectorSIMD::v) v;
|
|
} conv;
|
|
conv.v = b.v;
|
|
switch(perm){
|
|
case 3: conv.f = _mm512_swizzle_ps(conv.f,_MM_SWIZ_REG_CDAB); break;
|
|
case 2: conv.f = _mm512_swizzle_ps(conv.f,_MM_SWIZ_REG_BADC); break;
|
|
case 1 : conv.f = _mm512_permute4f128_ps(conv.f,(_MM_PERM_ENUM)_MM_SHUFFLE(2,3,0,1)); break;
|
|
case 0 : conv.f = _mm512_permute4f128_ps(conv.f,(_MM_PERM_ENUM)_MM_SHUFFLE(1,0,3,2)); break;
|
|
default: assert(0); break;
|
|
}
|
|
y.v=conv.v;
|
|
};
|
|
|
|
// Function name aliases
|
|
typedef Optimization::Vsplat VsplatSIMD;
|
|
typedef Optimization::Vstore VstoreSIMD;
|
|
typedef Optimization::Vset VsetSIMD;
|
|
typedef Optimization::Vstream VstreamSIMD;
|
|
template <typename S, typename T> using ReduceSIMD = Optimization::Reduce<S,T>;
|
|
|
|
|
|
// Arithmetic operations
|
|
typedef Optimization::Sum SumSIMD;
|
|
typedef Optimization::Sub SubSIMD;
|
|
typedef Optimization::Mult MultSIMD;
|
|
typedef Optimization::MultComplex MultComplexSIMD;
|
|
typedef Optimization::Conj ConjSIMD;
|
|
typedef Optimization::TimesMinusI TimesMinusISIMD;
|
|
typedef Optimization::TimesI TimesISIMD;
|
|
|
|
}
|