20 lines
390 B
Python
20 lines
390 B
Python
from .moe_align_block_size import moe_align_block_size as _moe_align_block_size
|
|
|
|
|
|
def moe_align_block_size(
|
|
topk_ids,
|
|
num_experts,
|
|
block_size,
|
|
sorted_token_ids,
|
|
experts_ids,
|
|
num_tokens_post_pad,
|
|
):
|
|
_moe_align_block_size(
|
|
topk_ids,
|
|
num_experts,
|
|
block_size,
|
|
sorted_token_ids,
|
|
experts_ids,
|
|
num_tokens_post_pad,
|
|
)
|