
    Pi                        d dl mZ d dlmZ d dlmZ d dlm	Z	 d dl
mZ ddlmZ e	edZ	 ed	         hZd
 ZdefdZdefdZdej        dej        fdZeddej        fd            Zeddej        fd            Z G d d          ZdS )    )core)builtin)proton)TritonSemantic)GluonSemantic   )get_instrumentation_on)tritongluonr   c                     t          | t          t                              s't          dt	          |            dt                     d S )NzUnsupported semantic type: z. Supported semantics are: )
isinstancetuple
_SEMANTICS	TypeErrortype)semantics    p/var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/triton/profiler/language.py_check_supported_semanticr      sg    hj 1 122 B Ad8nn A A4>A A B B 	BB B    semantic_namec                 P    t                               t          |                     d S N)r   add_ALL_SEMANTICSr   s    r   enable_semanticr      s    NN>-011111r   c                 P    t                               t          |                     d S r   )r   remover   r   s    r   disable_semanticr       s!    n]344444r   is_start
scope_namec                 (   t                      sd S t          |           t          j        |           } t          j        |          }|j                                        }t          j        t          j        || |          t          j	                  S r   )
r	   r   tl_unwrap_if_constexprbuilderget_op_buildertensortriton_protoncreate_proton_recordvoid)r    r!   r   
op_builders       r   recordr,   $   s|    !## h'''&x00H(44J!0022J9]7
HjYY[][bcccr   Nnamec                 *    t          d| |           d S )NTr    r!   r   r,   r-   	_semantics     r   enter_scoper3   .   s    
DTI>>>>>>r   c                 *    t          d| |           d S )NFr/   r0   r1   s     r   
exit_scoper5   3   s    
EdY??????r   c                   (    e Zd ZddefdZd Zd ZdS )scopeNr-   c                 "    || _         || _        d S r   )r-   r   )selfr-   r2   s      r   __init__zscope.__init__:   s    	!r   c                 <    t          | j        | j                   d S N)r2   )r3   r-   r   )r9   s    r   	__enter__zscope.__enter__>   s    DI777777r   c                 <    t          | j        | j                   d S r<   )r5   r-   r   )r9   exc_type	exc_value	tracebacks       r   __exit__zscope.__exit__A   s    49666666r   r   )__name__
__module____qualname__strr:   r=   rB    r   r   r7   r7   8   sR        " "S " " " "8 8 87 7 7 7 7r   r7   r   )triton.languager   r#   triton.language.corer   triton._C.libtritonr   r(   triton.language.semanticr   ,triton.experimental.gluon.language._semanticr   flagsr	   r   r   r   rF   r   r   	constexprr,   r3   r5   r7   rG   r   r   <module>rO      s   & & & & & & ( ( ( ( ( ( 7 7 7 7 7 7 3 3 3 3 3 3 F F F F F F ) ) ) ) ) )   W%&
B B B23 2 2 2 25C 5 5 5 5dR\ dr| d d d d 	? ?bl ? ? ? 	? 	@ @R\ @ @ @ 	@
7 
7 
7 
7 
7 
7 
7 
7 
7 
7r   