Mojo package

structured_kernels

Shared library components for SM100 structured kernels.

Modules

config: SM100 matmul configuration types and utilities.
epilogue_components: Low-level epilogue components for SM100 matrix multiplication.
output_writer: TileWriter for SM100 matmul output pipeline.
tile_loader: TMA tile loader for SM100 matrix multiplication.
tile_pipeline: Tile pipeline for SM100 producer-consumer synchronization.
tile_scheduler:
tile_scheduler_splitk:
tmem: Tensor Memory (TMEM) abstractions for SM100 Blackwell GPUs.
warp_context: RAII warp context managers for SM100 matmul kernel.