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); \