Mojo function
make_swizzle
make_swizzle[num_rows: Int, row_size: Int, access_size: Int]() -> Swizzle
Create a 2D swizzle to avoid bank conflicts.
This function creates a swizzle pattern for a 2D memory layout to minimize bank conflicts when accessing data in shared memory.
Parameters:
- num_rows (
Int): Number of rows in the minimum access pattern. - row_size (
Int): Size of each row in elements. - access_size (
Int): Number of elements accessed at once.
Returns:
A Swizzle object configured to avoid bank conflicts.
make_swizzle[type: DType, mode: TensorMapSwizzle]() -> Swizzle
Create a swizzle functor based on predefined swizzle modes.
This function returns a swizzle pattern based on standard swizzle modes (32B, 64B, 128B, or none) adjusted for the data type.
Parameters:
- type (
DType): The data type of the elements. - mode (
TensorMapSwizzle): The swizzle mode to use.
Returns:
A Swizzle object configured according to the mode.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!