Mojo module
moe
Functions
-
moe_create_indices
: -
moe_create_indices_bucket_sort_kernel
: The main goal of this kernel is to group tokens that use the same expert together. This allows for efficent batching when used by other kernels such as grouped matmul. -
moe_create_indices_kernel
:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!