tensor: Change B in-memory layout to column-major

2024-08-12 15:20:55 -07:00
parent 07dd9e35a0
commit 95e3e96c6c
6 changed files with 42 additions and 24 deletions
--- a/tests/regression/sgemm_tcore/util.hpp
+++ b/tests/regression/sgemm_tcore/util.hpp
@@ -20,7 +20,7 @@
 //   BM <= BK*TM*TN
 #define BM 64
 #define BN 64
-#define BK 64
+#define BK 128
 #define WM 16
 #define WN 8
 #define TCM 8