Support more hidden size

7b0c25f8 · Chenggang Zhao · 9d4f7ef8 · 7b0c25f8
Commit 7b0c25f8 authored Jun 20, 2025 by Chenggang Zhao
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
@@ -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); \