- 08 9月, 2023 1 次提交
-
-
由 Yichen Zhang 提交于
-
- 31 8月, 2023 1 次提交
-
-
由 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
-
- 25 8月, 2023 2 次提交
-
-
由 JZ-LIANG 提交于
* add embedding backward rule * update backward api * revert api * matmul inferbackward * update unitest
-
由 Yichen Zhang 提交于
* fix small bug in reshape spmd rule * small fix in unit test
-
- 14 8月, 2023 1 次提交
-
-
由 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
-
- 04 8月, 2023 1 次提交
-
-
由 JZ-LIANG 提交于
-
- 24 7月, 2023 2 次提交
-
-
由 Yichen Zhang 提交于
* [Semi-Auto] Add transpose spmd rule * add unit test in cmake file * log perm info
-
由 Yichen Zhang 提交于
* add split spmd rule * add pytest in cmake file * small fix
-
- 20 7月, 2023 1 次提交
-
-
由 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>
-
- 12 7月, 2023 1 次提交
-
-
由 JZ-LIANG 提交于
* resolute input sharding conflict maybe * fixed comment --------- Co-authored-by: NYichen Zhang <zhangyichen03@baidu.com> Co-authored-by: Nzhiqiu <chenqiuliang@baidu.com>
-
- 07 7月, 2023 2 次提交
-
-
由 Yichen Zhang 提交于
* add reduction spmd rule for auto parallel * fix the logic of handling partial * fix code style * fix the partial handling
-
由 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
-
- 04 7月, 2023 1 次提交
-
-
由 JZ-LIANG 提交于
* base rule * add sharidng merge * add sharidng axis merge * define unified data class for inferencing dist_attr --------- Co-authored-by: NYichen Zhang <zhangyichen03@baidu.com>
-
- 27 6月, 2023 1 次提交
-
-
由 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 * define unified data class for inferencing dist_attr --------- Co-authored-by: NYichen Zhang <zhangyichen03@baidu.com>
-