]> git.sesse.net Git - stockfish/blobdiff - src/nnue/layers/affine_transform.h
Fix AVX512 build with older compilers
[stockfish] / src / nnue / layers / affine_transform.h
index 985ee71a4193e571f9ecdddfc144ca4c2c571aea..8d2acd1852eabfa9dce873b803040742a2da75f9 100644 (file)
@@ -126,8 +126,7 @@ namespace Eval::NNUE::Layers {
             const auto iv256  = reinterpret_cast<const __m256i*>(&input_vector[kNumChunks]);
             const auto row256 = reinterpret_cast<const __m256i*>(&row[kNumChunks]);
             __m256i product256 = _mm256_maddubs_epi16(_mm256_loadA_si256(&iv256[0]), _mm256_load_si256(&row256[0]));
-            product256 = _mm256_madd_epi16(product256, _mm256_set1_epi16(1));
-            sum = _mm512_add_epi32(sum, _mm512_zextsi256_si512(product256));
+            sum = _mm512_add_epi32(sum, _mm512_cvtepi16_epi32(product256));
         }
         output[i] = _mm512_reduce_add_epi32(sum) + biases_[i];