Browse Source

Fix linux build error(#4265) (#4294)

Co-authored-by: wangyu <786794414@qq.com>
tags/20221128
bestpower GitHub 3 years ago
parent
commit
a116e005b8
No known key found for this signature in database GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      src/layer/x86/layernorm_x86.cpp

+ 2
- 2
src/layer/x86/layernorm_x86.cpp View File

@@ -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__


Loading…
Cancel
Save