integrations.expert_parallel.experts_fn
integrations.expert_parallel.experts_fn
DeepEP-backed registered functions for ALL_EXPERTS_FUNCTIONS.
Four names registered (eager / grouped_mm / scattermoe / sonicmoe) sharing one
_deep_ep_forward body. Templates ported from bench_deep_ep.py Stage 1
modes 3 and 4.
Functions
| Name | Description |
|---|---|
| kernel_to_registered_name | Map expert_parallel_local_kernel -> registered name. |
| register_all | Register the four names in ALL_EXPERTS_FUNCTIONS and whitelist them. |
kernel_to_registered_name
integrations.expert_parallel.experts_fn.kernel_to_registered_name(kernel)Map expert_parallel_local_kernel -> registered name.
register_all
integrations.expert_parallel.experts_fn.register_all()Register the four names in ALL_EXPERTS_FUNCTIONS and whitelist them.