wu-arch/kernels
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

sgemm_impl: barrier dumb dumb

Browse Source
This commit is contained in:
Hansung Kim
2024-09-11 00:01:56 -07:00
parent dc746272fb
commit ba66d2c2bd
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/regression/sgemm_tcore/sgemm_impl.hpp
View File

@@ -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, load_tile_to_smem<T, MemLayout::MN_major, MemLayout::MN_major, BN, BK,
threads_per_threadblock>(dim_n, block_n, block_k, B, threads_per_threadblock>(dim_n, block_n, block_k, B,
local_b, tid_in_threadblock); local_b, tid_in_threadblock);
threadblock_barrier(threadblock_id_in_cluster,
warps_per_threadblock_per_core);
#endif #endif
// consumer code: SMEM->RF and compute // consumer code: SMEM->RF and compute