Homec4science
Diffusion Lenstool-HPC (chi) Browse

R1448/src/simd_math_avx.h
43bf390e02a9chi

Last Change
Raw File

simd_math_avx.h
View Options

#ifndef SIMD_MATH
#define SIMD_MATH_
//
#include <immintrin.h>
//
//
//
#ifdef __INTEL_COMPILER
//
inline __m256d operator + (__m256d a, __m256d b) {return _mm256_add_pd(a, b);}
inline __m256d operator - (__m256d a, __m256d b) {return _mm256_sub_pd(a, b);}
inline __m256d operator * (__m256d a, __m256d b) {return _mm256_mul_pd(a, b);}
inline __m256d operator / (__m256d a, __m256d b) {return _mm256_div_pd(a, b);}
#endif
//#define __INV RCP
//#define __INV RCP_1NR
#define __INV RCP_2NR
#define __SQRT _mm256_sqrt_pd
//#define __SQRT SQRT
//#define __SQRT SQRT_1NR
//#define __SQRT SQRT_2NR
inline __m256d RCP(const __m256d d)
{
//asm volatile("# RCP begins");
const __m128 b = _mm256_cvtpd_ps(d);
const __m128 rcp = _mm_rcp_ps (b);
__m256d x0 = _mm256_cvtps_pd(rcp);
//
//asm volatile("# RCP ends");
return x0;
}
//
//
//
inline __m256d RCP_1NR(const __m256d d)
{
//asm volatile("# RCP_1NR begins");
const __m128 b = _mm256_cvtpd_ps(d);
const __m128 rcp = _mm_rcp_ps (b);
__m256d x0 = _mm256_cvtps_pd(rcp);
//
x0 = x0 + x0 - d*x0*x0;
// //
//asm volatile("# RCP_1NR ends");
return x0;
}
//
//
//
inline __m256d RCP_2NR(const __m256d d)
{
//asm volatile("# RCP_2NR begins");
const __m128 b = _mm256_cvtpd_ps(d);
const __m128 rcp = _mm_rcp_ps (b);
__m256d x0 = _mm256_cvtps_pd(rcp);
//
x0 = x0 + x0 - d*x0*x0;
x0 = x0 + x0 - d*x0*x0;
//
//asm volatile("# RCP_2NR ends");
return x0;
}
//
inline __m256d SQRT(const __m256d d)
{
//asm volatile("# SQRT begins");
const __m128 b = _mm256_cvtpd_ps(d);
const __m128 rcp = _mm_sqrt_ps (b);
__m256d x0 = _mm256_cvtps_pd(rcp);
//asm volatile("# SQRT ends");
return x0;
}
//
//
//
inline __m256d SQRT_1NR(const __m256d d)
{
//asm volatile("# SQRT_1NR begins");
const __m128 b = _mm256_cvtpd_ps(d);
const __m128 rcp = _mm_sqrt_ps (b);
__m256d x0 = _mm256_cvtps_pd(rcp);
__m256d half = _mm256_set1_pd(0.5);
__m256d three = _mm256_set1_pd(3.);
__m256d a = RCP_1NR(d);
x0 = half*x0*(three - x0*x0*a);
//asm volatile("# SQRT_1NR ends");
return x0;
}
//
//
//
inline __m256d SQRT_2NR(const __m256d d)
{
//asm volatile("# SQRT_2NR begins");
const __m128 b = _mm256_cvtpd_ps(d);
const __m128 rcp = _mm_sqrt_ps (b);
__m256d x0 = _mm256_cvtps_pd(rcp);
__m256d half = _mm256_set1_pd(0.5);
__m256d three = _mm256_set1_pd(3.);
__m256d a = RCP_2NR(d);
x0 = half*x0*(three - x0*x0*a);
x0 = half*x0*(three - x0*x0*a);
//asm volatile("# SQRT_2NR ends");
return x0;
}
//
#endif
c4science · Help