Skip to content

Commit b59b7e0

Browse files
committed
address gemini comment
Signed-off-by: Ming Yang <[email protected]>
1 parent 4468db2 commit b59b7e0

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

vllm/model_executor/layers/fused_moe/deepep_ht_prepare_finalize.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,7 @@ def _get_dispatch_config(self) -> Optional[deep_ep.Config]:
5252
return deep_ep.Buffer.get_dispatch_config(self.dp_size)
5353

5454
def _get_combine_config(self) -> Optional[deep_ep.Config]:
55-
if self.dp_size not in self.available_rank_configs:
55+
if self.num_dispatchers_ not in self.available_rank_configs:
5656
return None
5757
return deep_ep.Buffer.get_combine_config(self.num_dispatchers_)
5858

0 commit comments

Comments
 (0)