diff --git a/example/ck_tile/03_gemm/run_gemm_example.inc b/example/ck_tile/03_gemm/run_gemm_example.inc index fb67f944cc..de17628ab3 100644 --- a/example/ck_tile/03_gemm/run_gemm_example.inc +++ b/example/ck_tile/03_gemm/run_gemm_example.inc @@ -203,9 +203,9 @@ auto shuffle_b(const ck_tile::HostTensor& t) int divisor = 1; ck_tile::HostTensor t_view({n_ / GemmConfig::N_Warp_Tile, GemmConfig::N_Warp_Tile, - k_ / ItemsPerAccess, + k_ / GemmConfig::K_Warp_Tile, divisor, - ItemsPerAccess}); + GemmConfig::K_Warp_Tile / divisor}); std::copy(t.begin(), t.end(), t_view.begin()); return ck_tile::reference_permute(t_view, {0, 2, 3, 1, 4}); }