o
    TÃiØ  ã                   @   s6   d dl mZ ddlT G dd„ deƒZG dd„ dƒZdS )	é    )ÚDeepSpeedConfigModelé   )Ú*c                   @   sJ   e Zd ZU eZeed< eZeed< e	Z
eed< eZeed< eZeed< dS )ÚCommsLoggerConfigÚenabledÚprof_allÚprof_opsÚverboseÚdebugN)Ú__name__Ú
__module__Ú__qualname__ÚCOMMS_LOGGER_ENABLED_DEFAULTr   ÚboolÚ__annotations__ÚCOMMS_LOGGER_PROF_ALL_DEFAULTr   ÚCOMMS_LOGGER_PROF_OPS_DEFAULTr   ÚlistÚCOMMS_LOGGER_VERBOSE_DEFAULTr	   ÚCOMMS_LOGGER_DEBUG_DEFAULTr
   © r   r   úI/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/comm/config.pyr      s   
 r   c                   @   s   e Zd Zdd„ ZdS )ÚDeepSpeedCommsConfigc                 C   s,   d|v | _ | j rtdi |d ¤Ž| _d S d S )NÚcomms_loggerr   )Úcomms_logger_enabledr   r   )ÚselfÚ	ds_configr   r   r   Ú__init__   s   
ÿzDeepSpeedCommsConfig.__init__N)r   r   r   r   r   r   r   r   r      s    r   N)Údeepspeed.runtime.config_utilsr   Ú	constantsr   r   r   r   r   r   Ú<module>   s   