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

dispatch

SM100 Conv2D dispatch for the nn conv op.

This module provides a dtype-safe dispatch function that gates SM100 conv2d kernel instantiation to supported dtypes (bf16/fp16) only. Importing this module does NOT trigger kernel compilation -- the kernel is only compiled when dispatch_sm100_conv2d is called with a supported dtype inside a @parameter if guard.

Functions

• ​dispatch_sm100_conv2d: Dispatch to SM100 structured conv2d with filter transpose.
• ​test_alignment_sm100_conv2d: Checks whether the input and output channel counts meet SM100 conv2d alignment requirements.

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