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

@ -59,6 +59,9 @@ class All2AllManagerBase:
# usually the underlying implementation caches the handle
# and reuse it for the same config.
raise NotImplementedError
def get_handles(self, kwargs):
raise NotImplementedError
def dispatch(self, hidden_states: torch.Tensor,
router_logits: torch.Tensor):