This is an automated email from the ASF dual-hosted git repository.
haichen pushed a change to branch master
in repository https://gitbox.apache.org/repos/asf/incubator-tvm.git.
from a400f82 [TFLite Runtime] Fix bug and re-enable RPC execution test
(#5436)
add 674f58a [Relay][VM] Memory planner (part 1) (#5144)
No new revisions were added by this update.
Summary of changes:
include/tvm/runtime/ndarray.h | 2 +
include/tvm/runtime/vm.h | 12 +-
python/tvm/relay/__init__.py | 6 +
python/tvm/relay/expr.py | 1 +
python/tvm/relay/op/memory/memory.py | 7 +-
python/tvm/relay/transform/__init__.py | 1 -
python/tvm/relay/transform/memory_alloc.py | 22 +-
python/tvm/relay/transform/memory_plan.py | 355 +++++++++++++++++++++
src/relay/backend/vm/compiler.cc | 138 +++++---
src/relay/op/memory/memory.cc | 31 +-
src/runtime/ndarray.cc | 3 +
src/runtime/vm/executable.cc | 45 +--
src/runtime/vm/memory_manager.cc | 24 +-
src/runtime/vm/vm.cc | 66 +++-
tests/python/frontend/onnx/test_forward.py | 11 +-
..._pass_memory_alloc.py => test_memory_passes.py} | 51 ++-
16 files changed, 649 insertions(+), 126 deletions(-)
create mode 100644 python/tvm/relay/transform/memory_plan.py
rename tests/python/relay/{test_pass_memory_alloc.py => test_memory_passes.py}
(62%)
