You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by pa...@apache.org on 2020/12/12 04:33:20 UTC
[incubator-mxnet] branch master updated (87674ea -> 3746bab)
This is an automated email from the ASF dual-hosted git repository.
patriczhao pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git.
from 87674ea NP Large tensor tests batch 5 (#19541)
add 3746bab [FEATURE] Restore Quantization API to MXNet (#19587)
No new revisions were added by this update.
Summary of changes:
example/quantization/README.md | 191 ++++
example/quantization/imagenet_gen_qsym_onednn.py | 272 +++++
example/quantization/imagenet_inference.py | 187 +++
example/quantization/launch_inference_onednn.sh | 116 ++
python/mxnet/contrib/__init__.py | 2 +
python/mxnet/contrib/quantization.py | 938 +++++++++++++++
src/common/utils.h | 9 -
src/operator/nn/mkldnn/mkldnn_convolution-inl.h | 3 +
src/operator/quantization/calibrate.cc | 1 +
src/operator/quantization/quantize.cc | 1 +
src/operator/quantization/quantize_graph_pass.cc | 87 +-
src/operator/quantization/quantize_v2.cc | 1 +
src/operator/subgraph/mkldnn/mkldnn_conv.cc | 49 +-
.../subgraph/mkldnn/mkldnn_conv_property.h | 26 +-
.../mkldnn/mkldnn_post_quantize_property.h | 5 +-
.../test_quantization_mkldnn.py} | 16 +-
tests/python/mkl/test_subgraph.py | 923 ++++++++++++++-
tests/python/quantization/common.py | 1 +
tests/python/quantization/test_quantization.py | 1207 ++++++++++++++++++++
...test_tvm_op_gpu.py => test_quantization_gpu.py} | 11 +-
tools/rec2idx.py | 4 +-
21 files changed, 3982 insertions(+), 68 deletions(-)
create mode 100644 example/quantization/README.md
create mode 100644 example/quantization/imagenet_gen_qsym_onednn.py
create mode 100644 example/quantization/imagenet_inference.py
create mode 100755 example/quantization/launch_inference_onednn.sh
create mode 100644 python/mxnet/contrib/quantization.py
copy tests/python/{gpu/test_deferred_compute_gpu.py => mkl/test_quantization_mkldnn.py} (71%)
create mode 120000 tests/python/quantization/common.py
create mode 100644 tests/python/quantization/test_quantization.py
copy tests/python/{gpu/test_tvm_op_gpu.py => test_quantization_gpu.py} (91%)