DeepSeek/DeepEP
1
0
Fork 0
mirror of https://github.com/deepseek-ai/DeepEP synced 2025-05-03 11:41:13 +00:00
Code Issues Actions Packages Projects Releases Wiki Activity

Fix compilation

Browse Source
This commit is contained in:
Chenggang Zhao 2025-03-28 16:45:10 +08:00
parent dcf46f1c26
commit c4d12b4f8f
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/kernels/internode.cu
View File

@ -428,7 +428,7 @@ void notify_dispatch(const int* num_tokens_per_rank, int* moe_recv_counter_mappe
buffer_ptrs, task_fifo_ptrs, head, rank, \ buffer_ptrs, task_fifo_ptrs, head, rank, \
cpu_rdma_team); } break cpu_rdma_team); } break
constexpr int kNumThreads = std::max(256, 32 * num_channels); constexpr int kNumThreads = 512;
const auto num_rdma_ranks = num_ranks / NUM_MAX_NVL_PEERS; const auto num_rdma_ranks = num_ranks / NUM_MAX_NVL_PEERS;
// Get clean meta // Get clean meta