
    ci                         d dl mZmZ ddlmZ ddlmZ ddlmZ	 ddl
mZ dZ ed	d
      Z edd
      Z G d de      Zy
)   )enter_state
exit_state    )LazyDict   )Hook)proton)
ContextVar__proton_launch_metadataop_nameN)defaultidc                       e Zd ZdZdZg dZeD  cg c]  }d| 	 c}} dgz   dgz   Zd ZfdZde	d	e
d
e	ddfdZd Zd ZdeddfdZdeddfdZxZS c c}} w )
LaunchHookd   N)          @   flopsbytesc                      y N selfs    V/var/www/html/engine/venv/lib/python3.12/site-packages/triton/profiler/hooks/launch.py__init__zLaunchHook.__init__           c                 d    | j                   t        t        |   |       | _         | j                   S r   )	_instancesuperr   __new__)cls	__class__s    r   r$   zLaunchHook.__new__   s*    == !*c:3?CM}}r    namemetadata_grouphashreturnc                      y r   r   )r   modulefunctionr'   r(   r)   s         r   init_handlezLaunchHook.init_handle   r   r    c                      y r   r   r   s    r   activatezLaunchHook.activate    r   r    c                      y r   r   r   s    r   
deactivatezLaunchHook.deactivate#   r   r    metadatac                    t        t               |j                         }t                t        j
                  D ci c]  }||v s|||    }}t        j                  |d          t        j                  t        j                                t        j                  t        j                         |d          t        j                  t        j                         |       y c c}w )Nr'   )r   COMPUTE_METADATA_SCOPE_NAMEgetr   r   metricsr   setr   	libprotonrecord_scopeenter_opadd_metrics)r   r3   lazy_metadatak
fn_metricss        r   enterzLaunchHook.enter&   s    /0 3=3E3E\amI[aq))\
\M&)*
y%%'(2668]6%:;bffh
3	 ]s   	C#C#c                 x    t        j                  t        j                         t        j                                y r   )r9   exit_opr   r6   r   )r   r3   s     r   exitzLaunchHook.exit0   s    "&&(GKKM2r    )__name__
__module____qualname__priorityr"   flops_widthr7   r   r$   strdictr.   r0   r2   r   r@   rC   __classcell__).0widthr&   s   00@r   r   r      s    HI!K,785ug8G9DyPG
# t SV [_ 4h 44 43X 3$ 3; 9s   Ar   )stater   r   triton.compilerr   hookr   triton._C.libprotonr	   r9   contextvarsr
   r5   r   r   r   r   r    r   <module>rS      s>    + $  3 "8 
Y
-d#$3 $3r    