This is an automated email from the ASF dual-hosted git repository.
github-bot pushed a change to branch nightly
in repository https://gitbox.apache.org/repos/asf/tvm.git
from 90320b2424 [Frontend][PaddlePaddle] Fixed the bug that prevented the
model from being successfully converted to microTVM on MacOS (#16478)
add e7125a7af2 [Unity][Frontend] Add Sigmoid and Square Op (#16483)
add 5092c003de [BugTIR] fix thread_sync occurs in letstmt (#16454)
add ff9ebadf6a [wasm] Enlarge initial memory for emcc (#16485)
add 147ed5e27d [Unity][CodeGen] RunCodegen based on externally-exposed
functions (#16422)
add fd1f40f989 [Unity][MSC][Refactor] Reconstruct BYOC and runner (#16467)
No new revisions were added by this update.
Summary of changes:
python/tvm/contrib/emcc.py | 2 +-
python/tvm/contrib/msc/core/frontend/translate.py | 28 +-
python/tvm/contrib/msc/core/ir/graph.py | 13 +
python/tvm/contrib/msc/core/runtime/hook.py | 196 +++++++
python/tvm/contrib/msc/core/runtime/runner.py | 634 +++++++++++++--------
.../contrib/msc/core/tools/distill/distiller.py | 15 +-
python/tvm/contrib/msc/core/tools/prune/pruner.py | 82 +--
python/tvm/contrib/msc/core/tools/tool.py | 83 ++-
python/tvm/contrib/msc/core/tools/track/method.py | 10 +-
python/tvm/contrib/msc/core/tools/track/tracker.py | 12 +-
python/tvm/contrib/msc/core/transform/pattern.py | 97 +++-
python/tvm/contrib/msc/core/transform/transform.py | 4 +-
python/tvm/contrib/msc/core/utils/__init__.py | 1 +
python/tvm/contrib/msc/core/utils/arguments.py | 228 ++++++++
python/tvm/contrib/msc/core/utils/dataset.py | 6 +-
python/tvm/contrib/msc/core/utils/expr.py | 23 +-
python/tvm/contrib/msc/core/utils/file.py | 55 ++
python/tvm/contrib/msc/core/utils/info.py | 240 ++------
python/tvm/contrib/msc/core/utils/log.py | 2 +-
python/tvm/contrib/msc/core/utils/message.py | 2 +-
python/tvm/contrib/msc/core/utils/register.py | 41 +-
.../msc/framework/tensorflow/runtime/runner.py | 91 ++-
.../msc/framework/tensorrt/codegen/codegen.py | 48 +-
.../msc/framework/tensorrt/runtime/runner.py | 47 +-
.../msc/framework/tensorrt/transform/pattern.py | 26 +-
.../contrib/msc/framework/torch/runtime/runner.py | 132 ++++-
.../msc/framework/torch/tools/distill/distiller.py | 4 +-
.../contrib/msc/framework/tvm/runtime/runner.py | 157 ++++-
.../msc/framework/tvm/tools/track/tracker.py | 5 +-
python/tvm/contrib/msc/pipeline/manager.py | 569 ++++++++----------
python/tvm/ir/module.py | 9 +
python/tvm/relax/frontend/nn/op.py | 44 ++
python/tvm/relax/transform/transform.py | 3 +-
src/contrib/msc/core/codegen/codegen_utils.h | 9 +-
src/contrib/msc/core/codegen/py_codegen.h | 2 +-
src/contrib/msc/core/ir/graph.cc | 38 ++
src/contrib/msc/core/ir/graph.h | 2 +
src/contrib/msc/core/ir/graph_builder.cc | 348 +++++++----
src/contrib/msc/core/ir/graph_builder.h | 49 +-
src/contrib/msc/core/transform/fuse_tuple.cc | 36 +-
src/contrib/msc/core/transform/inline_params.cc | 192 +++++++
src/contrib/msc/core/transform/layout_utils.cc | 25 +-
src/contrib/msc/core/transform/set_byoc_attrs.cc | 39 +-
src/contrib/msc/core/transform/set_expr_name.cc | 105 ++--
.../msc/framework/tensorflow/codegen_utils.h | 7 +-
src/contrib/msc/framework/tensorrt/codegen.cc | 10 +-
.../msc/framework/tensorrt/transform_tensorrt.cc | 28 +-
src/contrib/msc/framework/torch/codegen_utils.h | 7 +-
src/ir/module.cc | 34 ++
src/relax/transform/run_codegen.cc | 40 +-
src/tir/transforms/storage_access.cc | 14 +
src/tir/transforms/storage_access.h | 1 +
tests/python/contrib/test_msc/test_manager.py | 16 +-
tests/python/contrib/test_msc/test_runner.py | 22 +-
tests/python/contrib/test_msc/test_tools.py | 18 +-
tests/python/contrib/test_msc/test_transform.py | 20 +-
.../contrib/test_msc/test_translate_tensorrt.py | 10 +-
tests/python/relax/test_frontend_nn_op.py | 27 +
tests/python/relax/test_transform_codegen_pass.py | 43 +-
.../test_tir_transform_thread_sync.py | 77 +++
60 files changed, 2884 insertions(+), 1244 deletions(-)
create mode 100644 python/tvm/contrib/msc/core/runtime/hook.py
create mode 100644 python/tvm/contrib/msc/core/utils/arguments.py
create mode 100644 src/contrib/msc/core/transform/inline_params.cc