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(-)