From 1d7e2f6fb859047d73a0eb013aa9e384f0337063 Mon Sep 17 00:00:00 2001 From: ZeyuanChen2000 <806126546@qq.com> Date: Mon, 1 Jun 2026 15:32:14 +0800 Subject: [PATCH] [NPU] fix normal DeepEP mode num_tokens_per_rdma_rank error caused by none (#22972) --- python/sglang/srt/eplb/expert_distribution.py | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/python/sglang/srt/eplb/expert_distribution.py b/python/sglang/srt/eplb/expert_distribution.py index 17d20fbbd..5e3adcedb 100644 --- a/python/sglang/srt/eplb/expert_distribution.py +++ b/python/sglang/srt/eplb/expert_distribution.py @@ -419,7 +419,11 @@ class _DetailSinglePassGatherer(_SinglePassGatherer): dict( layer_id=layer_idx, num_tokens_per_rank=num_tokens_per_rank.cpu().tolist(), - num_tokens_per_rdma_rank=num_tokens_per_rdma_rank.cpu().tolist(), + num_tokens_per_rdma_rank=( + num_tokens_per_rdma_rank.cpu().tolist() + if num_tokens_per_rdma_rank is not None + else None + ), num_tokens_per_expert=num_tokens_per_expert.cpu().tolist(), ) )