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

mork to support 13w tokens

Browse Source
This commit is contained in:
coderfeli
2025-03-17 01:19:00 +00:00
parent ef8c1333b9
commit 5eaa36be18
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp
View File

@@ -1250,6 +1250,10 @@ CK_TILE_HOST index_t moe_sorting_mp_get_workspace_size(int tokens_, int num_expe
CK_TILE_HOST index_t moe_sorting_get_workspace_size(int tokens_, int num_experts_)
{
#if 1
if(tokens_ >= 65536 * 2)
{
return 0;
}
if(moe_sorting_is_oneshot(tokens_, num_experts_))
{
return 0;