stateful local opkernel: return a temp parallel ctx (#4857)
* add temp parallel ctx for single card Signed-off-by:daquexian <daquexian566@gmail.com> * add TODO comment Signed-off-by:
daquexian <daquexian566@gmail.com> Co-authored-by:
oneflow-ci-bot <69100618+oneflow-ci-bot@users.noreply.github.com>
Showing
- oneflow/api/python/env/env.cpp 1 addition, 1 deletiononeflow/api/python/env/env.cpp
- oneflow/api/python/env/env.h 1 addition, 1 deletiononeflow/api/python/env/env.h
- oneflow/api/python/env/env_api.h 1 addition, 1 deletiononeflow/api/python/env/env_api.h
- oneflow/python/framework/distribute.py 1 addition, 1 deletiononeflow/python/framework/distribute.py
- oneflow/user/kernels/stateful_local_opkernel.cpp 10 additions, 1 deletiononeflow/user/kernels/stateful_local_opkernel.cpp
Please register or sign in to comment