Mojo package

layout

Modules

• ​element: Provides element-based access to memory using layout-driven vectorization.
• ​int_tuple: Hierarchical integer tuple data structures for high-performance tensor operations.
• ​layout: Provides a high-performance tensor layout system for memory mapping and indexing.
• ​layout_tensor:
• ​math:
• ​nd_buffer_stub:
• ​runtime_layout: Provides the RuntimeLayout type and functions for working with it. You can use RuntimeLayout to define a layout where the dimensions are not known at compile time.
• ​runtime_tuple:
• ​swizzle: Defines swizzle layouts for optimizing memory access patterns, particularly in shared memory for GPU kernels, to mitigate bank conflicts.
• ​tensor_builder:
• ​tensor_core: This module provides abstractions for using Tensor Cores do to arithmetic and matrix operations.
• ​tensor_core_async: This module provides abstractions for using Async Tensor Cores to perform asynchronous matrix multiplication operations.
• ​tma_async: Tensor Memory Accelerator (TMA) Asynchronous Operations Module

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