You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@mxnet.apache.org by zh...@apache.org on 2021/10/30 15:02:40 UTC
[incubator-mxnet] branch master updated (79e1753 -> fb1d395)
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 79e1753 Prospector checker initial commit (#20684)
add fb1d395 Add quantized batch_dot (#20680)
No new revisions were added by this update.
Summary of changes:
python/mxnet/amp/lists/symbol_bf16.py | 1 +
python/mxnet/amp/lists/symbol_fp16.py | 1 +
src/operator/nn/dnnl/dnnl_batch_dot-inl.h | 73 ++++++++-
src/operator/nn/dnnl/dnnl_batch_dot.cc | 157 ++++++++++++++----
src/operator/nn/dnnl/dnnl_ops-inl.h | 1 +
src/operator/subgraph/dnnl/dnnl_batch_dot.cc | 176 +++++++++++++++++++++
.../subgraph/dnnl/dnnl_batch_dot_property.h | 99 ++++++++++++
...erty.h => dnnl_matmul_post_quantize_property.h} | 34 ++--
.../subgraph/dnnl/dnnl_subgraph_property.cc | 10 +-
.../subgraph/dnnl/dnnl_transformer_qk_property.h | 2 +-
src/operator/tensor/dot-inl.h | 19 +--
src/operator/tensor/dot.cc | 5 +-
...sformer_subgraph.py => test_matmul_subgraph.py} | 51 +++++-
13 files changed, 551 insertions(+), 78 deletions(-)
create mode 100644 src/operator/subgraph/dnnl/dnnl_batch_dot.cc
create mode 100644 src/operator/subgraph/dnnl/dnnl_batch_dot_property.h
rename src/operator/subgraph/dnnl/{dnnl_transformer_post_quantize_property.h => dnnl_matmul_post_quantize_property.h} (82%)
rename tests/python/dnnl/subgraphs/{test_transformer_subgraph.py => test_matmul_subgraph.py} (67%)