ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 19:28:33 +00:00
Code Issues Packages Projects Releases Wiki Activity

avoid floating pointer exception for some cases

Browse Source
This commit is contained in:
Feng Shijie
2025-08-14 09:29:21 +00:00
parent cc9c7b9e58
commit af96372a04
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
example/ck_tile/21_moe_flatmm/run_moe_flatmm_example.inc
View File

@@ -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++)
{