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 7b349dd grouping large array tests based on type and updating nightly
CI function (#17305)
add 6b9a1da Multi-tensor LAMB (#16893)
No new revisions were added by this update.
Summary of changes:
python/mxnet/ndarray/contrib.py | 76 +++++++
python/mxnet/optimizer/optimizer.py | 121 ++++++++---
python/mxnet/test_utils.py | 77 ++++---
src/operator/contrib/multi_lamb-inl.h | 359 ++++++++++++++++++++++++++++++++
src/operator/contrib/multi_lamb.cc | 251 ++++++++++++++++++++++
src/operator/contrib/multi_lamb.cu | 261 +++++++++++++++++++++++
src/operator/contrib/multi_sum_sq-inl.h | 4 +
src/operator/contrib/multi_sum_sq.cc | 6 +
src/operator/contrib/multi_sum_sq.cu | 29 ++-
tests/python/unittest/test_optimizer.py | 46 +++-
10 files changed, 1157 insertions(+), 73 deletions(-)
mode change 100644 => 100755 python/mxnet/optimizer/optimizer.py
mode change 100644 => 100755 python/mxnet/test_utils.py
create mode 100644 src/operator/contrib/multi_lamb-inl.h
create mode 100644 src/operator/contrib/multi_lamb.cc
create mode 100644 src/operator/contrib/multi_lamb.cu
mode change 100644 => 100755 tests/python/unittest/test_optimizer.py
