Merge pull request #154 from sleepcoo/support-more-hidden

Support hidden size 4096

Merge pull request #154 from sleepcoo/support-more-hidden
Support hidden size 4096
bb393e77 · Chenggang Zhao · GitHub · 05104029 · a107266a · bb393e77
Unverified Commit bb393e77 authored May 12, 2025 by Chenggang Zhao Committed by GitHub May 12, 2025
Show 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"); \