You are viewing a plain text version of this content. The canonical link for it is here.
Posted to commits@tvm.apache.org by mo...@apache.org on 2022/02/14 09:54:44 UTC

[tvm] branch main updated (bb60ee9 -> 55849e6)

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

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


    from bb60ee9  [PyTorch] add var_mean support (#10233)
     add 55849e6  [USMP] adding support for U2 and U3 usecases (#10193)

No new revisions were added by this update.

Summary of changes:
 include/tvm/ir/memory_pools.h                      | 162 +++++++++++++++
 include/tvm/ir/module.h                            |   9 +
 include/tvm/tir/usmp/utils.h                       | 106 +---------
 python/tvm/__init__.py                             |   2 +
 python/tvm/ir/__init__.py                          |   1 +
 .../tvm/{tir/usmp/utils.py => ir/memory_pools.py}  |  83 ++------
 python/tvm/micro/model_library_format.py           |  13 +-
 python/tvm/relay/backend/executor_factory.py       |   2 +
 python/tvm/relay/build_module.py                   |  24 ++-
 python/tvm/tir/usmp/utils.py                       |  93 +--------
 src/ir/memory_pools.cc                             |  92 +++++++++
 src/relay/backend/aot_executor_codegen.cc          |   2 +-
 src/relay/backend/build_module.cc                  |  30 ++-
 src/relay/backend/graph_executor_codegen.cc        |   2 +-
 src/relay/backend/utils.cc                         |  19 ++
 src/relay/backend/utils.h                          |  33 ++-
 src/target/source/interface_c.cc                   |  79 ++++++--
 src/target/source/source_module.cc                 |  49 +++--
 src/tir/usmp/transform/assign_pool_info.cc         |  29 +--
 src/tir/usmp/utils.cc                              |  44 +---
 tests/cpp/relay_build_module_test.cc               |   4 +-
 tests/cpp/runtime_test.cc                          |   4 +-
 tests/cpp/target/source/interface_c_test.cc        | 183 +++++++++++++++--
 tests/micro/zephyr/test_utils.py                   |   2 +-
 tests/python/contrib/test_ethosu/test_networks.py  |   2 +-
 tests/python/relay/aot/aot_test_utils.py           |  94 +++++----
 tests/python/relay/aot/test_crt_aot.py             |   6 +-
 tests/python/relay/aot/test_crt_aot_usmp.py        | 223 ++++++++++++++++++++-
 28 files changed, 924 insertions(+), 468 deletions(-)
 create mode 100644 include/tvm/ir/memory_pools.h
 copy python/tvm/{tir/usmp/utils.py => ir/memory_pools.py} (64%)
 create mode 100644 src/ir/memory_pools.cc