Skip to content

Commit 1cc377e

Browse files
committed
Only convert B if M is greater or equal to 4.
1 parent 0acb60a commit 1cc377e

File tree

2 files changed

+12
-4
lines changed

2 files changed

+12
-4
lines changed

kernel/riscv64/sbgemm_kernel_16x8_zvl256b.c

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,9 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
5252
BLASLONG gvl = __riscv_vsetvl_e16m1(16);
5353
#ifdef BF16_WIDEN_ONE
5454
BLASLONG bi2 = K * 8;
55-
B_CONV(BB + (n_top*K), CONV, bi2);
55+
if (M >= 4) {
56+
B_CONV(BB + (n_top*K), CONV, bi2);
57+
}
5658
BLASLONG ai2 = K * 8;
5759
#endif
5860

@@ -474,7 +476,9 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
474476

475477
#ifdef BF16_WIDEN_ONE
476478
BLASLONG bi2 = K * 4;
477-
B_CONV(BB + (n_top*K), CONV, bi2);
479+
if (M >= 4) {
480+
B_CONV(BB + (n_top*K), CONV, bi2);
481+
}
478482
BLASLONG ai2 = K * 8;
479483
#endif
480484

kernel/riscv64/sbgemm_kernel_8x8_zvl128b.c

Lines changed: 6 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,7 +53,9 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
5353

5454
#ifdef BF16_WIDEN_ONE
5555
BLASLONG bi2 = K * 8;
56-
B_CONV(BB + (n_top*K), CONV, bi2);
56+
if (M >= 4) {
57+
B_CONV(BB + (n_top*K), CONV, bi2);
58+
}
5759
BLASLONG ai2 = K * 8;
5860
#endif
5961

@@ -377,7 +379,9 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
377379

378380
#ifdef BF16_WIDEN_ONE
379381
BLASLONG bi2 = K * 4;
380-
B_CONV(BB + (n_top*K), CONV, bi2);
382+
if (M >= 4) {
383+
B_CONV(BB + (n_top*K), CONV, bi2);
384+
}
381385
BLASLONG ai2 = K * 8;
382386
#endif
383387

0 commit comments

Comments
 (0)