• Z
    [AutoParallel] adapt for 2d laplace (#41601) · c043a21b
    zhaoyingli 提交于
    * add default_ctx in backward.py
    
    * record grad_var_to_var with grad_times
    
    * fix backward
    
    * update annotation
    
    * add complete_high_order_grad in complete_forward
    
    * add dist slice op
    
    * update grad_var_to_var type
    
    * update partition_block init mapping before loss op
    
    * update compatible for 'XShape' & update 'allreduce_vars'
    
    * add dist reshape op when input dim equal to output dim
    
    * update 'set_grad_var_shape' with grad_var_to_var
    
    * fix dist slice
    
    * fix set_grad_var_shape
    
    * add dist pnorm op
    
    * fix dist pnorm dist_attr
    
    * fix engine startprogram & adapt highorder grad
    
    * fix set_grad_var_shape when mp
    
    * update unittest
    
    * update cmakelist
    
    * default strategy in engine: dp
    
    * bug fix
    
    * tiny fix
    
    * flatten outputs
    
    * fix default strategy
    
    * init default ctx
    
    * tiny fix
    
    * test=allcase
    c043a21b
utils.py 55.7 KB