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 module

structuring

Provides structured sparsity utilities for scatter-gather operations on GPU.

comptime values

NVIDIASharedMemoryManager

comptime NVIDIASharedMemoryManager = SharedMemoryManager[NVIDIASharedMemoryBasePtr]

Structs

• ​IteratorScatterGatherAmd: Iterator-based AMD scatter-gather for DRAM-register data movement.
• ​NVIDIASharedMemoryBasePtr: NVIDIA implementation of SharedMemoryBasePtr using external memory.
• ​ScatterGatherAmd: AMD tile-based scatter-gather for DRAM-register data movement.
• ​SharedMemoryManager: Manages bump allocation of tiles and arrays from a shared memory base pointer.

Traits

• ​SharedMemoryBasePtr: Defines a base pointer into GPU shared memory with a fixed alignment.

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