• G
    [Auto Parallel]: Support std::vector<phi::Tensor> input and output for DistTensor. (#56602) · d2fedeac
    Ghost Screaming 提交于
    * [WIP] Support std::vector<phi::Tensor> input and output for DistTensor.
    Concat forward and backward are verified.
    
    * Polish code for new dist tensor implementation.
    
    * Fix bug of DistTensor upgrade. Add support functions for std::vector<Tensor> -> std::vector<Tensor>.
    
    * Add support for DistTensor type of std::vector<phi::Tensor> as input or output of operators.
    Following testcases are passed.
    1. concat: std::vector<phi::Tensor> -> phi::Tensor
    2. unbind: phi::Tensor -> std::vector<phi::Tensor>
    3. broadcast_tensors: std::vector<phi::Tensor> -> std::vector<phi::Tensor>
    
    * Polish code. Remove useless comments.
    
    * Add update_loss_scaling in skip_op_lists.
    
    * Polish code.
    d2fedeac
dist_api_gen.py 33.5 KB