Mojo package
grouped_block_scaled
Grouped block-scaled matmul kernel for SM100.
Modules
-
grouped_block_scaled_matmul: CPU entry points for grouped block-scaled SM100 matmul. -
grouped_block_scaled_matmul_kernel: Grouped block-scaled SM100 matmul kernel for multiple GEMM problems. -
grouped_block_scaled_smem: Shared memory layout for grouped block-scaled SM100 matmul. -
grouped_tile_scheduler: Grouped tile scheduler for SM100 structured block-scaled GEMM.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!