From 7b0c25f864cc7a2d95d7e3c3c2b7c13b22116063 Mon Sep 17 00:00:00 2001
From: Chenggang Zhao <chenggangz@deepseek.com>
Date: Fri, 20 Jun 2025 16:37:28 +0800
Subject: [PATCH] Support more hidden size

---
 csrc/kernels/launch.cuh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/csrc/kernels/launch.cuh b/csrc/kernels/launch.cuh
index 5b398bf..c9cca9b 100644
--- a/csrc/kernels/launch.cuh
+++ b/csrc/kernels/launch.cuh
@@ -84,6 +84,7 @@ cfg.dynamicSmemBytes = smem_size;
 
 #define SWITCH_HIDDEN(case_macro) \
     switch (hidden) { \
+        case 2048: case_macro(2048); \
         case 2560: case_macro(2560); \
         case 4096: case_macro(4096); \
         case 5120: case_macro(5120); \