From 5eaa36be180516b8f3c043bfcbf4b44ed7728dfb Mon Sep 17 00:00:00 2001
From: coderfeli <coderfeli@163.com>
Date: Mon, 17 Mar 2025 01:19:00 +0000
Subject: [PATCH] mork to support 13w tokens

---
 include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp b/include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp
index a1410d1f4f..3b1d91884f 100644
--- a/include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp
+++ b/include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp
@@ -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;