支持hidden size等于8192

043aa8f7 · lishen · cf4514dc · 043aa8f7
Commit 043aa8f7 authored Feb 04, 2026 by lishen
Show whitespace changes
Inline Side-by-side

Showing with 7 additions and 5 deletions

csrc/kernels/launch.cuh csrc/kernels/launch.cuh +7 -5

No files found.
--- a/csrc/kernels/launch.cuh
+++ b/csrc/kernels/launch.cuh
@@ -121,6 +121,8 @@ inline void LAUNCH_KERNEL_NON_COOPERATIVE(T &&config, Kern &&kernel, Args &&...a
        case_macro(4096);                                                                          \
    case 7168:                                                                                     \
        case_macro(7168);                                                                          \
+    case 8192:                                                                                     \
+        case_macro(8192);                                                                          \
    default:                                                                                       \
        EP_HOST_ASSERT(false and "Unsupported hidden");                                            \
    }                                                                                              \