
    ϦiX                     P    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_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   )          @   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   metadatak
fn_metricss       S/var/www/html/ai-image-ml/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                  c3  [        R                  [         l        [        R                  [         l        g g N)r   launch_enter_hookr
   r   r!   launch_exit_hookr$   r#   r   register_triton_hookr2      s-    ''/+5+;+;(*4//' 0r#   c                  v    [         R                  [        R                  :X  a  S [         l        S [         l        g g r/   )r   r0   r
   r   r1   r$   r#   r   unregister_triton_hookr4      s+    '':+;+;;+/(*.' <r#   )r   N)
scoper   r   triton.compilerr   r   r   r
   r2   r4   r$   r#   r   <module>r7      s%    * 48 # #":/r#   