fix sofmax seg fault in AVX, test=develop (#19487)

6 years ago · 11f2f78458
parent 1fe468d319
commit 11f2f78458
1 changed files with 1 additions and 1 deletions
--- a/paddle/fluid/operators/math/cpu_vec.h
+++ b/paddle/fluid/operators/math/cpu_vec.h
@ -160,7 +160,7 @@ inline void vec_sum<float, platform::avx>(const size_t n, const float* x,
  end = n & ~(block - 1);
  __m256 tmp = _mm256_setzero_ps();
  for (i = 0; i < end; i += block) {
-    tmp = _mm256_add_ps(tmp, _mm256_load_ps(x + i));
+    tmp = _mm256_add_ps(tmp, _mm256_loadu_ps(x + i));
  }

  __m256 hsum = _mm256_hadd_ps(tmp, tmp);