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

Add retired flag

Browse Source
This commit is contained in:
Chenggang Zhao
2025-06-20 14:35:15 +08:00
parent 74afd75df2
commit 47dd77ab5f
1 changed files with 4 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
csrc/kernels/internode.cu
View File

@@ -799,6 +799,10 @@ dispatch(int4* recv_x, float* recv_x_scales, int64_t* recv_topk_idx, float* recv
}
__syncwarp();
}
// Retire
if (lane_id == 0)
forward_channel_retired[src_rdma_rank] = 1;
} else if (warp_role == WarpRole::kForwarderCoordinator) {
// Forward warp coordinator
EP_STATIC_ASSERT(kNumRDMARanks <= 32, "Invalid number of RDMA peers");