triton.experimental.gluon.language.nvidia.ampere.mbarrier.wait

triton.experimental.gluon.language.nvidia.ampere.mbarrier.wait(mbarrier, phase, pred=True, deps=(), _semantic=None)

Wait until the mbarrier object completes its current phase.

Parameters:

• mbarrier (shared_memory_descriptor) – The barrier object to wait on.

• phase (int) – The phase index to wait for.

• pred (bool) – Predicate. Operation is skipped if predicate is False. Defaults to True.

• deps (Sequence[shared_memory_descriptor]) – Dependent allocations barrier is waiting on. Used to track liveness of dependent allocations. Defaults to ().