This is an automated email from the ASF dual-hosted git repository.
zhasheng pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 2d3ce93 CI: Add docker containers with Cuda 11.0. (#19219)
add 39ea683 AMP support for Numpy ops (#19036)
No new revisions were added by this update.
Summary of changes:
python/mxnet/contrib/amp/amp.py | 157 ++++++---
python/mxnet/contrib/amp/lists/symbol_fp16.py | 489 +++++++++++++++-----------
python/mxnet/contrib/amp/loss_scaler.py | 18 +-
python/mxnet/operator.py | 26 ++
src/operator/contrib/all_finite.cc | 2 +
src/operator/tensor/amp_cast.cc | 2 +
tests/python/gpu/test_contrib_amp.py | 69 ++--
7 files changed, 471 insertions(+), 292 deletions(-)
