From 134ba825def507ed7a4a63c4d15f4cc7e86c72aa Mon Sep 17 00:00:00 2001 From: Hansung Kim Date: Mon, 19 Aug 2024 18:02:00 -0700 Subject: [PATCH] sgemm_impl: Fix typo bug for BK_adjusted --- tests/regression/sgemm_tcore/sgemm_impl.hpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/regression/sgemm_tcore/sgemm_impl.hpp b/tests/regression/sgemm_tcore/sgemm_impl.hpp index f3cfa1a3..8f8be348 100644 --- a/tests/regression/sgemm_tcore/sgemm_impl.hpp +++ b/tests/regression/sgemm_tcore/sgemm_impl.hpp @@ -293,8 +293,7 @@ inline void wmma_load_b(const volatile T *smem_B, const int local_k, // see comment in wmma_load_a constexpr int packed_factor = (std::is_same_v ? 2 : 1); - constexpr int BK_adjusted = BN / packed_factor; - constexpr int BN_adjusted = BN / packed_factor; + constexpr int BK_adjusted = BK / packed_factor; const int local_k_adjusted = local_k / packed_factor; // B is stored N-major in smem