leavelet/cutlass
1
0
Fork 0
Code Issues Pull Requests Actions 28 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 22:02:59 -05:00
committed by GitHub
co-authored by Shreya Gaur Shreya Gaur
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
include/cutlass/gemm/group_array_problem_shape.hpp
+1 -2
View File
@@ -116,8 +116,7 @@ struct MoEProblemShape {
UnderlyingProblemShape const UnderlyingProblemShape const
get_host_problem_shape(int32_t group_idx=0) const { get_host_problem_shape(int32_t group_idx=0) const {
UnderlyingProblemShape expert_problem_dims = {max_m, max_n, max_k}; 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 && tokens_per_expert_host != nullptr) {
if (group_idx < num_groups) {
expert_problem_dims = {max_m, tokens_per_expert_host[group_idx], max_k}; expert_problem_dims = {max_m, tokens_per_expert_host[group_idx], max_k};
} }
return expert_problem_dims; return expert_problem_dims;