You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by tq...@apache.org on 2020/10/26 23:37:16 UTC

[incubator-tvm] branch main updated (9377a9a -> 8d56164)

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

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


    from 9377a9a  [FIX,MICROTVM] Add requires_micro decorators to microtvm tests (#6747)
     add 8d56164  [Relay] A set of utilities that allows a model to be run efficiently on tensorcores. (#6748)

No new revisions were added by this update.

Summary of changes:
 python/tvm/relay/analysis/__init__.py              |   3 +
 python/tvm/relay/analysis/count_layers.py          |  68 ++++++++++
 python/tvm/relay/op/nn/_nn.py                      |  14 +++
 python/tvm/relay/op/strategy/cuda.py               |   9 +-
 python/tvm/relay/transform/__init__.py             |   1 +
 python/tvm/relay/transform/recast.py               | 139 +++++++++++++++++++++
 python/tvm/relay/transform/transform.py            |  27 ++++
 src/runtime/graph/graph_runtime.cc                 |   2 +-
 .../test_layer_count.py}                           |  27 ++--
 tests/python/relay/test_pass_convert_op_layout.py  |  73 +++++++++++
 tests/python/relay/test_recast.py                  | 108 ++++++++++++++++
 11 files changed, 453 insertions(+), 18 deletions(-)
 create mode 100644 python/tvm/relay/analysis/count_layers.py
 create mode 100644 python/tvm/relay/transform/recast.py
 copy tests/python/{unittest/test_tir_analysis_expr_deep_equal.py => relay/test_layer_count.py} (63%)
 create mode 100644 tests/python/relay/test_recast.py