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/main.cpp
+++ b/tests/regression/sgemm_tcore/main.cpp
@@ -173,7 +173,8 @@ int main(int argc, char *argv[]) {
  uint32_t dim_n = 64;
  uint32_t dim_k = 64;

-  using float_type = float;
+  using float_type = half;
+
  generate_source_matrix<float_type>(dim_m, dim_n, dim_k);
  generate_reference_matmul<float_type>(dim_m, dim_n, dim_k);