• C
    [AutoParallel] Generate spmd rule and reshard impl in phi api (#56831) · e9364a38
    Chen Weihang 提交于
    * add spmd and reshard code gen
    
    * add backward reshard code gen
    
    * test matmul forward success
    
    * polish test impl
    
    * add unsafe mutable value
    
    * polish details and add test
    
    * fix unittest time out
    
    * fix typo
    
    * refactor reshard input generate impl
    
    * resolve conflict with develop
    
    * fix compile error
    e9364a38
semi_auto_parallel_for_matmul.py 3.0 KB