Ë
    ÏÍ:j†  ã                   ód   — d dl mZmZ d dlmZmZ ddlmZ ddlm	Z	 dZ
 G d„ d«      Zdd
„Zdd„Zy	)é   )Úenter_stateÚ
exit_state)Úenter_scopeÚ
exit_scopeé    )Úknobs)ÚLazyDictÚ__proton_launch_metadatac                   óŽ   — e Zd Zg d¢ZeD � �cg c]  }d|› �‘Œ	 c}} dgz   dgz   Zededdfd„«       Zededdfd„«       Zyc c}} w )	Ú
TritonHook)é   é   é    é@   ÚflopsÚbytesÚ	lazy_dictÚreturnNc                 óÎ   — t        t        «       | j                  «       }t        «        t        j
                  D �ci c]  }||v sŒ|||   “Œ }}t        |d   d|¬«       y c c}w )NÚnameT)Ú	triton_opÚmetrics)r   ÚCOMPUTE_METADATA_SCOPE_NAMEÚgetr   r   r   r   )r   ÚmetadataÚkÚ
fn_metricss       úi/home/mcse/projects/srt_converter/srt-converter-venv/lib/python3.12/site-packages/triton/profiler/hook.pyÚenterzTritonHook.enter   sY   € äÔ/Ô0Ø—=‘=“?ˆÜŒÜ.8×.@Ñ.@ÖR¨ÀAÈÂM�a˜ !™‘nÐRˆ
ÐRÜ�H˜VÑ$°¸jÖIùò Ss   ¼	A"ÁA"c                 ó   — t        d¬«       y )NT)r   )r   )r   s    r   ÚexitzTritonHook.exit   s
   € ä˜TÖ"ó    )	Ú__name__Ú
__module__Ú__qualname__Úflops_widthr   Ústaticmethodr	   r   r!   )Ú.0Úwidths   00r   r   r   	   sv   „ Ú!€KØ,7×8 5��u�gŠÓ8¸G¸9ÑDÈÀyÑP€GàðJ˜ð J dò Jó ðJð ð#˜ð # Tò #ó ñ#ùó 9s   ŽAr   Nc                  óÆ   — t         j                  j                  €Gt        j                  t         j                  _        t        j
                  t         j                  _        y y ©N)r   ÚruntimeÚlaunch_enter_hookr   r   r!   Úlaunch_exit_hook© r"   r   Úregister_triton_hookr0      s9   € Ü‡}�}×&Ñ&Ð.Ü*4×*:Ñ*:Œ�‰Ô'Ü)3¯©Œ�‰Õ&ð /r"   c                  ó°   — t         j                  j                  t        j                  k(  r+d t         j                  _        d t         j                  _        y y r+   )r   r,   r-   r   r   r.   r/   r"   r   Úunregister_triton_hookr2       s7   € Ü‡}�}×&Ñ&¬*×*:Ñ*:Ò:Ø*.Œ�‰Ô'Ø)-Œ�‰Õ&ð ;r"   )r   N)Ústater   r   Úscoper   r   Útritonr   Útriton.compilerr	   r   r   r0   r2   r/   r"   r   ú<module>r7      s+   ðß *ß *Ý Ý $à8Ð ÷#ñ #ó"9ô.r"   