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%)