Skip to content

Commit aebacdc

Browse files
committed
pplx cleanup
Signed-off-by: Sage Moore <[email protected]>
1 parent c1c003f commit aebacdc

File tree

1 file changed

+5
-7
lines changed

1 file changed

+5
-7
lines changed

vllm/model_executor/layers/fused_moe/pplx_prepare_finalize.py

Lines changed: 5 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -249,10 +249,8 @@ def finalize(
249249
if apply_router_weight_on_input:
250250
topk_weights = torch.ones_like(topk_weights)
251251

252-
self.a2a.combine(
253-
out_tokens=output,
254-
indices=topk_ids.view(dtype=torch.uint32),
255-
weights=topk_weights,
256-
expert_y=fused_expert_output,
257-
bound_m=bound_m,
258-
)
252+
self.a2a.combine(out_tokens=output,
253+
indices=topk_ids.view(dtype=torch.uint32),
254+
weights=topk_weights,
255+
expert_y=fused_expert_output,
256+
bound_m=bound_m)

0 commit comments

Comments
 (0)