Check the empty list

This commit is contained in:
Chenggang Zhao
2025-06-11 11:14:30 +08:00
parent a8299ca7c2
commit dd13c7145c
2 changed files with 4 additions and 2 deletions

View File

@@ -290,7 +290,8 @@ class Buffer:
recv_topk_idx: received expert indices.
recv_topk_weights: received expert weights.
num_recv_tokens_per_expert_list: Python list shaped `[num_local_experts]`, the received token count by
each local expert, aligned to the input `expert_alignment`.
each local expert, aligned to the input `expert_alignment`. If `num_worst_tokens` is specified, the list
will be empty.
handle: the returned communication handle.
event: the event after executing the kernel (valid only if `async_finish` is set).
"""