This is an automated email from the ASF dual-hosted git repository.
haibin pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 814be59 Add #include <sys/wait.h> needed for waitpid (#17078)
add f86a8d1 [API] unified API for custom kvstores (#17010)
No new revisions were added by this update.
Summary of changes:
ci/docker/runtime_functions.sh | 1 +
python/mxnet/__init__.py | 24 +-
python/mxnet/gluon/trainer.py | 50 ++-
.../{contrib/amp/lists => kvstore}/__init__.py | 9 +-
python/mxnet/kvstore/base.py | 455 +++++++++++++++++++++
python/mxnet/{ => kvstore}/kvstore.py | 183 ++++-----
python/mxnet/{ => kvstore}/kvstore_server.py | 6 +-
python/mxnet/model.py | 35 +-
src/kvstore/kvstore_local.h | 4 +-
tests/nightly/dist_device_sync_kvstore_custom.py | 96 +++++
tests/python/unittest/test_gluon_trainer.py | 33 +-
tests/python/unittest/test_kvstore_custom.py | 195 +++++++++
12 files changed, 937 insertions(+), 154 deletions(-)
copy python/mxnet/{contrib/amp/lists => kvstore}/__init__.py (84%)
create mode 100644 python/mxnet/kvstore/base.py
rename python/mxnet/{ => kvstore}/kvstore.py (85%)
rename python/mxnet/{ => kvstore}/kvstore_server.py (97%)
create mode 100644 tests/nightly/dist_device_sync_kvstore_custom.py
create mode 100644 tests/python/unittest/test_kvstore_custom.py
|