
    -iD                     &   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   )flags)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    b/home/obispo/Crisostomo_bridge/mision_env/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         j                  sy t        |       t        j                  |       } t        j                  |      }t        j
                  t        j                  |j                  | |      t        j                        S r   )
r	   instrumentation_onr   tl_unwrap_if_constexprtensortriton_protoncreate_proton_recordbuildervoidr    r!   r   s      r   recordr,   $   sd    ##h'&&x0H((4J99]778H8H(T^_acahahiir   Nnamec                      t        d| |       y )NTr+   r,   r-   	_semantics     r   enter_scoper2   -   s    
DTI>r   c                      t        d| |       y )NFr+   r/   r0   s     r   
exit_scoper4   2   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-   r1   s      r   __init__zscope.__init__9   s    	!r   c                 F    t        | j                  | j                         y N)r1   )r2   r-   r   )r8   s    r   	__enter__zscope.__enter__=   s    DII7r   c                 F    t        | j                  | j                         y r;   )r4   r-   r   )r8   exc_type	exc_value	tracebacks       r   __exit__zscope.__exit__@   s    4996r   r   )__name__
__module____qualname__strr9   r<   rA    r   r   r6   r6   7   s    "S "87r   r6   r   )triton.languager   r$   triton.language.corer   triton._C.libtritonr   r'   triton.language.semanticr   ,triton.experimental.gluon.language._semanticr   r	   r   r   r   rE   r   r   	constexprr,   r2   r4   r6   rF   r   r   <module>rM      s    & ( 7 3 F   W%&
B23 25C 5jR\\ jr|| j 	?bll ? 	? 	@R\\ @ 	@
7 
7r   