- 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 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
-
- 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>
-
- 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>
-