o
     i                     @   sz   U d dl Z ddlmZmZ dZzdZeed< e j	d e ddl
mZ dZW n ey6   G d	d
 d
ZY nw e jd = dS )    N   )	_ProfilerloggerFz /usr/local/dcgm/bindings/python3DCGM_PYTHON_PATH)DCGMProfilerTc                   @   sF   e Zd ZdZ			d			dddZddd	Zdd
dZdddZdS )r   zThe dummy DCGM Profiler.Nmain_profilerr   returnc                 C      d S N )selfr   gpus_to_profilefield_ids_to_profile
updateFreqr   r   S/home/ubuntu/.local/lib/python3.10/site-packages/xformers/profiler/profiler_dcgm.py__init__   s   zDCGMProfiler.__init__c                 C   s   t dt d d S )Nz"Unable to find python bindings at z. No data will be captured.)r   warningr   r   r   r   r   	__enter__    s   
zDCGMProfiler.__enter__c                 C   r	   r
   r   )r   exc_typeexc_valexc_tbr   r   r   __exit__&      zDCGMProfiler.__exit__c                 C   r	   r
   r   r   r   r   r   step)   r   zDCGMProfiler.step)NNN)r   r   r   N)r   N)__name__
__module____qualname____doc__r   r   r   r   r   r   r   r   r      s    

	
r   )sysprofilerr   r   DCGM_PROFILER_AVAILABLEr   str__annotations__pathinsertprofiler_dcgm_implr   ModuleNotFoundErrorr   r   r   r   <module>   s   
