[NPU] fix normal DeepEP mode num_tokens_per_rdma_rank error caused by none (#22972)

2026-06-01 15:32:14 +08:00
parent 20f47cfe8e
commit 1d7e2f6fb8
1 changed files with 5 additions and 1 deletions
@@ -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(),
            )
        )