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

sgemm_impl: Fix typo bug for BK_adjusted

Browse Source
This commit is contained in:
Hansung Kim
2024-08-19 18:02:00 -07:00
parent 3f4abc542c
commit 134ba825de
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -293,8 +293,7 @@ inline void wmma_load_b(const volatile T *smem_B, const int local_k,
// see comment in wmma_load_a // see comment in wmma_load_a
constexpr int packed_factor = (std::is_same_v<T, float16_t> ? 2 : 1); constexpr int packed_factor = (std::is_same_v<T, float16_t> ? 2 : 1);
constexpr int BK_adjusted = BN / packed_factor; constexpr int BK_adjusted = BK / packed_factor;
constexpr int BN_adjusted = BN / packed_factor;
const int local_k_adjusted = local_k / packed_factor; const int local_k_adjusted = local_k / packed_factor;
// B is stored N-major in smem // B is stored N-major in smem