
    (Th                     `    S SK JrJr  S SKJrJr  SSKJrJr  Sr	 " S S5      r
SS	 jrSS
 jrg)   )enter_state
exit_state)enter_scope
exit_scope    )CompiledKernelLazyDict__proton_launch_metadatac                       \ 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\SS4S j5       r\S\SS4S	 j5       r	S
r
gs  snn f )
TritonHook   )r          @   flopsbytes	lazy_dictreturnNc                     [        [        5        U R                  5       n[        5         [        R
                   Vs0 s H  o"U;   d  M
  X!U   _M     nn[        US   SUS9  g s  snf )NnameT)	triton_opmetrics)r   COMPUTE_METADATA_SCOPE_NAMEgetr   r   r   r   )r   metadatak
fn_metricss       L/var/www/auris/envauris/lib/python3.13/site-packages/triton/profiler/hook.pyenterTritonHook.enter   sZ    /0==?.8.@.@R.@Mna!n.@
RHV$jI Ss   	A#		A#c                     [        SS9  g )NT)r   )r   )r   s    r   exitTritonHook.exit   s
    T"     )__name__
__module____qualname____firstlineno__flops_widthr   staticmethodr	   r   r"   __static_attributes__).0widths   00r   r   r      sz    !K,78K5ugK8G9DyPGJ Jd J J # #T # # 9s   A
r   Nc                      [         R                  c3  [        R                  [         l        [        R                  [         l        g g N)r   launch_enter_hookr   r   r"   launch_exit_hookr%   r$   r   register_triton_hookr3      s-    ''/+5+;+;(*4//' 0r$   c                  v    [         R                  [        R                  :X  a  S [         l        S [         l        g g r0   )r   r1   r   r   r2   r%   r$   r   unregister_triton_hookr5      s+    '':+;+;;+/(*.' <r$   )r   N)stater   r   scoper   r   triton.compilerr   r	   r   r   r3   r5   r%   r$   r   <module>r9      s(    * * 48 # #":/r$   