Feat graph autograd engine (#5296)
* feat(GraphAutogradEngine): add GraphAutogradEngine
* feat(GraphEngine): support backward interface
* feat(GraphEngine): support autograd.backward()
* fix(GraphEngine): some grad_fn in next_functions but can not apply
* feat(GraphEngine): support autograd.grad interface
* style(*): add JUST
* fix(GraphEngine): fix autograd.grad bugs
* test(Autograd): add flow.autograd test
* fix(GraphEngine): fix autograd.grad bug
* refine codes
* style(*): remove comments
Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/autograd/autograd_engine.cpp 212 additions, 8 deletionsoneflow/core/autograd/autograd_engine.cpp
- oneflow/core/autograd/autograd_engine.h 72 additions, 21 deletionsoneflow/core/autograd/autograd_engine.h
- oneflow/python/autograd/autograd.py 1 addition, 1 deletiononeflow/python/autograd/autograd.py
- oneflow/python/test/modules/test_autograd.py 101 additions, 0 deletionsoneflow/python/test/modules/test_autograd.py
Please register or sign in to comment