o "╒jьу@sFgZddДZddДZddДZddДZGdd Дd ГZGd dДdeГZdS) cszt|ГЙЗfddДtИГDГ}dgИ}t|ГD] \}}t|ГD]\}}|а|бr7d|||<||d7<q q||fS)Ncsg|]}dgИСqS)щйй┌.0┌_й┌nsizer·p/var/www/html/Deteccion_Ine/venv/lib/python3.10/site-packages/paddle/distributed/fleet/base/strategy_compiler.py┌ єz create_graph..rщ)┌len┌range┌ enumerateZ_can_update)┌optimizer_list┌edge┌indegree┌i┌opt┌jZ opt_innerrrr┌create_graphs А¤rcCsжt|Г}dg|}t|ГD]C}d}||kr)||dkr)|d7}||kr)||dks||ks1JdГВ|||<d||<t|ГD]}|||dkrO||d8<q=q |S)Nщ rrz0The combination of meta optimizers contains ringйrr )rrrZtoporr┌krrr┌ topo_sort s А■rcsКt|ГЙd}ddg}ЗfddДtИГDГ}tИГD]*}tИГD]#}|||dkrB|g|||<||||krB|||}||g}qqtИГD]f}tИГD]_}tИГD]X}|||dkr_qT|||dkso|||dkrpqT|||||||||krм|||||||||<|||||||||<|||}||g}qTqNqH|dkr╢dgS|dg||d|dS)Nrcsg|]}ddДtИГDГСqS)cSsg|]}gСqSrrrrrrr 7sz$floyd...)r rrrrr 7szfloyd..rrr)r┌max_lenZmax_edge┌max_pathrrrrrr┌floyd2s>А· $ АЄ rcCszt|ГdkrdSt|Г\}}t||Гt|Г}g}|D] }|а||бqt|ddЕГD] \}}|а||dбq-|S)Nrrr)rrrr┌appendrZ_update_inner_optimizer)rrrr┌ candidate┌idxrrrr┌maximum_path_len_algoZs r c@seZdZddДZdS)┌StrategyCompilerBasecCsdSйNrй┌selfrrr┌__init__lszStrategyCompilerBase.__init__N)┌__name__┌ __module__┌__qualname__r%rrrrr!ksr!csJeZdZdZЗfddДZddДZddДZdd ДZd dДZ dd ДZ ЗZ S)┌StrategyCompileraЙ StrategyCompiler is responsible for meta optimizers combination Generally, a user can define serveral distributed strategies that can generate serveral meta optimizer. The combination of these meta optimizers should have the right order to apply the optimizers' minimize function. This class is responsible for the executable distributed optimizer generation. cs2tГабg|_g|_d|_d|_g|_g|_dSr")┌superr%┌_meta_optimizers┌_graph_optimizersZ_valid_optimizer_list┌_user_defined_strategy┌_meta_optimizer_candidates┌_graph_optimizer_candidatesr#й┌ __class__rrr%{s zStrategyCompiler.__init__cCs|jSr"йr+r#rrr┌_get_applied_meta_optimizerДsz,StrategyCompiler._get_applied_meta_optimizercCєddД|jDГS)NcSєg|]}t|ГjСqSrй┌typer&йrrrrrr Иr z;StrategyCompiler._get_applied_meta_list..r2r#rrr┌_get_applied_meta_listЗєz'StrategyCompiler._get_applied_meta_listcCr4)NcSr5rr6r8rrrr Лr z.)r,r#rrr┌_get_applied_graph_listКr:z(StrategyCompiler._get_applied_graph_listc CsДddl}|а|б}g}|jD]}d}|jD]}|jj|jjkr#d}nq|s+|а|бq|D]} | а|бq.|D]} | а|бq8|S)NrFT)┌copy┌deepcopyr.r+r1r&rZ_disable_strategy) r$Z dist_strategyZcan_not_apply_optimizer_listr<Zvalid_strategyZinvalid_optimizersrZis_valid┌validrrrr┌_get_valid_strategyНs$ ■ Аz$StrategyCompiler._get_valid_strategycCsр||_||_||_t|Гdkrt|Гdkr|dfSt|Г}t|Г}|dur'gn||_|dur0gn||_|dur9dn|d} |durCdn|d} |dusO|durS| | fSd}|D]}|D]} |jj| j vrfd}q[qW|sld} | | fS)NrTF) r-r.r/rr r+r,r1r&Zmeta_optimizers_black_list)r$ZlossZ role_makerZ optimizerZuser_defined_strategyZmeta_optimizer_listZgraph_optimizer_listZmeta_optimizersZgraph_optimizersZreturn_metaZreturn_graphZneed_graph_optZ graph_optZprogram_optrrr┌generate_optimizerмs: А√z#StrategyCompiler.generate_optimizer)r&r'r(┌__doc__r%r3r9r;r?r@┌ __classcell__rrr0rr)ps r)N)┌__all__rrrr r!r)rrrr┌s(