+
    ړLj                     p    ^ RI Ht ^RIHt ^ RIHt Rt ! R R4      t ! R R]4      tR	 R
 lt	R R lt
R# )    )proton)flagswraps__proton_launch_metadatac                   T   a  ] tR t^t o RtV 3R lR ltR tV 3R lR ltR tRt	V t
R	# )
statea:  
A context manager and decorator for entering and exiting a state.

Usage:
    context manager:
    ```python
    with proton.state("test0"):
        foo[1,](x, y)
    ```

    decorator:
    ```python
    @proton.state("test0")
    def foo(x, y):
        ...
    ```

Args:
    name (str): The name of the state.
c                $   < V ^8  d   QhRS[ RR/#    namereturnNstr)format__classdict__s   "A/app/.local/lib/python3.14/site-packages/triton/profiler/state.py__annotate__state.__annotate__   s      S T     c                    Wn         R # Nr   )selfr   s   &&r   __init__state.__init__   s    	r   c                v    \         P                  '       g   V # \        P                  ! V P                  4       V # r   )r   profiling_on	libprotonenter_stater   )r   s   &r   	__enter__state.__enter__!   s*    !!!Kdii(r   c                   < V ^8  d   QhRR/# r   r   N )r   r   s   "r   r   r   '   s      $ r   c                `    \         P                  '       g   R # \        P                  ! 4        R # r   )r   r   r   
exit_state)r   exc_type	exc_value	tracebacks   &&&&r   __exit__state.__exit__'   s    !!!r   c                4   a a \        S4      VV 3R  l4       pV# )c                     < \         P                  '       d!   \        P                  ! SP                  4       S! V / VB p\         P                  '       d   \        P
                  ! 4        V# r   )r   r   r   r    r   r'   )argskwargsretfuncr   s   *, r   wrapperstate.__call__.<locals>.wrapper.   sM    !!!%%dii0''C!!!$$&Jr   r   )r   r2   r3   s   ff r   __call__state.__call__,   s     	t	 
	 r   r   N)__name__
__module____qualname____firstlineno____doc__r   r!   r+   r5   __static_attributes____classdictcell__)r   s   @r   r	   r	      s-     *  
 r   r	   c                   >   a a ] tR t^:t oV3R lV 3R lltRtVtV ;t# )metadata_statec                   < V ^8  d   QhRR/# r$   r%   )r   r   s   "r   r   metadata_state.__annotate__<   s     6 6$ 6r   c                .   < \         SV `  \        4       R # r   )superr   COMPUTE_METADATA_SCOPE_NAME)r   	__class__s   &r   r   metadata_state.__init__<   s    45r   r%   )r7   r8   r9   r:   r   r<   r=   __classcell__)rE   r   s   @@r   r?   r?   :   s     6 6 6r   r?   c                (    V ^8  d   QhR\         RR/# r   r   )r   s   "r   r   r   @   s        c  d  r   c                 2    \         P                  ! V 4       R # r   )r   r    r   s   &r   r    r    @   s    $r   c                    V ^8  d   QhRR/# r$   r%   )r   s   "r   r   r   D   s      D r   c                  0    \         P                  ! 4        R # r   )r   r'   r%   r   r   r'   r'   D   s    r   N)triton._C.libprotonr   r   r   	functoolsr   rD   r	   r?   r    r'   r%   r   r   <module>rN      s5    3  8 / /d6U 6 r   