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