support hidden size 4096

Co-authored-by: zhyncs <me@zhyncs.com> Co-authored-by: yinfan98 <1106310035@qq.com>

support hidden size 4096
Co-authored-by: zhyncs <me@zhyncs.com> Co-authored-by: yinfan98 <1106310035@qq.com>
a107266a · sleepcoo · 05104029 · a107266a
Commit a107266a authored May 12, 2025 by sleepcoo
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

csrc/kernels/launch.cuh csrc/kernels/launch.cuh +1 -0

No files found.
--- 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"); \