triton.experimental.gluon.language.nvidia.hopper.mbarrier.MBarrierLayout

class triton.experimental.gluon.language.nvidia.hopper.mbarrier.MBarrierLayout(self, cga_layout=None)

Layout for mbarrier synchronization in Ampere and later architectures.

Parameters:

cga_layout (List[List[int]]) – CGA layout bases. Defaults to [].

__init__(self, cga_layout=None)

Methods

__init__(self[, cga_layout])

format_hardware_view(self, shape)

format_tensor_view(self, shape)

mangle(self)

Attributes

multicta

Create a multi-CTA mbarrier layout.

type

vec

per_phase

max_phase

order

cga_layout