Skip to content

Commit e31a7a4

Browse files
authored
fix MX bpreshuffle gemm B grid descriptor dimension error. (#3170)
1 parent d04eba4 commit e31a7a4

1 file changed

Lines changed: 2 additions & 2 deletions

File tree

include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdl_cshuffle_v3_mx_bpreshuffle.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -429,8 +429,8 @@ struct GridwiseGemmMX_xdl_cshuffle_v3_bpreshuffle
429429
constexpr index_t MWave = MPerBlock / (MXdlPerWave * MPerXdl);
430430
constexpr index_t WaveSize = BlockSize / (MWave * NWave);
431431
constexpr index_t NkSwizzleNumber = Number<WaveSize * KPack>{};
432-
return make_naive_tensor_descriptor_packed(
433-
make_tuple(N0 / NWave / NXdlPack, NWave, NXdlPack, K0, NkSwizzleNumber));
432+
return make_naive_tensor_descriptor_packed(make_tuple(
433+
math::integer_divide_ceil(N0, NWave * NXdlPack), NWave, NXdlPack, K0, NkSwizzleNumber));
434434
}
435435

436436
__host__ __device__ static auto MakeBGridDescriptor_BK0_N_BK1(

0 commit comments

Comments
 (0)