portelli/Grid
1
0
Fork 0
mirror of https://github.com/paboyle/Grid.git synced 2025-04-04 19:25:56 +01:00
Code Releases Activity

Fix to no USE_FP16 case

Browse Source
This commit is contained in:
paboyle 2017-04-14 14:20:54 +01:00
parent 951be75292
commit 3ca41458a3
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
lib/simd/Grid_avx.h
View File

@ -470,13 +470,14 @@ namespace Optimization {
return in; return in;
}; };
}; };
#define USE_FP16
struct PrecisionChange { struct PrecisionChange {
static inline __m256i StoH (__m256 a,__m256 b) { static inline __m256i StoH (__m256 a,__m256 b) {
__m256 h;
#ifdef USE_FP16 #ifdef USE_FP16
__m128i ha = _mm256_cvtps_ph(a,0); __m128i ha = _mm256_cvtps_ph(a,0);
__m128i hb = _mm256_cvtps_ph(b,0); __m128i hb = _mm256_cvtps_ph(b,0);
__m256 h = _mm256_castps128_ps256(ha); h = _mm256_castps128_ps256(ha);
h = _mm256_insertf128_ps(h,hb,1); h = _mm256_insertf128_ps(h,hb,1);
#else #else
assert(0); assert(0);

2
lib/simd/Grid_avx512.h
View File

@ -340,7 +340,7 @@ namespace Optimization {
}; };
}; };
#define USE_FP16
struct PrecisionChange { struct PrecisionChange {
static inline __m512i StoH (__m512 a,__m512 b) { static inline __m512i StoH (__m512 a,__m512 b) {
__m512i h; __m512i h;