Mojo package
extensibility
Surface needed to write GraphCompiler kernel entry points.
Provides the @register / @register_internal decorators, the
ManagedTensorSlice type and supporting tensor specs/IO enums, plus the
elementwise operation traits.
For registering custom operations, use the @compiler.register decorator instead.
Modules
-
decorators: Decorators for registering MAX Graph kernels. -
managed_tensor_slice: ManagedTensorSlice and the types kernel authors compose against. -
operation_traits:
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!