From af8d5dfa54ed5ef550ab18ccc68bde59bbea72fc Mon Sep 17 00:00:00 2001
From: Shreya Gaur <48754356+Shreya-gaur@users.noreply.github.com>
Date: Mon, 1 Dec 2025 19:02:59 -0800
Subject: [PATCH] bug fix for example 92 (#2830)

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>
---
 include/cutlass/gemm/group_array_problem_shape.hpp | 3 +--
 1 file changed, 1 insertion(+), 2 deletions(-)

diff --git a/include/cutlass/gemm/group_array_problem_shape.hpp b/include/cutlass/gemm/group_array_problem_shape.hpp
index 6d198e17..ac471ffa 100644
--- a/include/cutlass/gemm/group_array_problem_shape.hpp
+++ b/include/cutlass/gemm/group_array_problem_shape.hpp
@@ -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;