For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).

Mojo struct

TuningConfigSmallMNGemms

struct TuningConfigSmallMNGemms

Holds launch parameters for small-M, small-N GEMM/GEMV kernels.

Stores the tile dimensions, thread count, unroll factor, K-tile size, and GEMV algorithm kind that select an optimized kernel configuration for matmuls whose M dimension falls in the half-open interval [M, M_end).

Fields

• ​M (Int):
• ​M_end (Int):
• ​N (Int):
• ​K (Int):
• ​kernel_kind (GEMVAlgorithm):
• ​tile_m (Int):
• ​tile_n (Int):
• ​num_threads (Int):
• ​unroll_factor (Int):
• ​tile_k (Int):
• ​swapAB (Bool):

Implemented traits

AnyType, Copyable, ImplicitlyCopyable, ImplicitlyDeletable, Movable, RegisterPassable, TrivialRegisterPassable, TuningConfig, Writable

Methods

__init__

def __init__(M: Int, M_end: Int, N: Int, K: Int, tile_m: Int, tile_n: Int, num_threads: Int, kernel_kind: GEMVAlgorithm, unroll_factor: Int = Int(1), tile_k: Int = Int(128), swapAB: Bool = False) -> Self

write_to

def write_to(self, mut writer: T)

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit