`i;nddlmZmZmZmZddlmZmZgdZed dZ edddd d Z dS) )MBarrierLayoutinit invalidatewait)_unwrap_if_constexprbuiltin)arriveexpectrrrrTNct|}||}|j|j||jdS)am Expect a specific number of bytes being copied. When they are copied, the barrier is signaled. Args: mbarrier (shared_memory_descriptor): Barrier that will be signaled when the operation is complete. bytes (int): Expected byte count. pred (bool): Scalar predicate. Operation is skipped if predicate is False. Defaults to True. N)r to_tensorbuildercreate_mbarrier_expecthandle)mbarrierbytespred _semantics /home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/triton/experimental/gluon/language/nvidia/hopper/mbarrier.pyr r I ! ' 'E   t $ $D ,,X_eT[QQQQQ)countrrct|}||}|j|j||jdS)a' Arrive at an mbarrier with a specified count. Args: mbarrier (shared_memory_descriptor): Barrier to be signalled. count (int): Count to arrive with. Defaults to 1. pred (bool): Scalar predicate. Operation is skipped if predicate is False. Defaults to True. N)rr rcreate_mbarrier_arriver)rrrrs rr r rr)TN) ampere.mbarrierrrrr_corerr __all__r r rrr sDDDDDDDDDDDD22222222 N N N  R R R  R dd R R R R  R R Rr