Skip to content

Commit ce71457

Browse files
Fix bf16 move_ker bug (#29)
1 parent 31baca9 commit ce71457

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

torch_ipex/csrc/cpu/bf16/vec/bf16_vec_kernel.h

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -166,9 +166,9 @@ static inline void move_ker(at::BFloat16 *out, at::BFloat16 *in, int64_t len) {
166166
int64_t i = 0;
167167
for (; i < len - 63; i += 64) {
168168
auto in0 = _mm512_loadu_si512(in + i);
169-
auto in1 = _mm512_loadu_si512(in + i + 16);
169+
auto in1 = _mm512_loadu_si512(in + i + 32);
170170
_mm512_storeu_si512(out + i, in0);
171-
_mm512_storeu_si512(out + i + 16, in1);
171+
_mm512_storeu_si512(out + i + 32, in1);
172172
}
173173

174174
if (i < len - 31) {

0 commit comments

Comments
 (0)