This is an automated email from the ASF dual-hosted git repository.
junrushao pushed a change to branch main
in repository https://gitbox.apache.org/repos/asf/tvm.git.
from 3ad7c4a [Conv2DTransposed] Fix wrong shape check and add new TOPI
module to support groups (#9465)
add be03d62 Switch PlanDevices pass to be w.r.t. SEScopes instead of
DLDeviceTypes. (#9326)
No new revisions were added by this update.
Summary of changes:
include/tvm/ir/function.h | 12 +-
include/tvm/relay/attrs/annotation.h | 62 --
include/tvm/relay/attrs/device_copy.h | 16 +-
include/tvm/relay/attrs/memory.h | 7 +-
include/tvm/relay/attrs/on_device.h | 101 +++
include/tvm/relay/transform.h | 24 +-
include/tvm/runtime/vm/bytecode.h | 26 +-
include/tvm/runtime/vm/executable.h | 44 +-
include/tvm/runtime/vm/vm.h | 44 +-
python/tvm/relay/op/annotation/annotation.py | 21 +-
python/tvm/relay/op/tensor.py | 36 +-
python/tvm/relay/transform/transform.py | 25 +-
python/tvm/runtime/vm.py | 14 +-
src/relay/backend/aot_executor_codegen.cc | 27 +-
src/relay/backend/build_module.cc | 171 ++---
src/relay/backend/graph_executor_codegen.cc | 14 +-
src/relay/backend/graph_plan_memory.cc | 55 +-
src/relay/backend/interpreter.cc | 9 +-
src/relay/backend/te_compiler.cc | 119 ++--
src/relay/backend/te_compiler.h | 9 +-
src/relay/backend/utils.cc | 71 +-
src/relay/backend/utils.h | 19 +-
src/relay/backend/vm/compiler.cc | 264 ++++----
src/relay/backend/vm/compiler.h | 20 +-
src/relay/backend/vm/lambda_lift.cc | 9 +-
src/relay/ir/expr_functor.cc | 13 +-
src/relay/op/annotation/annotation.cc | 152 -----
src/relay/op/annotation/annotation.h | 106 ---
src/relay/op/memory/device_copy.cc | 53 +-
src/relay/op/memory/device_copy.h | 38 +-
src/relay/op/memory/memory.cc | 9 +-
src/relay/op/memory/memory.h | 4 +-
src/relay/op/memory/on_device.cc | 167 +++++
src/relay/op/memory/on_device.h | 144 ++++
src/relay/transforms/device_aware_visitors.cc | 98 +--
src/relay/transforms/device_aware_visitors.h | 81 +--
src/relay/transforms/device_domains.cc | 339 +++++-----
src/relay/transforms/device_domains.h | 229 ++++---
src/relay/transforms/device_planner.cc | 392 ++++++-----
src/relay/transforms/fold_constant.cc | 52 +-
src/relay/transforms/memory_alloc.cc | 119 ++--
src/relay/transforms/pass_utils.h | 1 +
src/relay/transforms/to_a_normal_form.cc | 20 +-
src/runtime/vm/bytecode.cc | 33 +-
src/runtime/vm/executable.cc | 64 +-
src/runtime/vm/profiler/vm.cc | 12 +-
src/runtime/vm/serialize_utils.h | 12 +-
src/runtime/vm/vm.cc | 102 +--
tests/cpp/relay/transforms/device_domains_test.cc | 22 +-
.../python/relay/op/annotation/test_annotation.py | 27 +-
tests/python/relay/op/test_tensor.py | 52 ++
tests/python/relay/test_pass_plan_devices.py | 736 ++++++++++++++-------
tests/python/relay/test_vm.py | 57 +-
.../python/unittest/test_auto_scheduler_measure.py | 17 +-
.../unittest/test_micro_model_library_format.py | 3 +-
tests/python/unittest/test_runtime_profiling.py | 5 +-
tests/python/unittest/test_runtime_vm_profiler.py | 5 +-
57 files changed, 2432 insertions(+), 1951 deletions(-)
create mode 100644 include/tvm/relay/attrs/on_device.h
create mode 100644 src/relay/op/memory/on_device.cc
create mode 100644 src/relay/op/memory/on_device.h
create mode 100644 tests/python/relay/op/test_tensor.py
