This is an automated email from the ASF dual-hosted git repository.
jcf94 pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from f4f525d [AMP] Disallow fp16 conversion for summation-like ops (#8810)
add 227bf7f [TOPI] [Relay] Sparse Conv2d Implementation for 3x3 kernels
(#8605)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/nn.h | 4 +
python/tvm/autotvm/measure/measure_methods.py | 15 +-
python/tvm/relay/analysis/sparse_conv2d.py | 58 ++++---
.../tvm/relay/data_dep_optimization/bsr_conv2d.py | 44 +++++-
python/tvm/relay/op/nn/_nn.py | 6 +-
python/tvm/relay/op/strategy/x86.py | 25 +++
python/tvm/relay/transform/transform.py | 24 ++-
python/tvm/topi/nn/sparse.py | 21 +--
python/tvm/topi/x86/sparse.py | 162 ++++++++++++++++++-
src/relay/op/nn/sparse.cc | 3 +-
src/relay/transforms/convert_sparse_conv2d.cc | 173 ++++++++++++++++++++-
tests/python/relay/test_sparse_conv2d_convert.py | 63 ++++++++
12 files changed, 548 insertions(+), 50 deletions(-)
