
    ϑi	                        S SK Jr  S SKJr  S SKJr  S SKJrJr  S SK	J
r
  S SKJr  SSKJr  \
" S	5      r\" S
5      r " S S5      r\SS j5       r S   SS jjrg)    )annotations)contextmanagerwraps)CallableTypeVar)	ParamSpec)core   )ENV_SOT_EVENT_LEVELPTc                  0    \ rS rSrS rS rSS jrS rSrg)	SotProfiler   c                $    U R                  5         g N)enableselfs    S/var/www/html/banglarbhumi/venv/lib/python3.13/site-packages/paddle/jit/profiler.py	__enter__SotProfiler.__enter__    s        c                $    U R                  5         g r   )disable)r   exc_typeexc_valexc_tbs       r   __exit__SotProfiler.__exit__#   s    r   Nc                X    [         R                  " 5         [         R                  " 5         g r   )r
   nvprof_startnvprof_enable_record_event)r   tags     r   r   SotProfiler.enable&   s    '')r   c                .    [         R                  " 5         g r   )r
   nvprof_stopr   s    r   r   SotProfiler.disable*   s    r    r   )	__name__
__module____qualname____firstlineno__r   r    r   r   __static_attributes__r*   r   r   r   r      s    *r   r   c              #     #    Sn [         R                  " 5       U:  a  [        R                  " U 5        SnS v   U(       a  [        R                  " 5         g g ! U(       a  [        R                  " 5         f f = f7f)NFT)r   getr
   nvprof_nvtx_pushnvprof_nvtx_pop)
event_nameevent_levelneed_pops      r   
EventGuardr7   .   s^     H#""$3!!*-H  " 8  " s   A;5A A;A88A;c                \   ^ ^ SUU 4S jjnSS jn[         R                  " 5       T:  a  U$ U$ )Nc                :   >^  [        T 5      SUUU 4S jj5       nU$ )Nc                    > [        T5      (       a  T" U 0 UD6OTn[        UTS9   T" U 0 UD6sS S S 5        $ ! , (       d  f       g = f)N)r5   )callabler7   )argskwargsr4   r5   event_name_formatterfuncs      r   call_with_event>event_register.<locals>.event_wrapper.<locals>.call_with_event?   sN     011 %d5f5) 
 JK@T,V, A@@s	   8
A)r<   zP.argsr=   zP.kwargsr   )r?   r@   r5   r>   s   ` r   event_wrapper%event_register.<locals>.event_wrapper>   s&    	t	- 	- 
	- r   c                    U $ r   r*   )r?   s    r   
do_nothing"event_register.<locals>.do_nothingK   s    r   )r?   Callable[P, T]returnrG   )r   r1   )r>   r5   rB   rE   s   ``  r   event_registerrI   ;   s1       K/r   N)r   )r>   zCallable[P, str] | strrH   z*Callable[[Callable[P, T]], Callable[P, T]])
__future__r   
contextlibr   	functoolsr   typingr   r   typing_extensionsr	   paddle.frameworkr
   dy2static.utilsr   r   r   r   r7   rI   r*   r   r   <module>rQ      sg    # %  $ ' ! 0cNCL  	# 	# ?@0/r   