Mojo struct
Swizzle
@register_passable(trivial)
struct Swizzle
A swizzle functor for memory access pattern optimization.
This struct implements a swizzling pattern to reduce bank conflicts in shared memory accesses by XORing specific bits of memory indices.
Aliases
has_shape = False: Indicates whether the layout has a valid shape. This is always False.
Fields
- bits (
Int): Number of bits in the mask. - base (
Int): Number of least significant bits to keep constant. - shift (
Int): Distance to shift the mask (positive shifts right, negative left). - yyy_mask (
Int): Mask for the bits to be shifted. - zzz_mask (
Int): Mask for the target bits.
Implemented traits
AnyType,
CollectionElement,
Copyable,
ExplicitlyCopyable,
LayoutTrait,
Movable,
Stringable,
UnknownDestructibility,
Writable
Methods
__init__
__init__(bits: Int, base: Int, shift: Int) -> Self
Initialize a Swizzle object.
Args:
- bits (
Int): Number of bits in the mask. - base (
Int): Number of least significant bits to keep constant. - shift (
Int): Distance to shift the mask.
__call__
__call__(self, index: IntTuple[origin]) -> Int
Apply swizzle to an IntTuple index.
Args:
- index (
IntTuple[origin]): The index to swizzle.
Returns:
The swizzled index value.
__call__(self, offset: Int) -> Int
Apply swizzle to an integer offset.
Args:
- offset (
Int): The offset to swizzle.
Returns:
The swizzled offset value.
__call__(self, offset: SIMD[type, 1]) -> SIMD[type, 1]
Apply swizzle to a scalar offset.
Args:
- offset (
SIMD[type, 1]): The scalar offset to swizzle.
Returns:
The swizzled scalar value.
size
size(self) -> Int
Get the size of the swizzle pattern.
Returns:
The size of the swizzle pattern.
cosize
cosize(self) -> Int
Get the cosize of the swizzle pattern.
Returns:
The cosize, which is the same as size for swizzle.
write_to
write_to[W: Writer](self, mut writer: W)
Write the swizzle to a writer.
Args:
- writer (
W): The writer to write to.
__str__
__str__(self) -> String
Convert the swizzle to a string.
Returns:
String representation of the swizzle.
Was this page helpful?
Thank you! We'll create more content like this.
Thank you for helping us improve!