From a107266a4ea91de14b8ec2d4c4a4daf443a02060 Mon Sep 17 00:00:00 2001
From: sleepcoo <sleepcoo@gmail.com>
Date: Mon, 12 May 2025 16:32:32 +0800
Subject: [PATCH] support hidden size 4096

Co-authored-by: zhyncs <me@zhyncs.com>
Co-authored-by: yinfan98 <1106310035@qq.com>
---
 csrc/kernels/launch.cuh | 1 +
 1 file changed, 1 insertion(+)

diff --git a/csrc/kernels/launch.cuh b/csrc/kernels/launch.cuh
index 8dc5ffd..3d7574c 100644
--- a/csrc/kernels/launch.cuh
+++ b/csrc/kernels/launch.cuh
@@ -54,6 +54,7 @@
 #define SWITCH_HIDDEN(case_macro) \
     switch (hidden) { \
         case 2560: case_macro(2560); \
+        case 4096: case_macro(4096); \
         case 5120: case_macro(5120); \
         case 7168: case_macro(7168); \
         default: EP_HOST_ASSERT(false && "Unsupported hidden"); \