sgemm_wg: Implement software barrier for inter-core synchronization

2024-03-12 15:34:42 -07:00
parent fbe872c831
commit 510a834db5
2 changed files with 41 additions and 10 deletions
--- a/tests/regression/sgemm_wg/main.cpp
+++ b/tests/regression/sgemm_wg/main.cpp
@@ -147,9 +147,9 @@ int main(int argc, char *argv[]) {
  RT_CHECK(vx_dev_open(&device));

  // FIXME: hardcoded
-  uint32_t dim_m = 64;
-  uint32_t dim_n = 64;
-  uint32_t dim_k = 64;
+  uint32_t dim_m = 32;
+  uint32_t dim_n = 32;
+  uint32_t dim_k = 32;

  generate_source_matrix(dim_m, dim_n, dim_k);
  generate_reference_matmul(dim_m, dim_n, dim_k);