sgemm_wg: Implement blocking over k-dimension

2024-02-16 16:20:45 -08:00
parent 5f79e8a3f1
commit 301f1ca260
3 changed files with 113 additions and 63 deletions
--- a/tests/regression/sgemm_wg/common.h
+++ b/tests/regression/sgemm_wg/common.h
@@ -7,8 +7,11 @@
 #define DEV_SMEM_START_ADDR 0xff000000

 typedef struct {
-  uint32_t matrix_dim;
+  uint32_t dim_m;
+  uint32_t dim_n;
+  uint32_t dim_k;
  uint64_t addr_a;
+  uint64_t addr_b;
  uint64_t addr_c;
 } kernel_arg_t;