- 25 8月, 2023 1 次提交
-
-
由 JZ-LIANG 提交于
* add embedding backward rule * update backward api * revert api * matmul inferbackward * update unitest
-
- 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
-
- 24 7月, 2023 1 次提交
-
-
由 Yichen Zhang 提交于
* [Semi-Auto] Add transpose spmd rule * add unit test in cmake file * log perm info
-
- 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 1 次提交
-
-
由 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>
-