§
    9˜ºh†  ã                   ól   — 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d
S )é   )Úenter_stateÚ
exit_state)Úenter_scopeÚ
exit_scopeé    )Úknobs)ÚLazyDictÚ__proton_launch_metadatac                   ó~   — e Zd Zg d¢Zd„ eD ¦   «         dgz   dgz   Zededdfd„¦   «         Zededdfd	„¦   «         ZdS )
Ú
TritonHook)é   é   é    é@   c                 ó   — g | ]}d |› �‘ŒS )Úflops© )Ú.0Úwidths     ú\/var/www/html/services/rag-service/venv/lib/python3.11/site-packages/triton/profiler/hook.pyú
<listcomp>zTritonHook.<listcomp>   s   € Ð8Ð8Ð8 5ˆ�uˆˆÐ8Ð8Ð8ó    Úbytesr   Ú	lazy_dictÚreturnNc                 óÔ   ‡— t          t          ¦  «         |                      ¦   «         Št          ¦   «          ˆfd„t          j        D ¦   «         }t          ‰d         d|¬¦  «         d S )Nc                 ó*   •— i | ]}|‰v ¯|‰|         “ŒS r   r   )r   ÚkÚmetadatas     €r   ú
<dictcomp>z$TritonHook.enter.<locals>.<dictcomp>   s$   ø€ ÐRÐRÐR¨ÀAÈÀMÀM�a˜ !œÀMÀMÀMr   ÚnameT)Ú	triton_opÚmetrics)r   ÚCOMPUTE_METADATA_SCOPE_NAMEÚgetr   r   r#   r   )r   Ú
fn_metricsr   s     @r   ÚenterzTritonHook.enter   sd   ø€ åÕ/Ñ0Ô0Ð0Ø—=’=‘?”?ˆÝ‰ŒˆØRÐRÐRÐR­jÔ.@ÐRÑRÔRˆ
Ý�H˜VÔ$°¸jÐIÑIÔIÐIÐIÐIr   c                 ó&   — t          d¬¦  «         d S )NT)r"   )r   )r   s    r   ÚexitzTritonHook.exit   s   € å˜TÐ"Ñ"Ô"Ð"Ð"Ð"r   )	Ú__name__Ú
__module__Ú__qualname__Úflops_widthr#   Ústaticmethodr	   r'   r)   r   r   r   r   r   	   sž   € € € € € Ø!�/�/€KØ8Ð8¨KÐ8Ñ8Ô8¸G¸9ÑDÈÀyÑP€GàðJ˜ð J dð Jð Jð Jñ „\ðJð ð#˜ð # Tð #ð #ð #ñ „\ð#ð #ð #r   r   r   Nc                  ó˜   — t           j        j        €8t          j        t           j        _        t          j        t           j        _        d S d S ©N)r   ÚruntimeÚlaunch_enter_hookr   r'   r)   Úlaunch_exit_hookr   r   r   Úregister_triton_hookr4      s4   € Ý„}Ô&Ð.Ý*4Ô*:�ŒÔ'Ý)3¬�ŒÔ&Ð&Ð&ð /Ð.r   c                  óŒ   — t           j        j        t          j        k    r$d t           j        _        d t           j        _        d S d S r0   )r   r1   r2   r   r'   r3   r   r   r   Úunregister_triton_hookr6       s7   € Ý„}Ô&­*Ô*:Ò:Ð:Ø*.�ŒÔ'Ø)-�ŒÔ&Ð&Ð&ð ;Ð:r   )r   N)Ústater   r   Úscoper   r   Útritonr   Útriton.compilerr	   r$   r   r4   r6   r   r   r   ú<module>r;      s¹   ðØ *Ð *Ð *Ð *Ð *Ð *Ð *Ð *Ø *Ð *Ð *Ð *Ð *Ð *Ð *Ð *Ø Ð Ð Ð Ð Ð Ø $Ð $Ð $Ð $Ð $Ð $à8Ð ð#ð #ð #ð #ð #ñ #ô #ð #ð"9ð 9ð 9ð 9ð.ð .ð .ð .ð .ð .r   