1. 08 9月, 2023 1 次提交
  2. 31 8月, 2023 1 次提交
    • 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
  3. 25 8月, 2023 2 次提交
  4. 14 8月, 2023 1 次提交
    • Y
      [Semi-Auto] Add reshape spmd rule (#55177) · a97b507e
      Yichen Zhang 提交于
      * add reshape spmd rule
      
      * add unit test for reshape spmd rule
      
      * bug fix
      
      * replace the print_info function with to_string
      
      * fix typo
      
      * bug fix
      
      * add handling for "0" in target shape
      
      * remove the part of computing size in dim_trans.cc
      a97b507e
  5. 04 8月, 2023 1 次提交
  6. 24 7月, 2023 2 次提交
  7. 20 7月, 2023 1 次提交
    • J
      [Semi Auto] Entropy SPMD Rule (#55394) · 5f376f00
      JZ-LIANG 提交于
      * base rule
      
      * add sharidng merge
      
      * add sharidng axis merge
      
      * define unified data class for inferencing dist_attr
      
      * test wrap DistTensorSpec in dygraph mode
      
      * matmul main logic done
      
      * shape int64
      
      * common cc
      
      * define unified data class for inferencing dist_attr
      
      * test wrap DistTensorSpec in dygraph mode
      
      * define python api and wrap function in static mode for DistTensorSpec
      
      * revise syntax
      
      * map bugfix
      
      * broadcast func
      
      * compile 1
      
      * add unitest
      
      * add registry
      
      * update unitest
      
      * bugfix
      
      * bugfix
      
      * add pybind
      
      * bugfix
      
      * bugfix macro gloabl name space
      
      * bugfix macro gloabl name space
      
      * pybind
      
      * pybind test
      
      * pybind bugfixed1
      
      * pybind bugfixed2
      
      * pybind unitest
      
      * merge dev
      
      * merge dev
      
      * merge dev
      
      * fixed cmake conflict
      
      * fixed cmake conflict
      
      * rename get method
      
      * revise inferforward output type
      
      * revise comment
      
      * replicated rule
      
      * replicated rule 2
      
      * revert bug deps
      
      * add rule
      
      * add unitest
      
      * add rule
      
      * add unitest
      
      * move ut of auto_parallel
      
      * fix ut
      
      * bugfix
      
      * bugfix
      
      * bugfix
      
      * bugfix
      
      * bugfix
      
      * bugfix
      
      * bugfix
      
      * resolute input sharding conflict maybe
      
      * fixed comment
      
      * add rule
      
      * add unitest
      
      * fixed typoes
      
      ---------
      Co-authored-by: NYichen Zhang <zhangyichen03@baidu.com>
      Co-authored-by: Nzhiqiu <chenqiuliang@baidu.com>
      5f376f00
  8. 12 7月, 2023 1 次提交
  9. 07 7月, 2023 2 次提交
    • Y
      [Semi-Auto] Add reduction spmd rule (#54991) · 35b72e87
      Yichen Zhang 提交于
      * add reduction spmd rule for auto parallel
      
      * fix the logic of handling partial
      
      * fix code style
      
      * fix the partial handling
      35b72e87
    • Y
      [Semi-auto] add elementwise spmd rule for auto parallel (#54373) · 8e5b0af5
      Yichen Zhang 提交于
      * add some basic functions
      
      * add elementwise rule for auto parallel
      
      * add unit test for elementwise rule
      
      * fix the lib name in spmd rule test cmake file
      
      * fix some bugs
      
      * add unit tests for elementwise spmd rule in python
      
      * bug fix
      
      * delete cpp unit test for elementwise spmd rule (use python ut now)
      
      * add cpp unit test for elementwise rule
      
      * use concrete op name in unit test
      
      * fix typo
      
      * fix code style
      
      * delete cpp unit test
      
      * add more details in log
      8e5b0af5
  10. 04 7月, 2023 1 次提交
  11. 27 6月, 2023 1 次提交