sbgemm: cooperlake: tuning for small matrix

guowangy · guowangy · commit 59a1114d03b5 · 2021-09-07T21:30:46.000+08:00
diff --git a/kernel/x86_64/sbgemm_small_kernel_permit_cooperlake.c b/kernel/x86_64/sbgemm_small_kernel_permit_cooperlake.c
@@ -38,5 +38,11 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 
 int CNAME(int transa, int transb, BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, FLOAT beta)
 {
-	return 1;
+	double MNK = (double) M * (double) N * (double) K;
+	if (MNK > 256.0*256.0*256.0)  // disable for big size matrix
+		return 0;
+	/* small matrix kernel works well for N = 8, 16, 32 */
+	if (N == 8 || N == 16 || N == 32)
+		return 1;
+	return 0;
 }

Original file line number	Diff line number	Diff line change
`@@ -38,5 +38,11 @@ USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.`
`38`	`38`
`39`	`39`	`int CNAME(int transa, int transb, BLASLONG M, BLASLONG N, BLASLONG K, FLOAT alpha, FLOAT beta)`
`40`	`40`	`{`
`41`		`- return 1;`
	`41`	`+ double MNK = (double) M * (double) N * (double) K;`
	`42`	`+ if (MNK > 256.0256.0256.0) // disable for big size matrix`
	`43`	`+ return 0;`
	`44`	`+ /* small matrix kernel works well for N = 8, 16, 32 */`
	`45`	`+ if (N == 8 \|\| N == 16 \|\| N == 32)`
	`46`	`+ return 1;`
	`47`	`+ return 0;`
`42`	`48`	`}`