File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed
vllm/model_executor/layers/fused_moe Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -49,14 +49,14 @@ def topk_indices_dtype(self) -> Optional[torch.dtype]:
4949 return torch .int64
5050
5151 def _get_dispatch_config (self ) -> Optional [deep_ep .Config ]:
52- if self .dp_size not in self .available_rank_configs :
52+ if self .num_dispatchers_ not in self .available_rank_configs :
5353 return None
54- return deep_ep .Buffer .get_dispatch_config (self .dp_size )
54+ return deep_ep .Buffer .get_dispatch_config (self .num_dispatchers_ )
5555
5656 def _get_combine_config (self ) -> Optional [deep_ep .Config ]:
57- if self .dp_size not in self .available_rank_configs :
57+ if self .num_dispatchers_ not in self .available_rank_configs :
5858 return None
59- return deep_ep .Buffer .get_combine_config (self .dp_size )
59+ return deep_ep .Buffer .get_combine_config (self .num_dispatchers_ )
6060
6161 def _do_dispatch (
6262 self ,
You can’t perform that action at this time.
0 commit comments