• Y
    [Auto Parallel] Improve the APIs (#45776) · b042a3b1
    Yulong Ao 提交于
    * [Auto Parallel] Use c++ dist attr in the completion process
    
    * [Auto Parallel] Add minor changes
    
    * [Auto Parallel] Use c++ dist attr in the completion process
    
    * [Auto Parallel] Add minor changes
    
    * [Auto Parallel] Add the serialization process for dist attrs
    
    * [Auto Parallel] Remove unnecessary comments
    
    * [Auto Parallel] Fix some bugs
    
    * [Auto Parallel] Fix the code style
    
    * [Auto Parallel] Remove unnecessary impls
    
    * [Auto Parallel] Fix the importing error
    
    * [Auto Parallel] Fix the copy from bugs of op dist attr
    
    * [Auto Parallel] Replace the use of constexpr if
    
    * [Auto Parallel] Redesign the shard_tensor, shard_op and ProcessMesh
    
    * [Auto Parallel] Change API of the completion unittest
    
    * [Auto Parallel] Fix the bug when set_attr an int
    
    * [Auto Parallel] Add the unittest for the serialization
    
    * [Auto Parallel] Add some unit tests
    
    * [Auto Paralle] Unify the strategy
    
    * [Auto Parallel] Improve the engine api
    
    * [Auto Parallel] Reset the changes made to the framework
    
    * [Auto Parallel] Change the engine unittest
    
    * [Auto Parallel] Update API of the completion and partitioner
    
    * [Auto Parallel] Update unit tests using engine api
    
    * update shard annotation
    
    * [Auto Parallel] Remove the modifications of other modules
    
    * [Auto Parallel] Add docs for APIs
    
    * add new strategy
    
    * [Auto Parallel] Replace the logger
    
    * [Auto Parallel] Restore the test_program.py
    
    * [Auto Parallel] Change the import rules
    
    * [Auto Parallel] Add the examples for Engine
    
    * [Auto Parallel] Do some minor changes
    
    * [Auto Parallel] Remove yaml dependency
    
    * [Auto Parallel] Fix the unittests
    
    * add valid after train
    
    * bug fix
    Co-authored-by: Nzhaoyingli <zhaoyingli@baidu.com>
    Co-authored-by: Ncaozhou <caozhou@radi.ac.cn>
    Co-authored-by: Ncaozhou <48191911+Caozhou1995@users.noreply.github.com>
    b042a3b1
utils.py 60.0 KB