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