triton.experimental.gluon.language.nvidia.ampere.mbarrier.MBarrierLayout
- class triton.experimental.gluon.language.nvidia.ampere.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
multictaCreate a multi-CTA mbarrier layout.
typevecper_phasemax_phaseordercga_layout