sgemm_tcore: Template-ize kernel code

2024-07-29 20:05:58 -07:00
parent 5f342914bd
commit 5d5a6fbad2
3 changed files with 117 additions and 111 deletions
--- a/tests/regression/sgemm_tcore/main.cpp
+++ b/tests/regression/sgemm_tcore/main.cpp
@@ -173,7 +173,7 @@ int main(int argc, char *argv[]) {
  uint32_t dim_n = 64;
  uint32_t dim_k = 64;

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