This is an automated email from the ASF dual-hosted git repository.
apeforest pushed a change to branch benchmark
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 169ed69 Speed fused_op compilation by caching ptx and jit-compiled
functions (#16783)
add a47b540 multi-precision lamb update operator
No new revisions were added by this update.
Summary of changes:
python/mxnet/optimizer/optimizer.py | 94 ++++++++++-----
src/operator/optimizer_op-inl.h | 202 ++++++++++++++++++++++++++++----
src/operator/optimizer_op.cc | 38 ++++++
src/operator/optimizer_op.cu | 5 +
tests/python/gpu/test_operator_gpu.py | 59 ++++++++++
tests/python/unittest/test_optimizer.py | 23 +++-
6 files changed, 365 insertions(+), 56 deletions(-)
