IMPORTANT: To view this page as Markdown, append `.md` to the URL (e.g. /docs/manual/basics.md). For the complete Mojo documentation index, see llms.txt.
Skip to main content
Version: Nightly
For the complete Mojo documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /docs/manual/basics.md).

select_k_atom

select_k_atom[dtype: 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:

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

Returns:

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