This is an automated email from the ASF dual-hosted git repository.
jroesch pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 01599d1 [SimplifyExpr] Simplify consecutive adds with constants
(#9671)
add e3379a6 [Relay] PlanDevices supports 'free' on_device annotations
(#9693)
No new revisions were added by this update.
Summary of changes:
include/tvm/relay/attrs/call.h | 4 +-
include/tvm/relay/attrs/on_device.h | 62 ++--
python/tvm/relay/op/annotation/annotation.py | 24 +-
src/printer/relay_text_printer.cc | 11 +-
src/printer/text_printer.cc | 57 ++--
.../contrib/example_target_hooks/relay_to_tir.cc | 9 +-
src/relay/backend/te_compiler.cc | 42 +--
src/relay/backend/vm/compiler.cc | 17 +-
src/relay/op/call/call.cc | 43 +--
src/relay/op/call/call.h | 34 ++-
src/relay/op/memory/on_device.cc | 81 +++---
src/relay/op/memory/on_device.h | 75 +++--
src/relay/transforms/device_aware_visitors.cc | 15 +-
src/relay/transforms/device_aware_visitors.h | 19 +-
src/relay/transforms/device_domains.cc | 56 +++-
src/relay/transforms/device_domains.h | 19 +-
src/relay/transforms/device_planner.cc | 314 +++++++++++++--------
src/relay/transforms/fold_constant.cc | 6 +-
src/relay/transforms/memory_alloc.cc | 77 +++--
src/relay/transforms/to_a_normal_form.cc | 8 +-
src/runtime/vm/executable.cc | 2 +-
src/target/se_scope.cc | 6 -
src/target/target.cc | 4 -
tests/cpp/relay/op/memory/on_device_test.cc | 89 ++++++
.../python/relay/op/annotation/test_annotation.py | 18 +-
tests/python/relay/test_pass_fold_constant.py | 2 +-
tests/python/relay/test_pass_plan_devices.py | 149 +++++++---
tests/python/relay/test_vm.py | 6 +-
28 files changed, 831 insertions(+), 418 deletions(-)
create mode 100644 tests/cpp/relay/op/memory/on_device_test.cc
