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 831d769f47 [Transform] Remove R.Object parameters after
LazyTransformParams (#16699)
add 3b25697e9d [MSC][M5.2] Enable quantize && prune with gym by wrapper
(#16702)
add 6a877df173 [CMake] Add "USE_FLASHINFER" to libinfo (#16710)
add dffdc3e592 [Relax][Frontend] Add op `tanh`, `exp`, `negative`, and
`permute` (#16711)
add 8023a981a4 [Relax] Normalize use of void-type variable to inline
R.tuple() (#16658)
add 981009d457 [Fix] PagedKVCache fetching compute stream when copy stream
is needed (#16714)
No new revisions were added by this update.
Summary of changes:
cmake/modules/LibInfo.cmake | 1 +
gallery/how_to/work_with_msc/using_tools.py | 11 +-
.../tvm/contrib/msc/core/gym/agent/base_agent.py | 49 +++---
python/tvm/contrib/msc/core/gym/agent/method.py | 11 +-
.../tvm/contrib/msc/core/gym/agent/search_agent.py | 11 +-
.../tvm/contrib/msc/core/gym/control/configer.py | 12 +-
.../tvm/contrib/msc/core/gym/control/controller.py | 6 +-
python/tvm/contrib/msc/core/gym/control/service.py | 46 ++++--
python/tvm/contrib/msc/core/gym/control/worker.py | 32 ++--
.../contrib/msc/core/gym/environment/base_env.py | 105 +++++++++---
.../tvm/contrib/msc/core/gym/environment/method.py | 14 +-
.../contrib/msc/core/gym/environment/prune_env.py | 51 +++---
.../msc/core/gym/environment/quantize_env.py | 58 ++-----
.../msc/core/gym/{control => }/namespace.py | 2 +-
python/tvm/contrib/msc/core/runtime/hook.py | 4 +-
python/tvm/contrib/msc/core/runtime/runner.py | 43 ++++-
python/tvm/contrib/msc/core/tools/configer.py | 20 +--
.../contrib/msc/core/tools/distill/distiller.py | 4 +-
.../tvm/contrib/msc/core/tools/distill/method.py | 6 +-
python/tvm/contrib/msc/core/tools/execute.py | 2 +-
python/tvm/contrib/msc/core/tools/prune/method.py | 6 +-
python/tvm/contrib/msc/core/tools/prune/pruner.py | 4 +-
.../tvm/contrib/msc/core/tools/quantize/method.py | 6 +-
.../contrib/msc/core/tools/quantize/quantizer.py | 4 +-
.../tvm/contrib/msc/core/tools/track/configer.py | 15 +-
python/tvm/contrib/msc/core/tools/track/method.py | 4 +-
python/tvm/contrib/msc/core/tools/track/tracker.py | 4 +-
python/tvm/contrib/msc/core/utils/expr.py | 25 ++-
python/tvm/contrib/msc/core/utils/file.py | 26 ++-
python/tvm/contrib/msc/core/utils/log.py | 8 +
python/tvm/contrib/msc/core/utils/message.py | 2 +
python/tvm/contrib/msc/core/utils/register.py | 181 +++++++++------------
.../tensorflow/tools/distill/distiller.py | 5 +-
.../msc/framework/tensorflow/tools/prune/pruner.py | 5 +-
.../tensorflow/tools/quantize/quantizer.py | 5 +-
.../framework/tensorflow/tools/track/tracker.py | 5 +-
.../msc/framework/tensorrt/runtime/runner.py | 23 +++
.../framework/tensorrt/tools/distill/distiller.py | 5 +-
.../msc/framework/tensorrt/tools/prune/pruner.py | 5 +-
.../framework/tensorrt/tools/quantize/method.py | 4 +-
.../framework/tensorrt/tools/quantize/quantizer.py | 5 +-
.../msc/framework/tensorrt/tools/track/tracker.py | 5 +-
.../msc/framework/torch/tools/distill/distiller.py | 5 +-
.../msc/framework/torch/tools/distill/method.py | 4 +-
.../msc/framework/torch/tools/prune/pruner.py | 5 +-
.../msc/framework/torch/tools/quantize/method.py | 4 +-
.../framework/torch/tools/quantize/quantizer.py | 5 +-
.../msc/framework/torch/tools/track/tracker.py | 5 +-
.../contrib/msc/framework/tvm/runtime/runner.py | 38 ++++-
.../msc/framework/tvm/tools/distill/distiller.py | 5 +-
.../msc/framework/tvm/tools/prune/pruner.py | 5 +-
.../msc/framework/tvm/tools/quantize/method.py | 4 +-
.../msc/framework/tvm/tools/quantize/quantizer.py | 5 +-
.../msc/framework/tvm/tools/track/tracker.py | 5 +-
python/tvm/contrib/msc/pipeline/config.py | 14 +-
python/tvm/contrib/msc/pipeline/manager.py | 140 +++++++++++-----
python/tvm/contrib/msc/pipeline/wrapper.py | 26 +--
python/tvm/relax/frontend/nn/op.py | 94 +++++++++++
src/relax/ir/block_builder.cc | 9 +-
src/runtime/relax_vm/paged_kv_cache.cc | 4 +-
src/script/ir_builder/relax/utils.h | 11 +-
src/script/printer/relax/binding.cc | 22 +--
src/support/libinfo.cc | 1 +
tests/python/relax/test_frontend_nn_op.py | 6 +
.../relax/test_transform_lift_transform_params.py | 10 +-
tests/python/relax/test_transform_normalize.py | 29 ++++
tests/python/relax/test_tvmscript_printer_relax.py | 5 -
67 files changed, 845 insertions(+), 456 deletions(-)
copy python/tvm/contrib/msc/core/gym/{control => }/namespace.py (96%)
