Skip to content

Commit 7a1d234

Browse files
committed
Add flag for not converting A & B - will be used in future to do conversion during packing.
1 parent 1cc377e commit 7a1d234

2 files changed

Lines changed: 12 additions & 0 deletions

File tree

kernel/riscv64/sbgemm_kernel_16x8_zvl256b.c

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,9 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
4242
if ((M >= 4) && (N >= 4) && (K > 0)) {
4343
CONV = (FLOAT *)(malloc((K * (8 + (M & -4))) * sizeof(FLOAT)));
4444
if (!CONV) return 1;
45+
#ifndef BF16_DONT_CONV
4546
B_CONV(AA, CONV + (K * 8), (M & -4) * K);
47+
#endif
4648
}
4749
#endif
4850

@@ -52,9 +54,11 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
5254
BLASLONG gvl = __riscv_vsetvl_e16m1(16);
5355
#ifdef BF16_WIDEN_ONE
5456
BLASLONG bi2 = K * 8;
57+
#ifndef BF16_DONT_CONV
5558
if (M >= 4) {
5659
B_CONV(BB + (n_top*K), CONV, bi2);
5760
}
61+
#endif
5862
BLASLONG ai2 = K * 8;
5963
#endif
6064

@@ -476,9 +480,11 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
476480

477481
#ifdef BF16_WIDEN_ONE
478482
BLASLONG bi2 = K * 4;
483+
#ifndef BF16_DONT_CONV
479484
if (M >= 4) {
480485
B_CONV(BB + (n_top*K), CONV, bi2);
481486
}
487+
#endif
482488
BLASLONG ai2 = K * 8;
483489
#endif
484490

kernel/riscv64/sbgemm_kernel_8x8_zvl128b.c

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -42,7 +42,9 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
4242
if ((M >= 4) && (N >= 4) && (K > 0)) {
4343
CONV = (FLOAT *)(malloc((K * (8 + (M & -4))) * sizeof(FLOAT)));
4444
if (!CONV) return 1;
45+
#ifndef BF16_DONT_CONV
4546
B_CONV(AA, CONV + (K * 8), (M & -4) * K);
47+
#endif
4648
}
4749
#endif
4850

@@ -53,9 +55,11 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
5355

5456
#ifdef BF16_WIDEN_ONE
5557
BLASLONG bi2 = K * 8;
58+
#ifndef BF16_DONT_CONV
5659
if (M >= 4) {
5760
B_CONV(BB + (n_top*K), CONV, bi2);
5861
}
62+
#endif
5963
BLASLONG ai2 = K * 8;
6064
#endif
6165

@@ -379,9 +383,11 @@ int CNAME(BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, IFLOAT *A, IFLOAT *B,
379383

380384
#ifdef BF16_WIDEN_ONE
381385
BLASLONG bi2 = K * 4;
386+
#ifndef BF16_DONT_CONV
382387
if (M >= 4) {
383388
B_CONV(BB + (n_top*K), CONV, bi2);
384389
}
390+
#endif
385391
BLASLONG ai2 = K * 8;
386392
#endif
387393

0 commit comments

Comments
 (0)