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

decorators

Decorators for registering MAX Graph kernels.

@register is the public decorator for DPS-style custom kernels. @register_internal is used by built-in MAX Graph operations.

Functions

• ​register: Registers a struct as a kernel implementation for a MAX Graph op.
• ​register_internal: Registers the given mojo function as an implementation of an mo op or a mo.custom op. Used by built-in MAX Graph operations.
• ​register_shape_function: Registers a standalone shape function for a MAX Graph op.
• ​view_kernel: Marks a DPS kernel as a view operation that aliases its input memory.