From 4e2abe99b7e2d6daa048d0c48f58d38070119191 Mon Sep 17 00:00:00 2001 From: Zhuohan Li Date: Sun, 19 Oct 2025 23:31:03 -0700 Subject: [PATCH] minor fix Signed-off-by: Zhuohan Li --- vllm/model_executor/layers/moe/ep_kernels_no_abstraction.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/vllm/model_executor/layers/moe/ep_kernels_no_abstraction.py b/vllm/model_executor/layers/moe/ep_kernels_no_abstraction.py index 0157a2ca7d28d..28a391da99789 100644 --- a/vllm/model_executor/layers/moe/ep_kernels_no_abstraction.py +++ b/vllm/model_executor/layers/moe/ep_kernels_no_abstraction.py @@ -183,11 +183,11 @@ if __name__ == "__main__": all_topk_idx = torch.cat(all_topk_idx, dim=0) all_topk_weights = torch.cat(all_topk_weights, dim=0) + # Verification expert_range_start = rank * local_num_experts expert_range_end = (rank + 1) * local_num_experts recv_i = -1 expert_inputs_ground_truth = [] - # Verification for i in range(batch_size): activated_on_this_rank = False for j in range(topk):