Mojo package

structured_kernels

Shared library components for SM100 structured kernels.

Modules

barriers: Barrier abstractions for SM100 structured matmul kernels.
config: SM100 matmul configuration types and utilities.
epilogue_components: Low-level epilogue components for SM100 matrix multiplication.
kernel_common: Shared kernel components for SM100 warp-specialized matmul kernels.
output_writer: TileWriter for SM100 matmul output pipeline.
pipeline: Producer-consumer pipeline utilities for SM100 structured kernels.
pipeline_storage: Unified Pipeline Storage Framework for SM100 Structured Kernels.
tile_loader: TMA tile loader for SM100 matrix multiplication.
tile_pipeline: Tile pipeline for SM100 producer-consumer synchronization.
tile_scheduler:
tile_scheduler_splitk:
tile_types: Native TileTensor types for SM100 structured kernels.
tmem: Tensor Memory (TMEM) abstractions for SM100 Blackwell GPUs.
warp_context: RAII warp context managers for SM100 matmul kernel.