Mojo package

amd

Provides the AMD GPU backend implementations for matmuls.

Modules

amd_target: AMD GPU target definitions for the pipeline scheduling framework.
matmul:
matmul_schedule: Declarative software pipeline schedule for the default AMD matmul kernel.
pingpong_kernel:
pingpong_schedule: Ping-pong schedule for AMD GPU matmul kernels.
pipeline_body: Builder for declarative pipeline body specifications.
ring_buffer: Ring Buffer implementation for producer-consumer synchronization in GPU kernels.
ring_buffer_traits: Trait definitions and utilities for ring buffer synchronization strategies.
structured:
warp_spec_matmul: AMD Warp-Specialized Matrix Multiplication