From 6d1e97cec9caecee95d46542a90de54a21d799e2 Mon Sep 17 00:00:00 2001 From: "Chen, Jian Ping" Date: Wed, 27 May 2020 05:57:44 +0800 Subject: [PATCH] Fix bf16 move_ker bug --- torch_ipex/csrc/cpu/bf16/vec/bf16_vec_kernel.h | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/torch_ipex/csrc/cpu/bf16/vec/bf16_vec_kernel.h b/torch_ipex/csrc/cpu/bf16/vec/bf16_vec_kernel.h index fd90899a9..974015324 100644 --- a/torch_ipex/csrc/cpu/bf16/vec/bf16_vec_kernel.h +++ b/torch_ipex/csrc/cpu/bf16/vec/bf16_vec_kernel.h @@ -166,9 +166,9 @@ static inline void move_ker(at::BFloat16 *out, at::BFloat16 *in, int64_t len) { int64_t i = 0; for (; i < len - 63; i += 64) { auto in0 = _mm512_loadu_si512(in + i); - auto in1 = _mm512_loadu_si512(in + i + 16); + auto in1 = _mm512_loadu_si512(in + i + 32); _mm512_storeu_si512(out + i, in0); - _mm512_storeu_si512(out + i + 16, in1); + _mm512_storeu_si512(out + i + 32, in1); } if (i < len - 31) {