o Me°у@s:ddlmZd ddДZddДZddДZd d ДZddДZdS)щй┌OrderedDictNcCs<|durtГ}nt|tГsJВ|D] }||vrd||<q|S)NT)r┌ isinstance)┌list_objZordered_dict┌objйr·TD:\Projects\ConvertPro\env\Lib\site-packages\paddle/distributed/passes/pass_utils.py┌list_to_ordered_dictsАr cCs\tГ}g}|абjD]!}|jD]}||vr|а|б|а|бq|jD]}|а|бq#q |SйN)┌set┌global_block┌ops┌input_arg_names┌append┌add┌output_arg_names)┌programZvisited_vars┌ input_vars┌op┌in_var_name┌out_var_namerrr┌get_inputs_of_programs А rcCs.tГ}|абjD]}t|j|Гqt|абГSr )rrr r r┌list┌keys)r┌output_varsrrrr┌get_outputs_of_program-srcCsZt|абjГ}|dkr||7}|dkr||ksJВ|dkr!||7}|dkr)||ks-J|ГВ||ks3JВ|аб}t|d|ddГD]}|абj|ddНqAt|dddГD]}|абj|ddНqU|абtГ}|абjD]}|jD]}|а |бqr|j D]}|а |бq}qmg} |абjD]} | |vrШ| а| бqН| D]} |абj | ddНqЫ|аб|S)Nrщщ F)Zsync)┌lenrr ┌clone┌rangeZ _remove_opZ_sync_with_cpprrrr┌varsrZ_remove_var)rZstart_op_idxZ end_op_idx┌op_num┌idxZ valid_varsrrrZvars_to_remove┌varrrr┌ prune_program4s< Аr%cs||sJdГВt|абjГЙИdksJdГВЗfddД|DГ}|ddkr)dg|}|dИkr4|аИбtt|ГdГD]}||||dksLJdГВqgs z!split_program..rrz"op_indices must be strictly sortedcSsg|]}t|ГСqSr)rйr&┌prrrr(xscSsg|]}tt|ГГСqSr)r rr)rrrr(ys cSsg|]}tГСqSrr)r&┌_rrrr(}sTcSsg|]}t|абГСqSr)rr)r&┌itemrrrr(Еs)rrr rr r%┌reversed)rZ op_indicesr#Zsplitted_programsZ new_splitZ num_splitrrZvalid_output_vars┌ir┌jrr'r┌ split_programWsH ■■А■ r0r )┌collectionsrr rrr%r0rrrr┌s #