triton.experimental.gluon.language.SwizzledSharedLayout

class triton.experimental.gluon.language.SwizzledSharedLayout(self, vec: int, per_phase: int, max_phase: int, order: ~typing.List[int], cga_layout: ~typing.List[~typing.List[int]] = <factory>)

Represents a generic swizzled shared memory layout.

Parameters:

• vec (int) – Vector width for swizzling.

• per_phase (int) – Elements per swizzle phase.

• max_phase (int) – Maximum number of swizzle phases.

• order (List[int]) – Dimension ordering for swizzling.

• cga_layout (Optional[List[List[int]]]) – Bases describing CTA tiling.

__init__(self, vec: int, per_phase: int, max_phase: int, order: ~typing.List[int], cga_layout: ~typing.List[~typing.List[int]] = <factory>) → None

Methods

__init__(self, vec, per_phase, max_phase, ...)
format_hardware_view(self, shape)
format_tensor_view(self, shape)
mangle(self)

Attributes

type
vec
per_phase
max_phase
order
cga_layout