Skip to content

vllm_omni.core.sched.utils

Shared utilities for omni schedulers.

omni_routed_experts_for_request

omni_routed_experts_for_request(
    routed_experts: RoutedExpertsLists, request
) -> ndarray | None

Extract per-request routed experts from RoutedExpertsLists using slot_mapping.

Matches upstream RoutedExpertsManager.get() pattern — filters routing_data rows whose slot_mapping entries belong to this request's block_table.