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

operation_traits

Elementwise operation traits for MAX Graph kernel implementations.

Defines the ElementwiseUnaryOp, ElementwiseBinaryOp, and related traits that structured kernel implementations satisfy so the graph compiler can fuse and dispatch elementwise epilogues.

Traits

• ​ElementwiseBinaryComparisonOp: Requires an elementwise op that compares two SIMD values and yields a boolean mask.
• ​ElementwiseBinaryOp: Requires an elementwise op that combines two SIMD values of the same dtype.
• ​ElementwiseUnaryMixedOp: Requires an elementwise op that produces an output dtype different from its input.
• ​ElementwiseUnaryOp: Requires an elementwise op that transforms one SIMD value into another of the same dtype.

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