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

Fix bugs

Browse Source
This commit is contained in:
Chenggang Zhao
2025-06-20 13:44:49 +08:00
parent 8da790e3f3
commit 371df2da52
2 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
csrc/kernels/internode.cu
View File

@@ -793,7 +793,7 @@ dispatch(int4* recv_x, float* recv_x_scales, int64_t* recv_topk_idx, float* recv
if (lane_id == 0) {
nvshmemi_ibgda_amo_nonfetch_add(rdma_channel_head.buffer(rdma_rank),
cached_rdma_channel_tail - cached_rdma_channel_head,
translate_dst_rdma_rank<kLowLatencyMode>(lane_id, nvl_rank),
translate_dst_rdma_rank<kLowLatencyMode>(src_rdma_rank, nvl_rank),
channel_id + num_channels, src_rdma_rank == rdma_rank);
cached_rdma_channel_head = cached_rdma_channel_tail;
}