minor changes

Signed-off-by: Sage Moore <sage@neuralmagic.com>
This commit is contained in:
Sage Moore 2025-07-03 13:29:32 +00:00
parent 631be12edb
commit 6e2a3c0841

View File

@ -60,6 +60,9 @@ class All2AllManagerBase:
# 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):
raise NotImplementedError raise NotImplementedError