Skip to content

Commit 52c6abc

Browse files
committed
FIX: in avx sgemm, use setr to put the parameters in natural order
This is just a style change, the set/setr functions just take their parameters in opposite order.
1 parent c5627b5 commit 52c6abc

1 file changed

Lines changed: 2 additions & 1 deletion

File tree

src/sgemm_kernel.rs

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -310,7 +310,8 @@ unsafe fn kernel_x86_avx(k: usize, alpha: T, a: *const T, b: *const T,
310310
if csc == 1 {
311311
loop_m!(i, cv[i] = _mm256_loadu_ps(c![i, 0]));
312312
} else {
313-
loop_m!(i, cv[i] = _mm256_set_ps(*c![i, 7], *c![i, 6], *c![i, 5], *c![i, 4], *c![i, 3], *c![i, 2], *c![i, 1], *c![i, 0]));
313+
loop_m!(i, cv[i] = _mm256_setr_ps(*c![i, 0], *c![i, 1], *c![i, 2], *c![i, 3],
314+
*c![i, 4], *c![i, 5], *c![i, 6], *c![i, 7]));
314315
}
315316
// Compute β C
316317
loop_m!(i, cv[i] = _mm256_mul_ps(cv[i], betav));

0 commit comments

Comments
 (0)