leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 1 Packages Projects Releases Wiki Activity

bug fix for example 92 (#2830)

Browse Source 
Co-authored-by: Shreya Gaur <shgaur@dc2-container-xterm-012.prd.it.nvidia.com>
Co-authored-by: Shreya Gaur <shgaur@2u2g-spr-0015.ipp4a1.colossus.nvidia.com>
This commit is contained in:
Shreya Gaur
2025-12-01 19:02:59 -08:00
committed by GitHub
parent ec8daf642d
commit af8d5dfa54
1 changed files with 1 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/cutlass/gemm/group_array_problem_shape.hpp
View File

@@ -116,8 +116,7 @@ struct MoEProblemShape {
UnderlyingProblemShape const
get_host_problem_shape(int32_t group_idx=0) const {
UnderlyingProblemShape expert_problem_dims = {max_m, max_n, max_k};
assert(tokens_per_expert_host != nullptr); //tokens_per_expert_host should not be null
if (group_idx < num_groups) {
if (group_idx < num_groups && tokens_per_expert_host != nullptr) {
expert_problem_dims = {max_m, tokens_per_expert_host[group_idx], max_k};
}
return expert_problem_dims;