For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /get-started.md).
Mojo module
dispatch
RDNA dispatch for 2-D convolution.
Provides two paths for 2-D convolution on RDNA 3+:
-
Implicit GEMM (preferred): Fuses im2col into the WMMA matmul kernel's A-tile loader, eliminating the large intermediate buffer. Requires C_in % BLOCK_K == 0 for vectorized im2col loads.
-
Explicit im2col + matmul (fallback): Materializes the im2col buffer then calls the standard RDNA matmul. Used when C_in alignment requirements aren't met.
Functions
-
dispatch_rdna_conv2d: Try to dispatch Conv2D on RDNA via implicit GEMM (im2col fused into WMMA).