diff --git a/src/layer/x86/layernorm_x86.cpp b/src/layer/x86/layernorm_x86.cpp index 3f6a66a5ec0..27ded2b4e3a 100644 --- a/src/layer/x86/layernorm_x86.cpp +++ b/src/layer/x86/layernorm_x86.cpp @@ -119,7 +119,7 @@ static NCNN_FORCEINLINE void fast_mean(float* ptr, float* mean, int elempack, in #if __SSE2__ #if __AVX__ #if __AVX512F__ - sum += _mm512_reduce_add_ps(_sum_512); + sum += _mm512_comp_reduce_add_ps(_sum_512); #endif // __AVX512F__ sum += _mm256_reduce_add_ps(_sum_256); #endif // __AVX__ @@ -230,7 +230,7 @@ static NCNN_FORCEINLINE void fast_var(float* ptr, float* var, const float* mean, #if __SSE2__ #if __AVX__ #if __AVX512F__ - sq_sum += _mm512_reduce_add_ps(_sq_sum_512); + sq_sum += _mm512_comp_reduce_add_ps(_sq_sum_512); #endif // __AVX512F__ sq_sum += _mm256_reduce_add_ps(_sq_sum_256); #endif // __AVX__