-
Notifications
You must be signed in to change notification settings - Fork 6.8k
Add graph pass and backward option #20837
base: zero_sharding
Are you sure you want to change the base?
Add graph pass and backward option #20837
Conversation
Hey @xinyual , Thanks for submitting the PR
CI supported jobs: [unix-gpu, centos-cpu, website, unix-cpu, windows-gpu, windows-cpu, centos-gpu, miscellaneous, clang, sanity, edge] Note: |
Thank you @xinyual! Could you please make sure the CI tests (ci/jenkins/mxnet-validation/sanity) pass, then we can merge this code into the zero_sharding feature branch for future devolpment |
Hi leo, I don't know where is the problem since the error information only shows the wrong command. Is this related to third party ondnn? It seems I don't update it to latest version. |
Thank you @xinyual. Please rebase on top of the apache/incubator-mxnet master branch and force push to your xinyual/incubator-mxnet add_graph_pass_and_backward_option branch |
Description
This is the branch to add the zero sharding (especially partitioning gradient). The changes can be divided into three part:
Current problem