You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by wu...@apache.org on 2020/04/20 05:15:46 UTC
[incubator-tvm] branch master updated (4c0a53d -> 3f03869)

This is an automated email from the ASF dual-hosted git repository.

wuwei pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.


    from 4c0a53d  [TIR][REFACTOR] RewriteForTensorCore -> te/schedule (#5379)
     add 3f03869  [Blocksparse] Pipeline for lowering dense model to sparse-dense (#5377)

No new revisions were added by this update.

Summary of changes:
 python/setup.py                                    |   1 +
 python/tvm/relay/__init__.py                       |   1 +
 python/tvm/relay/analysis/__init__.py              |   1 +
 python/tvm/relay/analysis/analysis.py              |  18 +++
 python/tvm/relay/analysis/sparse_dense.py          |  93 ++++++++++++
 .../data_dep_optimization}/__init__.py             |   7 +-
 .../tvm/relay/data_dep_optimization/bsr_dense.py   |  57 ++++++++
 .../data_dep_optimization/simplify_fc_transpose.py |  60 ++++++++
 .../tvm/relay/data_dep_optimization/utils.py       |  26 ++--
 python/tvm/relay/transform/transform.py            |  40 ++++++
 src/relay/transforms/convert_sparse_dense.cc       | 159 +++++++++++++++++++++
 src/relay/transforms/simplify_fc_transpose.cc      | 154 ++++++++++++++++++++
 tests/python/relay/test_simplify_fc_transpose.py   |  67 +++++++++
 tests/python/relay/test_sparse_dense_convert.py    |  86 +++++++++++
 14 files changed, 756 insertions(+), 14 deletions(-)
 create mode 100644 python/tvm/relay/analysis/sparse_dense.py
 copy python/tvm/{contrib/tf_op => relay/data_dep_optimization}/__init__.py (82%)
 create mode 100644 python/tvm/relay/data_dep_optimization/bsr_dense.py
 create mode 100644 python/tvm/relay/data_dep_optimization/simplify_fc_transpose.py
 copy topi/python/topi/generic/search.py => python/tvm/relay/data_dep_optimization/utils.py (63%)
 create mode 100644 src/relay/transforms/convert_sparse_dense.cc
 create mode 100644 src/relay/transforms/simplify_fc_transpose.cc
 create mode 100644 tests/python/relay/test_simplify_fc_transpose.py
 create mode 100644 tests/python/relay/test_sparse_dense_convert.py