mirror of
https://git.datalinker.icu/vllm-project/vllm.git
synced 2026-05-29 23:17:10 +08:00
minor changes
Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
parent
631be12edb
commit
6e2a3c0841
@ -59,6 +59,9 @@ class All2AllManagerBase:
|
|||||||
# usually the underlying implementation caches the handle
|
# usually the underlying implementation caches the handle
|
||||||
# and reuse it for the same config.
|
# and reuse it for the same config.
|
||||||
raise NotImplementedError
|
raise NotImplementedError
|
||||||
|
|
||||||
|
def get_handles(self, kwargs):
|
||||||
|
raise NotImplementedError
|
||||||
|
|
||||||
def dispatch(self, hidden_states: torch.Tensor,
|
def dispatch(self, hidden_states: torch.Tensor,
|
||||||
router_logits: torch.Tensor):
|
router_logits: torch.Tensor):
|
||||||
|
|||||||
Loading…
x
Reference in New Issue
Block a user