From 68cd6455fe3fa9835909f70d1b3d9a2d544f0b81 Mon Sep 17 00:00:00 2001
From: Hansung Kim <hansung_kim@berkeley.edu>
Date: Wed, 2 Oct 2024 15:17:44 -0700
Subject: [PATCH] sgemm_impl: Add mmio reconverge barrier to avoid slip-off;
 switch to FP32

---
 tests/regression/sgemm_tcore/sgemm_impl.hpp | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/tests/regression/sgemm_tcore/sgemm_impl.hpp b/tests/regression/sgemm_tcore/sgemm_impl.hpp
index d1b9d76e..0134d6e5 100644
--- a/tests/regression/sgemm_tcore/sgemm_impl.hpp
+++ b/tests/regression/sgemm_tcore/sgemm_impl.hpp
@@ -6,7 +6,7 @@
 #include "include/gemmini.h"
 #include "gemmini_mmio.h"
 
-#define FP_SIZE 16
+#define FP_SIZE 32
 
 // "fake" fp16 type that only has the correct data width.
 using float16_t = uint16_t;
@@ -822,6 +822,10 @@ __attribute__((always_inline)) inline void thread_block_gemm_single_tile(
     if (tid_in_threadblock == 0) {
       gemmini_fence();
     }
+
+    // reconverge after mmio
+    threadblock_barrier(threadblock_id_in_cluster,
+                        warps_per_threadblock_per_core);
   }
 
   if constexpr (write_to_mem) {