
    9i                     z    S SK JrJr  SSKJr  SSKJr  SSKJr	  SSK
Jr  Sr\" S	S
S9r\" SS
S9r " S S\5      rg
)   )enter_state
exit_state    )LazyDict   )Hook)proton)
ContextVar__proton_launch_metadataop_nameN)defaultidc                      ^ \ rS rSrSrSr/ SQr\ V Vs/ s H  nSU 3PM
     snn S/-   S/-   rS rU4S jr	S	\
S
\S\
SS4S jrS rS rS\SS4S jrS\SS4S jrSrU=r$ s  snn f )
LaunchHook   d   N)          @   flopsbytesc                     g N selfs    \/var/www/html/land-doc-ocr/venv/lib/python3.13/site-packages/triton/profiler/hooks/launch.py__init__LaunchHook.__init__           c                 f   > U R                   c  [        [        U ]  U 5      U l         U R                   $ r   )	_instancesuperr   __new__)cls	__class__s    r   r&   LaunchHook.__new__   s*    == !*c:3?CM}}r"   namemetadata_grouphashreturnc                     g r   r   )r   modulefunctionr*   r+   r,   s         r   init_handleLaunchHook.init_handle   r!   r"   c                     g r   r   r   s    r   activateLaunchHook.activate    r!   r"   c                     g r   r   r   s    r   
deactivateLaunchHook.deactivate#   r!   r"   metadatac                    [        [        5        UR                  5       n[        5         [        R
                   Vs0 s H  o3U;   d  M
  X2U   _M     nn[        R                  US   5        [        R                  [        R                  " 5       5        [        R                  " [        R                  5       US   5        [        R                  " [        R                  5       U5        g s  snf )Nr*   )r   COMPUTE_METADATA_SCOPE_NAMEgetr   r   metricsr   setr   	libprotonrecord_scopeenter_opadd_metrics)r   r9   lazy_metadatak
fn_metricss        r   enterLaunchHook.enter&   s    /0 3=3E3E\3EamI[)aq))3E
\M&)*
y%%'(2668]6%:;bffh
3	 ]s   	C*		C*c                 z    [         R                  " [        R                  5       [        R                  5       5        g r   )r?   exit_opr   r<   r   )r   r9   s     r   exitLaunchHook.exit0   s    "&&(GKKM2r"   r   )__name__
__module____qualname____firstlineno__priorityr$   flops_widthr=   r   r&   strdictr1   r4   r7   r   rF   rJ   __static_attributes____classcell__).0widthr(   s   00@r   r   r      s    HI!K,78K5ugK8G9DyPG
# t SV [_ 4h 44 43X 3$ 3 3; 9s   A$r   )stater   r   triton.compilerr   hookr   triton._C.libprotonr	   r?   contextvarsr
   r;   r   r   r   r   r"   r   <module>r]      s>    + $  3 "8 
Y
-d#$3 $3r"   