From 11f2f784589c32add30f42c97eaaac2f6c98fe68 Mon Sep 17 00:00:00 2001 From: Zeng Jinle <32832641+sneaxiy@users.noreply.github.com> Date: Thu, 29 Aug 2019 14:57:53 +0800 Subject: [PATCH] fix sofmax seg fault in AVX, test=develop (#19487) --- paddle/fluid/operators/math/cpu_vec.h | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/fluid/operators/math/cpu_vec.h b/paddle/fluid/operators/math/cpu_vec.h index 4406a558718..8940a41424b 100644 --- a/paddle/fluid/operators/math/cpu_vec.h +++ b/paddle/fluid/operators/math/cpu_vec.h @@ -160,7 +160,7 @@ inline void vec_sum(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); -- GitLab