• C
    [AutoParallel] Adapt static spmd rules for dynamic graph (#56367) · 54fcd9a9
    Chen Weihang 提交于
    * move matmul spmd rules into phi
    
    * add basic infer spmd utils
    
    * addspmd factory
    
    * fix compile error
    
    * add unittest
    
    * refine infer spmd test and utils
    
    * debug infer spmd test
    
    * adapt python test
    
    * poish details
    
    * change to vector attr arg
    
    * revert needless change
    
    * update matmul spmd rule test
    
    * remove original rule
    
    * polish details
    
    * fix marco error
    
    * add comment
    
    * pass backward test
    
    * fix compile error
    
    * add cmake rule for spmd_rules_test
    
    * add dist meta tensor
    
    * update pybind impl
    
    * add marco for rules
    54fcd9a9
rules.h 1.6 KB