DeepSeek/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP synced 2025-06-26 18:28:11 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Support more hidden size

Browse Source
This commit is contained in:
Chenggang Zhao
2025-06-20 16:37:28 +08:00
parent 9d4f7ef8ee
commit 7b0c25f864
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
csrc/kernels/launch.cuh
View File

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