o "j!@sddlZddlmZGdddZGdddeZGdd d eZGd d d eZGd d d eZGdddeZGdddeZ GdddeZ GdddeZ GdddeZ GdddeZ GdddeZGdddeZGdddeZGd d!d!eZdS)"N) constantsc@s@eZdZdddZddZddZdd Zd d Zdd d ZdS) BaseConfigNcCsx||_d|_|durt|tr||_ntd|t|j}|D] \}}t|||q$|jr:| |jdSdSNz%Expected a dictionary. But received: ) _category _config_dict isinstancedict ValueErrorrget_category_default_configitemssetattr from_dict)selfcategory config_dictconfigfield default_valuerj/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/distributed/auto_parallel/strategy.py__init__s  zBaseConfig.__init__cCsBt|j}|D]}||tj}|tjkrt|||q dSN)rr rkeysget NOT_FOUNDr )rrrrvaluerrrr(s    zBaseConfig.from_dictcCs^i}t|j}|D] }t||}|||<q |jD]\}}t|tr,| ||<q|Sr) rr rrgetattr__dict__r rrto_dict)r result_dictrrrrrrr0s      zBaseConfig.to_dictcCs<|}d}|D]\}}|d|d|d7}q |dS)N{"z":"z",})rr )rr stringkvrrr__repr__;s zBaseConfig.__repr__cCsH|j}||}||t|<|jD]\}}t||t||q|Sr) __class____new__idrr r copydeepcopy)rmemoclsresultr%r&rrr __deepcopy__Bs   zBaseConfig.__deepcopy__cCs|}|||Sr)rr)rr%dr rrrrJs zBaseConfig.getr) __name__ __module__ __qualname__rrrr'r0rrrrrrs  rceZdZdfdd ZZS)RecomputeConfigNctj}t||dSr)r RECOMPUTEsuperrrrrr(rrrPzRecomputeConfig.__init__rr2r3r4r __classcell__rrr;rr6Or6cr5)FusedLinearPromotionConfigNcr7r)rFUSEDLINEARPROMOTIONr9rr:r;rrrVr<z#FusedLinearPromotionConfig.__init__rr=rrr;rr@Ur?r@cr5) AMPConfigNcr7r)rAMPr9rr:r;rrr\r<zAMPConfig.__init__rr=rrr;rrB[r?rBcr5)ShardingConfigNcr7r)rSHARDINGr9rr:r;rrrbr<zShardingConfig.__init__rr=rrr;rrDar?rDcr5)GradientMergeConfigNcr7r)rGRADIENT_MERGEr9rr:r;rrrhr<zGradientMergeConfig.__init__rr=rrr;rrFgr?rFcr5)PipelineConfigNcr7r)rPIPELINEr9rr:r;rrrnr<zPipelineConfig.__init__rr=rrr;rrHmr?rHcr5) QATConfigNcr7r)rQATr9rr:r;rrrtr<zQATConfig.__init__rr=rrr;rrJsr?rJcr5) TuningConfigNcr7r)rTUNINGr9rr:r;rrrzr<zTuningConfig.__init__rr=rrr;rrLyr?rLcr5) DatasetConfigNcr7r)rDATASETr9rr:r;rrrr<zDatasetConfig.__init__rr=rrr;rrNr?rNcr5)FusedPassesConfigNcr7r)r FUSED_PASSESr9rr:r;rrrr<zFusedPassesConfig.__init__rr=rrr;rrPr?rPcr5)DPOptimizationConfigNcr7r)rDP_OPTIMIZATIONr9rr:r;rrrr<zDPOptimizationConfig.__init__rr=rrr;rrRr?rRcr5)MPOptimizationConfigNcr7r)rMP_OPTIMIZATIONr9rr:r;rrrr<zMPOptimizationConfig.__init__rr=rrr;rrTr?rTcr5)SPOptimizationConfigNcr7r)rSP_OPTIMIZATIONr9rr:r;rrrr<zSPOptimizationConfig.__init__rr=rrr;rrVr?rVcs"eZdZdZdfdd ZZS)Strategya The `Strategy` object is used to configure the parallelization and optimization for static graph. Args: config (dict|string, optional): If this is None, the default configurations will used. If this is a dictionary, the recognized key-value of it will be used to override the default configurations while other default configurations are left unchanged. If this is a string, it is interpreted as the path to a YAML configuration and will be loaded to override the corresponding default configurations. Examples: .. code-block:: python >>> import paddle >>> from paddle.distributed.fleet import auto >>> strategy = auto.Strategy() >>> sharding = strategy.sharding >>> assert sharding.enable == False >>> assert sharding.stage == 1 >>> assert sharding.degree == 8 >>> sharding.enable = True >>> sharding.stage = 2 >>> sharding.degree = 2 >>> assert sharding.enable == True >>> assert sharding.stage == 2 >>> assert sharding.degree == 2 Ncs|durt|trt||_n td|i|_tj}t ||j|j tj d}t ||_ |j tjd}t||_|j tjd}t||_|j tjd}t||_|j tjd}t||_|j tjd}t||_|j tjd}t||_|j tj d}t!||_"|j tj#d}t$||_%|j tj&d}t'||_(|j tj)d}t*||_+|j tj,d}t-||_.|j tj/d}t0||_1dSr)2rr r+r,rr rZBASEr9rrr8r6Z recomputerCrBamprErDZshardingrGrFZgradient_mergerIrHZpipelinerKrJZqatrMrLZtuningrOrNZdatasetrQrPZ fused_passesrAr@Zfused_linear_promotionrSrRZdp_optimizationrUrTZmp_optimizationrWrVZsp_optimization)rrrrr;rrrsJ             zStrategy.__init__r)r2r3r4__doc__rr>rrr;rrXsrX)r+rrr6r@rBrDrFrHrJrLrNrPrRrTrVrXrrrrs" ;