o
    "j                     @   s   d dl Z d dlmZ d dlmZ d dlmZ ee j	dda
G dd dZedd	d
Zt
dkr:edddZeZdddZdS )    N)contextmanagerwraps)coreZEVENT_LEVELz-1c                   @   s.   e Zd Zdd Zdd Zd
ddZdd	 ZdS )SotProfilerc                 C      |    d S N)enableself r   X/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/jit/sot/profiler.py	__enter__      zSotProfiler.__enter__c                 C   r   r   )disable)r   exc_typeexc_valexc_tbr   r   r   __exit__   r   zSotProfiler.__exit__Nc                 C   s   t   t   d S r   )r   Znvprof_startZnvprof_enable_record_event)r   tagr   r   r   r	      s   zSotProfiler.enablec                 C   s   t   d S r   )r   Znvprof_stopr
   r   r   r   r   #   r   zSotProfiler.disabler   )__name__
__module____qualname__r   r   r	   r   r   r   r   r   r      s
    
r   c                 c   sJ    zd}t |krt|  d}d V  W |rt  d S d S |r$t  w w )NFT)_event_levelr   Znvprof_nvtx_pushZnvprof_nvtx_pop)
event_nameevent_levelZneed_popr   r   r   
EventGuard'   s   

r   c                 c   s    d V  d S r   r   )r   r   r   r   r   _EmptyEventGuard7   s   
r   c                    s$    fdd}dd }t |kr|S |S )Nc                    s   t   fdd}|S )Nc                     s>   t  dd | i |W  d    S 1 sw   Y  d S )Nr   )r   )r   )argskwargs)r   funcr   r   call_with_event@   s   $z>event_register.<locals>.event_wrapper.<locals>.call_with_eventr   )r!   r"   r   r!   r   event_wrapper?   s   z%event_register.<locals>.event_wrapperc                 S   s   | S r   r   r$   r   r   r   
do_nothingG   s   z"event_register.<locals>.do_nothing)r   )r   r   r%   r&   r   r#   r   event_register>   s
   r'   )r   )os
contextlibr   	functoolsr   Zpaddle.frameworkr   intenvirongetr   r   r   r   r'   r   r   r   r   <module>   s   