From 47dd77ab5f8c9b1f29017a4c8e593a3731b8a857 Mon Sep 17 00:00:00 2001
From: Chenggang Zhao <chenggangz@deepseek.com>
Date: Fri, 20 Jun 2025 14:35:15 +0800
Subject: [PATCH] Add retired flag

---
 csrc/kernels/internode.cu | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/csrc/kernels/internode.cu b/csrc/kernels/internode.cu
index 687af57..9887a58 100644
--- a/csrc/kernels/internode.cu
+++ b/csrc/kernels/internode.cu
@@ -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");