You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by jc...@apache.org on 2021/05/19 10:53:09 UTC
[tvm] branch main updated (53c3456 -> 17bbf02)
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 53c3456 [AutoScheduler] Make RecordReader error-free (#8066)
add 17bbf02 [Autoscheduler] Add sparse conv2d(1*1) support for auto_scheduler (#8065)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/nn.h | 12 +
python/tvm/relay/analysis/__init__.py | 1 +
.../analysis/{sparse_dense.py => sparse_conv2d.py} | 46 +++-
python/tvm/relay/data_dep_optimization/__init__.py | 1 +
.../{bsr_dense.py => bsr_conv2d.py} | 12 +-
python/tvm/relay/op/nn/_nn.py | 11 +
python/tvm/relay/op/op_attrs.py | 5 +
python/tvm/relay/op/strategy/generic.py | 23 ++
python/tvm/relay/transform/transform.py | 23 ++
python/tvm/topi/generic/nn.py | 17 ++
python/tvm/topi/nn/sparse.py | 264 +++++++++++++++++++++
python/tvm/topi/sparse/utils.py | 240 ++++++++++++++++++-
src/relay/op/nn/sparse.cc | 64 +++++
...rt_sparse_dense.cc => convert_sparse_conv2d.cc} | 80 ++++---
...se_convert.py => test_sparse_conv2d_convert.py} | 75 +++---
tests/python/topi/python/test_topi_sparse.py | 76 +++++-
tutorials/auto_scheduler/tune_network_x86.py | 24 +-
17 files changed, 874 insertions(+), 100 deletions(-)
copy python/tvm/relay/analysis/{sparse_dense.py => sparse_conv2d.py} (73%)
copy python/tvm/relay/data_dep_optimization/{bsr_dense.py => bsr_conv2d.py} (85%)
copy src/relay/transforms/{convert_sparse_dense.cc => convert_sparse_conv2d.cc} (55%)
copy tests/python/relay/{test_sparse_dense_convert.py => test_sparse_conv2d_convert.py} (50%)