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

types

Core types for the pipeline scheduling framework.

Contains ResourceKind, ScheduleOps, _Ops, KOffsetKind, Phase, OpRole, OpCost, TargetCostModel, OpDesc, DepKind, DepEdge, EdgeRule, and the annotate_ops helper.

Structs

• ​DepEdge: A dependency between two schedule entries.
• ​DepKind: Type of dependency between operations.
• ​EdgeRule: Declarative edge derivation rule.
• ​KOffsetKind: Describes how to compute the K offset for a load operation.
• ​OpCost: Hardware cost annotation for a single operation kind.
• ​OpDesc: Describes a single operation in the pipeline schedule.
• ​OpRole: Role of an operation in the pipeline data flow.
• ​Phase: Which phase of the pipeline an entry belongs to.
• ​ResourceKind: Hardware resource that an operation occupies.
• ​TargetCostModel: Maps op tags to hardware cost annotations.

Traits

• ​ScheduleOps: Base 'enum' for pipeline schedule operations.

Functions

• ​annotate_ops: Apply a target cost model to logical ops.

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