From af96372a044f97b1fb522884efc5191fd11cfb3a Mon Sep 17 00:00:00 2001
From: Feng Shijie <Shijie.Feng@amd.com>
Date: Thu, 14 Aug 2025 09:29:21 +0000
Subject: [PATCH] avoid floating pointer exception for some cases

---
 example/ck_tile/21_moe_flatmm/run_moe_flatmm_example.inc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/example/ck_tile/21_moe_flatmm/run_moe_flatmm_example.inc b/example/ck_tile/21_moe_flatmm/run_moe_flatmm_example.inc
index 6e80bc53f7..b361d5e88e 100644
--- a/example/ck_tile/21_moe_flatmm/run_moe_flatmm_example.inc
+++ b/example/ck_tile/21_moe_flatmm/run_moe_flatmm_example.inc
@@ -195,12 +195,12 @@ int run_moe_gemm_example_with_layouts(int argc,
 
     for(int i = 0; i < sorted_tile_num; i++)
     {
-        expert_ids.mData[i] = i / (valid_tile_num / experts);
+        expert_ids.mData[i] = i / ((valid_tile_num + experts - 1) / experts);
     }
 
     int token_per_tile = (num_tokens * topk + valid_tile_num - 1) / valid_tile_num;
     // int token_per_tile = num_tokens * topk / valid_tile_num;
-    int tokenid        = 0;
+    int tokenid = 0;
     // sorted_token_ids.mData[0] = 0;
     for(int i = 0; i < sorted_tile_num * MPerBlock; i++)
     {