§
    Pi;  ã                   ón   — d dl mZmZmZmZ ddlmZmZ g d¢Zedd„¦   «         Z	edddd	œd
„¦   «         Z
dS )é   )ÚMBarrierLayoutÚinitÚ
invalidateÚwaité   )Ú_unwrap_if_constexprÚbuiltin)ÚarriveÚexpectr   r   r   r   TNc                 óš   — t          |¦  «        }|                     |¦  «        }|j                             | j        ||j        ¦  «         dS )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_tensorÚbuilderÚcreate_mbarrier_expectÚhandle)ÚmbarrierÚbytesÚpredÚ	_semantics       ú‘/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/triton/experimental/gluon/language/nvidia/hopper/mbarrier.pyr   r      óI   € õ ! Ñ'Ô'€EØ×Ò˜tÑ$Ô$€DØÔ×,Ò,¨X¬_¸eÀTÄ[ÑQÔQÐQÐQÐQó    é   )Úcountr   r   c                óš   — t          |¦  «        }|                     |¦  «        }|j                             | j        ||j        ¦  «         dS )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)r   r   r   Úcreate_mbarrier_arriver   )r   r   r   r   s       r   r
   r
      r   r   )TN)Úampere.mbarrierr   r   r   r   Ú_corer   r	   Ú__all__r   r
   © r   r   ú<module>r       s­   ðØ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DÐ DØ 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2Ð 2à
NÐ
NÐ
N€ð 	ðRð Rð Rñ 	„ðRð 	Ø d°dð Rð Rð Rð Rñ 	„ðRð Rð Rr   