sgemm_impl: barrier dumb dumb

2024-09-11 00:01:56 -07:00
parent dc746272fb
commit ba66d2c2bd
1 changed files with 3 additions and 0 deletions
--- 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