Fix concat backward bug (#5443)
* add argmax test * fix ci error * fix docstring warning * fix tensor greater and less bug * fix conflict * add test_flow_xxx_against_pytorch func * fix concat backward bug * auto format by CI * format * Add autograd engine warning (#5444) * add autograd engine warning * fix bug Co-authored-by:oneflow-ci-bot <ci-bot@oneflow.org> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/core/autograd/autograd_engine.cpp 5 additions, 1 deletiononeflow/core/autograd/autograd_engine.cpp
- oneflow/core/autograd/gradient_funcs/concat.cpp 5 additions, 11 deletionsoneflow/core/autograd/gradient_funcs/concat.cpp
- oneflow/python/test/modules/test_concat.py 23 additions, 0 deletionsoneflow/python/test/modules/test_concat.py
Please register or sign in to comment