
    h                     h    S SK JrJr  S SKJrJr  SSKJr  SSK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    )knobs)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	   )          @   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       N/var/www/fran/franai/venv/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                  R                  cG  [        R                  [         R                  l        [        R
                  [         R                  l        g g N)r   runtimelaunch_enter_hookr   r    r#   launch_exit_hookr&   r%   r   register_triton_hookr5      s9    }}&&.*4*:*:')3& /r%   c                      [         R                  R                  [        R                  :X  a+  S [         R                  l        S [         R                  l        g g r1   )r   r2   r3   r   r    r4   r&   r%   r   unregister_triton_hookr7       s7    }}&&**:*::*.')-& ;r%   )r   N)stater   r   scoper   r   tritonr   triton.compilerr	   r   r   r5   r7   r&   r%   r   <module>r<      s+    * *  $8 # #"9.r%   