IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /get-started.md). For the complete documentation index, see llms.txt.
Skip to main content
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 trait

WeightLoader

B/weight-side comptime policy for the shared Apple GEMM body.

Carries the B operand dtype (b_type), the MMA accumulator dtype (accum_type; fp32 for dense/fp8/fp4, int32 for a later int8 slice), and the SMEM-staging policy (needs_smem / smem_elems). The A loader still issues the MMA; this policy tells the body how to source and size the B side.

Implemented traits

AnyType

comptime members

accum_type

comptime accum_type

b_type

comptime b_type

needs_smem

comptime needs_smem

smem_elems

comptime smem_elems