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 module

mma_nvidia_sm100

This module includes utilities for working with the SM100 MMA instructions.

Structs

MMASmemDescriptor: Descriptor for shared memory operands tcgen05 mma instructions.
MMASmemDescriptorPair: Descriptor for shared memory operands tcgen05 mma instructions.
UMMAInsDescriptor: Descriptor for UMMA instructions.
UMMAKind: Struct for UMMA instruction types.

Functions

mma: Perform a matrix multiply-accumulate operation using the tcgen05.mma instruction.
mma_arrive: Arrive at the mbar pointer for the MMA instruction.
mma_arrive_multicast: Arrive at the mbar pointer for the MMA instruction for multiple ctas.

Structs
Functions