o
    TÃi@  ã                   @   s4   d dl mZ dZdZdZdd„ ZG dd„ deƒZd	S )
é    )ÚDeepSpeedConfigModelzp
Timers should be enabled as:
"timers": {
  "throughput": {
    "enabled": true,
    "synchronized": true
  }
}
ÚtimersÚ
throughputc                 C   s8   | rt | v rt| t  v r| t  t }ni }tdi |¤ŽS )N© )ÚTIMERSÚTIMERS_THROUGHPUTÚDeepSpeedThroughputTimerConfig)Ú
param_dictÚtimers_config_dictr   r   úJ/home/ubuntu/.local/lib/python3.10/site-packages/deepspeed/utils/config.pyÚget_timers_config   s   r   c                   @   s,   e Zd ZU dZdZeed< 	 dZeed< dS )r   z Configure throughput timers TÚenabledÚsynchronizedN)Ú__name__Ú
__module__Ú__qualname__Ú__doc__r   ÚboolÚ__annotations__r   r   r   r   r   r   #   s   
 r   N)Údeepspeed.runtime.config_utilsr   ÚTIMERS_FORMATr   r   r   r   r   r   r   r   Ú<module>   s   
