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

triton.experimental.gluon.language.nvidia.hopper.mbarrier.expect(mbarrier, bytes_per_cta=None, pred=True, _semantic=None)

Expect a specific number of bytes being copied. When they are copied, the barrier is signaled.

Parameters:
  • mbarrier (shared_memory_descriptor) – Barrier that will be signaled when the operation is complete.

  • bytes_per_cta (int) – Expected byte count per CTA.

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