Ë
    ÛéñiÕ  ã                   ó†   — d dl mZmZ ddlmZ ddlmZ ddlmZ	 ddl
mZ dZ ed	d
¬«      Z edd
¬«      Z G d„ de«      Zy
)é   )Úenter_stateÚ
exit_stateé    )ÚLazyDicté   )ÚHook)Úproton)Ú
ContextVarÚ__proton_launch_metadataÚop_nameN)ÚdefaultÚidc                   ó¾   ‡— e Zd ZdZdZg d¢ZeD � �cg c]  }d|› �‘Œ	 c}} dgz   dgz   Zd„ Zˆfd„Zde	d	e
d
e	ddfd„Zd„ Zd„ Zdeddfd„Zdeddfd„ZˆxZS c c}} w )Ú
LaunchHookéd   N)é   é   é    é@   ÚflopsÚbytesc                  ó   — y ©N© ©Úselfs    ú^/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/triton/profiler/hooks/launch.pyÚ__init__zLaunchHook.__init__   ó   € Øó    c                 ód   •— | j                   €t        t        | �  | «      | _         | j                   S r   )Ú	_instanceÚsuperr   Ú__new__)ÚclsÚ	__class__s    €r   r$   zLaunchHook.__new__   s*   ø€ Ø�=‰=Ð Ü!¤*¨cÑ:¸3Ó?ˆCŒMØ�}‰}Ðr    ÚnameÚmetadata_groupÚhashÚreturnc                  ó   — y r   r   )r   ÚmoduleÚfunctionr'   r(   r)   s         r   Úinit_handlezLaunchHook.init_handle   r   r    c                  ó   — y r   r   r   s    r   ÚactivatezLaunchHook.activate    r   r    c                  ó   — y r   r   r   s    r   Ú
deactivatezLaunchHook.deactivate#   r   r    Úmetadatac                 óÐ  — t        t        «       |j                  «       }t        «        t        j
                  D �ci c]  }||v sŒ|||   “Œ }}t        j                  |d   «       t        j                  t        j                  «       «       t        j                  t        j                  «       |d   «       t        j                  t        j                  «       |«       y c c}w )Nr'   )r   ÚCOMPUTE_METADATA_SCOPE_NAMEÚgetr   r   Úmetricsr   Úsetr   Ú	libprotonÚrecord_scopeÚenter_opÚadd_metrics)r   r3   Úlazy_metadataÚkÚ
fn_metricss        r   ÚenterzLaunchHook.enter&   s¢   € ÜÔ/Ô0Ø Ÿ™›ˆÜŒÜ3=×3EÑ3EÖ\¨aÈÈmÒI[�a˜ qÑ)Ñ)Ð\ˆ
Ð\Ü�‰�M &Ñ)Ô*Ü
�‰Œy×%Ñ%Ó'Ô(Ü×Ñœ2Ÿ6™6›8 ]°6Ñ%:Ô;Ü×ÑœbŸf™f›h¨
Õ3ùò	 ]s   ¼	C#ÁC#c                 óx   — t        j                  t        j                  «       t        j                  «       «       y r   )r9   Úexit_opr   r6   r   )r   r3   s     r   ÚexitzLaunchHook.exit0   s   € Ü×Ñœ"Ÿ&™&›(¤G§K¡K£MÕ2r    )Ú__name__Ú
__module__Ú__qualname__Úpriorityr"   Úflops_widthr7   r   r$   ÚstrÚdictr.   r0   r2   r   r@   rC   Ú__classcell__)Ú.0Úwidthr&   s   00@r   r   r      s˜   ø„ à€Hà€IÚ!€KØ,7×8 5��u�gŠÓ8¸G¸9ÑDÈÀyÑP€Gòôð
°#ð Àtð ÐSVð Ð[_ó òòð4˜hð 4¨4ó 4ð3˜Xð 3¨$÷ 3ùó; 9s   “Ar   )Ústater   r   Útriton.compilerr   Úhookr   Útriton._C.libprotonr	   r9   Úcontextvarsr
   r5   r   r   r   r   r    r   ú<module>rS      s>   ðß +Ý $Ý Ý 3Ý "à8Ð á
�Y¨Ô
-€Ù�˜dÔ#€ô$3�õ $3r    