Ë
    Ûéñi¹  ã                   óp   — d dl mZ d dlmZ d dlmZmZmZ d dlm	Z	 d dl
mZ  G d„ d«      Z G d„ d	«      Zy)
é    )ÚLazyDict)Úabstractmethod)ÚDictÚAnyÚOptional)ÚdefaultdictNc                   ó¨   — e Zd ZU dZeed<   edededede	eef   dedd	fd
„«       Z
ededd	fd„«       Zededd	fd„«       Zedd„«       Zedd„«       Zy	)ÚHookr   ÚpriorityÚmoduleÚfunctionÚnameÚmetadata_groupÚhashÚreturnNc                 ó   — t         ‚©N©ÚNotImplementedError)Úselfr   r   r   r   r   s         ú\/var/www/pod-logistic/pod-ai/venv/lib/python3.12/site-packages/triton/profiler/hooks/hook.pyÚinit_handlezHook.init_handle   s
   € ô "Ð!ó    Úmetadatac                 ó   — t         ‚r   r   ©r   r   s     r   Úenterz
Hook.enter   ó   € ä!Ð!r   c                 ó   — t         ‚r   r   r   s     r   Úexitz	Hook.exit   r   r   c                 ó   — t         ‚r   r   ©r   s    r   ÚactivatezHook.activate   r   r   c                 ó   — t         ‚r   r   r"   s    r   Ú
deactivatezHook.deactivate   r   r   )r   N)Ú__name__Ú
__module__Ú__qualname__r   ÚintÚ__annotations__r   r   Ústrr   r   r   r   r    r#   r%   © r   r   r
   r
      sÃ   … Ø€HˆcÓàð" #ð "°ð "¸Cð "ÐQUÐVYÐ[^ÐV^ÑQ_ð "Øð"Ø"&ò"ó ð"ð ð"˜hð "¨4ò "ó ð"ð ð"˜Xð "¨$ò "ó ð"ð ò"ó ð"ð ò"ó ñ"r   r
   c                   ó>  — e Zd ZU g Zee   ed<    ed„ «      Ze	e
e	eef   f   ed<   edededede	eef   ded	d
fd„«       Zeded	d
fd„«       Zeded	d
fd„«       Zeddee
   d	d
fd„«       Zeddee
   d	d
fd„«       Zedede
d	d
fd„«       Zeddee
   d	d
fd„«       Zy
)ÚHookManagerÚactive_hooksc                  ó    — t        t        «      S r   )r   Úboolr,   r   r   ú<lambda>zHookManager.<lambda>%   s   € Ä[ÔQUÓEV€ r   Úsession_hooksr   r   r   r   r   r   Nc                 óX   — t         j                  D ]  }|j                  | ||||«       Œ y r   )r.   r/   r   )r   r   r   r   r   Úhooks         r   r   zHookManager.init_handle'   s/   € ä×,Ñ,ò 	KˆDØ×Ñ˜V X¨t°^ÀTÕJñ	Kr   r   c                 óP   — t         j                  D ]  }|j                  | «       Œ y r   )r.   r/   r   ©r   r5   s     r   r   zHookManager.enter,   s#   € ä×,Ñ,ò 	!ˆDØ�J‰J�xÕ ñ	!r   c                 ób   — t        t        j                  «      D ]  }|j                  | «       Œ y r   )Úreversedr.   r/   r    r7   s     r   r    zHookManager.exit1   s*   € ô œ[×5Ñ5Ó6ò 	 ˆDØ�I‰I�hÕñ	 r   Úsessionc                 ó|  — | €t         j                  j                  «       }n| g}|D ]q  } t         j                  |    D ]Y  }|t         j                  vr/|j	                  «        t         j                  j                  |«       dt         j                  |    |<   Œ[ Œs t         j                  j                  d„ d¬«       y )NTc                 ó   — | j                   S r   ©r   ©Úxs    r   r2   z&HookManager.activate.<locals>.<lambda>E   ó
   € °A·J±J€ r   ©ÚkeyÚreverse)r.   r3   Úkeysr/   r#   ÚappendÚsort)r:   Úsessionsr5   s      r   r#   zHookManager.activate7   s«   € àˆ?Ü"×0Ñ0×5Ñ5Ó7‰Hà�yˆHàò 	@ˆGÜ#×1Ñ1°'Ñ:ò @�Øœ{×7Ñ7Ñ7Ø—M‘M”OÜ×,Ñ,×3Ñ3°DÔ9Ø;?”×)Ñ)¨'Ñ2°4Ò8ñ	@ð	@ô 	× Ñ ×%Ñ%Ñ*>ÈÐ%ÕMr   c                 óà  ‡— | €t         j                  j                  «       }n| g}t        «       }|D ]S  } t         j                  |    D ];  Š‰t         j                  v r|j                  ‰«       dt         j                  |    ‰<   Œ= ŒU |D ]b  Št        ˆfd„t         j                  j                  «       D «       «      rŒ4‰j                  «        t         j                  j                  ‰«       Œd y )NFc              3   ó(   •K  — | ]	  }|‰   –— Œ y ­wr   r,   ©Ú.0r3   r5   s     €r   ú	<genexpr>z)HookManager.deactivate.<locals>.<genexpr>W   s   øè ø€ Òc¨}�} TÕ*Ñcùó   ƒ)
r.   r3   rD   Úsetr/   ÚaddÚanyÚvaluesr%   Úremove)r:   rG   Údeactivated_hooksr5   s      @r   r%   zHookManager.deactivateG   sÕ   ø€ àˆ?Ü"×0Ñ0×5Ñ5Ó7‰Hà�yˆHä›EÐØò 	AˆGÜ#×1Ñ1°'Ñ:ò A�Øœ;×3Ñ3Ñ3Ø%×)Ñ)¨$Ô/Ø;@”×)Ñ)¨'Ñ2°4Ò8ñAð	Að &ò 	6ˆDÜÓcÄ×@YÑ@Y×@`Ñ@`Ó@bÔcÕcØ—‘Ô!Ü×(Ñ(×/Ñ/°Õ5ñ	6r   r5   c                 ó@  — dt         j                  |   | <   | t         j                  vr/| j                  «        t         j                  j	                  | «       t         j                  j                  d„ d¬«       t        j                  j                  j                  t         j                  «       t        j                  j                  j                  t         j                  «       t        j                  j                  j                  t         j                  «       y )NTc                 ó   — | j                   S r   r=   r>   s    r   r2   z&HookManager.register.<locals>.<lambda>b   r@   r   rA   )r.   r3   r/   r#   rE   rF   ÚknobsÚruntimeÚkernel_load_end_hookrO   r   Úlaunch_enter_hookr   Úlaunch_exit_hookr    )r5   r:   s     r   ÚregisterzHookManager.register[   s¸   € à37Œ×!Ñ! 'Ñ*¨4Ñ0Ø”{×/Ñ/Ñ/Ø�M‰MŒOÜ×$Ñ$×+Ñ+¨DÔ1ä× Ñ ×%Ñ%Ñ*>ÈÐ%ÔMô 	�‰×*Ñ*×.Ñ.¬{×/FÑ/FÔGÜ�‰×'Ñ'×+Ñ+¬K×,=Ñ,=Ô>Ü�‰×&Ñ&×*Ñ*¬;×+;Ñ+;Õ<r   c                 ó˜  ‡— | �| t         j                  vry | €bt         j                  D ]  Š‰j                  «        Œ t         j                  j	                  «        t         j                  j	                  «        nšt         j                  j                  | «      }|j                  «       D ]h  \  Š}|sŒ	t        ˆfd„t         j                  j                  «       D «       «      rŒ:‰j                  «        t         j                  j                  ‰«       Œj t         j                  s¦t        j                  j                  j                  t         j                  «       t        j                  j                  j                  t         j                  «       t        j                  j                   j                  t         j"                  «       y y )Nc              3   ó(   •K  — | ]	  }|‰   –— Œ y ­wr   r,   rJ   s     €r   rL   z)HookManager.unregister.<locals>.<genexpr>y   s   øè ø€ Òg°=˜=¨Õ.ÑgùrM   )r.   r3   r/   r%   ÚclearÚpopÚitemsrP   rQ   rR   rV   rW   rX   r   rY   r   rZ   r    )r:   Úpopped_hooksÚactiver5   s      @r   Ú
unregisterzHookManager.unregisteri   s>  ø€ àÐ 7´+×2KÑ2KÑ#KØàˆ?Ü#×0Ñ0ò "�Ø—‘Õ!ð"ä×$Ñ$×*Ñ*Ô,Ü×%Ñ%×+Ñ+Õ-ä&×4Ñ4×8Ñ8¸ÓAˆLà ,× 2Ñ 2Ó 4ò :‘��fÙØÜÓgÄK×D]ÑD]×DdÑDdÓDfÔgÕgØ—O‘OÔ%Ü×,Ñ,×3Ñ3°DÕ9ð:ô ×'Ò'Ü�M‰M×.Ñ.×5Ñ5´k×6MÑ6MÔNÜ�M‰M×+Ñ+×2Ñ2´;×3DÑ3DÔEÜ�M‰M×*Ñ*×1Ñ1´+×2BÑ2BÕCð (r   r   )r&   r'   r(   r/   Úlistr
   r*   r   r3   r   r)   r1   Ústaticmethodr   r+   r   r   r   r    r   r#   r%   r[   rc   r,   r   r   r.   r.   !   sk  … à!€L�$�t‘*Ó!á1<Ñ=VÓ1W€M�4˜˜T $¨ *Ñ-Ð-Ñ.ÓWàðK˜Cð K¨3ð K°cð KÈ4ÐPSÐUXÐPXÉ>ð KÐadð KÐimò Kó ðKð ð!˜ð ! Tò !ó ð!ð ð �xð   Dò  ó ð ð
 ñN˜( 3™-ð N°4ò Nó ðNð ñ6˜H S™Mð 6°Tò 6ó ð6ð& ð=�tð = cð =¨dò =ó ð=ð ñD˜H S™Mð D°Tò Dó ñDr   r.   )Útriton.compilerr   Úabcr   Útypingr   r   r   Úcollectionsr   Útriton.knobsrV   r
   r.   r,   r   r   ú<module>rk      s/   ðÝ $Ý ß &Ñ &Ý #Ý ÷"ñ "÷2_Dò _Dr   