
    c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y)    )core)builtin)proton)TritonSemantic)GluonSemantic   )get_instrumentation_on)tritongluonr   c                 r    t        | t        t                    st        dt	        |        dt               y )NzUnsupported semantic type: z. Supported semantics are: )
isinstancetuple
_SEMANTICS	TypeErrortype)semantics    R/var/www/html/engine/venv/lib/python3.12/site-packages/triton/profiler/language.py_check_supported_semanticr      sB    hj 125d8n5E F44><A B 	B 3    semantic_namec                 <    t         j                  t        |           y N)r   add_ALL_SEMANTICSr   s    r   enable_semanticr      s    NN>-01r   c                 <    t         j                  t        |           y r   )r   remover   r   s    r   disable_semanticr       s    n]34r   is_start
scope_namec                 (   t               sy t        |       t        j                  |       } t        j                  |      }|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,   $   sn    !#h'&&x0H((4J!!002J99]77
HjY[][b[bccr   Nnamec                      t        d| |       y )NTr    r!   r   r,   r-   	_semantics     r   enter_scoper3   .   s    
DTI>r   c                      t        d| |       y )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y)scopeNr-   c                      || _         || _        y r   )r-   r   )selfr-   r2   s      r   __init__zscope.__init__:   s    	!r   c                 F    t        | j                  | j                         y N)r2   )r3   r-   r   )r9   s    r   	__enter__zscope.__enter__>   s    DII7r   c                 F    t        | j                  | j                         y r<   )r5   r-   r   )r9   exc_type	exc_value	tracebacks       r   __exit__zscope.__exit__A   s    4996r   r   )__name__
__module____qualname__strr:   r=   rB    r   r   r7   r7   8   s    "S "8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 3 F )  W%&
B23 25C 5dR\\ dr|| d 	?bll ? 	? 	@R\\ @ 	@
7 
7r   