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

shmem_context

SHMEM runtime context and multi-GPU launch helper.

Provides SHMEMContext, which wraps the SHMEM runtime lifecycle and exposes a DeviceContext-compatible interface for submitting kernels and managing symmetric-heap buffers. Use shmem_launch to run a per-GPU function across all attached GPUs.

Structs

• ​SHMEMContext: Usable as a context manager to run kernels on a GPU with SHMEM support, on exit it will finalize SHMEM and clean up resources.

Functions

• ​shmem_launch: Takes a function defining a SHMEM program and launches it on one thread for each GPU you have attached.

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