From ba66d2c2bd23b2a54e50d0c6fae7874ae14047b9 Mon Sep 17 00:00:00 2001
From: Hansung Kim <hansung_kim@berkeley.edu>
Date: Wed, 11 Sep 2024 00:01:56 -0700
Subject: [PATCH] sgemm_impl: barrier dumb dumb

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

diff --git a/tests/regression/sgemm_tcore/sgemm_impl.hpp b/tests/regression/sgemm_tcore/sgemm_impl.hpp
index f0998873..d1b9d76e 100644
--- a/tests/regression/sgemm_tcore/sgemm_impl.hpp
+++ b/tests/regression/sgemm_tcore/sgemm_impl.hpp
@@ -1042,6 +1042,9 @@ inline void thread_block_gemm(const T *A, const T *B, float *C,
         load_tile_to_smem<T, MemLayout::MN_major, MemLayout::MN_major, BN, BK,
                           threads_per_threadblock>(dim_n, block_n, block_k, B,
                                                    local_b, tid_in_threadblock);
+
+        threadblock_barrier(threadblock_id_in_cluster,
+                            warps_per_threadblock_per_core);
 #endif
 
         // consumer code: SMEM->RF and compute