Mojo function

select_k_atom

select_k_atom[type: DType, swizzle_mode: TensorMapSwizzle]() -> Layout

Creates a core matrix layout for tensor core operations.

Constructs the fundamental atomic layout for tensor core operations based on the specified data type and swizzle mode. This layout represents the minimal dense matrix structure that can be efficiently processed by tensor cores.

Parameters:

• ​type (DType): Element data type of the tensor.
• ​swizzle_mode (TensorMapSwizzle): Memory access pattern swizzling mode.

Returns:

Layout - A core matrix layout optimized for tensor core operations.

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