o
    MeM  ã                   @   sž  d dl mZ eeƒZdd„ Zdd„ Zdd„ Zdd	„ Zd
ZdZ	ee	ddƒ ee	ddƒ ee	ddƒ ee	ddƒ ee	ddƒ ee	ddƒ ee	ddƒ ee	ddƒ dZ
ee
ddƒ ee
ddƒ ee
ddƒ dZeeddƒ eeddƒ eedd ƒ eed!d"ƒ eed#d$ƒ eed%d&ƒ eed'dƒ eed(g ƒ eed)g ƒ eed*g ƒ eed+dƒ eed,dƒ eed-dƒ d.Zeeddƒ eed/d0ƒ eed1d2ƒ eed3d4ƒ eeddƒ eed5g ƒ d6Zeeddƒ eed7d0ƒ eed8dƒ d9Zeeddƒ eed:dƒ eed;d2ƒ eed<d2ƒ eed=d>gƒ eed?dƒ d@Zeeddƒ eedAd0ƒ eedBdƒ eedCd0ƒ eedDd0ƒ eedEdƒ eedFdƒ dBZeeddƒ eedGd0ƒ dS )Hé    )Údefaultdictc                 C   s   t |  S ©N©Ú_g_default_config)Úcategory© r   úZD:\Projects\ConvertPro\env\Lib\site-packages\paddle/distributed/auto_parallel/constants.pyÚget_category_default_config   s   r	   c                 C   s   |t | < d S r   r   )r   Údefault_valuer   r   r   Úset_category_default_config   ó   r   c                 C   s   t |  | S r   r   )r   Úfieldr   r   r   Úget_field_default_config   r   r   c                 C   s   |t |  |< d S r   r   )r   r   r
   r   r   r   Úset_field_default_config!   s   r   Ú	not_foundÚbaseZ	auto_modeÚsemiZgradient_scaleTZ	use_cacheZreturn_numpyZ	all_ranksFZ
split_dataÚseedNZreinitZ	recomputeÚenableZcheckpointsZenable_tuningÚampZinit_loss_scalingg      à@Zincr_every_n_stepsiè  Zdecr_every_n_nan_or_infé   Z
incr_ratiog       @Z
decr_ratiogš™™™™™é?Zuse_dynamic_loss_scalingZcustom_white_listZcustom_black_listZcustom_black_varnamesZuse_pure_fp16Zuse_fp16_guardZuse_optimizer_fp16ZshardingZstageé   Zdegreeé   Zsegment_broadcast_MBg      @@Ztuning_rangeZgradient_mergeZk_stepsZavgZqatZchannel_wise_abs_maxZweight_bitsZactivation_bitsZnot_quant_patternZ
skip_quantÚalgoZtuningZ
batch_sizeZdatasetZprofile_start_stepZprofile_end_stepZrun_after_tuningÚverboseZ
num_shards)Úcollectionsr   Údictr   r	   r   r   r   Ú	NOT_FOUNDZBASEZ	RECOMPUTEÚAMPZSHARDINGZGRADIENT_MERGEZQATZTUNINGZDATASETr   r   r   r   Ú<module>   s~   