Mojo module

grouped_matmul

Functions

• ​dispatch_amd_matmul_by_block_shape: Dispatches to the best kernel configuration based on runtime M dimension.
• ​get_swizzle_mode:
• ​grouped_matmul: TileTensor primary implementation of grouped_matmul.
• ​grouped_matmul_amd:
• ​grouped_matmul_amd_kernel_launcher:
• ​grouped_matmul_kernel_sm100:
• ​grouped_matmul_sm100:
• ​grouped_matmul_vendor: TileTensor primary implementation of grouped_matmul_vendor.
• ​naive_epilogue:
• ​naive_epilogue_kernel:
• ​naive_grouped_matmul: TileTensor primary implementation of naive_grouped_matmul.
• ​naive_grouped_matmul_kernel:

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit